/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package thrift

import (
	"bytes"
	"container/list"
	"strconv"
)

/**
 * Type constants in the Thrift protocol.
 */
type TType byte

const (
	STOP    = 0
	VOID    = 1
	BOOL    = 2
	BYTE    = 3
	I08     = 3
	DOUBLE  = 4
	I16     = 6
	I32     = 8
	I64     = 10
	STRING  = 11
	UTF7    = 11
	STRUCT  = 12
	MAP     = 13
	SET     = 14
	LIST    = 15
	ENUM    = 16
	UTF8    = 16
	UTF16   = 17
	BINARY  = 18
	GENERIC = 127
)

func (p TType) ThriftTypeId() byte {
	switch p {
	default:
		return byte(p)
	case BINARY:
		return byte(STRING)
	}
	return byte(0)
}

func (p TType) String() string {
	switch p {
	case STOP:
		return "STOP"
	case VOID:
		return "VOID"
	case BOOL:
		return "BOOL"
	case BYTE:
		return "BYTE"
	case DOUBLE:
		return "DOUBLE"
	case I16:
		return "I16"
	case I32:
		return "I32"
	case I64:
		return "I64"
	case STRING:
		return "STRING"
	case STRUCT:
		return "STRUCT"
	case MAP:
		return "MAP"
	case SET:
		return "SET"
	case LIST:
		return "LIST"
	case ENUM:
		return "ENUM"
	case UTF16:
		return "UTF16"
	case GENERIC:
		return "GENERIC"
	case BINARY:
		return "BINARY"
	}
	return "Unknown"
}

