fix: properly handle custom resource file extensions

Author: mattkhan

lib/anchor/type_script/file_structure.rb

@@ -24,14 +24,15 @@ def self.export_code(identifier)
       end
     end
 
-    def initialize(definition)
+    def initialize(definition, extension: ".ts")
       @definition = definition
       @name = definition.name
       @object = definition.object
+      @extension = extension
     end
 
     def name
-      "#{@definition.name}.ts"
+      "#{@definition.name}#{@extension}"
     end
 
     def to_code(manually_editable: false)
@@ -66,7 +67,7 @@ def shared_imports
     def relationship_imports
       relationships_to_import
         .reject { |type| type.anchor_schema_name == @name }
-        .map { |type| Import.new(file_name: "#{type.anchor_schema_name}.ts", type:) }
+        .map { |type| Import.new(file_name: "#{type.anchor_schema_name}#{@extension}", type:) }
     end
 
     def relationships_to_import

lib/anchor/type_script/multifile_save_service.rb

@@ -8,11 +8,10 @@ def self.call(...)
       new(...).call
     end
 
-    def initialize(generator:, folder_path:, force: false, resource_file_extension: ".ts")
+    def initialize(generator:, folder_path:, force: false)
       @generator = generator
       @folder_path = folder_path
       @force = force
-      @resource_file_extension = "." + resource_file_extension.sub(/^\./, "")
     end
 
     def call
@@ -24,13 +23,7 @@ def call
     private
 
     def save_result(result)
-      filename = if result.type == MultifileSchemaGenerator::FileType::RESOURCE
-        resource_file_name(result.name)
-      else
-        result.name
-      end
-
-      path = Rails.root.join(@folder_path, filename)
+      path = Rails.root.join(@folder_path, result.name)
 
       if @force || !File.exist?(path)
         File.open(path, "w") { |f| f.write(result.text) }
@@ -51,10 +44,6 @@ def save_result(result)
       File.open(path, "w") { |f| f.write(new_content) }
     end
 
-    def resource_file_name(name)
-      File.basename(name, ".ts") + @resource_file_extension
-    end
-
     def manually_editable?(text)
       !text.match(REGEX).nil?
     end

lib/anchor/type_script/multifile_schema_generator.rb

@@ -7,12 +7,20 @@ module FileType
       UTIL = "util"
     end
 
-    def initialize(register:, context: {}, include_all_fields: false, exclude_fields: nil, manually_editable: true) # rubocop:disable Lint/MissingSuper
+    def initialize( # rubocop:disable Lint/MissingSuper
+      register:,
+      context: {},
+      include_all_fields: false,
+      exclude_fields: nil,
+      manually_editable: true,
+      resource_file_extension: ".ts"
+    )
       @register = register
       @context = context
       @include_all_fields = include_all_fields
       @exclude_fields = exclude_fields
       @manually_editable = manually_editable
+      @resource_file_extension = "." + resource_file_extension.sub(/^\./, "")
     end
 
     def call
@@ -37,7 +45,7 @@ def resource_files
           exclude_fields: @exclude_fields.nil? ? [] : @exclude_fields[r.anchor_schema_name.to_sym],
         )
 
-        file_structure = ::Anchor::TypeScript::FileStructure.new(definition)
+        file_structure = ::Anchor::TypeScript::FileStructure.new(definition, extension: @resource_file_extension)
         text = file_structure.to_code(manually_editable: @manually_editable)
         name = file_structure.name
         Result.new(name:, text:, type: FileType::RESOURCE)