111 lines
2.3 KiB
Go
111 lines
2.3 KiB
Go
package workspace
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/json"
|
|
"errors"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/google/uuid"
|
|
"github.com/labstack/echo/v4"
|
|
"net/http"
|
|
"tesseract/internal/service"
|
|
"tesseract/internal/template"
|
|
"time"
|
|
)
|
|
|
|
type createWorkspaceRequestBody struct {
|
|
ImageID string `json:"imageId"`
|
|
}
|
|
|
|
func fetchAllWorkspaces(c echo.Context) error {
|
|
db := service.Database(c)
|
|
|
|
var workspaces []workspace
|
|
err := db.NewSelect().Model(&workspaces).Scan(c.Request().Context())
|
|
if err != nil {
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return c.JSON(http.StatusOK, make([]workspace, 0))
|
|
}
|
|
return err
|
|
}
|
|
|
|
if len(workspaces) == 0 {
|
|
return c.JSON(http.StatusOK, make([]workspace, 0))
|
|
}
|
|
|
|
return c.JSON(http.StatusOK, workspaces)
|
|
}
|
|
|
|
func createWorkspace(c echo.Context) error {
|
|
workspaceName := c.Param("workspaceName")
|
|
if workspaceName == "" {
|
|
return echo.NewHTTPError(http.StatusNotFound)
|
|
}
|
|
|
|
if !workspaceNameRegex.MatchString(workspaceName) {
|
|
return echo.NewHTTPError(http.StatusNotFound)
|
|
}
|
|
|
|
body := createWorkspaceRequestBody{}
|
|
err := json.NewDecoder(c.Request().Body).Decode(&body)
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest)
|
|
}
|
|
|
|
db := service.Database(c)
|
|
|
|
tx, err := db.BeginTx(c.Request().Context(), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var img template.TemplateImage
|
|
err = tx.NewSelect().Model(&img).
|
|
Where("image_id = ?", body.ImageID).
|
|
Scan(c.Request().Context())
|
|
if err != nil {
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "image id not found")
|
|
}
|
|
return err
|
|
}
|
|
|
|
docker := service.DockerClient(c)
|
|
|
|
res, err := docker.ContainerCreate(c.Request().Context(), &container.Config{
|
|
Image: img.ImageID,
|
|
}, nil, nil, nil, workspaceName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = docker.ContainerStart(c.Request().Context(), res.ID, container.StartOptions{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
id, err := uuid.NewV7()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
w := workspace{
|
|
ID: id,
|
|
Name: workspaceName,
|
|
ContainerID: res.ID,
|
|
ImageTag: img.ImageTag,
|
|
CreatedAt: time.Now().Format(time.RFC3339),
|
|
}
|
|
_, err = tx.NewInsert().Model(&w).Exec(c.Request().Context())
|
|
if err != nil {
|
|
_ = tx.Rollback()
|
|
return err
|
|
}
|
|
|
|
if err = tx.Commit(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.JSON(http.StatusOK, w)
|
|
}
|