netmaker/turnserver/internal/auth/auth.go

package auth

import (
	"encoding/base64"
	"encoding/json"
	"os"
	"path/filepath"
	"sync"

	"github.com/gravitl/netmaker/logger"
	"github.com/gravitl/netmaker/turnserver/config"
	"github.com/pion/turn/v2"
)

var (
	authMapLock    = &sync.RWMutex{}
	HostMap        = make(map[string]string)
	authBackUpFile = "auth.json"
	backUpFilePath = filepath.Join("/etc/config", authBackUpFile)
)

func init() {
	os.MkdirAll("/etc/config", os.ModePerm)
	// reads creds from disk if any present
	loadCredsFromFile()
}

// RegisterNewHostWithTurn - add new host's creds to map and dumps it to the disk
func RegisterNewHostWithTurn(hostID, hostPass string) {
	authMapLock.Lock()
	HostMap[hostID] = base64.StdEncoding.EncodeToString(turn.GenerateAuthKey(hostID, config.GetTurnHost(), hostPass))
	dumpCredsToFile()
	authMapLock.Unlock()
}

// UnRegisterNewHostWithTurn - deletes the host creds
func UnRegisterNewHostWithTurn(hostID string) {
	authMapLock.Lock()
	delete(HostMap, hostID)
	dumpCredsToFile()
	authMapLock.Unlock()
}

// dumpCredsToFile - saves the creds to file
func dumpCredsToFile() {
	d, err := json.MarshalIndent(HostMap, "", " ")
	if err != nil {
		logger.Log(0, "failed to dump creds to file: ", err.Error())
		return
	}

	err = os.WriteFile(backUpFilePath, d, os.ModePerm)
	if err != nil {
		logger.Log(0, "failed to backup auth data: ", err.Error())
	}
}

// loadCredsFromFile - loads the creds from disk
func loadCredsFromFile() error {
	d, err := os.ReadFile(backUpFilePath)
	if err != nil {
		return err
	}
	return json.Unmarshal(d, &HostMap)
}