Gen liquid templates

lib/gyro/liquidgen/generator.rb

@@ -24,81 +24,56 @@ module Liquidgen
     #
     class Generator
 
-      attr_accessor :root_template, :filename_template, :params, :output_dir
-      attr_accessor :enum_filename_template, :enum_filename_template_path, :enum_template_path, :enum_template
+      attr_accessor :params, :output_dir, :template_dir
 
       # PUBLIC METHODS #######################################################
       def initialize(template_dir, output_dir, params)
         Gyro::Log.title('Generating Model')
 
         @params = params
         @output_dir = output_dir
-
-        # Parse object template name
-        root_template_path = template_dir + 'root.liquid'
-        Gyro::Error.exit_with_error('Bad template directory content ! Your template need to include root.liquid file') unless root_template_path.exist?
-        root_template_string = root_template_path.read
-        @root_template = Liquid::Template.parse(root_template_string)
+        @template_dir = template_dir
 
         # Define Template path for Liquid file system to use Include Tag
         Liquid::Template.file_system = Liquid::LocalFileSystem.new(template_dir)
-        # Parse object template
-        filename_template_path = (template_dir + 'filename.liquid')
-        Gyro::Error.exit_with_error('Bad template directory content ! Your template need to include filename.liquid file') unless filename_template_path.exist?
-        filename_template_string = filename_template_path.readlines.first
-        @filename_template = Liquid::Template.parse(filename_template_string)
-
-        # Parse enum template
-        @enum_template_path = (template_dir + 'enum.liquid')
-
-        enum_template_string = enum_template_path.read
-        @enum_template = Liquid::Template.parse(enum_template_string)
-
-        # Parse enum template name
-        @enum_filename_template_path = (template_dir + 'enum_filename.liquid').exist? ? (template_dir + 'enum_filename.liquid') : (template_dir + 'filename.liquid')
-        enum_filename_template_string = enum_filename_template_path.readlines.first
-        @enum_filename_template = Liquid::Template.parse(enum_filename_template_string)
+      end
 
+      def generate(xcdatamodel)
+        generate_root_template(xcdatamodel)
         Gyro::Log.success('Model objects are generated !')
       end
 
-      def generate(xcdatamodel)
+      private ################################################################
 
+      def generate_root_template(xcdatamodel) 
         xcdatamodel.to_h['entities'].each do |entity|
           entity_context = { 'params' => @params, 'entity' => entity }
           # Rendering template using entity and params context
-          output = @root_template.render(entity_context, filters: [CustomFilters])
-                                .gsub(/^ +$/, '')
+          output = root_template_rendering(entity_context)
           # Don't generate empty output
           next if output.delete("\n").empty?
 
           filename_context = { 'params' => @params, 'name' => entity['name'] }
           # Rendering filename template using entity name and params context
-          filename = @filename_template.render(filename_context).chomp
+          filename = filename_rendering(filename_context)
           Gyro::Log.success("#{filename} is created !")
           # Write model object
           Gyro.write_file_with_name(@output_dir, filename, output)
           # Generate model object enums
           generate_enums(entity['attributes'])
         end
-
-        Gyro::Log.success('Model objects are generated !')
       end
-
-      private ################################################################
-
+
       def generate_enums(attributes)
         enums = []
         attributes.each do |attribute|
           enum_type = attribute['enum_type']
           next unless !enums.include?(enum_type) && !enum_type.empty?
           enums.push(enum_type)
-          next unless @enum_template_path.exist?
 
           enum_context = { 'params' => @params, 'attribute' => attribute }
           # Rendering enum template using attribute and params context
-          output = @enum_template.render(enum_context, filters: [CustomFilters])
-                                .gsub(/^ +$/, '')
+          output = enum_template_rendering(enum_context)
           # Don't generate empty output
           next if output.delete("\n").empty?
 
@@ -107,14 +82,49 @@ def generate_enums(attributes)
       end
 
       def generate_enum(enum_name, output)
-
-        Gyro::Error.exit_with_error('Bad template directory content ! Your template need to have enum_filename.liquid or filename.liquid file !') unless @enum_filename_template_path.exist?
         # Rendering enum filename template using enum name and params context
         enum_filename_context = { 'params' => @params, 'name' => enum_name }
-        enum_filename = @enum_filename_template.render(enum_filename_context).chomp
-
+        enum_filename = enum_filename_template_rendering(enum_filename_context)
         Gyro.write_file_with_name(@output_dir, enum_filename, output)
       end
+
+      def root_template_rendering(context)
+        # Parse object template
+        root_template_path = @template_dir + 'root.liquid'
+        Gyro::Error.exit_with_error('Bad template directory content ! Your template need to include root.liquid file') unless root_template_path.exist?
+        root_template_string = root_template_path.read
+        root_template = Liquid::Template.parse(root_template_string)
+        root_template.render(context, filters: [CustomFilters])
+                                .gsub(/^ +$/, '')
+      end
+
+      def filename_rendering(context)
+        # Parse object template name
+        filename_template_path = (@template_dir + 'filename.liquid')
+        Gyro::Error.exit_with_error('Bad template directory content ! Your template need to include filename.liquid file') unless filename_template_path.exist?
+        filename_template_string = filename_template_path.readlines.first
+        filename_template = Liquid::Template.parse(filename_template_string)
+        filename_template.render(context).chomp
+      end
+
+      def enum_template_rendering(context)
+        # Parse enum template
+        enum_template_path = (@template_dir + 'enum.liquid')
+        Gyro::Error.exit_with_error('Bad template directory content ! Your template need to have enum.liquid file !') unless enum_template_path.exist?
+        enum_template_string = enum_template_path.read
+        enum_template = Liquid::Template.parse(enum_template_string)
+        enum_template.render(context, filters: [CustomFilters])
+                                .gsub(/^ +$/, '')
+      end
+
+      def enum_filename_template_rendering(context)
+        # Parse enum template name
+        enum_filename_template_path = (@template_dir + 'enum_filename.liquid').exist? ? (@template_dir + 'enum_filename.liquid') : (@template_dir + 'filename.liquid')
+        Gyro::Error.exit_with_error('Bad template directory content ! Your template need to have enum_filename.liquid or filename.liquid file !') unless enum_filename_template_path.exist?
+        enum_filename_template_string = enum_filename_template_path.readlines.first
+        enum_filename_template = Liquid::Template.parse(enum_filename_template_string)
+        enum_filename_template.render(context).chomp
+      end
     end
   end
 end