123 lines
3.0 KiB
Go
123 lines
3.0 KiB
Go
package servers
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"acooldomain.co/backend/models"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/docker/docker/api/types/filters"
|
|
"github.com/docker/docker/api/types/volume"
|
|
"github.com/docker/docker/client"
|
|
"github.com/gin-gonic/gin"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
)
|
|
|
|
type Connection struct {
|
|
connection *mongo.Client
|
|
apiClient *client.Client
|
|
}
|
|
|
|
type ContainerLabels struct {
|
|
OwnerId string `json:"user_id"`
|
|
ImageId string `json:"image_id"`
|
|
VolumeId string `json:"volume_id"`
|
|
Type string `json:"type"`
|
|
}
|
|
|
|
type VolumeLabels struct {
|
|
OwnerId string `json:"user_id"`
|
|
ImageId string `json:"image_id"`
|
|
}
|
|
|
|
type ImageLabels struct {
|
|
Type string `json:"type"`
|
|
}
|
|
|
|
func transformContainerPortsToModel(ports []types.Port) []models.Port {
|
|
modelPorts := make([]models.Port, len(ports))
|
|
for index, port := range ports {
|
|
modelPorts[index] = models.Port{
|
|
Number: int(port.PublicPort),
|
|
Protocol: port.Type,
|
|
}
|
|
}
|
|
return modelPorts
|
|
}
|
|
|
|
func (con Connection) GetServers(ctx *gin.Context) {
|
|
volumes, err := con.apiClient.VolumeList(
|
|
context.TODO(),
|
|
volume.ListOptions{
|
|
Filters: filters.NewArgs(filters.Arg("label", "type=GAME")),
|
|
},
|
|
)
|
|
if err != nil {
|
|
ctx.AbortWithError(500, err)
|
|
}
|
|
var volumeLabels VolumeLabels
|
|
var servers []models.ServerInfo
|
|
println("Found %d Containers", len(volumes.Volumes))
|
|
for _, volume := range volumes.Volumes {
|
|
jsonData, err := json.Marshal(volume.Labels)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
json.Unmarshal(jsonData, &volumeLabels)
|
|
|
|
imageNameAndVersion := strings.Split(volumeLabels.ImageId, ":")
|
|
|
|
imageName := imageNameAndVersion[0]
|
|
imageVersion := imageNameAndVersion[1]
|
|
containers, err := con.apiClient.ContainerList(context.Background(), container.ListOptions{
|
|
All: true,
|
|
Filters: filters.NewArgs(filters.Arg("label", "type=GAME"), filters.Arg("label", fmt.Sprintf("volume_id=%s", volume.Name))),
|
|
})
|
|
var state bool
|
|
var ports []models.Port
|
|
if err != nil || len(containers) == 0 {
|
|
state = false
|
|
ports = nil
|
|
} else {
|
|
container := containers[0]
|
|
state = container.State == "running"
|
|
ports = container.Ports
|
|
}
|
|
var serverData models.ServerData
|
|
con.connection.Database("backend").Collection("servers").FindOne(context.TODO(), bson.D{{Key: "volume_id", Value: volume.Name}}).Decode(&serverData)
|
|
|
|
servers = append(servers, models.ServerInfo{
|
|
Id: volume.Name,
|
|
Image: models.ImageInfo{
|
|
Name: imageName,
|
|
Version: imageVersion,
|
|
},
|
|
OwnerId: volumeLabels.OwnerId,
|
|
On: state,
|
|
Ports: ports,
|
|
})
|
|
|
|
}
|
|
|
|
if err != nil {
|
|
ctx.AbortWithError(500, err)
|
|
}
|
|
ctx.JSON(200, servers)
|
|
|
|
}
|
|
|
|
func LoadGroup(group *gin.RouterGroup, mongo_client *mongo.Client) {
|
|
apiClient, err := client.NewClientWithOpts(client.FromEnv)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer apiClient.Close()
|
|
|
|
connection := Connection{connection: mongo_client, apiClient: apiClient}
|
|
group.GET("/", connection.GetServers)
|
|
}
|