package sip

import (
	"bytes"
	"fmt"
	"strings"

	"gitee.com/zhjun2512/gosip/util"
)

const (
	MTU uint = 1500

	DefaultHost     = "127.0.0.1"
	DefaultProtocol = "UDP"

	DefaultUdpPort Port = 5060
	DefaultTcpPort Port = 5060
	DefaultTlsPort Port = 5061
	DefaultWsPort  Port = 80
	DefaultWssPort Port = 443
)

// TODO should be refactored, currently here the pit

type Address struct {
	DisplayName MaybeString
	Uri         Uri
	Params      Params
}

func NewAddressFromFromHeader(from *FromHeader) *Address {
	addr := &Address{
		DisplayName: from.DisplayName,
	}
	if from.Address != nil {
		addr.Uri = from.Address.Clone()
	}
	if from.Params != nil {
		addr.Params = from.Params.Clone()
	}

	return addr
}

func NewAddressFromToHeader(to *ToHeader) *Address {
	addr := &Address{
		DisplayName: to.DisplayName,
	}
	if to.Address != nil {
		addr.Uri = to.Address.Clone()
	}
	if to.Params != nil {
		addr.Params = to.Params.Clone()
	}

	return addr
}

func NewAddressFromContactHeader(cnt *ContactHeader) *Address {
	addr := &Address{
		DisplayName: cnt.DisplayName,
	}
	if cnt.Address != nil {
		addr.Uri = cnt.Address.Clone()
	}
	if cnt.Params != nil {
		addr.Params = cnt.Params.Clone()
	}

	return addr
}

func (addr *Address) String() string {
	var buffer bytes.Buffer

	if addr == nil {
		return "<nil>"
	}

	if addr.DisplayName != nil {
		if displayName, ok := addr.DisplayName.(String); ok && displayName.String() != "" {
			buffer.WriteString(fmt.Sprintf("\"%s\" ", displayName))
		}
	}

	buffer.WriteString(fmt.Sprintf("<%s>", addr.Uri))

	if addr.Params != nil && addr.Params.Length() > 0 {
		buffer.WriteString(";")
		buffer.WriteString(addr.Params.ToString(';'))
	}

	return buffer.String()
}

func (addr *Address) Clone() *Address {
	var name MaybeString
	var uri Uri
	var params Params

	if addr.DisplayName != nil {
		name = String{Str: addr.DisplayName.String()}
	}
	if addr.Uri != nil {
		uri = addr.Uri.Clone()
	}
	if addr.Params != nil {
		params = addr.Params.Clone()
	}

	return &Address{
		DisplayName: name,
		Uri:         uri,
		Params:      params,
	}
}

func (addr *Address) Equals(other interface{}) bool {
	otherPtr, ok := other.(*Address)
	if !ok {
		return false
	}

	if addr == otherPtr {
		return true
	}
	if addr == nil && otherPtr != nil || addr != nil && otherPtr == nil {
		return false
	}

	res := true

	if addr.DisplayName != otherPtr.DisplayName {
		if addr.DisplayName == nil {
			res = res && otherPtr.DisplayName == nil
		} else {
			res = res && addr.DisplayName.Equals(otherPtr.DisplayName)
		}
	}

	if addr.Uri != otherPtr.Uri {
		if addr.Uri == nil {
			res = res && otherPtr.Uri == nil
		} else {
			res = res && addr.Uri.Equals(otherPtr.Uri)
		}
	}

	if addr.Params != otherPtr.Params {
		if addr.Params == nil {
			res = res && otherPtr.Params == nil
		} else {
			res = res && addr.Params.Equals(otherPtr.Params)
		}
	}

	return res
}

func (addr *Address) AsToHeader() *ToHeader {
	to := &ToHeader{
		DisplayName: addr.DisplayName,
	}
	if addr.Uri != nil {
		to.Address = addr.Uri.Clone()
	}
	if addr.Params != nil {
		to.Params = addr.Params.Clone()
	}

	return to
}

func (addr *Address) AsFromHeader() *FromHeader {
	from := &FromHeader{
		DisplayName: addr.DisplayName,
	}
	if addr.Uri != nil {
		from.Address = addr.Uri.Clone()
	}
	if addr.Params != nil {
		from.Params = addr.Params.Clone()
	}

	return from
}

func (addr *Address) AsContactHeader() *ContactHeader {
	cnt := &ContactHeader{
		DisplayName: addr.DisplayName,
	}
	if addr.Uri != nil {
		cnt.Address = addr.Uri.Clone()
	}
	if addr.Params != nil {
		cnt.Params = addr.Params.Clone()
	}

	return cnt
}

// Port number
type Port uint16

func (port *Port) Clone() *Port {
	if port == nil {
		return nil
	}
	newPort := *port
	return &newPort
}

