version_info.rb in rake-funnel-0.4.0.pre

- old
+ new
@@ -1,72 +1,103 @@
-require 'ostruct'
-
-module Rake::Funnel::Support
-  class VersionInfo < OpenStruct
-    include Enumerable
-
-    def initialize(hash = nil)
-      super(hash)
-      freeze
-    end
-
-    def each(&block)
-      each_pair(&block)
-    end
-
-    class << self
-      def parse(context)
-        VersionInfo.new({
-            assembly_version: assembly_version(context),
-            assembly_file_version: assembly_file_version(context),
-            assembly_informational_version: assembly_informational_version(context)
-          })
-      end
-
-      def read_version_from(file)
-        File.open(file, &:readline).strip
-      end
-
-      private
-      def default_version(context)
-        context[:version] || '0'
-      end
-
-      def assembly_version(context)
-        strip_trailing_non_numeric(default_version(context))
-      end
-
-      def assembly_file_version(context)
-        numeric_build_number = strip_leading_non_numeric(context[:build_number])
-
-        [
-          assembly_version(context),
-          numeric_build_number
-        ].compact.join('.')
-      end
-
-      def assembly_informational_version(context)
-        build_number = context[:build_number]
-        join_using = '.'
-        join_using = '' if build_number =~ /^\D/
-
-        prefix = [default_version(context), build_number].compact.join(join_using)
-        sha = context[:sha]
-
-        [prefix, sha].compact.join('-')
-      end
-
-      def strip_trailing_non_numeric(str)
-        return nil if str.nil?
-        str.to_s.gsub(/[^\d\.].*/, '')
-      end
-
-      def strip_leading_non_numeric(str)
-        return nil if str.nil?
-        str = str.to_s.gsub(/[^\d\.]/, '')
-
-        return nil if str.empty?
-        str
-      end
-    end
-  end
-end
+require 'ostruct'
+
+module Rake
+  module Funnel
+    module Support
+      class VersionInfo < OpenStruct
+        include Enumerable
+
+        def initialize(hash = nil)
+          super(hash)
+          freeze
+        end
+
+        def each(&block)
+          each_pair(&block)
+        end
+
+        class << self
+          def parse(context)
+            VersionInfo.new({
+                assembly_version: assembly_version(context),
+                assembly_file_version: assembly_file_version(context),
+                assembly_informational_version: assembly_informational_version(context)
+              })
+          end
+
+          def read_version_from(file)
+            File.open(file, &:readline).strip
+          end
+
+          private
+          def default_version(context)
+            context[:version].to_s || '0'
+          end
+
+          def pad(version, parts)
+            numerics = version.split('.').take(parts).map { |part| part.to_i }
+
+            template = Array.new(parts) { 0 }.map.with_index do |part, index|
+              numerics[index] || part
+            end
+            template.join('.')
+          end
+
+          def assembly_version(context)
+            version = default_version(context)
+            pad(version, 4)
+          end
+
+          def assembly_file_version(context)
+            version = assembly_version(context)
+            build_number = numeric(context.fetch(:metadata, {})[:build])
+            return version.sub(/\.0$/, ".#{build_number}") if build_number
+            version
+          end
+
+          def numeric(str)
+            return str if str.to_s =~ /^\d+$/
+            nil
+          end
+
+          def assembly_informational_version(context)
+            version = default_version(context)
+            numeric_version = pad(version, 3)
+            alpha_version = version.sub(/^[\d\.]*/, '')
+
+            semver = [
+              numeric_version,
+              alpha_version,
+              pre(context),
+            ].join
+
+            [
+              semver,
+              metadata(context)
+            ].compact.join('+')
+          end
+
+          def pre(context)
+            pre = context.fetch(:metadata, {})[:pre]
+
+            if pre && pre.to_s !~ /^-/
+              pre = "-#{pre}"
+            end
+
+            pre
+          end
+
+          def metadata(context)
+            metadata = context.fetch(:metadata, {}).reject { |k, _| k == :pre }
+
+            metadata = metadata.map { |key, value|
+              [key.to_s, value.to_s] if value
+            }.compact
+
+            return nil if metadata.empty?
+            metadata.join('.')
+          end
+        end
+      end
+    end
+  end
+end