require "goon_model_gen"

require "goon_model_gen/golang/modifier"

require "active_support/core_ext/string"

module GoonModelGen
  module Golang
    class Packages < Array

      def name_to_type_map
        each_with_object({}) do |pkg, d|
          d.update(pkg.name_to_type_map)
        end
      end

      def dup
        Packages.new.add(*self)
      end

      def add(*packages)
        packages.each do |i|
          self << i
        end
        self
      end

      def detect_by(basename)
        detect{|pkg| pkg.basename == basename}
      end

      def resolve_type_names(extra_packages = [])
        candidates = dup.add(*extra_packages)
        map(&:types).flatten.each do |t|
          t.resolve(candidates)
        end
        return self
      end

      # @param type_name [string]
      # @param [Type]
      def lookup(type_name)
        if type_name.include?('.')
          pkg_name, name = type_name.split('.', 2)
          pkg = detect_by(pkg_name)
          return pkg ? pkg.lookup(name) : nil
        else
          each do |pkg|
            t = pkg.lookup(type_name)
            return t if t
          end
          return nil
        end
      end

      def type_for(expression)
        return nil if expression.blank?
        Modifier.parse(expression) do |name|
          lookup(name)
        end
      end
    end
  end
end