func (port *Port) String() string {
	if port == nil {
		return ""
	}
	return fmt.Sprintf("%d", *port)
}

func (port *Port) Equals(other interface{}) bool {
	if p, ok := other.(*Port); ok {
		return util.Uint16PtrEq((*uint16)(port), (*uint16)(p))
	}

	return false
}

// String wrapper
type MaybeString interface {
	String() string
	Equals(other interface{}) bool
}

type String struct {
	Str string
}

func (str String) String() string {
	return str.Str
}

func (str String) Equals(other interface{}) bool {
	if v, ok := other.(string); ok {
		return str.Str == v
	}
	if v, ok := other.(String); ok {
		return str.Str == v.Str
	}

	return false
}

type CancelError interface {
	Canceled() bool
}

type ExpireError interface {
	Expired() bool
}

type MessageError interface {
	error
	// Malformed indicates that message is syntactically valid but has invalid headers, or
	// without required headers.
	Malformed() bool
	// Broken or incomplete message, or not a SIP message
	Broken() bool
}

// Broken or incomplete messages, or not a SIP message.
type BrokenMessageError struct {
	Err error
	Msg string
}

func (err *BrokenMessageError) Malformed() bool { return false }
func (err *BrokenMessageError) Broken() bool    { return true }
func (err *BrokenMessageError) Error() string {
	if err == nil {
		return "<nil>"
	}

	s := "BrokenMessageError: " + err.Err.Error()
	if err.Msg != "" {
		s += fmt.Sprintf("\nMessage dump:\n%s", err.Msg)
	}

	return s
}

// syntactically valid but logically invalid message
type MalformedMessageError struct {
	Err error
	Msg string
}

func (err *MalformedMessageError) Malformed() bool { return true }
func (err *MalformedMessageError) Broken() bool    { return false }
func (err *MalformedMessageError) Error() string {
	if err == nil {
		return "<nil>"
	}

	s := "MalformedMessageError: " + err.Err.Error()
	if err.Msg != "" {
		s += fmt.Sprintf("\nMessage dump:\n%s", err.Msg)
	}

	return s
}

type UnsupportedMessageError struct {
	Err error
	Msg string
}

func (err *UnsupportedMessageError) Malformed() bool { return true }
func (err *UnsupportedMessageError) Broken() bool    { return false }
func (err *UnsupportedMessageError) Error() string {
	if err == nil {
		return "<nil>"
	}

	s := "UnsupportedMessageError: " + err.Err.Error()
	if err.Msg != "" {
		s += fmt.Sprintf("\nMessage dump:\n%s", err.Msg)
	}

	return s
}

type UnexpectedMessageError struct {
	Err error
	Msg string
}

func (err *UnexpectedMessageError) Broken() bool    { return false }
func (err *UnexpectedMessageError) Malformed() bool { return false }
func (err *UnexpectedMessageError) Error() string {
	if err == nil {
		return "<nil>"
	}

	s := "UnexpectedMessageError: " + err.Err.Error()
	if err.Msg != "" {
		s += fmt.Sprintf("\nMessage dump:\n%s", err.Msg)
	}

	return s
}

const RFC3261BranchMagicCookie = "z9hG4bK"

// GenerateBranch returns random unique branch ID.
func GenerateBranch() string {
	return strings.Join([]string{
		RFC3261BranchMagicCookie,
		util.RandString(32),
	}, ".")
}

// DefaultPort returns protocol default port by network.
func DefaultPort(protocol string) Port {
	switch strings.ToLower(protocol) {
	case "tls":
		return DefaultTlsPort
	case "tcp":
		return DefaultTcpPort
	case "udp":
		return DefaultUdpPort
	case "ws":
		return DefaultWsPort
	case "wss":
		return DefaultWssPort
	default:
		return DefaultTcpPort
	}
}

func MakeDialogIDFromMessage(msg Message) (string, error) {
	callID, ok := msg.CallID()
	if !ok {
		return "", fmt.Errorf("missing Call-ID header")
	}

	to, ok := msg.To()
	if !ok {
		return "", fmt.Errorf("missing To header")
	}

	toTag, ok := to.Params.Get("tag")
	if !ok {
		return "", fmt.Errorf("missing tag param in To header")
	}

	from, ok := msg.From()
	if !ok {
		return "", fmt.Errorf("missing To header")
	}

	fromTag, ok := from.Params.Get("tag")
	if !ok {
		return "", fmt.Errorf("missing tag param in From header")
	}

	return MakeDialogID(string(*callID), toTag.String(), fromTag.String()), nil
}

func MakeDialogID(callID, innerID, externalID string) string {
	return strings.Join([]string{callID, innerID, externalID}, "__")
}