func (p TType) IsBaseType() bool {
	switch p {
	case BOOL, BYTE, DOUBLE, I16, I32, I64, STRING, UTF8, UTF16, BINARY:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsEmptyType() bool {
	switch p {
	case STOP, VOID:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsEnum() bool {
	switch p {
	case ENUM:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsNumericType() bool {
	switch p {
	case ENUM, BOOL, BYTE, DOUBLE, I16, I32, I64:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsStringType() bool {
	switch p {
	case STRING, UTF8, UTF16, BINARY:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsContainer() bool {
	switch p {
	case MAP, SET, LIST:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsStruct() bool {
	switch p {
	case STRUCT:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsMap() bool {
	switch p {
	case MAP:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsList() bool {
	switch p {
	case LIST:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsSet() bool {
	switch p {
	case SET:
		return true
	default:
		return false
	}
	return false
}

func (p TType) IsInt() bool {
	switch p {
	case BYTE, I16, I32, I64:
		return true
	default:
		return false
	}
	return false
}

func (p TType) Coerce(other interface{}) TType {
	if other == nil {
		return TType(STOP)
	}
	switch b := other.(type) {
	default:
		return TType(STOP)
	case nil:
		return TType(STOP)
	case TType:
		return b
	case byte:
		return TType(b)
	case int:
		return TType(byte(b))
	case int8:
		return TType(byte(b))
	case int32:
		return TType(byte(b))
	case int64:
		return TType(byte(b))
	case uint:
		return TType(byte(b))
	case uint32:
		return TType(byte(b))
	case uint64:
		return TType(byte(b))
	case float32:
		return TType(byte(int(b)))
	case float64:
		return TType(byte(int(b)))
	}
	return TType(STOP)
}

func (p TType) LessType(other interface{}) bool {
	return p < p.Coerce(other)
}

func (p TType) Less(i, j interface{}) bool {
	cmp, ok := p.Compare(i, j)
	return ok && cmp > 0
}

func (p TType) Compare(i, j interface{}) (int, bool) {
	if p != BINARY {
		if i == j {
			return 0, true
		}
	}
	if i == nil {
		if j == nil {
			return 0, true
		}
		return -1, true
	}
	if j == nil {
		return 1, true
	}
	ci, iok := p.CoerceData(i)
	cj, jok := p.CoerceData(j)
	if iok && !jok {
		return 1, true
	}
	if !iok && jok {
		return -1, true
	}
	// hopefully this doesn't happen as Compare() would continuously return 0, false
	if !iok && !jok {
		return 0, false
	}
	if p != BINARY {
		if ci == cj {
			return 0, true
		}
	}
	if ci == nil {
		if cj == nil {
			return 0, true
		}
		return -1, true
	}
	if cj == nil {
		return 1, true
	}
	switch p {
	case STOP, VOID:
		// hopefully this doesn't happen as Compare() would continuously return 0, false
		return 0, false
	case BOOL:
		vi, iok := ci.(bool)
		vj, jok := cj.(bool)
		if !iok || !jok {
			return 0, false
		}
		if vi == vj {
			return 0, true
		}
		if vi == false {
			return -1, true
		}
		return 1, true
	case BYTE:
		vi, iok := ci.(byte)
		vj, jok := cj.(byte)
		if !iok || !jok {
			return 0, false
		}
		if vi == vj {
			return 0, true
		}
		if vi < vj {
			return -1, true
		}
		return 1, true
	case DOUBLE:
		vi, iok := ci.(float64)
		vj, jok := cj.(float64)
		if !iok || !jok {
			return 0, false
		}
		if vi == vj {
			return 0, true
		}
		if vi < vj {
			return -1, true
		}
		return 1, true
	case I16:
		vi, iok := ci.(int16)
		vj, jok := cj.(int16)
		if !iok || !jok {
			return 0, false
		}
		if vi == vj {
			return 0, true
		}
		if vi < vj {
			return -1, true
		}
		return 1, true
	case I32:
		vi, iok := ci.(int32)
		vj, jok := cj.(int32)
		if !iok || !jok {
			return 0, false
		}
		if vi == vj {
			return 0, true
		}
		if vi < vj {
			return -1, true
		}
		return 1, true
	case I64:
		vi, iok := ci.(int64)
		vj, jok := cj.(int64)
		if !iok || !jok {
			return 0, false
		}
		if vi == vj {
			return 0, true
		}
		if vi < vj {
			return -1, true
		}
		return 1, true
	case STRING, UTF8, UTF16:
		vi, iok := ci.(string)
		vj, jok := cj.(string)
		if !iok || !jok {
			return 0, false
		}
		if vi == vj {
			return 0, true
		}
		if vi < vj {
			return -1, true
		}
		return 1, true
	case BINARY:
		vi, iok := ci.([]byte)
		vj, jok := cj.([]byte)
		if !iok || !jok {
			return 0, false
		}
		return bytes.Compare(vi, vj), true
	case STRUCT:
		si, iok := ci.(TStruct)
		sj, jok := cj.(TStruct)
		if !iok || !jok {
			return 0, false
		}
		if cmp := CompareString(si.ThriftName(), sj.ThriftName()); cmp != 0 {
			return cmp, true
		}
		if cmp, ok := si.TStructFields().CompareTo(sj.TStructFields()); !ok || cmp != 0 {
			return cmp, ok
		}
		for field := range si.TStructFields().Iter() {
			a := si.AttributeFromFieldId(field.Id())
			b := sj.AttributeFromFieldId(field.Id())
			if cmp, ok := field.TypeId().Compare(a, b); !ok || cmp != 0 {
				return cmp, ok
			}
		}
		return 0, true
	case MAP:
		mi, iok := ci.(TMap)
		mj, jok := cj.(TMap)
		if !iok || !jok {
			return 0, false
		}
		ki := mi.KeyType()
		if kj := mj.KeyType(); ki != kj {
			return CompareInt(int(ki), int(kj)), true
		}
		vi := mi.ValueType()
		if vj := mj.ValueType(); vi != vj {
			return CompareInt(int(vi), int(vj)), true
		}
		if size := mi.Len(); size != mj.Len() {
			return CompareInt(size, mj.Len()), true
		}
		if c, cok := ki.CompareValueArrays(mi.Keys(), mj.Keys()); c != 0 || !cok {
			return c, cok
		}
		return vi.CompareValueArrays(mi.Values(), mj.Values())
	case LIST:
		li, iok := ci.(TList)
		lj, jok := cj.(TList)
		if !iok || !jok {
			return 0, false
		}
		ei := li.ElemType()
		ej := lj.ElemType()
		if ei != ej {
			return CompareInt(int(ei), int(ej)), true
		}
		size := li.Len()
		if size != lj.Len() {
			return CompareInt(size, lj.Len()), true
		}
		for k := 0; k < size; k++ {
			vi := li.At(k)
			vj := lj.At(k)
			c, cok := ei.Compare(vi, vj)
			if c != 0 || !cok {
				return c, cok
			}
		}
		return 0, true
	case SET:
		li, iok := ci.(TSet)
		lj, jok := cj.(TSet)
		if !iok || !jok {
			return 0, false
		}
		ei := li.ElemType()
		ej := lj.ElemType()
		if ei != ej {
			return CompareInt(int(ei), int(ej)), true
		}
		size := li.Len()
		if size != lj.Len() {
			return CompareInt(size, lj.Len()), true
		}
		return ei.CompareValueArrays(li.Values(), lj.Values())
	default:
		panic("Invalid thrift type to coerce")
	}
	return 0, false
}

func (p TType) CompareValueArrays(li, lj []interface{}) (int, bool) {
	size := len(li)
	if cmp := CompareInt(size, len(lj)); cmp != 0 {
		return cmp, true
	}
	for i := 0; i < size; i++ {
		vi := li[i]
		vj := lj[i]
		c, cok := p.Compare(vi, vj)
		if c != 0 || !cok {
			return c, cok
		}
	}
	return 0, true
}

func (p TType) Equals(other interface{}) bool {
	return p == p.Coerce(other)
}

type Stringer interface {
	String() string
}

type Enumer interface {
	String() string
	Value() int
	IsEnum() bool
}

func TypeFromValue(data interface{}) TType {
	switch data.(type) {
	default:
		return STOP
	case nil:
		return VOID
	case bool:
		return BOOL
	case float32, float64:
		return DOUBLE
	case int, int32:
		return I32
	case byte:
		return BYTE
	case int8:
		return I08
	case int16:
		return I16
	case int64:
		return I64
	case string:
		return STRING
	case []byte:
		return BINARY
	case TStruct:
		return STRUCT
	case TMap:
		return MAP
	case TSet:
		return SET
	case []interface{}, *list.List, TList:
		return LIST
	}
	return STOP
}

func (p TType) CoerceData(data interface{}) (interface{}, bool) {
	if data == nil {
		switch p {
		case STOP:
			return nil, true
		case VOID:
			return nil, true
		case BOOL:
			return false, true
		case BYTE:
			return byte(0), true
		case DOUBLE:
			return float64(0), true
		case I16:
			return int16(0), true
		case I32:
			return int32(0), true
		case I64:
			return int64(0), true
		case STRING, UTF8, UTF16:
			return "", true
		case BINARY:
			return nil, true
		case STRUCT:
			return NewTStructEmpty(""), true
		case MAP:
			return NewTMapDefault(), true
		case LIST:
			return NewTListDefault(), true
		case SET:
			return NewTSetDefault(), true
		default:
			panic("Invalid thrift type to coerce")
		}
	}
	switch p {
	case STOP:
		return nil, true
	case VOID:
		return nil, true
	case BOOL:
		switch b := data.(type) {
		default:
			return false, false
		case bool:
			return b, true
		case Numeric:
			return bool(b.Int() != 0), true
		case int:
			return b != 0, true
		case byte:
			return b != 0, true
		case int8:
			return b != 0, true
		case int16:
			return b != 0, true
		case int32:
			return b != 0, true
		case int64:
			return b != 0, true
		case uint:
			return b != 0, true
		case uint16:
			return b != 0, true
		case uint32:
			return b != 0, true
		case uint64:
			return b != 0, true
		case float32:
			return b != 0, true
		case float64:
			return b != 0, true
		case []byte:
			return len(b) > 1 || (len(b) == 1 && b[0] != 0), true
		case Stringer:
			v := b.String()
			if v == "false" || v == "0" || len(v) == 0 {
				return false, true
			}
			return true, true
		case string:
			if b == "false" || b == "0" || len(b) == 0 {
				return false, true
			}
			return true, true
		}
	case BYTE:
		if b, ok := data.(byte); ok {
			return b, true
		}
		if b, ok := data.(Numeric); ok {
			return b.Byte(), true
		}
		if b, ok := data.(bool); ok {
			if b {
				return byte(1), true
			}
			return byte(0), true
		}
		if b, ok := data.(int); ok {
			return byte(b), true
		}
		if b, ok := data.(int8); ok {
			return byte(b), true
		}
		if b, ok := data.(int16); ok {
			return byte(b), true
		}
		if b, ok := data.(int32); ok {
			return byte(b), true
		}
		if b, ok := data.(int64); ok {
			return byte(b), true
		}
		if b, ok := data.(uint); ok {
			return byte(b), true
		}
		if b, ok := data.(uint8); ok {
			return byte(b), true
		}
		if b, ok := data.(uint16); ok {
			return byte(b), true
		}
		if b, ok := data.(uint32); ok {
			return byte(b), true
		}
		if b, ok := data.(uint64); ok {
			return byte(b), true
		}
		if b, ok := data.(float32); ok {
			return byte(int(b)), true
		}
		if b, ok := data.(float64); ok {
			return byte(int(b)), true
		}
		if b, ok := data.([]byte); ok {
			if len(b) > 0 {
				return b[0], true
			}
			return byte(0), true
		}
		if b, ok := data.(Stringer); ok {
			data = b.String()
		}
		if b, ok := data.(string); ok {
			i1, err := strconv.Atoi(b)
			if err != nil {
				return byte(int(i1)), true
			}
		}
		return byte(0), false
	case DOUBLE:
		if b, ok := data.(float32); ok {
			return float64(b), true
		}
		if b, ok := data.(float64); ok {
			return b, true
		}
		if b, ok := data.(Numeric); ok {
			return bool(b.Float64() != 0.0), true
		}
		if b, ok := data.(byte); ok {
			return float64(b), true
		}
		if b, ok := data.(bool); ok {
			if b {
				return float64(1.0), true
			}
			return float64(0.0), true
		}
		if b, ok := data.(int); ok {
			return float64(b), true
		}
		if b, ok := data.(int8); ok {
			return float64(b), true
		}
		if b, ok := data.(int16); ok {
			return float64(b), true
		}
		if b, ok := data.(int32); ok {
			return float64(b), true
		}
		if b, ok := data.(int64); ok {
			return float64(b), true
		}
		if b, ok := data.(uint); ok {
			return float64(b), true
		}
		if b, ok := data.(uint8); ok {
			return float64(b), true
		}
		if b, ok := data.(uint16); ok {
			return float64(b), true
		}
		if b, ok := data.(uint32); ok {
			return float64(b), true
		}
		if b, ok := data.(uint64); ok {
			return float64(b), true
		}
		if b, ok := data.([]byte); ok {
			if len(b) > 0 {
				return float64(b[0]), true
			}
			return float64(0), true
		}
		if b, ok := data.(Stringer); ok {
			data = b.String()
		}
		if b, ok := data.(string); ok {
			d1, err := strconv.ParseFloat(b, 64)
			if err != nil {
				return d1, true
			}
		}
		return float64(0), false
	case I16:
		if b, ok := data.(int16); ok {
			return b, true
		}
		if b, ok := data.(int); ok {
			return int16(b), true
		}
		if b, ok := data.(Numeric); ok {
			return bool(b.Int16() != 0), true
		}
		if b, ok := data.(byte); ok {
			return int16(b), true
		}
		if b, ok := data.(bool); ok {
			if b {
				return int16(1.0), true
			}
			return int16(0.0), true
		}
		if b, ok := data.(int8); ok {
			return int16(b), true
		}
		if b, ok := data.(int32); ok {
			return int16(b), true
		}
		if b, ok := data.(int64); ok {
			return int16(b), true
		}
		if b, ok := data.(uint); ok {
			return int16(b), true
		}
		if b, ok := data.(uint8); ok {
			return int16(b), true
		}
		if b, ok := data.(uint16); ok {
			return int16(b), true
		}
		if b, ok := data.(uint32); ok {
			return int16(b), true
		}
		if b, ok := data.(uint64); ok {
			return int16(b), true
		}
		if b, ok := data.(float32); ok {
			return int16(int(b)), true
		}
		if b, ok := data.(float64); ok {
			return int16(int(b)), true
		}
		if b, ok := data.(Stringer); ok {
			data = b.String()
		}
		if b, ok := data.(string); ok {
			i1, err := strconv.Atoi(b)
			if err != nil {
				return int16(i1), true
			}
		}
		return int16(0), false
	case I32:
		if b, ok := data.(int32); ok {
			return b, true
		}
		if b, ok := data.(int); ok {
			return int32(b), true
		}
		if b, ok := data.(Numeric); ok {
			return bool(b.Int32() != 0), true
		}
		if b, ok := data.(byte); ok {
			return int32(b), true
		}
		if b, ok := data.(bool); ok {
			if b {
				return int32(1.0), true
			}
			return int32(0.0), true
		}
		if b, ok := data.(int8); ok {
			return int32(b), true
		}
		if b, ok := data.(int16); ok {
			return int32(b), true
		}
		if b, ok := data.(int64); ok {
			return int32(b), true
		}
		if b, ok := data.(uint); ok {
			return int32(b), true
		}
		if b, ok := data.(uint8); ok {
			return int32(b), true
		}
		if b, ok := data.(uint16); ok {
			return int32(b), true
		}
		if b, ok := data.(uint32); ok {
			return int32(b), true
		}
		if b, ok := data.(uint64); ok {
			return int32(b), true
		}
		if b, ok := data.(float32); ok {
			return int32(int(b)), true
		}
		if b, ok := data.(float64); ok {
			return int32(int(b)), true
		}
		if b, ok := data.(Stringer); ok {
			data = b.String()
		}
		if b, ok := data.(string); ok {
			i1, err := strconv.Atoi(b)
			if err != nil {
				return int32(i1), true
			}
		}
		return int32(0), false
	case I64:
		if b, ok := data.(int64); ok {
			return b, true
		}
		if b, ok := data.(int32); ok {
			return int64(b), true
		}
		if b, ok := data.(int); ok {
			return int64(b), true
		}
		if b, ok := data.(Numeric); ok {
			return bool(b.Int64() != 0), true
		}
		if b, ok := data.(byte); ok {
			return int64(b), true
		}
		if b, ok := data.(bool); ok {
			if b {
				return int64(1.0), true
			}
			return int64(0.0), true
		}
		if b, ok := data.(int8); ok {
			return int64(b), true
		}
		if b, ok := data.(int16); ok {
			return int64(b), true
		}
		if b, ok := data.(uint); ok {
			return int64(b), true
		}
		if b, ok := data.(uint8); ok {
			return int64(b), true
		}
		if b, ok := data.(uint16); ok {
			return int64(b), true
		}
		if b, ok := data.(uint32); ok {
			return int64(b), true
		}
		if b, ok := data.(uint64); ok {
			return int64(b), true
		}
		if b, ok := data.(float32); ok {
			return int64(b), true
		}
		if b, ok := data.(float64); ok {
			return int64(b), true
		}
		if b, ok := data.(Stringer); ok {
			data = b.String()
		}
		if b, ok := data.(string); ok {
			i1, err := strconv.ParseInt(b, 10, 64)
			if err != nil {
				return i1, true
			}
		}
		return int64(0), false
	case STRING, UTF8, UTF16:
		if b, ok := data.([]byte); ok {
			return string(b), true
		}
		if b, ok := data.(Enumer); ok {
			if i1, ok := data.(int); ok {
				return string(i1), true
			}
			return b.String(), true
		}
		if b, ok := data.(Stringer); ok {
			return b.String(), true
		}
		if b, ok := data.(string); ok {
			return b, true
		}
		if b, ok := data.(int); ok {
			return string(b), true
		}
		if b, ok := data.(byte); ok {
			return string(b), true
		}
		if b, ok := data.(bool); ok {
			if b {
				return "true", true
			}
			return "false", true
		}
		if b, ok := data.(int8); ok {
			return string(b), true
		}
		if b, ok := data.(int16); ok {
			return string(b), true
		}
		if b, ok := data.(int32); ok {
			return string(b), true
		}
		if b, ok := data.(int64); ok {
			return string(b), true
		}
		if b, ok := data.(uint); ok {
			return string(b), true
		}
		if b, ok := data.(uint8); ok {
			return string(b), true
		}
		if b, ok := data.(uint16); ok {
			return string(b), true
		}
		if b, ok := data.(uint32); ok {
			return string(b), true
		}
		if b, ok := data.(uint64); ok {
			return string(b), true
		}
		if b, ok := data.(float32); ok {
			return strconv.FormatFloat(float64(b), 'g', -1, 32), true
		}
		if b, ok := data.(float64); ok {
			return strconv.FormatFloat(b, 'g', -1, 64), true
		}
		return "", false
	case BINARY:
		if b, ok := data.([]byte); ok {
			return b, true
		}
		if b, ok := data.(Enumer); ok {
			if i1, ok := data.(int); ok {
				return []byte(string(i1)), true
			}
			return []byte(b.String()), true
		}
		if b, ok := data.(Stringer); ok {
			return []byte(b.String()), true
		}
		if b, ok := data.(string); ok {
			return []byte(b), true
		}
		if b, ok := data.(int); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(byte); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(bool); ok {
			if b {
				return []byte("true"), true
			}
			return []byte("false"), true
		}
		if b, ok := data.(int8); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(int16); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(int32); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(int64); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(uint); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(uint8); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(uint16); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(uint32); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(uint64); ok {
			return []byte(string(b)), true
		}
		if b, ok := data.(float32); ok {
			return []byte(strconv.FormatFloat(float64(b), 'g', -1, 32)), true
		}
		if b, ok := data.(float64); ok {
			return []byte(strconv.FormatFloat(b, 'g', -1, 64)), true
		}
		return "", false
	case STRUCT:
		if b, ok := data.(TStruct); ok {
			return b, true
		}
		return NewTStructEmpty(""), true
	case MAP:
		if b, ok := data.(TMap); ok {
			return b, true
		}
		return NewTMapDefault(), false
	case LIST:
		if b, ok := data.(TList); ok {
			return b, true
		}
		return NewTListDefault(), false
	case SET:
		if b, ok := data.(TSet); ok {
			return b, true
		}
		return NewTSetDefault(), false
	default:
		panic("Invalid thrift type to coerce")
	}
	return nil, false
}

type EqualsOtherInterface interface {
	Equals(other interface{}) bool
}

type EqualsMap interface {
	Equals(other TMap) bool
}

type EqualsSet interface {
	Equals(other TSet) bool
}

type EqualsList interface {
	Equals(other TList) bool
}

type EqualsStruct interface {
	Equals(other TStruct) bool
}