wehub-resource-sync
110 lines
3.1 KiB
Go
110 lines
3.1 KiB
Go
package explorer
|
|
|
|
import (
|
|
"cmp"
|
|
"encoding/base64"
|
|
"net/http"
|
|
"slices"
|
|
"strings"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/mudler/LocalAI/core/explorer"
|
|
"github.com/mudler/LocalAI/core/http/middleware"
|
|
"github.com/mudler/LocalAI/internal"
|
|
)
|
|
|
|
func Dashboard() echo.HandlerFunc {
|
|
return func(c echo.Context) error {
|
|
summary := map[string]any{
|
|
"Title": "LocalAI API - " + internal.PrintableVersion(),
|
|
"Version": internal.PrintableVersion(),
|
|
"BaseURL": middleware.BaseURL(c),
|
|
}
|
|
|
|
contentType := c.Request().Header.Get("Content-Type")
|
|
accept := c.Request().Header.Get("Accept")
|
|
if strings.Contains(contentType, "application/json") || (accept != "" && !strings.Contains(accept, "html")) {
|
|
// The client expects a JSON response
|
|
return c.JSON(http.StatusOK, summary)
|
|
} else {
|
|
// Render index
|
|
return c.Render(http.StatusOK, "views/explorer", summary)
|
|
}
|
|
}
|
|
}
|
|
|
|
type AddNetworkRequest struct {
|
|
Token string `json:"token"`
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
}
|
|
|
|
type Network struct {
|
|
explorer.TokenData
|
|
Token string `json:"token"`
|
|
}
|
|
|
|
func ShowNetworks(db *explorer.Database) echo.HandlerFunc {
|
|
return func(c echo.Context) error {
|
|
results := []Network{}
|
|
for _, token := range db.TokenList() {
|
|
networkData, exists := db.Get(token) // get the token data
|
|
hasWorkers := false
|
|
for _, cluster := range networkData.Clusters {
|
|
if len(cluster.Workers) > 0 {
|
|
hasWorkers = true
|
|
break
|
|
}
|
|
}
|
|
if exists && hasWorkers {
|
|
results = append(results, Network{TokenData: networkData, Token: token})
|
|
}
|
|
}
|
|
|
|
// order by number of clusters
|
|
slices.SortFunc(results, func(a, b Network) int {
|
|
return cmp.Compare(len(b.Clusters), len(a.Clusters))
|
|
})
|
|
|
|
return c.JSON(http.StatusOK, results)
|
|
}
|
|
}
|
|
|
|
func AddNetwork(db *explorer.Database) echo.HandlerFunc {
|
|
return func(c echo.Context) error {
|
|
request := new(AddNetworkRequest)
|
|
if err := c.Bind(request); err != nil {
|
|
return c.JSON(http.StatusBadRequest, map[string]any{"error": "Cannot parse JSON"})
|
|
}
|
|
|
|
if request.Token == "" {
|
|
return c.JSON(http.StatusBadRequest, map[string]any{"error": "Token is required"})
|
|
}
|
|
|
|
if request.Name == "" {
|
|
return c.JSON(http.StatusBadRequest, map[string]any{"error": "Name is required"})
|
|
}
|
|
|
|
if request.Description == "" {
|
|
return c.JSON(http.StatusBadRequest, map[string]any{"error": "Description is required"})
|
|
}
|
|
|
|
// TODO: check if token is valid, otherwise reject
|
|
// try to decode the token from base64
|
|
_, err := base64.StdEncoding.DecodeString(request.Token)
|
|
if err != nil {
|
|
return c.JSON(http.StatusBadRequest, map[string]any{"error": "Invalid token"})
|
|
}
|
|
|
|
if _, exists := db.Get(request.Token); exists {
|
|
return c.JSON(http.StatusBadRequest, map[string]any{"error": "Token already exists"})
|
|
}
|
|
err = db.Set(request.Token, explorer.TokenData{Name: request.Name, Description: request.Description})
|
|
if err != nil {
|
|
return c.JSON(http.StatusInternalServerError, map[string]any{"error": "Cannot add token"})
|
|
}
|
|
|
|
return c.JSON(http.StatusOK, map[string]any{"message": "Token added"})
|
|
}
|
|
}
|