squared 0.5.23 → 0.6.0

data/lib/squared/common/system.rb

@@ -6,10 +6,28 @@ require 'rake'
 module Squared
   module Common
     module System
+      class << self
+        private
+
+        def parse_link(val)
+          case val
+          when ::TrueClass, 's'
+            1
+          when 'r'
+            2
+          when 'h'
+            3
+          else
+            raise ArgumentError, "unrecognized 'link' flag: #{val}" if val
+
+            0
+          end
+        end
+      end
+
       module_function
 
       def shell(*args, name: :system, **kwargs)
-        kwargs.delete(:exception) unless name == :system
         if RUBY_ENGINE == 'jruby' && Rake::Win32.windows?
           e = kwargs[:exception]
           if (dir = kwargs[:chdir]) && ((pwd = Dir.pwd) != dir)
@@ -25,16 +43,16 @@ module Squared
         else
           return Kernel.send(name, *args, **kwargs)
         end
-        return ret unless e && !ret
+        return ret unless e && !ret && name == :system
 
         raise $?.to_s
       end
 
-      def copy_dir(src, dest, glob = ['**/*'], create: false, link: nil, force: false, pass: nil, hidden: false,
-                   verbose: true)
+      def copy_dir(src, dest, glob = ['**/*'], create: false, link: nil, preserve: nil, force: false, verbose: true,
+                   pass: nil, hidden: false)
         base = Pathname.new(src)
         target = Pathname.new(dest)
-        raise "#{target.cleanpath} (not found)" if !create && !target.parent.exist?
+        raise Errno::ENOENT, dest.cleanpath.to_s unless create || target.parent.exist?
 
         subdir = {}
         target.mkpath if create
@@ -58,53 +76,67 @@ module Squared
         end
         count = 0
         soft = 0
+        type = System.send :parse_link, link
         subdir.each do |dir, files|
-          if link
-            files.dup.tap do |items|
-              files.clear
-              items.each do |file|
-                if file.exist?
-                  if !file.symlink?
-                    files << file
-                  elsif !force
-                    next
-                  end
-                end
-                if link == 'hard'
-                  FileUtils.ln(file, dir, force: force, verbose: false)
-                else
-                  FileUtils.ln_s(file, dir, force: force, verbose: false)
+          unless type == 0
+            items = files.dup
+            files.clear
+            items.each do |file|
+              if file.exist?
+                if !file.symlink?
+                  files << file
+                elsif !force
+                  next
                 end
-                soft += 1
               end
+              case type
+              when 1
+                FileUtils.ln_s(file, dir, force: force, verbose: false)
+              when 2
+                FileUtils.ln_s(file.relative_path_from(dir), dir, force: force, verbose: false)
+              else
+                FileUtils.ln(file, dir, force: force, verbose: false)
+              end
+              soft += 1
             end
           end
           next if files.empty?
 
-          out = FileUtils.cp(files, dir, verbose: false)
+          out = FileUtils.cp(files, dir, preserve: preserve, verbose: false)
           count += out.size
         end
         puts [target.realpath, subdir.size, soft > 0 ? "#{count}+#{soft}" : count].join(' => ') if verbose
       end
 
-      def copy_guard(src, dest, link: nil, force: false, verbose: true)
-        unless force
-          target = Pathname.new(dest)
-          if target.directory?
-            src = Array(src).reject { |val| target.join(File.basename(val)).exist? }
-            return if src.empty?
-          elsif target.exist?
-            return
+      def copy_guard(*src, dest, base: '.', create: false, link: nil, preserve: nil, force: false, verbose: true)
+        src = src.compact.flatten
+        dest = Pathname.new(dest).realdirpath
+        base = Pathname.new(base).realpath
+        dir = if dest.directory?
+                true
+              elsif src.size > 1
+                raise Errno::ENOENT, dest.cleanpath.to_s unless create && !dest.exist?
+
+                dest.mkpath
+                true
+              end
+        targets = src.map! { |file| [base + file, dir ? dest + File.basename(file) : dest] }
+        return if !force && (targets = targets.reject { |to| to[1].exist? }).empty?
+
+        type = System.send :parse_link, link
+        targets.each do |file, to|
+          case type
+          when 0
+            FileUtils.cp(file, to, preserve: preserve, verbose: verbose)
+          when 1
+            FileUtils.ln_s(file, to, force: force, verbose: verbose)
+          when 2
+            FileUtils.ln_s(file.relative_path_from(dir ? to.dirname : to), to, force: force, verbose: verbose)
+          else
+            FileUtils.ln(file, to, force: force, verbose: verbose)
           end
         end
-        case link
-        when 'hard', 1
-          FileUtils.ln(src, dest, force: force, verbose: verbose)
-        when ::TrueClass, 'soft', 0
-          FileUtils.ln_s(src, dest, force: force, verbose: verbose)
-        else
-          FileUtils.cp(src, dest, verbose: verbose)
-        end
+        nil
       end
     end
   end

