aye_var 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 13bedf4da163ee9b20c83ed235f78825ced3e3a6547abc98af2de5e08d06401c
-  data.tar.gz: 4cbc787d135c70ea6c1b725ad01af3c2be374959fb1a52667f9db6271e458b32
+  metadata.gz: d1738b022486bc41367e9dee76d2fc4a0792a47dccb5059ac872a479b192ab8c
+  data.tar.gz: 41ba0ed417967bea1af88f4c828a7bae04b586d4749e93b50ce91972ad4e55bd
 SHA512:
-  metadata.gz: eb4d35e722d2f31d4953a03687421c18b1aea499e69e77ef614285622e41e8ab23c0422b6649dfad1e54066d77362ccad1a5851c3f2a9a1a7dafe6bf44fc877f
-  data.tar.gz: bd303daa537463d610e8b37b356e3bc44e8bd46e4e4734acc58b24ca3a262748ae39e2169cbe328b3daddf4bd84767d94514fd6ed7ebdd54aa1ad980da4a8a4a
+  metadata.gz: 9626b16ccb6c22998e885816359a0c304db95adff5657c69e59cdc83d3549717928b8425f69beee37269e69b8f1152c43330153c88267b0e701faa7b378ab84b
+  data.tar.gz: '089409f803c9dbae0d834795c373617088eef192a1f12bbb719fec2af28cb810884f8c4c3b90275b34fd92f88b04c0152f3c1a400094a3121d5734156033d288'

data/README.md

@@ -1,14 +1,20 @@
-## AyeVar
+# AyeVar 🏴‍☠️
 
-Note, this gem is very new and experimental. The API will probably change.
+Avast ye, this gem be fresh off th’ plunderin’ an’ experimental as a new recruit, me hearties! Th’ API be shiftin’ like th’ treacherous seas beneath yer barnacled hull.
 
-Add this gem to your gemfile.
+## What does it do?
+
+It keelhauls them scurvy undefined instance variables by transformin’ yer code as it loads into th’ hold, savvy? Davy Jones ’imself would approve!
+
+## Setup
+
+Add this precious booty to yer gemfile, shiver me timbers!
 
 ```ruby
 gem "aye_var", require: false
 ```
 
-Then require and initialize it in your app as early as possible. If you’re using Bootsnap, it should be right after Bootsnap.
+Then be requirin’ an’ initializin’ it in yer galleon as early as ye can hoist sail, ye scallywag! If ye be usin’ Bootsnap, place it right after, or by the powers, ye’ll be swimmin’ with th’ fishes!
 
 ```ruby
 require "aye_var"
@@ -16,4 +22,4 @@ require "aye_var"
 AyeVar.init(include: ["#{Dir.pwd}/**/*"])
 ```
 
-You can pass in an array of globs to `include:` and `exclude:`.
+Ye can pass an array o’ globs to `include:` an’ `exclude:`, or I’ll feed ye to th’ kraken, ye mangy bilge-suckin’ swab!

data/lib/aye_var/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AyeVar
-	VERSION = "0.1.0"
+	VERSION = "0.2.0"
 end

data/lib/aye_var.rb

@@ -1,77 +1,136 @@
 # frozen_string_literal: true
 
 require "require-hooks/setup"
+require "prism"
 
 module AyeVar
+	NameError = Class.new(::NameError)
+
+	def self.init(include: [], exclude: [])
+		RequireHooks.source_transform(patterns: include, exclude_patterns: exclude) do |path, source|
+			source ||= File.read(path)
+			Processor.call(source)
+		end
+	end
+
 	class Processor < Prism::Visitor
 		def self.call(source)
-			object = new(source)
-			object.visit(Prism.parse(source).value)
+			visitor = new
+			visitor.visit(Prism.parse(source).value)
 
 			buffer = source.dup
 
-			object.ivars.sort_by(&:first).reverse_each do |offset, action, name|
+			visitor.annotations.sort_by(&:first).reverse_each do |offset, action, name|
 				case action
 				when :start
-					buffer.insert(offset, "((raise ::AyeVar::NameError.new('Undefined instance variable #{name}') unless defined?(#{name}));")
+					buffer.insert(offset, "((raise ::AyeVar::NameError.new('Undefined instance variable #{name}') unless defined?(#{name})); ")
 				when :end
 					buffer.insert(offset, ")")
+				else
+					raise "Invalid annotation"
 				end
 			end
 
 			buffer
 		end
 
-		def initialize(source)
-			@initializer = false
-			@ivars = []
+		def initialize
+			@this = nil
+			@context = Set[]
+			@annotations = []
+		end
+
+		attr_reader :annotations
+
+		def visit_class_node(node)
+			new_context(node) { super }
+		end
+
+		def visit_module_node(node)
+			new_context(node) { super }
+		end
+
+		def visit_block_node(node)
+			new_context(node) { super }
+		end
+
+		def visit_singleton_class_node(node)
+			new_context(node) { super }
 		end
 
-		attr_reader :ivars
+		def visit_defined_node(node)
+			if Prism::InstanceVariableReadNode === node.value
+				context << node.value.name
+			end
+
+			super
+		end
 
 		def visit_def_node(node)
-			if node.name == :initialize
-				begin
-					@initializer = true
-					super
-				ensure
-					@initializer = false
-				end
-			else
-				super
+			new_context(node) { super }
+		end
+
+		def visit_if_node(node)
+			visit(node.predicate)
+
+			branch { visit(node.statements) }
+			branch { visit(node.subsequent) }
+		end
+
+		def visit_case_node(node)
+			visit(node.predicate)
+
+			node.conditions.each do |condition|
+				branch { visit(condition) }
 			end
+
+			branch { visit(node.else_clause) }
 		end
 
 		def visit_instance_variable_read_node(node)
-			location = node.location
 			name = node.name
 
-			@ivars << [location.start_character_offset, :start, name]
-			@ivars << [location.end_character_offset, :end, name]
-			super
-		end
-
-		def visit_instance_variable_write_node(node)
-			unless @initializer
+			unless context.include?(name)
 				location = node.location
-				name = node.name
 
-				@ivars << [location.start_character_offset, :start, name]
-				@ivars << [location.end_character_offset, :end, name]
+				context << name
+
+				@annotations << [location.start_character_offset, :start, name]
+				@annotations << [location.end_character_offset, :end, name]
 			end
 
 			super
 		end
-	end
 
-	private_constant :Processor
+		private def new_context(this)
+			original_this = @this
+			original_context = @context
 
-	NameError = Class.new(::NameError)
+			@this = this
+			@context = Set[]
 
-	def self.init(include: [], exclude: [])
-		RequireHooks.source_transform(patterns: include, exclude_patterns: exclude) do |path, source|
-			source ||= File.read(path)
-			Processor.call(source)
+			begin
+				yield
+			ensure
+				@this = original_this
+				@context = original_context
+			end
+		end
+
+		private def branch
+			original_context = @context
+			@context = original_context.dup
+
+			begin
+				yield
+			ensure
+				@context = original_context
+			end
+		end
+
+		# The current context on the stack
+		private def context
+			@context
 		end
 	end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aye_var
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Joel Drapper
@@ -23,6 +23,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: prism
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Raise an exception when using undefined instance variables.
 email:
 - joel@drapper.me