/*
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
Copyright (C) ITsysCOM GmbH

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>
*/

package agents

import (
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"strconv"
	"strings"
	"time"

	"github.com/cgrates/birpc"
	"github.com/cgrates/birpc/context"
	"github.com/cgrates/cgrates/config"
	"github.com/cgrates/cgrates/engine"
	"github.com/cgrates/cgrates/sessions"
	"github.com/cgrates/cgrates/utils"
	janus "github.com/cgrates/janusgo"
	"nhooyr.io/websocket"
)

// NewJanusAgent will construct a JanusAgent
func NewJanusAgent(cgrCfg *config.CGRConfig,
	connMgr *engine.ConnManager,
	filterS *engine.FilterS) (*JanusAgent, error) {
	jsa := &JanusAgent{
		cgrCfg:  cgrCfg,
		connMgr: connMgr,
		filterS: filterS,
	}
	srv, err := birpc.NewServiceWithMethodsRename(jsa, utils.AgentV1, true, func(oldFn string) (newFn string) {
		return strings.TrimPrefix(oldFn, "V1")
	})
	if err != nil {
		return nil, err
	}
	jsa.ctx = context.WithClient(context.TODO(), srv)
	return jsa, nil
}

// JanusAgent is a gateway between HTTP and Janus Server over Websocket
type JanusAgent struct {
	cgrCfg  *config.CGRConfig
	connMgr *engine.ConnManager
	filterS *engine.FilterS
	jnsConn *janus.Gateway
	adminWs *websocket.Conn
	ctx     *context.Context
}

// Connect will create the connection to the Janus Server
func (ja *JanusAgent) Connect() (err error) {
	ja.jnsConn, err = janus.Connect(
		fmt.Sprintf("ws://%s", ja.cgrCfg.JanusAgentCfg().JanusConns[0].Address))
	if err != nil {
		return
	}
	ja.adminWs, _, err = websocket.Dial(context.Background(), fmt.Sprintf("ws://%s", ja.cgrCfg.JanusAgentCfg().JanusConns[0].AdminAddress), &websocket.DialOptions{
		Subprotocols: []string{utils.JanusAdminSubProto},
	})

	return
}

// Shutdown will close the connection to the Janus Server
func (ja *JanusAgent) Shutdown() (err error) {
	if err = ja.jnsConn.Close(); err != nil {
		utils.Logger.Err(fmt.Sprintf("<%s> Error on disconnecting janus server: %s", utils.JanusAgent, err.Error()))
	}
	if err = ja.adminWs.CloseNow(); err != nil {
		utils.Logger.Err(fmt.Sprintf("<%s> Error on disconnecting janus admin: %s", utils.JanusAgent, err.Error()))
	}
	return
}

// ServeHTTP implements http.Handler interface
func (ja *JanusAgent) CORSOptions(w http.ResponseWriter, req *http.Request) {
	janusAccessControlHeaders(w, req)
}

// CreateSession will create a new session within janusgo
func (ja *JanusAgent) CreateSession(w http.ResponseWriter, req *http.Request) {
	janusAccessControlHeaders(w, req)
	var msg janus.BaseMsg
	if err := json.NewDecoder(req.Body).Decode(&msg); err != nil {
		http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
		return
	}
	if err := ja.authSession(strings.Split(req.RemoteAddr, ":")[0]); err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	resp, err := ja.jnsConn.CreateSession(ctx, msg)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	json.NewEncoder(w).Encode(&resp)
}

func (ja *JanusAgent) authSession(origIP string) (err error) {
	authArgs := &sessions.V1AuthorizeArgs{
		GetMaxUsage:   true,
		GetAttributes: true,
		CGREvent: &utils.CGREvent{
			Tenant: ja.cgrCfg.GeneralCfg().DefaultTenant,
			ID:     utils.Sha1(),
			Time:   utils.TimePointer(time.Now()),
			Event: map[string]any{
				utils.AccountField: origIP,
				utils.Destination:  "echotest",
			},
		}}
	rply := new(sessions.V1AuthorizeReply)
	err = ja.connMgr.Call(ja.ctx, ja.cgrCfg.JanusAgentCfg().SessionSConns,
		utils.SessionSv1AuthorizeEvent,
		authArgs, rply)
	return
}

