redbreast 0.1.0 → 1.0.2

data/lib/redbreast/commands/image_test_generator.rb

@@ -0,0 +1,57 @@
+module Redbreast
+  module Command
+    # Class for generating image tests
+    class ImageTestGenerator
+      include Helper::Terminal
+      include Helper::General
+
+      def self.init
+        new.call
+      end
+
+      def call
+        filtered_bundles = bundles.select { |bundle| bundle[:outputTestPathImages] }
+        return if filtered_bundles.empty?
+
+        prompt.say('Generating images test resources...')
+        generate_test_sources(bundles, programming_language, app_name)
+        success('Images test resources generated!')
+      end
+
+      private
+
+      def generate_test_sources(bundles, programming_language, app_name)
+        bundles.each do |bundle|
+          image_names = pull_asset_names(bundle[:assetsSearchPath])
+          write_tests(image_names, bundle, programming_language, app_name)
+        end
+      end
+
+      # Serializing data
+
+      def write_tests(image_names, bundle, programming_language, app_name)
+        output_path = bundle[:outputTestPathImages]
+
+        return if output_path.to_s.empty?
+
+        case programming_language.downcase
+        when 'objc'
+          serializer = Redbreast::Serializer::ObjC
+          template_generator = Redbreast::TemplateGenerator::ImageTest::ObjC
+        when 'swift'
+          serializer = Redbreast::Serializer::Swift
+          template_generator = Redbreast::TemplateGenerator::ImageTest::Swift
+        end
+        serializer.new(image_names, bundle, app_name).save(output_source_path: output_path, template_generator: template_generator.new, generate_colors: false)
+
+      end
+
+      # Pulling data
+
+      def pull_asset_names(assetsSearchPath)
+        Redbreast::Crawler::Image
+          .image_names_uniq(assetsSearchPath)
+      end
+    end
+  end
+end

data/lib/redbreast/commands/setup.rb

@@ -2,115 +2,163 @@ require 'commander/blank'
 require 'commander/command'
 
 module Redbreast
