class Cpe23

# File lib/cpe23.rb, line 202
def attr_match?(first, second)
  first == '*' || second == '*' || first == second
end

# File lib/cpe23.rb, line 118
def initialize(part: nil, vendor: nil, product: nil, version: nil,
               update: nil, edition: nil, language: nil, sw_edition: nil,
               target_sw: nil, target_hw: nil, other: nil)
  @part = part
  @vendor = vendor
  @product = product
  @version = version
  @update = update
  @edition = edition
  @language = language
  @sw_edition = sw_edition
  @target_sw = target_sw
  @target_hw = target_hw
  @other = other
end

# File lib/cpe23.rb, line 189
def parse(str)
  str = str.strip
  if str.start_with? 'wfn:'
    parse_wfn(str)
  elsif str.start_with? 'cpe:/'
    parse_uri(str)
  elsif str.start_with? 'cpe:2.3:'
    parse_str(str)
  else
    raise ArgumentError, 'CPE malformed'
  end
end

# File lib/cpe23.rb, line 252
def parse_cpe23(str)
  raise ArgumentError, 'Not a CPE str' unless str.start_with?('cpe:2.3')

  index = 7
  size = str.size
  char = str[index]
  attr = 11.times.map do
    raise ArgumentError, 'CPE formatted string malformed' if char != ':'

    index += 1
    attr_index = index
    until index >= size || (char = str[index]) == ':'
      index += 1
      index += 1 if char == '\\' # Skip escaped characters
    end
    str[attr_index...index]
  end
  raise ArgumentError, 'CPE formatted string malformed' if index != size

  attr
end

# File lib/cpe23.rb, line 274
def parse_str(str)
  # attr = Cpe23.parse(str)[:attr].map(&:value)
  attr = parse_cpe23(str)
  data = {}
  data[:part], data[:vendor], data[:product], data[:version],
    data[:update], data[:edition], data[:language], data[:sw_edition],
    data[:target_sw], data[:target_hw], data[:other], remainder = attr

  # All attributes MUST appear
  if data.any? { |_k, v| v.nil? } || !remainder.nil?
    raise ArgumentError, 'CPE formatted string malformed'
  end

  # Remove empty attributes to avoid confusing the constructor
  data.reject! { |_k, v| v.empty? }

  new(**data)
end

# File lib/cpe23.rb, line 234
def parse_uri(str)
  tag, body = str.split(':/', 2)
  raise ArgumentError, 'Not a CPE URI' if tag != 'cpe' || body.nil?

  body.sub!('%01', '?')
  body.sub!('%02', '*')

  data = {}
  data[:part], data[:vendor], data[:product], data[:version], data[:update],
    data[:edition], data[:language], remainder = body.split(':')

  raise ArgumentError, 'CPE URI malformed' unless remainder.nil?

  # All attributes are optional.
  new(**data)
end

# File lib/cpe23.rb, line 208
def parse_wfn(str)
  data = {}
  WFN.parse(str)[:attr].each do |attr|
    key = attr.capture(:symbol).value.to_sym
    value = attr.capture(:value).then do |val|
      if (str = val.capture(:string))
        str.capture(:content).value
      else
        # Translate WFN special values (only applies to non-string)
        case (str = val.value)
        when 'ANY' then '*'
        when 'NA' then nil
        else str
        end
      end
    end
    if data.include? key
      raise ArgumentError, 'Attribute defined multiple times'
    end

    data[key] = value
  end

  new(**data)
end

# File lib/cpe23.rb, line 134
def <=>(other)
  unless other.is_a? Cpe23
    begin
      other = Cpe23.parse(other)
    rescue StandardError
      return nil
    end
  end
  return nil unless
    Cpe23.attr_match?(part, other.part) &&
    Cpe23.attr_match?(vendor, other.vendor) &&
    Cpe23.attr_match?(product, other.product) &&
    Cpe23.attr_match?(update, other.update) &&
    Cpe23.attr_match?(edition, other.edition) &&
    Cpe23.attr_match?(language, other.language) &&
    Cpe23.attr_match?(target_sw, other.target_sw) &&
    Cpe23.attr_match?(target_hw, other.target_hw) &&
    Cpe23.attr_match?(self.other, other.other)

  version <=> other.version
end

# File lib/cpe23.rb, line 181
def to_str
  ['cpe', '2.3', @part, @vendor, @product, @version, @update, @edition,
   @language, @sw_edition, @target_sw, @target_hw, @other].join(':').downcase
end

# File lib/cpe23.rb, line 170
def to_uri
  fields = [@part, @vendor, @product, @version, @update, @edition, @language]
  # Strip trailing empty fields
  fields = fields[0...-1] while fields.any? && fields[-1].nil?
  fields.map! do |f|
    f.sub('?', '%01')
     .sub('*', '%02')
  end
  'cpe:/' + fields.join(':').downcase
end

# File lib/cpe23.rb, line 156
def to_wfn
  attrs = %i[part vendor product version update edition language sw_edition
             target_sw target_hw other].map do |key|
    value = instance_variable_get("@#{key}")
    str = case value
          when nil then 'NA'
          when '*' then 'ANY'
          else "\"#{value.downcase}\""
          end
    "#{key}=#{str}"
  end
  "wfn:[#{attrs.join(',')}]"
end

# File lib/cpe23.rb, line 46
def version
  Cpe23::Version.new(@version)
end

# File lib/cpe23.rb, line 42
def version_raw
  @version
end