func (ja *JanusAgent) acntStartSession(s *janus.Session) (err error) {
	initArgs := &sessions.V1InitSessionArgs{
		GetAttributes: true,
		InitSession:   true,
		CGREvent: &utils.CGREvent{
			Tenant: ja.cgrCfg.GeneralCfg().DefaultTenant,
			ID:     utils.Sha1(),
			Time:   utils.TimePointer(time.Now()),
			Event: map[string]any{
				utils.AccountField: s.Data[utils.AccountField],
				utils.OriginHost:   s.Data[utils.OriginHost],
				utils.OriginID:     s.Data[utils.OriginID],
				utils.Destination:  s.Data[utils.Destination],
				utils.AnswerTime:   s.Data[utils.AnswerTime],
			},
		},
		ForceDuration: true,
	}
	rply := new(sessions.V1InitSessionReply)
	err = ja.connMgr.Call(ja.ctx, ja.cgrCfg.JanusAgentCfg().SessionSConns,
		utils.SessionSv1InitiateSession,
		initArgs, rply)
	return
}

func (ja *JanusAgent) acntStopSession(s *janus.Session) (err error) {
	terminateArgs := &sessions.V1TerminateSessionArgs{
		TerminateSession: true,
		CGREvent: &utils.CGREvent{
			Tenant: ja.cgrCfg.GeneralCfg().DefaultTenant,
			ID:     utils.Sha1(),
			Time:   utils.TimePointer(time.Now()),
			Event: map[string]any{
				utils.AccountField: s.Data[utils.AccountField],
				utils.OriginHost:   s.Data[utils.OriginHost],
				utils.OriginID:     s.Data[utils.OriginID],
				utils.Destination:  s.Data[utils.Destination],
				utils.AnswerTime:   s.Data[utils.AnswerTime],
				utils.Usage:        s.Data[utils.Usage],
			},
		},
		ForceDuration: true,
	}
	var rply string
	err = ja.connMgr.Call(ja.ctx, ja.cgrCfg.JanusAgentCfg().SessionSConns,
		utils.SessionSv1TerminateSession,
		terminateArgs, &rply)
	return
}

func (ja *JanusAgent) cdrSession(s *janus.Session) (err error) {
	cgrEv := &utils.CGREvent{
		Tenant: ja.cgrCfg.GeneralCfg().DefaultTenant,
		ID:     utils.Sha1(),
		Time:   utils.TimePointer(time.Now()),
		Event: map[string]any{
			utils.AccountField: s.Data[utils.AccountField],
			utils.OriginHost:   s.Data[utils.OriginHost],
			utils.OriginID:     s.Data[utils.OriginID],
			utils.Destination:  s.Data[utils.Destination],
			utils.AnswerTime:   s.Data[utils.AnswerTime],
			utils.Usage:        s.Data[utils.Usage],
		},
	}
	var rply string
	err = ja.connMgr.Call(ja.ctx, ja.cgrCfg.JanusAgentCfg().SessionSConns,
		utils.SessionSv1ProcessCDR,
		cgrEv, &rply)
	return
}

// SessioNKeepalive sends keepalive once OPTIONS are coming for the session from HTTP
func (ja *JanusAgent) SessionKeepalive(w http.ResponseWriter, r *http.Request) {
	janusAccessControlHeaders(w, r)
	sessionID, err := strconv.ParseUint(r.PathValue("sessionID"), 10, 64)
	if err != nil {
		http.Error(w, "Invalid session ID", http.StatusBadRequest)
		return
	}
	ja.jnsConn.RLock()
	session, has := ja.jnsConn.Sessions[sessionID]
	ja.jnsConn.RUnlock()
	if !has {
		http.Error(w, "Session not found", http.StatusNotFound)
		return
	}
	msg := janus.BaseMsg{
		Session: session.ID,
		Type:    "keepalive",
	}
	var resp any
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	resp, err = session.KeepAlive(ctx, msg)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return

	}
	json.NewEncoder(w).Encode(resp)
}

