tf/semmy

module Semmy
  module VersionString
    extend self

    class SuffixNotFound < Error; end

    class UnexpectedSuffix < Error; end

    class NoPreviousMinor < Error; end

    def remove_suffix(version, suffix)
      new_version = version.dup

      unless new_version.gsub!(/.#{suffix}$/, '')
        fail(SuffixNotFound, "Suffix #{suffix} not found in #{version}")
      end

      new_version
    end

    def bump_major(version, suffix)
      [bump(0, version), suffix].join('.')
    end

    def bump_minor(version, suffix)
      [bump(1, version), suffix].join('.')
    end

    def bump_patch_level(version)
      bump(2, version)
    end

    def minor_only(version)
      version.split('.')[0..1].join('.')
    end

    def patch_level?(version)
      components(version)[:patch].to_i > 0
    end

    def components(version)
      components = version.split('.')

      {
        major: components[0],
        minor: components[1],
        patch: components[2]
      }
    end

    def previous_version(version)
      components = version.split('.').map(&:to_i)

      if components[2] > 0
        components[2] -= 1
      elsif components[1] > 0
        components[1] -= 1
      else
        fail(NoPreviousMinor, "Cannot get previous version of #{version}.")
      end

      components.join('.')
    end

    def stable_branch_name(version, stable_branch_name_pattern)
      stable_branch_name_pattern % VersionString.components(version)
    end

    def previous_stable_branch_name(version, stable_branch_name_pattern)
      stable_branch_name_pattern % previous_minor_version_components(version)
    end

    private

    def bump(component_index, version)
      components = version.split('.')[0..2].map!(&:to_i)

      [
        components[0...component_index],
        components[component_index] + 1,
        [0] * (2 - component_index)
      ].flatten.join('.')
    end

    def previous_minor_version_components(version)
      components = version.split('.').map(&:to_i)

      if components[1].zero?
        {
          major: components[0] - 1,
          minor: 'x'
        }
      else
        {
          major: components[0],
          minor: components[1] - 1
        }
      end
    end
  end
end