server/admin/api/health.go

package api

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"path"
	"strings"
	"time"

	"srs.epita.fr/fic-server/admin/pki"

	"github.com/gin-gonic/gin"
)

var TimestampCheck = "submissions"

func declareHealthRoutes(router *gin.RouterGroup) {
	router.GET("/timestamps.json", func(c *gin.Context) {
		stat, err := os.Stat(TimestampCheck)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusNotFound, gin.H{"errmsg": fmt.Sprintf("timestamp.json: %s", err.Error())})
			return
		}
		now := time.Now().UTC()
		c.JSON(http.StatusOK, gin.H{
			"frontend": stat.ModTime().UTC(),
			"backend":  now,
			"diffFB":   now.Sub(stat.ModTime()),
		})
	})
	router.GET("/health.json", GetHealth)
}

type healthFileReport struct {
	IdTeam string `json:"id_team,omitempty"`
	Path   string `json:"path"`
	Error  string `json:"error"`
}

func getHealth(pathname string) (ret []healthFileReport) {
	if ds, err := ioutil.ReadDir(pathname); err != nil {
		ret = append(ret, healthFileReport{
			Path:  strings.TrimPrefix(pathname, TimestampCheck),
			Error: fmt.Sprintf("unable to ReadDir: %s", err),
		})
		return
	} else {
		for _, d := range ds {
			p := path.Join(pathname, d.Name())
			if d.IsDir() && d.Name() != ".tmp" && d.Mode()&os.ModeSymlink == 0 {
				ret = append(ret, getHealth(p)...)
			} else if !d.IsDir() && d.Mode()&os.ModeSymlink == 0 && time.Since(d.ModTime()) > 2*time.Second {
				teamDir := strings.TrimPrefix(pathname, TimestampCheck)
				idteam, _ := pki.GetAssociation(path.Join(TeamsDir, teamDir))
				ret = append(ret, healthFileReport{
					IdTeam: idteam,
					Path:   path.Join(teamDir, d.Name()),
					Error:  "existing untreated file",
				})
			}
		}
		return
	}
}

func GetHealth(c *gin.Context) {
	if _, err := os.Stat(TimestampCheck); err != nil {
		c.AbortWithStatusJSON(http.StatusNotFound, gin.H{"errmsg": fmt.Sprintf("health.json: %s", err.Error())})
		return
	}

	c.JSON(http.StatusOK, getHealth(TimestampCheck))
}