// PollSession will create a long-poll request to be notified about events and incoming messages from session
func (ja *JanusAgent) PollSession(w http.ResponseWriter, req *http.Request) {
	janusAccessControlHeaders(w, req)
	sessionID, err := strconv.ParseUint(req.PathValue("sessionID"), 10, 64)
	if err != nil {
		http.Error(w, "Invalid session ID", http.StatusBadRequest)
		return
	}
	ja.jnsConn.RLock()
	session, has := ja.jnsConn.Sessions[sessionID]
	ja.jnsConn.RUnlock()
	if !has {
		http.Error(w, "Session not found", http.StatusNotFound)
		return
	}
	maxEvs, err := strconv.Atoi(req.URL.Query().Get("maxev"))
	if err != nil {
		http.Error(w, fmt.Sprintf("Invalid maxev, err: %s", err.Error()),
			http.StatusBadRequest)
		return
	}
	msg := janus.BaseMsg{
		Session: session.ID,
		Type:    "keepalive",
	}
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()
	events, err := session.LongPoll(ctx, maxEvs, msg)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
	}
	for _, evIface := range events {
		upEv, isWebrtcup := evIface.(*janus.WebRTCUpMsg)
		if isWebrtcup {
			ja.jnsConn.RLock()
			s := ja.jnsConn.Sessions[upEv.Session]
			ja.jnsConn.RUnlock()
			if s == nil {
				continue
			}
			s.Data[utils.AccountField] = strings.Split(req.RemoteAddr, ":")[0]
			s.Data[utils.OriginHost] = strings.Split(req.Host, ":")[0]
			s.Data[utils.OriginID] = strconv.Itoa(int(s.ID))
			s.Data[utils.Destination] = "echotest"
			s.Data[utils.AnswerTime] = time.Now()
			go func() { ja.acntStartSession(s) }()
		}
	}
	json.NewEncoder(w).Encode(events)
}

// AttachPlugin will attach a plugin to a session
func (ja *JanusAgent) AttachPlugin(w http.ResponseWriter, r *http.Request) {
	janusAccessControlHeaders(w, r)
	sessionID, err := strconv.ParseUint(r.PathValue("sessionID"), 10, 64)
	if err != nil {
		http.Error(w, "Invalid session ID", http.StatusBadRequest)
		return
	}
	ja.jnsConn.RLock()
	session, has := ja.jnsConn.Sessions[sessionID]
	ja.jnsConn.RUnlock()
	if !has {
		http.Error(w, "Session not found", http.StatusNotFound)
		return
	}
	var msg janus.BaseMsg
	if err := json.NewDecoder(r.Body).Decode(&msg); err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}
	msg.Session = session.ID

	var resp any
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()
	if msg.Type == "destroy" {
		answerTime, _ := utils.IfaceAsTime(session.Data[utils.AnswerTime], ja.cgrCfg.GeneralCfg().DefaultTimezone)
		var totalDur time.Duration
		if !answerTime.IsZero() {
			totalDur = time.Since(answerTime)
		}
		session.Data[utils.Usage] = totalDur // toDo: lock session RW

		go func() {
			ja.acntStopSession(session)
			ja.cdrSession(session)
		}() // CGRateS accounting stop
		resp, err = session.DestroySession(ctx, msg)
	} else {
		resp, err = session.AttachSession(ctx, msg)
	}
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return

	}
	json.NewEncoder(w).Encode(resp)
}

