# encoding: utf-8

require 'stringio'
require_relative '../../hocon/impl'
require_relative '../../hocon/impl/token'
require_relative '../../hocon/impl/token_type'
require_relative '../../hocon/impl/config_number'
require_relative '../../hocon/impl/config_string'
require_relative '../../hocon/impl/config_null'
require_relative '../../hocon/impl/config_boolean'
require_relative '../../hocon/config_error'
require_relative '../../hocon/impl/resolve_status'
require_relative '../../hocon/config_value_type'

# FIXME the way the subclasses of Token are private with static isFoo and accessors is kind of ridiculous.
class Hocon::Impl::Tokens
  Token = Hocon::Impl::Token
  TokenType = Hocon::Impl::TokenType
  ConfigNumber = Hocon::Impl::ConfigNumber
  ConfigInt = Hocon::Impl::ConfigInt
  ConfigDouble = Hocon::Impl::ConfigDouble
  ConfigString = Hocon::Impl::ConfigString
  ConfigNull = Hocon::Impl::ConfigNull
  ConfigBoolean = Hocon::Impl::ConfigBoolean
  ResolveStatus = Hocon::Impl::ResolveStatus

  ConfigBugOrBrokenError = Hocon::ConfigError::ConfigBugOrBrokenError

  class Value < Token
    def initialize(value, orig_text = nil)
      super(TokenType::VALUE, value.origin, orig_text)
      @value = value
    end

    attr_reader :value

    def to_s
      if value.resolve_status == ResolveStatus::RESOLVED
        "'#{value.unwrapped}' (#{Hocon::ConfigValueType.value_type_name(value.value_type)})"
      else
        "'<unresolved value>' ((#{Hocon::ConfigValueType.value_type_name(value.value_type)})"
      end

    end

    def can_equal(o)
      o.is_a?(Value)
    end

    def ==(other)
      super(other) && other.value == @value
    end

    def hash
      41 * (41 + super) + value.hash
    end
  end

  class Line < Token
    def initialize(origin)
      super(TokenType::NEWLINE, origin)
    end

    def can_equal(other)
      o.is_a?(Line)
    end

    def ==(other)
      super(other) && other.line_number == line_number
    end

    def hash
      41 * (41 + super) + line_number
    end

    def token_text
      "\n"
    end
  end

  class UnquotedText < Token
    def initialize(origin, s)
      super(TokenType::UNQUOTED_TEXT, origin)
      @value = s
    end
    attr_reader :value

    def to_s
      "'#{value}'"
    end

    def can_equal(o)
      o.is_a?(UnquotedText)
    end

    def ==(other)
      super(other) && other.value == @value
    end

    def hash
      41 * (41 + super) + value.hash
    end

    def token_text
      @value
    end
  end

  class IgnoredWhitespace < Token
    def initialize(origin, s)
      super(TokenType::IGNORED_WHITESPACE, origin)
      @value = s
    end
    attr_reader :value

    def to_s
      "'#{value}' (WHITESPACE)"
    end

    def can_equal(o)
      o.is_a?(IgnoredWhitespace)
    end

    def ==(other)
      super(other) && other.value == value
    end

    def hash
      41 * (41 + super) + value.hash
    end

    def token_text
      @value
    end
  end

  class Problem < Token
    def initialize(origin, what, message, suggest_quotes, cause)
      super(TokenType::PROBLEM, origin)
      @what = what
      @message = message
      @suggest_quotes = suggest_quotes
      @cause = cause
    end

    def what
      @what
    end

    def message
      @message
    end

    def suggest_quotes
      @suggest_quotes
    end

    def cause
      @cause
    end

    def to_s
      sb = StringIO.new
      sb << "'"
      sb << what
      sb << "'"
      sb << " ("
      sb << message
      sb << ")"
      sb.string
    end

    def can_equal(other)
      other.is_a?(Problem)
    end

    def ==(other)
      super(other) && other.what == @what &&
          other.message == @message &&
          other.suggest_quotes == @suggest_quotes &&
          Hocon::Impl::ConfigImplUtil.equals_handling_nil?(other.cause, @cause)
    end

    def hash
      hashcode = 41 * (41 + super)
      hashcode = 41 * (hashcode + @what.hash)
      hashcode = 41 * (hashcode + @message.hash)
      hashcode = 41 * (hashcode + @suggest_quotes.hash)
      unless @cause.nil?
        hashcode = 41 * (hashcode + @cause.hash)
      end

      hashcode
    end
  end

  class Comment < Token
    def initialize(origin, text)
      super(TokenType::COMMENT, origin)
      @text = text
    end

    class DoubleSlashComment < Comment
      def initialize(origin, text)
        super(origin, text)
      end

      def token_text
        "//" + @text
      end
    end

    class HashComment < Comment
      def initialize(origin, text)
        super(origin, text)
      end

      def token_text
        "#" + @text
      end
    end

    attr_reader :text

    def to_s
      sb = StringIO.new
      sb << "'#"
      sb << text
      sb << "' (COMMENT)"
      sb.string
    end

    def can_equal(other)
      other.is_a?(Comment)
    end

    def ==(other)
      super(other) && other.text == @text
    end

    def hash
      hashcode = 41 * (41 + super)
      hashcode = 41 * (hashcode + @text.hash)

      hashcode
    end
  end

  # This is not a Value, because it requires special processing
  class Substitution < Token
    def initialize(origin, optional, expression)
      super(TokenType::SUBSTITUTION, origin)
      @optional = optional
      @value = expression
    end

    def optional?
      @optional
    end

    attr_reader :value

    def token_text
      sub_text = ""
      @value.each { |t| sub_text << t.token_text }
      "${" + (@optional ? "?" : "") + sub_text + "}"
    end

    def to_s
      sb = StringIO.new
      value.each do |t|
        sb << t.to_s
      end
      "'${#{sb.to_s}}'"
    end

    def can_equal(other)
      other.is_a?(Substitution)
    end

    def ==(other)
      super(other) && other.value == @value
    end

    def hash
      41 * (41 + super + @value.hash)
    end
  end

  def self.value?(token)
    token.is_a?(Value)
  end

  def self.value(token)
    if token.is_a?(Value)
      token.value
    else
      raise ConfigBugOrBrokenError, "tried to get value of non-value token #{token}"
    end
  end

  def self.value_with_type?(t, value_type)
    value?(t) && (value(t).value_type == value_type)
  end

  def self.newline?(t)
    t.is_a?(Line)
  end

  def self.problem?(t)
    t.is_a?(Problem)
  end

  def self.get_problem_what(token)
    if token.is_a?(Problem)
      token.what
    else
      raise ConfigBugOrBrokenError, "tried to get problem what from #{token}"
    end
  end

  def self.get_problem_message(token)
    if token.is_a?(Problem)
      token.message
    else
      raise ConfigBugOrBrokenError.new("tried to get problem message from #{token}")
    end
  end

  def self.get_problem_suggest_quotes(token)
    if token.is_a?(Problem)
      token.suggest_quotes
    else
      raise ConfigBugOrBrokenError.new("tried to get problem suggest_quotes from #{token}")
    end
  end

  def self.get_problem_cause(token)
    if token.is_a?(Problem)
      token.cause
    else
      raise ConfigBugOrBrokenError.new("tried to get problem cause from #{token}")
    end
  end

  def self.comment?(t)
    t.is_a?(Comment)
  end

  def self.comment_text(token)
    if comment?(token)
      token.text
    else
      raise ConfigBugOrBrokenError, "tried to get comment text from #{token}"
    end
  end

  def self.unquoted_text?(token)
    token.is_a?(UnquotedText)
  end

  def self.unquoted_text(token)
    if unquoted_text?(token)
      token.value
    else
      raise ConfigBugOrBrokenError, "tried to get unquoted text from #{token}"
    end
  end

  def self.ignored_whitespace?(token)
    token.is_a?(IgnoredWhitespace)
  end

  def self.substitution?(token)
    token.is_a?(Substitution)
  end

  def self.get_substitution_path_expression(token)
    if token.is_a?(Substitution)
      token.value
    else
      raise ConfigBugOrBrokenError, "tried to get substitution from #{token}"
    end
  end

  def self.get_substitution_optional(token)
    if token.is_a?(Substitution)
      token.optional?
    else
      raise ConfigBugOrBrokenError, "tried to get substitution optionality from #{token}"
    end
  end

  START = Token.new_without_origin(TokenType::START, "start of file", "")
  EOF = Token.new_without_origin(TokenType::EOF, "end of file", "")
  COMMA = Token.new_without_origin(TokenType::COMMA, "','", ",")
  EQUALS = Token.new_without_origin(TokenType::EQUALS, "'='", "=")
  COLON = Token.new_without_origin(TokenType::COLON, "':'", ":")
  OPEN_CURLY = Token.new_without_origin(TokenType::OPEN_CURLY, "'{'", "{")
  CLOSE_CURLY = Token.new_without_origin(TokenType::CLOSE_CURLY, "'}'", "}")
  OPEN_SQUARE = Token.new_without_origin(TokenType::OPEN_SQUARE, "'['", "[")
  CLOSE_SQUARE = Token.new_without_origin(TokenType::CLOSE_SQUARE, "']'", "]")
  PLUS_EQUALS = Token.new_without_origin(TokenType::PLUS_EQUALS, "'+='", "+=")

  def self.new_line(origin)
    Line.new(origin)
  end

  def self.new_problem(origin, what, message, suggest_quotes, cause)
    Problem.new(origin, what, message, suggest_quotes, cause)
  end

  def self.new_comment_double_slash(origin, text)
    Comment::DoubleSlashComment.new(origin, text)
  end

  def self.new_comment_hash(origin, text)
    Comment::HashComment.new(origin, text)
  end

  def self.new_unquoted_text(origin, s)
    UnquotedText.new(origin, s)
  end

  def self.new_ignored_whitespace(origin, s)
    IgnoredWhitespace.new(origin, s)
  end

  def self.new_substitution(origin, optional, expression)
    Substitution.new(origin, optional, expression)
  end

  def self.new_value(value, orig_text = nil)
    Value.new(value, orig_text)
  end

  def self.new_string(origin, value, orig_text)
    new_value(ConfigString::Quoted.new(origin, value), orig_text)
  end

  def self.new_int(origin, value, orig_text)
    new_value(ConfigNumber.new_number(origin, value, orig_text), orig_text)
  end

  def self.new_double(origin, value, orig_text)
    new_value(ConfigNumber.new_number(origin, value, orig_text), orig_text)
  end

  def self.new_long(origin, value, orig_text)
    new_value(ConfigNumber.new_number(origin, value, orig_text), orig_text)
  end

  def self.new_null(origin)
    new_value(ConfigNull.new(origin), "null")
  end

  def self.new_boolean(origin, value)
    new_value(ConfigBoolean.new(origin, value), value.to_s)
  end
end
