collect-set-words.red

Red [
	title:    "COLLECT-SET-WORDS mezzanine"
	purpose:  "Helpful in localizing a piece of code"
	author:   @hiiamboris
	license:  BSD-3
	provides: collect-set-words
	notes: {
		- function-based version also collects word!-s from `foreach`, `remove-each`, `repeat` (side effect)
		- parse-based version's result may require a call to `unique` (depending on your needs)
	}
]


collect-set-words: function [
	"Deeply collect set-words from a block of code"
	code [block!]
	/local w
][
	spec: spec-of function [] code
	parse spec [remove /local any change set w word! (to set-word! w)]
	spec
]

comment {
	;-- this version is 4x slower, but uses 10x less RAM
	collect-set-words: function [
		"Deeply collect set-words from a block of code"
		code [block!]
	][
		rule: [any [
			ahead [block! | paren!] into rule
		|	keep set-word!
		|	skip
		]]
		parse code [collect rule]
	]
}