goldgorilla/app.go

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"github.com/pion/webrtc/v3"
	"io"
	"net/http"
	"os"
	"os/signal"
	"sourcecode.social/greatape/goldgorilla/controllers"
	"sourcecode.social/greatape/goldgorilla/models"
	"sourcecode.social/greatape/goldgorilla/repositories"
	"sourcecode.social/greatape/goldgorilla/routers"
	"syscall"
	"time"
)

type App struct {
	conf   *models.ConfigModel
	router *routers.Router
	src    string
}

func (a *App) Init(srcListenAddr string, svcAddr string, logjamBaseUrl string, targetRoom string, iceTCPMUXListenPort uint, customICEHostCandidateIP string) {
	println("initializing ..")
	a.src = srcListenAddr
	var iceServers []webrtc.ICEServer
	iceconfjson, err := os.ReadFile("./ice.servers.json")
	if err != nil {
		println("[E] error reading ice.servers.json: " + err.Error())
	} else {
		err = json.Unmarshal(iceconfjson, &iceServers)
		if err != nil {
			panic("[E] can't parse ice.servers.json: " + err.Error())
		}
	}
	startRejoinCH := make(chan bool, 2)
	a.conf = &models.ConfigModel{
		LogjamBaseUrl:            logjamBaseUrl + "/auxiliary-node",
		TargetRoom:               targetRoom,
		ServiceAddress:           svcAddr,
		ICEServers:               iceServers,
		ICETCPMUXListenPort:      iceTCPMUXListenPort,
		CustomICEHostCandidateIP: customICEHostCandidateIP,
		StartRejoinCH:            &startRejoinCH,
	}
	roomRepo := repositories.NewRoomRepository(a.conf)
	a.router = &routers.Router{}
	respHelper := controllers.NewResponseHelper()
	roomCtrl := controllers.NewRoomController(respHelper, roomRepo, a.conf)

	err = a.router.RegisterRoutes(roomCtrl)
	panicIfErr(err)

	{
		sigs := make(chan os.Signal, 1)
		signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM, syscall.SIGKILL, syscall.SIGILL)
		go func() {
			a.onDie(<-sigs)
		}()
	}
}

func (a *App) Run() {
	go func() {
		*a.conf.StartRejoinCH <- true
		for simplyJoin := range *a.conf.StartRejoinCH {
			if simplyJoin {
				buffer, _ := json.Marshal(map[string]any{"roomId": a.conf.TargetRoom, "svcAddr": a.conf.ServiceAddress})
				body := bytes.NewReader(buffer)
				c := &http.Client{
					Timeout: 8 * time.Second,
				}
				res, err := c.Post(a.conf.LogjamBaseUrl+"/join", "application/json", body)
				if err != nil {
					println(err.Error())
					time.Sleep(4 * time.Second)
					*a.conf.StartRejoinCH <- true
					continue
				}
				if res.StatusCode > 204 {
					resbody, _ := io.ReadAll(res.Body)
					println("get /join "+res.Status, string(resbody))
					time.Sleep(4 * time.Second)
					*a.conf.StartRejoinCH <- true
					continue
				}
			} else {
				reqModel := struct {
					RoomId string `json:"roomId"`
				}{
					RoomId: a.conf.TargetRoom,
				}
				serializedReqBody, err := json.Marshal(reqModel)
				if err != nil {
					println(err.Error())
					*a.conf.StartRejoinCH <- true
					continue
				}

				resp, err := http.Post(a.conf.LogjamBaseUrl+"/rejoin", "application/json", bytes.NewReader(serializedReqBody))
				if err != nil {
					println(err.Error())
					*a.conf.StartRejoinCH <- true
					continue
				}
				if resp.StatusCode > 204 {
					println("/rejoin", resp.Status)
					*a.conf.StartRejoinCH <- true
				}
			}
		}
	}()
	err := a.router.Serve(a.src)
	panicIfErr(err)
}

func (a *App) onDie(sig os.Signal) {
	fmt.Println("<-", sig)

	os.Exit(0)
}

func panicIfErr(err error) {
	if err != nil {
		panic(err)
	}
}