108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package apps
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/texm/dokku-go"
|
|
"gitlab.com/texm/shokku/internal/env"
|
|
"gitlab.com/texm/shokku/internal/server/dto"
|
|
"net/http"
|
|
)
|
|
|
|
func isMountInSlice(mount dokku.StorageBindMount, mounts []dokku.StorageBindMount) bool {
|
|
for _, m := range mounts {
|
|
if m == mount {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func GetAppStorage(e *env.Env, c echo.Context) error {
|
|
var req dto.GetAppStorageRequest
|
|
if err := dto.BindRequest(c, &req); err != nil {
|
|
return err.ToHTTP()
|
|
}
|
|
|
|
report, err := e.Dokku.GetAppStorageReport(req.Name)
|
|
if err != nil {
|
|
return fmt.Errorf("getting app storage report: %w", err)
|
|
}
|
|
|
|
var allMounts []dokku.StorageBindMount
|
|
allMounts = append(allMounts, report.RunMounts...)
|
|
allMounts = append(allMounts, report.DeployMounts...)
|
|
allMounts = append(allMounts, report.BuildMounts...)
|
|
|
|
seenMap := map[string]bool{}
|
|
mounts := []dto.StorageMount{}
|
|
for _, dokkuMount := range allMounts {
|
|
if _, seen := seenMap[dokkuMount.String()]; seen {
|
|
continue
|
|
}
|
|
seenMap[dokkuMount.String()] = true
|
|
mounts = append(mounts, dto.StorageMount{
|
|
HostDir: dokkuMount.HostDir,
|
|
ContainerDir: dokkuMount.ContainerDir,
|
|
IsBuildMount: isMountInSlice(dokkuMount, report.BuildMounts),
|
|
IsRunMount: isMountInSlice(dokkuMount, report.RunMounts),
|
|
IsDeployMount: isMountInSlice(dokkuMount, report.DeployMounts),
|
|
})
|
|
}
|
|
|
|
return c.JSON(http.StatusOK, dto.GetAppStorageResponse{
|
|
Mounts: mounts,
|
|
})
|
|
}
|
|
|
|
func MountAppStorage(e *env.Env, c echo.Context) error {
|
|
var req dto.AlterAppStorageRequest
|
|
if err := dto.BindRequest(c, &req); err != nil {
|
|
return err.ToHTTP()
|
|
}
|
|
|
|
if req.StorageType == "New Storage" {
|
|
// TODO: actually chown properly
|
|
err := e.Dokku.EnsureStorageDirectory(req.Name, dokku.StorageChownOptionHerokuish)
|
|
if err != nil {
|
|
return fmt.Errorf("ensuring app storage dir: %w", err)
|
|
}
|
|
}
|
|
|
|
mount := dokku.StorageBindMount{
|
|
HostDir: req.HostDir,
|
|
ContainerDir: req.ContainerDir,
|
|
}
|
|
if err := e.Dokku.MountAppStorage(req.Name, mount); err != nil {
|
|
return fmt.Errorf("mounting app storage dir: %w", err)
|
|
}
|
|
|
|
return c.NoContent(http.StatusOK)
|
|
}
|
|
|
|
func UnmountAppStorage(e *env.Env, c echo.Context) error {
|
|
var req dto.AlterAppStorageRequest
|
|
if err := dto.BindRequest(c, &req); err != nil {
|
|
return err.ToHTTP()
|
|
}
|
|
|
|
mount := dokku.StorageBindMount{
|
|
HostDir: req.HostDir,
|
|
ContainerDir: req.ContainerDir,
|
|
}
|
|
if err := e.Dokku.UnmountAppStorage(req.Name, mount); err != nil {
|
|
return fmt.Errorf("unmounting app storage dir: %w", err)
|
|
}
|
|
|
|
if req.RestartApp {
|
|
go func() {
|
|
if _, err := e.Dokku.RestartApp(req.Name, nil); err != nil {
|
|
log.Error().Err(err).Msg("error while restarting app")
|
|
}
|
|
}()
|
|
}
|
|
|
|
return c.NoContent(http.StatusOK)
|
|
}
|