-    module Command
-      class Setup
-        include Helper::Terminal
-        include Helper::General
-        include Helper::Hash
-  
-        def self.init(options = Commander::Command::Options.new)
-          new(options).call
-        end
-      
-        def initialize(options = Commander::Command::Options.new)
-          @options = options
-        end
-  
-        def call
-          language = language_prompt
-          bundle_names = bundle_names_prompt(language).split(" ")
-          bundles = bundle_names.map do |bundle| 
-            reference = bundle_reference(bundle, language)
-            assets_search_path = assets_search_path_prompt(bundle)
-            output_source_path = images_sources_path_prompt(bundle, language)
-            include_tests = create_tests_path_prompt?(bundle)
-            {
-                name: bundle,
-                reference: reference,
-                assetsSearchPath: assets_search_path,
-                outputSourcePath: output_source_path,
-                outputTestPath: include_tests ? tests_path_prompt(bundle, language) : nil,
-                testableImport: include_tests ? testable_import_prompt(bundle, language) : nil
-            }.compact
-          end
-          config = {
-            language: language,
-            bundles: bundles
+  module Command
+    # Class for setting up the program
+    class Setup
+      include Helper::Terminal
+      include Helper::General
+      include Helper::HashHelper
+
+      def self.init(options = Commander::Command::Options.new)
+        new(options).call
+      end
+
+      def initialize(options = Commander::Command::Options.new)
+        @options = options
+      end
+
+      def call
+        language = language_prompt
+        app_name = app_name_prompt
+        bundle_names = bundle_names_prompt(language).split(' ')
+        assets_types = assets_types_prompt
+        bundles = bundle_names.map do |bundle|
+          reference = bundle_reference(bundle, language)
+          assets_search_path = assets_search_path_prompt(bundle)
+          output_source_path_images = assets_types == 1 ? nil : images_sources_path_prompt(bundle, language)
+          output_source_path_colors = assets_types.zero? ? nil : colors_sources_path_prompt(bundle, language)
+          include_tests = create_tests_path_prompt?(bundle)
+          output_test_path_images = assets_types != 1 && include_tests ? images_tests_path_prompt(bundle, language) : nil
+          output_test_path_colors = assets_types != 0 && include_tests ? colors_tests_path_prompt(bundle, language) : nil
+          fields = {
+            name: bundle,
+            reference: reference,
+            assetsSearchPath: assets_search_path,
+            outputSourcePathImages: output_source_path_images,
+            outputSourcePathColors: output_source_path_colors,
+            outputTestPathImages: output_test_path_images,
+            outputTestPathColors: output_test_path_colors,
+            testableImport: include_tests ? testable_import_prompt(bundle, app_name, language) : nil
           }
-          Redbreast::IO::Config.write(config)
-          success
-        end
-  
-        private
-        
-        # Language
-  
-        def language_prompt
-          languages = {'Swift' => 'swift', 'Objective-C' => 'objc'}
-          prompt.select('Choose a language: ', languages)
+          compact fields
         end
-  
-        # Assets source path
+        config = {
+          language: language,
+          bundles: bundles,
+          app_name: app_name
+        }
+        Redbreast::IO::Config.write(config)
+        success
+      end
+
+      private
+
+      # Language
+
+      def language_prompt
+        languages = { 'Swift' => 'swift', 'Objective-C' => 'objc' }
+        prompt.select('Choose a language: ', languages)
+      end
+
+      # Assets source path
+
+      def assets_search_path_prompt(bundle)
+        prompt_text = "Please enter assets folder search paths for bundle #{bundle}?"
+        prompt.ask(prompt_text, default: '**/*.xcassets')
+      end
 
-        def assets_search_path_prompt(bundle)
-          prompt.ask("Please enter assets folder search paths for bundle #{bundle}?", default: '**/*.xcassets')
+      # Bundle names promt
+
+      def bundle_names_prompt(language)
+        prompt_text = 'Please enter bundle names that you use separated by spaces'
+        case language
+        when 'objc'
+          prompt.ask(prompt_text, default: 'mainBundle')
+        when 'swift'
+          prompt.ask(prompt_text, default: 'main')
         end
+      end
 
-        # Bundle names promt
-        
-        def bundle_names_prompt(language)
-            case language
-            when "objc"
-              prompt.ask('Please enter bundle names that you use separated by spaces', default: 'mainBundle')
-            when "swift"
-              prompt.ask('Please enter bundle names that you use separated by spaces', default: 'main')
-            end
+      def bundle_reference(bundle_name, language)
+        case language
+        when 'objc'
+          "[NSBundle #{bundle_name}]"
+        when 'swift'
+          ".#{bundle_name}"
         end
+      end
+
+      # Images source path
 
-        def bundle_reference(bundle_name, language)
-          case language
-          when "objc"
-            "[NSBundle #{bundle_name}]"
-          when "swift"
-            ".#{bundle_name}"
-          end
+      def images_sources_path_prompt(bundle, language)
+        prompt_text = "Where would you like to store images resources files for bundle #{bundle}?"
+        case language
+        when 'objc'
+          prompt.ask(prompt_text, default: './Common/Categories/Images')
+        when 'swift'
+          prompt.ask(prompt_text, default: './Common/Extensions/UIImageExtension.swift')
         end
+      end
 
-        # Images source path
+      # Colors source path
 
-        def images_sources_path_prompt(bundle, language)
-          case language
-          when "objc"
-            prompt.ask("Where would you like to store images resources files for bundle #{bundle}?", default: './Common/Categories/Images')
-          when "swift"
-            prompt.ask("Where would you like to store images resources files for bundle #{bundle}?", default: './Common/Extensions/UIImageExtension.swift')
-          end
+      def colors_sources_path_prompt(bundle, language)
+        prompt_text = "Where would you like to store colors resources files for bundle #{bundle}?"
+        case language
+        when 'objc'
+          prompt.ask(prompt_text, default: './Common/Categories/Colors')
+        when 'swift'
+          prompt.ask(prompt_text, default: './Common/Extensions/UIColorExtension.swift')
         end
+      end
 
-        # Tests
+      # Tests
+
+      def create_tests_path_prompt?(bundle)
+        prompt.yes?("Would you like to create tests for bundle #{bundle}?")
+      end
 
-        def create_tests_path_prompt?(bundle)
-          prompt.yes?("Would you like to create tests for bundle #{bundle}?")
+      def images_tests_path_prompt(bundle, language)
+        prompt_text = "Where would you like to store tests for bundle #{bundle}?"
+        case language
+        when 'objc'
+          prompt.ask(prompt_text, default: './Common/Categories/ImagesTest')
+        when 'swift'
+          prompt.ask(prompt_text, default: './Common/Extensions/UIImageExtensionTest.swift')
         end
+      end
 
-        def tests_path_prompt(bundle, language)
-          case language
-          when "objc"
-            prompt.ask("Where would you like to store tests for bundle #{bundle}?", default: './Common/Categories/ImagesTest')
-          when "swift"
-            prompt.ask("Where would you like to store tests for bundle #{bundle}?", default: './Common/Extensions/UIImageExtensionTest.swift')
-          end
+      def colors_tests_path_prompt(bundle, language)
+        prompt_text = "Where would you like to store tests for bundle #{bundle}?"
+        case language
+        when 'objc'
+          prompt.ask(prompt_text, default: './Common/Categories/ColorsTest')
+        when 'swift'
+          prompt.ask(prompt_text, default: './Common/Extensions/UIColorExtensionTest.swift')
         end
+      end
 
-        def testable_import_prompt(bundle, language)
-          case language
-          when "objc"
-            nil
-          when "swift"
-            prompt.ask("Please enter testable import name for bundle #{bundle}?", required: true)
-          end
+      def testable_import_prompt(bundle, app_name, language)
+        case language
+        when 'objc'
+          nil
+        when 'swift'
+          prompt.ask("Please enter a name that will be used for testable import in #{bundle} bundle?", default: "#{app_name}", required: true)
         end
+      end
+
+      # Application name propmt
+
+      def app_name_prompt
+        prompt.ask('Please enter application name')
+      end
+
+      # Assets type prompt
 
+      def assets_types_prompt
+        types = { 'Images' => 0, 'Colors' => 1, 'Both' => 2 }
+        prompt.select('Choose a type: ', types)
       end
     end
-  end
+  end
+end

data/lib/redbreast/crawlers/color_crawler.rb

@@ -0,0 +1,38 @@
+module Redbreast
+  module Crawler
+    # Class used for finding colors
+    class Color
+      def self.color_names_uniq(assets_search_path)
+        Dir.glob(assets_search_path).flat_map do |asset_folder|
+          Dir.glob("#{asset_folder}/**/*.colorset").map do |color_name|
+            name_to_split = color_name
+            split_name = name_to_split.split('.xcassets/')
+            current_color_name = split_name[0] + '.xcassets/'
+            current_iterating_name = split_name[0] + '.xcassets/'
+
+            split_name[1].split('/').each do |folder|
+              if folder.include? '.colorset'
+                current_color_name += folder
+                next
+              end
+
+              current_iterating_name += folder + '/'
+
+              Dir.glob("#{current_iterating_name}*.json").map do |path_name|
+                File.open path_name do |file|
+                  unless file.find { |line| line =~ /provides/ }.nil?
+                    current_color_name += folder + '/'
+                    next
+                  end
+                end
+              end
+            end
+
+            current_color_name.split('.xcassets/')[-1].chomp('.colorset')
+          end
+        end
+           .uniq
+      end
+    end
+  end
+end

data/lib/redbreast/crawlers/image_crawler.rb

@@ -1,18 +1,38 @@
 module Redbreast
   module Crawler
+    # Class for finding images
     class Image
-
       def self.image_names_uniq(assets_search_path)
-        Dir.glob(assets_search_path).flat_map { |asset_folder|
-            Dir.glob("#{asset_folder}/**/*.imageset").map { |image_name|
-                image_name.split("/")[-1].chomp(".imageset")
-            }
-          }
-          .uniq
-      end
+        Dir.glob(assets_search_path).flat_map do |asset_folder|
+          Dir.glob("#{asset_folder}/**/*.imageset").map do |image_name|
+            name_to_split = image_name
+            split_name = name_to_split.split('.xcassets/')
+            current_image_name = split_name[0] + '.xcassets/'
+            current_iterating_name = split_name[0] + '.xcassets/'
+
+            split_name[1].split('/').each do |folder|
+              if folder.include?  '.imageset'
+                current_image_name += folder
+                next
+              end
+
+              current_iterating_name += folder + '/'
 
+              Dir.glob("#{current_iterating_name}*.json").map do |path_name|
+                File.open path_name do |file|
+                  unless file.find { |line| line =~ /provides/ }.nil?
+                    current_image_name += folder + '/'
+                    next
+                  end
+                end
+              end
+            end
+
+            current_image_name.split('.xcassets/')[-1].chomp('.imageset')
+          end
+        end
+           .uniq
+      end
     end
   end
 end
-
-      

data/lib/redbreast/error_handler.rb

@@ -1,24 +1,25 @@
 module Redbreast
+  # Class for handling errors that occurr
   class ErrorHandler
     extend Helper::Terminal
 
     class << self
       def rescuable
         yield
-      rescue => e
+      rescue StandardError => e
         handle(e)
       end
 
-      def handle(e)
+      def handle(error)
         prompt.error(
-          case e
+          case error
           when Errno::ENOENT
-            "We could not find a file that we need:\n\n#{e.message}"
+            "We could not find a file that we need:\n\n#{error.message}"
           else
-            "An error happened. This might help:\n\n#{e.message}"
+            "An error happened. This might help:\n\n#{error.message}"
           end
         )
       end
     end
   end
-end
+end

data/lib/redbreast/helpers/general.rb

@@ -1,67 +1,77 @@
 module Redbreast
-    module Helper
-      module General
-        ESCAPE_KEYWORDS = ["associatedtype", "class", "deinit", "enum", "extension", "fileprivate", "func", 
-                           "import", "init", "inout", "internal", "let", "operator", "private", "protocol", 
-                           "public", "static", "struct", "subscript", "typealias", "var", "break", "case", 
-                           "continue", "default", "defer", "do", "else", "fallthrough", "for", "guard", "if", 
-                           "in", "repeat", "return", "switch", "where", "while", "as", "Any", "catch", "false", 
-                           "is", "nil", "rethrows", "super", "self", "Self", "throw", "throws", "true", "try", "_"]
-    
-        def config
-          @config ||= Redbreast::IO::Config.read
-        end
-  
-        def programming_language
-          @programming_language ||= config[:language]
-        end
+  module Helper
+    # Module with general metods used in creating files
+    module General
+      ESCAPE_KEYWORDS = ['associatedtype', 'class', 'deinit', 'enum', 'extension',
+                         'fileprivate', 'func', 'import', 'init', 'inout',
+                         'internal', 'let', 'operator', 'private', 'protocol',
+                         'public', 'static', 'struct', 'subscript', 'typealias',
+                         'var', 'break', 'case', 'continue', 'default',
+                         'defer', 'do', 'else', 'fallthrough', 'for',
+                         'guard', 'if', 'in', 'repeat', 'return',
+                         'switch', 'where', 'while', 'as', 'Any',
+                         'catch', 'false', 'is', 'nil', 'rethrows',
+                         'super', 'self', 'Self', 'throw', 'throws',
+                         'true', 'try', '_'].freeze
+      def config
+        @config ||= Redbreast::IO::Config.read
+      end
+
+      def programming_language
+        @programming_language ||= config[:language]
+      end
+
+      def bundles
+        @bundles ||= config[:bundles]
+      end
+
+      def app_name
+        @app_name ||= config[:app_name]
+      end
+
+      def indent(level = 0, initial = '')
+        (1..level)
+          .to_a.reduce('') { |result, _| result + '    ' }
+          .concat(initial)
+      end
+
+      def clean_enum_name(name)
+        clean_name = name
+                     .split(/[^0-9a-zA-Z]/)
+                     .reject(&:empty?)
+                     .map(&:capitalize)
+                     .join
+
+        escape_with_underscore_if_needed(clean_name)
+      end
+
+      def clean_case_name(name)
+        clean_variable_name(name)
+      end
+
+      def clean_variable_name(name)
+        clean_name = name
+                     .split(/[^0-9a-zA-Z]/)
+                     .reject(&:empty?)
+                     .each_with_index
+                     .map { |v, i| i.zero? ? v.tap { |char| char[0] = char[0].downcase } : v.capitalize }
+                     .join
+
+        escaped_underscore = escape_with_underscore_if_needed(clean_name)
+        escape_keyword_if_needed(escaped_underscore)
+      end
+
+      def escape_with_underscore_if_needed(name)
+        return name if name.match(/^[A-Za-z_]/)
+
+        '_' + name
+      end
+
+      def escape_keyword_if_needed(name)
+        return name unless ESCAPE_KEYWORDS.include? name
 
-        def bundles
-          @bundles ||= config[:bundles]
-        end
-  
-        def indent(level = 0, initial = "")
-          (1..level)
-            .to_a.reduce("") { |result, value| result + "    " }
-            .concat(initial)
-        end
-  
-        def clean_enum_name(name)
-          clean_name = name
-            .split(/[^0-9a-zA-Z]/)
-            .reject { |c| c.empty? }
-            .map { |value| value.capitalize }
-            .join
-  
-          escape_with_underscore_if_needed(clean_name)
-        end
-  
-        def clean_case_name(name)
-          clean_variable_name(name)
-        end
-  
-        def clean_variable_name(name)
-          clean_name = name
-            .split(/[^0-9a-zA-Z]/)
-            .reject { |c| c.empty? }
-            .each_with_index
-            .map { |value, index| index == 0 ? value.downcase : value.capitalize }
-            .join
-  
-          escaped_underscore = escape_with_underscore_if_needed(clean_name)
-          escape_keyword_if_needed(escaped_underscore)
-        end
-        
-        def escape_with_underscore_if_needed(name)
-          return name if name.match(/^[A-Za-z_]/)
-          "_" + name
-        end
-  
-        def escape_keyword_if_needed(name)
-          return name if !ESCAPE_KEYWORDS.include? name
-          "`#{name}`"
-        end
-  
+        "`#{name}`"
       end
     end
-  end
+  end
+end