Only compile methods that are still owned by the class in the file

Author: joeldrapper

lib/phlex.rb

@@ -27,6 +27,8 @@ module Phlex
 	CACHED_FILES = Set.new
 	ATTRIBUTE_CACHE = FIFO.new
 
+	UNBOUND_INSTANCE_METHOD_METHOD = Module.instance_method(:instance_method)
+
 	def self.__expand_attribute_cache__(file_path)
 		unless CACHED_FILES.include?(file_path)
 			CACHED_FILES << file_path

lib/phlex/compiler.rb

@@ -27,7 +27,7 @@ def self.compile_file(path)
 
 		starting_line = last_line + 1
 
-		results = FileCompiler.new.compile(refract)
+		results = FileCompiler.new(path).compile(refract)
 
 		result = Refract::StatementsNode.new(
 			body: results.map do |result|

lib/phlex/compiler/class_compiler.rb

@@ -1,9 +1,10 @@
 # frozen_string_literal: true
 
 class Phlex::Compiler::ClassCompiler < Refract::Visitor
-	def initialize(component)
+	def initialize(component, path)
 		super()
 		@component = component
+		@path = path
 		@compiled_snippets = []
 	end
 
@@ -16,6 +17,17 @@ def compile(node)
 		return if node.name == :initialize
 		return if node.receiver
 
+		method = begin
+			Phlex::UNBOUND_INSTANCE_METHOD_METHOD.bind_call(@component, node.name)
+		rescue NameError
+			nil
+		end
+
+		return unless method
+		path, lineno = method.source_location
+		return unless @path == path
+		return unless node.start_line == lineno
+
 		@compiled_snippets << Phlex::Compiler::MethodCompiler.new(
 			@component
 		).compile(node)

lib/phlex/compiler/file_compiler.rb

@@ -3,8 +3,9 @@
 class Phlex::Compiler::FileCompiler < Refract::Visitor
 	Result = Data.define(:namespace, :compiled_snippets)
 
-	def initialize
-		super
+	def initialize(path)
+		super()
+		@path = path
 		@current_namespace = []
 		@results = []
 	end
@@ -32,7 +33,7 @@ def compile(node)
 		if Class === const && Phlex::SGML > const
 			@results << Result.new(
 				namespace: @current_namespace.dup.freeze,
-				compiled_snippets: Phlex::Compiler::ClassCompiler.new(const).compile(node)
+				compiled_snippets: Phlex::Compiler::ClassCompiler.new(const, @path).compile(node)
 			)
 		else
 			super(node)

lib/phlex/compiler/method_compiler.rb

@@ -430,7 +430,7 @@ def compile_raw_helper(node)
 
 		private def standard_element?(node)
 			if (tag = Phlex::HTML::StandardElements.__registered_elements__[node.name]) &&
-					(Phlex::HTML::StandardElements == @component.instance_method(node.name).owner)
+					(Phlex::HTML::StandardElements == Phlex::UNBOUND_INSTANCE_METHOD_METHOD.bind_call(@component, node.name).owner)
 
 				tag
 			else
@@ -440,7 +440,7 @@ def compile_raw_helper(node)
 
 		private def void_element?(node)
 			if (tag = Phlex::HTML::VoidElements.__registered_elements__[node.name]) &&
-					(Phlex::HTML::VoidElements == @component.instance_method(node.name).owner)
+					(Phlex::HTML::VoidElements == Phlex::UNBOUND_INSTANCE_METHOD_METHOD.bind_call(@component, node.name).owner)
 
 				tag
 			else
@@ -506,7 +506,7 @@ def compile_raw_helper(node)
 
 		ALLOWED_OWNERS = Set[Phlex::SGML, Phlex::HTML, Phlex::SVG]
 		private def own_method_without_scope?(node)
-			ALLOWED_OWNERS.include?(@component.instance_method(node.name).owner)
+			ALLOWED_OWNERS.include?(Phlex::UNBOUND_INSTANCE_METHOD_METHOD.bind_call(@component, node.name).owner)
 		end
 
 		private def state_local