package main

import (
	"fmt"
	"github.com/ghettovoice/gosip/sip"
	"golang.org/x/text/encoding/simplifiedchinese"
	"golang.org/x/text/transform"
	"net"
	"strconv"
	"strings"
)

const (
	XmlHeaderGBK = `<?xml version="1.0" encoding="GB2312"?>` + "\r\n"
)

func BuildSDP(userName, sessionName, ip string, port uint16, startTime, stopTime, setup string, speed int, ssrc string) string {
	format := "v=0\r\n" +
		"o=%s 0 0 IN IP4 %s\r\n" +
		"s=%s\r\n" +
		"c=IN IP4 %s\r\n" +
		"t=%s %s\r\n" +
		"m=video %d %s 96\r\n" +
		"a=%s\r\n" +
		"a=rtpmap:96 PS/90000\r\n"

	tcpFormat := "a=setup:%s\r\n" +
		"a=connection:new\r\n"

	var tcp bool
	var mediaProtocol string
	if "active" == setup || "passive" == setup {
		mediaProtocol = "TCP/RTP/AVP"
		tcp = true
	} else {
		mediaProtocol = "RTP/AVP"
	}

	sdp := fmt.Sprintf(format, userName, ip, sessionName, ip, startTime, stopTime, port, mediaProtocol, "recvonly")
	if tcp {
		sdp += fmt.Sprintf(tcpFormat, setup)
	}

	if speed > 0 {
		sdp += fmt.Sprintf("a=downloadspeed:%d\r\n", speed)
	}

	sdp += fmt.Sprintf("y=%s\r\n", ssrc)
	return sdp
}

func NewSIPRequestBuilderWithTransport(transport string) *sip.RequestBuilder {
	builder := sip.NewRequestBuilder()
	hop := sip.ViaHop{
		Transport: transport,
	}

	builder.AddVia(&hop)
	return builder
}

func BuildMessageRequest(from, fromRealm, to, toAddr, transport, body string) (sip.Request, error) {
	if !strings.HasPrefix(body, "<?xml") {
		body = XmlHeaderGBK + body
	}

	gbkBody, _, err := transform.String(simplifiedchinese.GBK.NewEncoder(), body)
	if err != nil {
		panic(err)
	}

	builder := NewRequestBuilder(sip.MESSAGE, from, fromRealm, to, toAddr, transport)
	builder.SetContentType(&XmlMessageType)
	builder.SetBody(gbkBody)
	return builder.Build()
}

func NewRequestBuilder(method sip.RequestMethod, fromUser, realm, toUser, toAddr, transport string) *sip.RequestBuilder {
	builder := NewSIPRequestBuilderWithTransport(transport)
	builder.SetMethod(method)

	host, p, _ := net.SplitHostPort(toAddr)
	port, _ := strconv.Atoi(p)
	sipPort := sip.Port(port)

	requestUri := &sip.SipUri{
		FUser: sip.String{Str: toUser},
		FHost: host,
		FPort: &sipPort,
	}

	builder.SetRecipient(requestUri)

	fromAddress := &sip.Address{
		Uri: &sip.SipUri{
			FUser: sip.String{Str: fromUser},
			FHost: realm,
		},
	}

	fromAddress.Params = sip.NewParams().Add("tag", sip.String{Str: GenerateTag()})
	builder.SetFrom(fromAddress)
	builder.SetTo(&sip.Address{
		Uri: requestUri,
	})

	return builder
}