data/lib/squared/common/utils.rb

@@ -9,21 +9,44 @@ module Squared
     module Utils
       module_function
 
-      def split_escape(val, char: ',')
-        val.split(/\s*(?<!\\)#{char}\s*/)
+      def as_a(obj, *meth, flat: nil, compact: false, &blk)
+        return [] if obj.nil?
+
+        unless obj.is_a?(::Array)
+          obj = if obj.respond_to?(:to_ary)
+                  obj.to_ary
+                elsif obj.respond_to?(:to_a) && !obj.is_a?(::Hash) && (val = obj.to_a).is_a?(::Array)
+                  val
+                else
+                  [obj]
+                end
+        end
+        obj = flat.is_a?(::Numeric) ? obj.flatten(flat) : obj.flatten if flat
+        obj = obj.compact if compact
+        obj = obj.map(&meth.shift) until meth.empty?
+        return obj unless block_given?
+
+        obj.select(&blk)
+      end
+
+      def split_escape(val, char: ',', &blk)
+        ret = val.split(/\s*(?<!\\)#{char}\s*/)
+        return ret unless block_given?
+
+        ret.each(&blk)
       end
 
       def split_option(val)
         val = val.strip
         return [val, '', ''] unless (i = val.index('='))
 
-        last = val[(i + 1)..-1].strip
+        last = val[i.succ..-1].strip
         quote = ''
         if last =~ /\A(["'])(.+)\1\z/
           last = $2
           quote = $1
         end
-        [val[0..(i - 1)], last, quote]
+        [val[0..i.pred], last, quote]
       end
 
       def task_invoke(*cmd, args: [], exception: true, warning: true)
@@ -105,9 +128,9 @@ module Squared
         end
       end
 
-      def env(key, default = nil, suffix: nil, strict: false, equals: nil, ignore: nil)
+      def env(key, default = nil, suffix: nil, strict: false, equals: nil, ignore: nil, **)
         ret = env_value(key, suffix: suffix, strict: strict)
-        return ret == equals.to_s unless equals.nil?
+        return Array(equals).any? { |val| val.to_s == ret } unless equals.nil?
 
         ret.empty? || (ignore && Array(ignore).any? { |val| val.to_s == ret }) ? default : ret
       end

data/lib/squared/config.rb

@@ -12,18 +12,19 @@ module Squared
       include Rake::DSL
 
       class << self
-        def parse(gem, namespace, ext = [gem])
+        def parse(gem, namespace, ext = [pkg])
           require gem
-          [eval(namespace), Array(ext)].tap do |data|
-            data.last.each { |key| @@mime_obj[key] = data }
-          end
+          obj = eval namespace
+          Array(ext).each { |val| @@mime_obj[val] = [obj, ext] }
         rescue LoadError, NameError => e
           warn e
           nil
+        else
+          @@mime_obj[ext.first]
         end
 
-        def link(project, main = project.dependfile&.basename, name = nil, **kwargs, &blk)
-          return unless project.enabled? && main
+        def link(project, main = project.dependfile.basename, name = nil, **kwargs, &blk)
+          return unless project.enabled?
 
           ret = new(main, name, project: project, **kwargs)
           ret.instance_eval(&blk) if block_given?
@@ -85,7 +86,7 @@ module Squared
                       ['path not found', realpath]
                     else
                       @required = true
-                      project ? [project, 'not found'] : %w[name missing]
+                      project ? [project, 'missing'] : %w[name missing]
                     end
         warn log_message(Logger::WARN, msg, subject: self.class, hint: hint)
       end
@@ -105,7 +106,7 @@ module Squared
                   next unless (data = Viewer.parse(type, type.upcase, ext))
                 end
                 obj, ext = data
-                target = file || target? ? file || realpath : nil
+                target = file || (realpath if target?)
 
                 task_desc(command, *ext, target: target)
                 task type, [:keys] do |_, args|
@@ -159,7 +160,7 @@ module Squared
       end
 
       def also(path, type = nil, name: nil, **kwargs)
-        return self unless (file = basepath(path)).exist? && !@mime.frozen?
+        return self if @mime.frozen? || !(file = basepath(path)).exist?
 
         ext = mimetype file
         type ||= ext
@@ -204,7 +205,7 @@ module Squared
 
       def read_keys(reader, type, file, keys, ext: [type], opts: {})
         if file && (mime = mimetype(file)) && basepath(file).exist?
-          raise_error(file, mime, hint: 'invalid') unless ext.include?(mime)
+          raise_error file, mime, hint: 'invalid' unless ext.include?(mime)
         else
           if ext.include?(mime)
             alt = file
@@ -219,7 +220,7 @@ module Squared
             args = { hint: 'no keys' }
           end
           unless file
-            args ||= { hint: 'not found', kind: LoadError }
+            args ||= { kind: Errno::ENOENT }
             raise_error(reader.name, "#{File.basename(alt, '.*')}.#{ext.first}", **args)
           end
         end
@@ -236,22 +237,18 @@ module Squared
         title = Pathname.new(file)
                         .realpath
                         .to_s
-                        .sub(/\A#{Regexp.escape(File.join(Dir.pwd, ''))}/, '')
+                        .sub(/^#{Regexp.escape(File.join(Dir.pwd, ''))}/, '')
         emphasize(lines, title: title, sub: unless stdin?
                                               [
-                                                { pat: /\A((?:[^:]|(?<! ):(?! ))+)\z/, styles: theme[:banner] },
-                                                { pat: /\A(.*?)(<[^>]+>)(.+)\z/m, styles: theme[:undefined], index: 2 },
-                                                { pat: /\A((?~ : ))( : (?!undefined).+)\z/m, styles: theme[:key] },
-                                                { pat: /\A((?~: ): )(-?[\d.]+)(\s*)\z/m, styles: theme[:number],
-                                                  index: 2 },
-                                                { pat: /\A((?~: ): ")(.+)("\s*)\z/m, styles: theme[:string], index: 2 },
-                                                { pat: /\A((?~: ): \{)(.+)(}\s*)\z/m, styles: theme[:hash], index: 2 },
-                                                { pat: /\A((?~: ): \[)(.+)(\]\s*)\z/m, styles: theme[:array],
-                                                  index: 2 },
-                                                { pat: /\A((?~: ): )(true|false)(\s*)\z/m, styles: theme[:boolean],
-                                                  index: 2 },
-                                                { pat: /\A((?~: ): (?!undefined))([^"\[{].*)\z/m, styles: theme[:value],
-                                                  index: 2 }
+                                                opt_style(theme[:banner], /\A((?:[^:]|(?<! ):(?! ))+)\z/),
+                                                opt_style(theme[:undefined], /\A(.*?)(<[^>]+>)(.+)\z/m, 2),
+                                                opt_style(theme[:key], /\A((?~ : ))( : (?!undefined).+)\z/m),
+                                                opt_style(theme[:number], /\A((?~: ): )(-?[\d.]+)(\s*)\z/m, 2),
+                                                opt_style(theme[:string], /\A((?~: ): ")(.+)("\s*)\z/m, 2),
+                                                opt_style(theme[:hash], /\A((?~: ): \{)(.+)(\}\s*)\z/m, 2),
+                                                opt_style(theme[:array], /\A((?~: ): \[)(.+)(\]\s*)\z/m, 2),
+                                                opt_style(theme[:boolean], /\A((?~: ): )(true|false)(\s*)\z/m, 2),
+                                                opt_style(theme[:value], /\A((?~: ): (?!undefined))([^"\[{].*)\z/m, 2)
                                               ]
                                             end, border: theme[:border])
       end
@@ -262,7 +259,7 @@ module Squared
         symbolize = opts[:symbolize_names]
         keys.each do |key|
           begin
-            items = key.split('.').flat_map { |name| name =~ /^(.+)\[(\d+)\]$/ ? [$1, $2.to_i] : name }
+            items = key.split('.')
             items = items.map(&:to_sym) if symbolize
             val = data.dig(*items)
             if val.nil?
@@ -293,7 +290,7 @@ module Squared
         if stdin?
           puts out.map!(&:last).join("\n")
         else
-          out.map! { |a, b| '%-*s : %s' % [pad, a, b] }
+          out.map! { |item| '%-*s : %s' % [pad, item[0], item[1]] }
         end
       end
 
@@ -308,7 +305,7 @@ module Squared
       def task_desc(command, *ext, target: nil)
         return unless Rake::TaskManager.record_task_metadata
 
-        val = "#{ext.first}[#{target ? '' : "file?=#{File.basename(main)}.#{ext.last},"}keys+]"
+        val = "#{ext.first}[#{"file?=#{File.basename(main)}.#{ext.last}," if target}keys+]"
         args = *name.split(':').push(command, val)
         if project
           project.workspace.task_desc(*args)
@@ -322,9 +319,7 @@ module Squared
       end
 
       def warning?
-        return true unless project
-
-        project.workspace.warning
+        project ? project.workspace.warning : true
       end
 
       def stdin?
@@ -349,9 +344,7 @@ module Squared
       end
 
       def basepath(*args)
-        return Pathname.pwd.join(*args) unless project
-
-        project.basepath(*args)
+        project ? project.basepath(*args) : Pathname.pwd.join(*args)
       end
     end
   end

data/lib/squared/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Squared
-  VERSION = '0.5.23'
+  VERSION = '0.6.0'
 end