module Mirah
  module JVM
    module Types
      class Type
        java_import 'org.mirah.typer.ErrorType'
        java_import 'org.mirah.typer.SpecialType'
        java_import 'org.mirah.typer.InlineCode'
        java_import 'org.mirah.typer.ResolvedType'

        class SpecialType
          def full_name
            name
          end
        end

        include Mirah::JVM::MethodLookup
        include Mirah::Logging::Logged

        include ResolvedType
        begin
          java_import 'org.mirah.jvm.types.JVMType'
          include JVMType
        rescue NameError
          $CLASSPATH << File.dirname(__FILE__) + '/../../../../javalib/mirah-compiler.jar'
          begin
            java_import 'org.mirah.jvm.types.JVMType'
            include JVMType
          rescue
  	        puts "Unable to load new type interface"
          end
        end

        attr_reader :name, :type_system
        attr_writer :inner_class

        def initialize(type_system, mirror_or_name)
          @type_system = type_system
          raise ArgumentError if type_system.nil?
          if mirror_or_name.kind_of?(BiteScript::ASM::ClassMirror)
            @type = mirror_or_name
            @name = mirror_or_name.type.class_name
          else
            @name = mirror_or_name.to_s
          end
          raise ArgumentError, "Bad type #{mirror_or_name}" if name =~ /Java::/
        end

        def full_name
          desc = BiteScript::Signature.class_id(self)
          BiteScript::ASM::Type.get_type(desc).class_name
        end
        
        def internal_name
          full_name.tr('.', '/')
        end
        def class_id
          BiteScript::Signature.class_id(self)
        end
        def getAsmType
          BiteScript::ASM::Type.get_type(class_id)
        end
        def isAnnotation
          jvm_type ? jvm_type.annotation? : false
        end
        def isEnum
          jvm_type ? jvm_type.enum? : false
        end
        def flags
          flags = BiteScript::ASM::Opcodes::ACC_PUBLIC
          flags |= BiteScript::ASM::Opcodes::ACC_ANNOTATION if isAnnotation
          flags |= BiteScript::ASM::Opcodes::ACC_ENUM if isEnum
          flags |= BiteScript::ASM::Opcodes::ACC_INTERFACE if self.interface?
          flags |= BiteScript::ASM::Opcodes::ACC_ABSTRACT if self.abstract?
          flags
        end
        def retention
          if jvm_type.respond_to?(:getDeclaredAnnotation)
            retention = jvm_type.getDeclaredAnnotation('java.lang.annotation.Retention')
            retention.value.name
          else
            nil
          end
        end

        def jvm_type
          @type
        end

        def void?
          false
        end

        def meta?
          false
        end
        def isMeta
          self.meta?
        end


        def generic?
          false
        end
        def isGeneric
          self.generic?
        end

        def isBlock
          false
        end

        def array?
          false
        end
        def isArray
          self.array?
        end

        def primitive?
          false
        end
        def isPrimitive
          self.primitive?
        end

        def interface?
          # FIXME: Don't do rescue nil. Figure out a cleaner way to handle
          # mirrors for all incoming types without blowing up on e.g. 'boolean' or 'int'
          (@type || BiteScript::ASM::ClassMirror.for_name(@name)).interface? rescue nil
        end

        def abstract?
          (@type || BiteScript::ASM::ClassMirror.for_name(@name)).abstract? rescue nil
        end

        def dynamic?
          false
        end

        def inner_class?
          @inner_class
        end

        def is_parent(other)
          assignable_from?(other)
        end

        def compatible?(other)
          assignable_from?(other)
        end

        def error?
          false
        end
        def isError
          false
        end

        def assignable_from?(other)
          return false unless other
          return true if !primitive? && other.kind_of?(NullType)
          return true if other == self
          return true if other.matchesAnything
          return true if other.kind_of?(InlineCode)

          return interface? || abstract? if other.isBlock

          return true if jvm_type && (jvm_type == other.jvm_type)

          return assignable_from?(other.ungeneric) if other.generic?

          return other.convertible_to?(self) if other.primitive?

          assignable_from?(other.superclass) ||
              other.interfaces.any? {|i| assignable_from?(i)}
        end

        def assignableFrom(other)
          assignable_from?(other)
        end

        def widen(other)
          return self if assignable_from?(other)
          common_parent = (ancestors_and_interfaces & ([other] + other.ancestors_and_interfaces))[0]
          common_parent || ErrorType.new([["Incompatible types #{self} and #{other}."]])
        end

        def matchesAnything; false; end

        def iterable?
          ['java.lang.Iterable',
           'java.util.Iterator',
           'java.util.Enumeration'].any? {|n| @type_system.type(nil, n).assignable_from(self)}
        end

        def component_type
          @type_system.type(nil, 'java.lang.Object') if iterable?
        end

        def meta
          @meta ||= MetaType.new(self)
        end

        def unmeta
          self
        end

        def generic
          @generic ||= GenericType.new(self)
        end

        def ungeneric
          self
        end

        def type_parameters
          return nil unless jvm_type
          jvm_type.type_parameters
        end

        def basic_type
          self
        end

        def array_type
          @array_type ||= Mirah::JVM::Types::ArrayType.new(self)
        end

        def prefix
          'a'
        end

        # is this a 64 bit type?
        def wide?
          false
        end

        def inspect(indent=0)
          "#{' ' * indent}#<#{self.class.name.split(/::/)[-1]} #{name}>"
        end

        def to_s
          inspect
        end

        def newarray(method)
          method.anewarray(self)
        end

        def pop(method)
          if wide?
            method.pop2
          else
            method.pop
          end
        end

        def superclass
          raise "Incomplete type #{self}" unless jvm_type
          @type_system.type(nil, jvm_type.superclass) if jvm_type.superclass
        end

        def interfaces(include_parent=true)
          raise "Incomplete type #{self} (#{self.class})" unless jvm_type
          @interfaces ||= begin
            interfaces = jvm_type.interfaces.map {|i| @type_system.type(nil, i)}.to_set
            if superclass && include_parent
              interfaces |= superclass.interfaces
            end
            interfaces.to_a
          end
          @interfaces
        end

        def ancestors_and_interfaces
          if self.primitive?
            []
          else
            ancestors = []
            get_ancestors = lambda {|c| [c.superclass] + c.interfaces(false)}
            new_ancestors = get_ancestors.call(self)
            until new_ancestors.empty?
              klass = new_ancestors.shift
              next if klass.nil? || klass.name == 'java.lang.Object' || klass.isError
              ancestors << klass
              new_ancestors.concat(get_ancestors.call(klass))
            end
            ancestors << @type_system.type(nil, 'java.lang.Object')
            ancestors
          end
        end

        def astore(builder)
          if primitive?
            builder.send "#{name[0,1]}astore"
          else
            builder.aastore
          end
        end

        def aload(builder)
          if primitive?
            builder.send "#{name[0,1]}aload"
          else
            builder.aaload
          end
        end
      end
    end
  end
end