// HandlePlugin will handle requests towards a plugin
func (ja *JanusAgent) HandlePlugin(w http.ResponseWriter, r *http.Request) {
	janusAccessControlHeaders(w, r)
	sessionID, err := strconv.ParseUint(r.PathValue("sessionID"), 10, 64)
	if err != nil {
		http.Error(w, "Invalid session ID", http.StatusBadRequest)
		return
	}
	ja.jnsConn.RLock()
	session, has := ja.jnsConn.Sessions[sessionID]
	ja.jnsConn.RUnlock()
	if !has {
		http.Error(w, "Session not found", http.StatusNotFound)
		return
	}
	handleID, err := strconv.ParseUint(r.PathValue("handleID"), 10, 64)
	if err != nil {
		http.Error(w, "Invalid handle ID", http.StatusBadRequest)
		return
	}
	handle, has := session.Handles[handleID]
	if !has {
		if !has {
			http.Error(w, "Handle not found", http.StatusNotFound)
			return
		}
	}
	rBody, err := io.ReadAll(r.Body)
	if err != nil {
		http.Error(w, "Cannot read body", http.StatusBadRequest)
		return
	}
	var msg janus.BaseMsg
	if err := json.Unmarshal(rBody, &msg); err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	var resp any
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	// handle message, depending on it's type
	switch msg.Type {
	case "message":
		var hMsg janus.HandlerMessageJsep
		if err := json.Unmarshal(rBody, &hMsg); err != nil {
			http.Error(w, err.Error(), http.StatusBadRequest)
			return
		}
		hMsg.Session = session.ID
		hMsg.BaseMsg.Handle = handle.ID
		hMsg.Handle = handle.ID
		resp, err = handle.Message(ctx, hMsg)
	case "trickle":
		var hMsg janus.TrickleOne
		if err := json.Unmarshal(rBody, &hMsg); err != nil {
			http.Error(w, err.Error(), http.StatusBadRequest)
			return
		}
		hMsg.Session = session.ID
		hMsg.Handle = handle.ID
		hMsg.HandleR = handle.ID
		resp, err = handle.Trickle(ctx, hMsg)
	default:
		return
	}

	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return

	}
	json.NewEncoder(w).Encode(resp)
}

func (ja *JanusAgent) V1GetActiveSessionIDs(ctx *context.Context, ignParam string,
	sessionIDs *[]*sessions.SessionID) error {
	var sIDs []*sessions.SessionID
	msg := struct {
		janus.BaseMsg
		AdminSecret string `json:"admin_secret"`
	}{
		BaseMsg: janus.BaseMsg{
			Type: "list_sessions",
			ID:   utils.GenUUID(),
		},
		AdminSecret: ja.cgrCfg.JanusAgentCfg().JanusConns[0].AdminPassword,
	}
	byteMsg, err := json.Marshal(msg)
	if err != nil {
		return err
	}
	if err = ja.adminWs.Write(context.Background(), websocket.MessageText, byteMsg); err != nil {
		return err
	}
	_, rpl, err := ja.adminWs.Read(context.Background())
	if err != nil {
		return err
	}
	var sucessMsg struct {
		janus.SuccessMsg
		Sessions []uint64 `json:"sessions"`
	}

	if err = json.Unmarshal(rpl, &sucessMsg); err != nil {
		return err
	}
	for _, sId := range sucessMsg.Sessions {
		sess, has := ja.jnsConn.Sessions[sId]
		if !has {
			continue
		}
		sIDs = append(sIDs, &sessions.SessionID{
			OriginHost: sess.Data[utils.OriginHost].(string),
			OriginID:   sess.Data[utils.OriginID].(string),
		})
	}
	if len(sIDs) == 0 {
		return utils.ErrNoActiveSession
	}
	*sessionIDs = sIDs
	return nil
}

func (ja *JanusAgent) V1DisconnectSession(ctx *context.Context, cgrEv utils.CGREvent, reply *string) (err error) {
	sessionID, err := engine.NewMapEvent(cgrEv.Event).GetTInt64(utils.OriginID)
	if err != nil {
		return err
	}
	session, has := ja.jnsConn.Sessions[uint64(sessionID)]
	if has {
		id := utils.GenUUID()
		_, err := session.DestroySession(context.Background(), janus.BaseMsg{Type: "destroy", ID: id, Session: uint64(sessionID)})
		if err != nil {
			return err
		}
	}
	*reply = utils.OK
	return nil
}

func (ja *JanusAgent) V1AlterSession(*context.Context, utils.CGREvent, *string) error {
	return utils.ErrNotImplemented
}

func (ja *JanusAgent) V1DisconnectPeer(*context.Context, *utils.DPRArgs, *string) error {
	return utils.ErrNotImplemented
}

func (ja *JanusAgent) V1WarnDisconnect(*context.Context, map[string]any, *string) error {
	return utils.ErrNotImplemented
}