feat: wip implement port forwarding
This commit is contained in:
@@ -1,20 +1,11 @@
|
||||
package workspace
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"github.com/docker/docker/api/types/container"
|
||||
"github.com/docker/go-connections/nat"
|
||||
"github.com/google/uuid"
|
||||
"fmt"
|
||||
"github.com/labstack/echo/v4"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"sync"
|
||||
"tesseract/internal/docker"
|
||||
"tesseract/internal/service"
|
||||
"tesseract/internal/template"
|
||||
"time"
|
||||
)
|
||||
|
||||
type createWorkspaceRequestBody struct {
|
||||
@@ -22,72 +13,16 @@ type createWorkspaceRequestBody struct {
|
||||
}
|
||||
|
||||
type updateWorkspaceRequestBody struct {
|
||||
Status string `json:"status"`
|
||||
Status string `json:"status"`
|
||||
PortMappings []portMapping `json:"ports"`
|
||||
}
|
||||
|
||||
func fetchAllWorkspaces(c echo.Context) error {
|
||||
db := service.Database(c)
|
||||
ctx := c.Request().Context()
|
||||
|
||||
var workspaces []workspace
|
||||
err := db.NewSelect().Model(&workspaces).Scan(c.Request().Context())
|
||||
mgr := workspaceManagerFrom(c)
|
||||
workspaces, err := mgr.findAllWorkspaces(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))
|
||||
}
|
||||
|
||||
dockerClient := service.DockerClient(c)
|
||||
sshProxy := service.SSHProxy(c)
|
||||
|
||||
var wg sync.WaitGroup
|
||||
var mu sync.Mutex
|
||||
var errs []error
|
||||
for i, w := range workspaces {
|
||||
wg.Add(1)
|
||||
i, w := i, w
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
|
||||
inspect, err := dockerClient.ContainerInspect(ctx, w.ContainerID)
|
||||
if err != nil {
|
||||
mu.Lock()
|
||||
errs = append(errs, err)
|
||||
mu.Unlock()
|
||||
} else {
|
||||
switch inspect.State.Status {
|
||||
case "running":
|
||||
workspaces[i].Status = statusRunning
|
||||
case "exited":
|
||||
workspaces[i].Status = statusStopped
|
||||
case "paused":
|
||||
workspaces[i].Status = statusPaused
|
||||
case "restarting":
|
||||
workspaces[i].Status = statusRestarting
|
||||
default:
|
||||
workspaces[i].Status = statusUnknown
|
||||
}
|
||||
|
||||
if internalPort := docker.ContainerSSHHostPort(ctx, inspect); internalPort > 0 {
|
||||
if port := sshProxy.FindExternalPort(internalPort); port > 0 {
|
||||
workspaces[i].SSHPort = port
|
||||
}
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
|
||||
if err = errors.Join(errs...); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return c.JSON(http.StatusOK, workspaces)
|
||||
}
|
||||
|
||||
@@ -101,59 +36,18 @@ func updateOrCreateWorkspace(c echo.Context) error {
|
||||
return echo.NewHTTPError(http.StatusNotFound)
|
||||
}
|
||||
|
||||
db := service.Database(c)
|
||||
ctx := c.Request().Context()
|
||||
mgr := workspaceManagerFrom(c)
|
||||
|
||||
var w workspace
|
||||
err := db.NewSelect().Model(&w).
|
||||
Where("name = ?", workspaceName).
|
||||
Scan(ctx)
|
||||
exists, err := mgr.hasWorkspace(ctx, workspaceName)
|
||||
if err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return createWorkspace(c, workspaceName)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
var body updateWorkspaceRequestBody
|
||||
if err = json.NewDecoder(c.Request().Body).Decode(&body); err != nil {
|
||||
return err
|
||||
if !exists {
|
||||
return createWorkspace(c, workspaceName)
|
||||
}
|
||||
|
||||
dockerClient := service.DockerClient(c)
|
||||
sshProxy := service.SSHProxy(c)
|
||||
|
||||
switch status(body.Status) {
|
||||
case statusStopped:
|
||||
if err = stopContainer(ctx, dockerClient, workspaceName); err != nil {
|
||||
return err
|
||||
}
|
||||
w.Status = statusStopped
|
||||
break
|
||||
|
||||
case statusRunning:
|
||||
if err = startContainer(ctx, dockerClient, workspaceName); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
inspect, err := dockerClient.ContainerInspect(ctx, w.ContainerID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
sshPort := docker.ContainerSSHHostPort(ctx, inspect)
|
||||
if sshPort > 0 {
|
||||
if err = sshProxy.NewProxyEntryTo(sshPort); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
w.Status = statusRunning
|
||||
|
||||
break
|
||||
}
|
||||
|
||||
return c.JSON(http.StatusOK, w)
|
||||
return updateWorkspace(c, workspaceName)
|
||||
}
|
||||
|
||||
func createWorkspace(c echo.Context, workspaceName string) error {
|
||||
@@ -162,169 +56,77 @@ func createWorkspace(c echo.Context, workspaceName string) error {
|
||||
return echo.NewHTTPError(http.StatusBadRequest)
|
||||
}
|
||||
|
||||
db := service.Database(c)
|
||||
ctx := c.Request().Context()
|
||||
mgr := workspaceManagerFrom(c)
|
||||
|
||||
tx, err := db.BeginTx(ctx, nil)
|
||||
w, err := mgr.createWorkspace(c.Request().Context(), createWorkspaceOptions{
|
||||
name: workspaceName,
|
||||
imageID: body.ImageID,
|
||||
})
|
||||
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")
|
||||
if errors.Is(err, errImageNotFound) {
|
||||
return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("no image with id %v exists", body.ImageID))
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
dockerClient := service.DockerClient(c)
|
||||
|
||||
containerSSHPort := nat.Port("22/tcp")
|
||||
containerConfig := &container.Config{
|
||||
Tty: true,
|
||||
Image: img.ImageID,
|
||||
ExposedPorts: nat.PortSet{
|
||||
containerSSHPort: {},
|
||||
},
|
||||
}
|
||||
|
||||
hostConfig := &container.HostConfig{
|
||||
PortBindings: nat.PortMap{
|
||||
containerSSHPort: {
|
||||
{"127.0.0.1", ""},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
res, err := dockerClient.ContainerCreate(ctx, containerConfig, hostConfig, nil, nil, workspaceName)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = dockerClient.ContainerStart(ctx, res.ID, container.StartOptions{})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
inspect, err := dockerClient.ContainerInspect(ctx, res.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
ports := inspect.NetworkSettings.Ports[containerSSHPort]
|
||||
if len(ports) == 0 {
|
||||
return errors.New("failed to bind ssh port for container")
|
||||
}
|
||||
|
||||
sshProxy := service.SSHProxy(c)
|
||||
hostPort, err := strconv.Atoi(ports[0].HostPort)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err = sshProxy.NewProxyEntryTo(hostPort); 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),
|
||||
SSHPort: hostPort,
|
||||
Status: statusRunning,
|
||||
}
|
||||
_, 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)
|
||||
}
|
||||
|
||||
func updateWorkspace(c echo.Context, workspaceName string) error {
|
||||
ctx := c.Request().Context()
|
||||
|
||||
var body updateWorkspaceRequestBody
|
||||
err := json.NewDecoder(c.Request().Body).Decode(&body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
mgr := workspaceManagerFrom(c)
|
||||
|
||||
workspace, err := mgr.findWorkspace(ctx, workspaceName)
|
||||
if err != nil {
|
||||
if errors.Is(err, errWorkspaceNotFound) {
|
||||
return echo.NewHTTPError(http.StatusNotFound)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
switch status(body.Status) {
|
||||
case statusStopped:
|
||||
if err = mgr.stopWorkspace(ctx, workspace); err != nil {
|
||||
return err
|
||||
}
|
||||
break
|
||||
|
||||
case statusRunning:
|
||||
if err = mgr.startWorkspace(ctx, workspace); err != nil {
|
||||
return err
|
||||
}
|
||||
break
|
||||
}
|
||||
|
||||
if len(body.PortMappings) > 0 {
|
||||
if err = mgr.addPortMappings(ctx, workspace, body.PortMappings); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return c.JSON(http.StatusOK, workspace)
|
||||
}
|
||||
|
||||
func deleteWorkspace(c echo.Context) error {
|
||||
workspaceName := c.Param("workspaceName")
|
||||
if workspaceName == "" {
|
||||
return echo.NewHTTPError(http.StatusNotFound)
|
||||
}
|
||||
|
||||
db := service.Database(c)
|
||||
dockerClient := service.DockerClient(c)
|
||||
ctx := c.Request().Context()
|
||||
|
||||
tx, err := db.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return echo.NewHTTPError(http.StatusNotFound)
|
||||
}
|
||||
|
||||
var w workspace
|
||||
if err = tx.NewSelect().Model(&w).Scan(ctx); err != nil {
|
||||
_ = tx.Rollback()
|
||||
return echo.NewHTTPError(http.StatusNotFound)
|
||||
}
|
||||
|
||||
inspect, err := inspectContainer(ctx, dockerClient, w.ContainerID)
|
||||
if err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
|
||||
if inspect.State.Running {
|
||||
if err = stopContainer(ctx, dockerClient, w.ContainerID); err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if err = deleteContainer(ctx, dockerClient, w.ContainerID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
res, err := tx.NewDelete().
|
||||
Table("workspaces").
|
||||
Where("name = ?", workspaceName).
|
||||
Exec(ctx)
|
||||
if err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
mgr := workspaceManagerFrom(c)
|
||||
if err := mgr.deleteWorkspace(c.Request().Context(), workspaceName); err != nil {
|
||||
if errors.Is(err, errWorkspaceNotFound) {
|
||||
return echo.NewHTTPError(http.StatusNotFound)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
count, err := res.RowsAffected()
|
||||
if err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
|
||||
if count == 0 {
|
||||
_ = tx.Rollback()
|
||||
return echo.NewHTTPError(http.StatusNotFound)
|
||||
}
|
||||
if count != 1 {
|
||||
_ = tx.Rollback()
|
||||
return echo.NewHTTPError(http.StatusNotFound)
|
||||
}
|
||||
|
||||
if err = tx.Commit(); err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
|
||||
return c.NoContent(http.StatusOK)
|
||||
}
|
||||
|
25
internal/workspace/middleware.go
Normal file
25
internal/workspace/middleware.go
Normal file
@@ -0,0 +1,25 @@
|
||||
package workspace
|
||||
|
||||
import (
|
||||
"github.com/labstack/echo/v4"
|
||||
"tesseract/internal/service"
|
||||
)
|
||||
|
||||
func newWorkspaceManagerMiddleware(services service.Services) echo.MiddlewareFunc {
|
||||
mgr := workspaceManager{
|
||||
db: services.Database,
|
||||
dockerClient: services.DockerClient,
|
||||
reverseProxy: services.ReverseProxy,
|
||||
sshProxy: services.SSHProxy,
|
||||
}
|
||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||
return func(c echo.Context) error {
|
||||
c.Set("workspaceManager", mgr)
|
||||
return next(c)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func workspaceManagerFrom(c echo.Context) workspaceManager {
|
||||
return c.Get("workspaceManager").(workspaceManager)
|
||||
}
|
@@ -2,9 +2,11 @@ package workspace
|
||||
|
||||
import (
|
||||
"github.com/labstack/echo/v4"
|
||||
"tesseract/internal/service"
|
||||
)
|
||||
|
||||
func DefineRoutes(g *echo.Group) {
|
||||
func DefineRoutes(g *echo.Group, services service.Services) {
|
||||
g.Use(newWorkspaceManagerMiddleware(services))
|
||||
g.GET("/workspaces", fetchAllWorkspaces)
|
||||
g.POST("/workspaces/:workspaceName", updateOrCreateWorkspace)
|
||||
g.DELETE("/workspaces/:workspaceName", deleteWorkspace)
|
||||
|
@@ -4,13 +4,16 @@ import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/docker/docker/api/types/container"
|
||||
"github.com/docker/docker/client"
|
||||
"github.com/google/uuid"
|
||||
"github.com/uptrace/bun"
|
||||
"net/url"
|
||||
"regexp"
|
||||
"sync"
|
||||
"tesseract/internal/docker"
|
||||
"tesseract/internal/reverseproxy"
|
||||
"tesseract/internal/service"
|
||||
)
|
||||
|
||||
@@ -31,6 +34,18 @@ type workspace struct {
|
||||
SSHPort int `bun:"-" json:"sshPort,omitempty"`
|
||||
|
||||
Status status `bun:"-" json:"status"`
|
||||
|
||||
PortMappings []portMapping `bun:"rel:has-many,join:id=workspace_id" json:"ports,omitempty"`
|
||||
}
|
||||
|
||||
type portMapping struct {
|
||||
bun.BaseModel `bun:"table:port_mappings,alias:port_mapping"`
|
||||
|
||||
WorkspaceID uuid.UUID `bun:",type:uuid,pk" json:"-"`
|
||||
ContainerPort int `json:"port"`
|
||||
Subdomain string `json:"subdomain"`
|
||||
|
||||
Workspace workspace `bun:"rel:belongs-to,join:workspace_id=id"`
|
||||
}
|
||||
|
||||
// status represents the status of a workspace.
|
||||
@@ -122,5 +137,66 @@ func SyncAll(ctx context.Context, services service.Services) error {
|
||||
return err
|
||||
}
|
||||
|
||||
if err = initializeHTTPProxies(ctx, services.Database, services.DockerClient, services.ReverseProxy); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func initializeHTTPProxies(ctx context.Context, db *bun.DB, dockerClient *client.Client, proxy *reverseproxy.ReverseProxy) error {
|
||||
var mappings []portMapping
|
||||
if err := db.NewSelect().
|
||||
Model(&mappings).
|
||||
Relation("Workspace", func(q *bun.SelectQuery) *bun.SelectQuery {
|
||||
return q.Column("container_id")
|
||||
}).
|
||||
Scan(ctx); err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return nil
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
if len(mappings) == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
var wg sync.WaitGroup
|
||||
var errs []error
|
||||
var mu sync.Mutex
|
||||
|
||||
for _, m := range mappings {
|
||||
m := m
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
var err error
|
||||
defer wg.Done()
|
||||
defer func() {
|
||||
mu.Lock()
|
||||
errs = append(errs, err)
|
||||
mu.Unlock()
|
||||
}()
|
||||
|
||||
inspect, err := dockerClient.ContainerInspect(ctx, m.Workspace.ContainerID)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
u, err := url.Parse(fmt.Sprintf("http://%s:%d", inspect.NetworkSettings.IPAddress, m.ContainerPort))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
proxy.AddEntry(m.Subdomain, u)
|
||||
}()
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
|
||||
if err := errors.Join(errs...); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
359
internal/workspace/workspace_manager.go
Normal file
359
internal/workspace/workspace_manager.go
Normal file
@@ -0,0 +1,359 @@
|
||||
package workspace
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/docker/docker/api/types/container"
|
||||
"github.com/docker/docker/client"
|
||||
"github.com/docker/go-connections/nat"
|
||||
"github.com/google/uuid"
|
||||
"github.com/uptrace/bun"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"sync"
|
||||
"tesseract/internal/docker"
|
||||
"tesseract/internal/reverseproxy"
|
||||
"tesseract/internal/sshproxy"
|
||||
"tesseract/internal/template"
|
||||
"time"
|
||||
)
|
||||
|
||||
// workspaceManager provides functions to manipulate workspaces.
|
||||
type workspaceManager struct {
|
||||
db *bun.DB
|
||||
dockerClient *client.Client
|
||||
reverseProxy *reverseproxy.ReverseProxy
|
||||
sshProxy *sshproxy.SSHProxy
|
||||
}
|
||||
|
||||
type createWorkspaceOptions struct {
|
||||
name string
|
||||
imageID string
|
||||
}
|
||||
|
||||
var errImageNotFound = errors.New("image not found")
|
||||
var errWorkspaceNotFound = errors.New("workspace not found")
|
||||
|
||||
func (mgr workspaceManager) findAllWorkspaces(ctx context.Context) ([]workspace, error) {
|
||||
var workspaces []workspace
|
||||
err := mgr.db.NewSelect().Model(&workspaces).Scan(ctx)
|
||||
if err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return make([]workspace, 0), nil
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(workspaces) == 0 {
|
||||
return make([]workspace, 0), nil
|
||||
}
|
||||
|
||||
var wg sync.WaitGroup
|
||||
var mu sync.Mutex
|
||||
var errs []error
|
||||
|
||||
for i := range workspaces {
|
||||
i := i
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
|
||||
inspect, err := mgr.dockerClient.ContainerInspect(ctx, workspaces[i].ContainerID)
|
||||
if err != nil {
|
||||
mu.Lock()
|
||||
errs = append(errs, err)
|
||||
mu.Unlock()
|
||||
} else {
|
||||
switch inspect.State.Status {
|
||||
case "running":
|
||||
workspaces[i].Status = statusRunning
|
||||
case "exited":
|
||||
workspaces[i].Status = statusStopped
|
||||
case "paused":
|
||||
workspaces[i].Status = statusPaused
|
||||
case "restarting":
|
||||
workspaces[i].Status = statusRestarting
|
||||
default:
|
||||
workspaces[i].Status = statusUnknown
|
||||
}
|
||||
|
||||
if internalPort := docker.ContainerSSHHostPort(ctx, inspect); internalPort > 0 {
|
||||
if port := mgr.sshProxy.FindExternalPort(internalPort); port > 0 {
|
||||
workspaces[i].SSHPort = port
|
||||
}
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
|
||||
if err = errors.Join(errs...); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return workspaces, nil
|
||||
}
|
||||
|
||||
func (mgr workspaceManager) findWorkspace(ctx context.Context, name string) (*workspace, error) {
|
||||
var w workspace
|
||||
err := mgr.db.NewSelect().Model(&w).
|
||||
Where("name = ?", name).
|
||||
Scan(ctx)
|
||||
if err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return nil, errWorkspaceNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return &w, nil
|
||||
}
|
||||
|
||||
func (mgr workspaceManager) hasWorkspace(ctx context.Context, name string) (bool, error) {
|
||||
exists, err := mgr.db.NewSelect().Table("workspaces").
|
||||
Where("name = ?", name).
|
||||
Exists(ctx)
|
||||
if err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return false, nil
|
||||
}
|
||||
return false, err
|
||||
}
|
||||
return exists, nil
|
||||
}
|
||||
|
||||
func (mgr workspaceManager) createWorkspace(ctx context.Context, opts createWorkspaceOptions) (*workspace, error) {
|
||||
tx, err := mgr.db.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var img template.TemplateImage
|
||||
err = tx.NewSelect().Model(&img).
|
||||
Where("image_id = ?", opts.imageID).
|
||||
Scan(ctx)
|
||||
if err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return nil, errImageNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
containerSSHPort := nat.Port("22/tcp")
|
||||
containerConfig := &container.Config{
|
||||
Tty: true,
|
||||
Image: img.ImageID,
|
||||
ExposedPorts: nat.PortSet{
|
||||
containerSSHPort: {},
|
||||
},
|
||||
}
|
||||
|
||||
hostConfig := &container.HostConfig{
|
||||
PortBindings: nat.PortMap{
|
||||
containerSSHPort: {
|
||||
{"127.0.0.1", ""},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
res, err := mgr.dockerClient.ContainerCreate(ctx, containerConfig, hostConfig, nil, nil, opts.name)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = mgr.dockerClient.ContainerStart(ctx, res.ID, container.StartOptions{})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
inspect, err := mgr.dockerClient.ContainerInspect(ctx, res.ID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
ports := inspect.NetworkSettings.Ports[containerSSHPort]
|
||||
if len(ports) == 0 {
|
||||
return nil, errors.New("failed to bind ssh port for container")
|
||||
}
|
||||
|
||||
hostPort, err := strconv.Atoi(ports[0].HostPort)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if err = mgr.sshProxy.NewProxyEntryTo(hostPort); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
id, err := uuid.NewV7()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
w := workspace{
|
||||
ID: id,
|
||||
Name: opts.name,
|
||||
ContainerID: res.ID,
|
||||
ImageTag: img.ImageTag,
|
||||
CreatedAt: time.Now().Format(time.RFC3339),
|
||||
SSHPort: hostPort,
|
||||
Status: statusRunning,
|
||||
}
|
||||
_, err = tx.NewInsert().Model(&w).Exec(ctx)
|
||||
if err != nil {
|
||||
_ = tx.Rollback()
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if err = tx.Commit(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &w, nil
|
||||
}
|
||||
|
||||
func (mgr workspaceManager) deleteWorkspace(ctx context.Context, name string) error {
|
||||
tx, err := mgr.db.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var workspace workspace
|
||||
if err = tx.NewSelect().
|
||||
Model(&workspace).
|
||||
Where("name = ?", name).
|
||||
Scan(ctx); err != nil {
|
||||
_ = tx.Rollback()
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return errWorkspaceNotFound
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
inspect, err := mgr.dockerClient.ContainerInspect(ctx, workspace.ContainerID)
|
||||
if err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
|
||||
if inspect.State.Running {
|
||||
if err = mgr.dockerClient.ContainerStop(ctx, workspace.ContainerID, container.StopOptions{}); err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if err = mgr.dockerClient.ContainerRemove(ctx, workspace.ContainerID, container.RemoveOptions{
|
||||
RemoveVolumes: true,
|
||||
}); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
res, err := tx.NewDelete().
|
||||
Model(&workspace).
|
||||
WherePK().
|
||||
Exec(ctx)
|
||||
if err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
|
||||
count, err := res.RowsAffected()
|
||||
if err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
|
||||
if count == 0 {
|
||||
_ = tx.Rollback()
|
||||
return errWorkspaceNotFound
|
||||
}
|
||||
if count != 1 {
|
||||
_ = tx.Rollback()
|
||||
return errors.New("unexpected number of workspaces deleted")
|
||||
}
|
||||
|
||||
if err = tx.Commit(); err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (mgr workspaceManager) startWorkspace(ctx context.Context, workspace *workspace) error {
|
||||
err := mgr.dockerClient.ContainerStart(ctx, workspace.ContainerID, container.StartOptions{})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
inspect, err := mgr.dockerClient.ContainerInspect(ctx, workspace.ContainerID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
sshPort := docker.ContainerSSHHostPort(ctx, inspect)
|
||||
if sshPort <= 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
if err = mgr.sshProxy.NewProxyEntryTo(sshPort); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
workspace.Status = statusRunning
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (mgr workspaceManager) stopWorkspace(ctx context.Context, workspace *workspace) error {
|
||||
err := mgr.dockerClient.ContainerStop(ctx, workspace.ContainerID, container.StopOptions{})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
workspace.Status = statusStopped
|
||||
return nil
|
||||
}
|
||||
|
||||
func (mgr workspaceManager) addPortMappings(ctx context.Context, workspace *workspace, portMappings []portMapping) error {
|
||||
inspect, err := mgr.dockerClient.ContainerInspect(ctx, workspace.ContainerID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
containerIP := inspect.NetworkSettings.IPAddress
|
||||
|
||||
urls := make([]*url.URL, len(portMappings))
|
||||
for i, m := range portMappings {
|
||||
u, err := url.Parse(fmt.Sprintf("http://%s:%d", containerIP, m.ContainerPort))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
urls[i] = u
|
||||
}
|
||||
|
||||
tx, err := mgr.db.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for i := range portMappings {
|
||||
portMappings[i].WorkspaceID = workspace.ID
|
||||
mgr.reverseProxy.AddEntry(portMappings[i].Subdomain, urls[i])
|
||||
}
|
||||
|
||||
_, err = tx.NewInsert().Model(&portMappings).Exec(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err = tx.Commit(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
workspace.PortMappings = portMappings
|
||||
|
||||
return nil
|
||||
}
|
Reference in New Issue
Block a user