helpers.go

package main

import (
	"encoding/json"
	"errors"
	"fmt"
	"log"
	"net/http"
	"strings"

	authz "github.com/CHESSComputing/golib/authz"
	srvConfig "github.com/CHESSComputing/golib/config"
	utils "github.com/CHESSComputing/golib/utils"
	"github.com/gin-gonic/gin"
)

// helper function to extract schema name from schema file name
func schemaName(fname string) string {
	arr := strings.Split(fname, "/")
	return strings.Split(arr[len(arr)-1], ".")[0]
}

// helper function to write data in NDJSON data format
func handleNDJSON(c *gin.Context, records []map[string]any) {
	// Set the Content-Type header to NDJSON
	c.Header("Content-Type", "application/x-ndjson")
	c.Status(http.StatusOK)

	// Use the Gin context's Writer to stream the response
	for _, record := range records {
		// Marshal each record to JSON
		line, err := json.Marshal(record)
		if err != nil {
			// Handle JSON marshalling error
			c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to serialize record"})
			return
		}
		// Write each JSON line followed by a newline
		_, _ = c.Writer.Write(append(line, '\n'))
	}
}

// helper function to adjust timestamp in spec
func adjustTimestamp(specPtr *map[string]any) {
	spec := *specPtr
	if val, ok := spec["date"]; ok {
		switch ts := val.(type) {
		case string:
			epoch, err := utils.RFC3339ToEpoch(ts)
			if err == nil {
				spec["date"] = epoch
			}
		}
	}
	if Verbose > 2 {
		log.Printf("adjusted spec: %+v", spec)
	}
}

func specRecord(tmplRec map[string]any) map[string]any {
	rec := make(map[string]any)
	// TODO: implement logic of getting spec record from tmpl record
	return rec
}

// checkUserBtrAffilication validates user's btr affiliation
func checkUserBtrAffilication(btr string, req *http.Request) error {
	tokenStr := authz.RequestToken(req)
	claims, err := authz.TokenClaims(tokenStr, srvConfig.Config.Authz.ClientID)
	if err != nil {
		return err
	}
	btrs := claims.CustomClaims.Btrs
	user := claims.CustomClaims.User
	groups := claims.CustomClaims.Groups
	if utils.InList("foxdenadmins", groups) {
		// FOXDEN admins are allowed to perform any actions on metadata record
		return nil
	}
	if utils.InList(btr, btrs) {
		// user btr is in btrs list and therefore this user is allowed to perform any actions on metadata record
		return nil
	}
	msg := fmt.Sprintf("[MetaData.main.checkUserBtrAffilication] user '%s' with btrs %+v is not authorized to perform %s action on record with btr=%s", user, btrs, req.Method, btr)
	return errors.New(msg)
}