Add restore execution functionality to dashboard

internal/view/web/dashboard/executions/list_executions.go

@@ -74,6 +74,7 @@ func listExecutions(
 				html.Div(
 					html.Class("flex justify-start space-x-1"),
 					showExecutionButton(execution),
+					restoreExecutionButton(execution),
 				),
 			),
 			html.Td(component.StatusBadge(execution.Status)),

internal/view/web/dashboard/executions/restore_execution.go

@@ -0,0 +1,246 @@
+package executions
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+
+	lucide "github.com/eduardolat/gomponents-lucide"
+	"github.com/eduardolat/pgbackweb/internal/database/dbgen"
+	"github.com/eduardolat/pgbackweb/internal/util/echoutil"
+	"github.com/eduardolat/pgbackweb/internal/validate"
+	"github.com/eduardolat/pgbackweb/internal/view/web/alpine"
+	"github.com/eduardolat/pgbackweb/internal/view/web/component"
+	"github.com/eduardolat/pgbackweb/internal/view/web/htmx"
+	"github.com/google/uuid"
+	"github.com/labstack/echo/v4"
+	"github.com/maragudk/gomponents"
+	"github.com/maragudk/gomponents/html"
+)
+
+func (h *handlers) restoreExecutionHandler(c echo.Context) error {
+	ctx := c.Request().Context()
+
+	var formData struct {
+		ExecutionID uuid.UUID `form:"execution_id" validate:"required,uuid"`
+		DatabaseID  uuid.UUID `form:"database_id" validate:"omitempty,uuid"`
+		ConnString  string    `form:"conn_string" validate:"omitempty"`
+	}
+	if err := c.Bind(&formData); err != nil {
+		return htmx.RespondToastError(c, err.Error())
+	}
+	if err := validate.Struct(&formData); err != nil {
+		return htmx.RespondToastError(c, err.Error())
+	}
+
+	if formData.DatabaseID == uuid.Nil && formData.ConnString == "" {
+		return htmx.RespondToastError(
+			c, "Database or connection string is required",
+		)
+	}
+
+	if formData.DatabaseID != uuid.Nil && formData.ConnString != "" {
+		return htmx.RespondToastError(
+			c, "Database and connection string cannot be both set",
+		)
+	}
+
+	execution, err := h.servs.ExecutionsService.GetExecution(
+		ctx, formData.ExecutionID,
+	)
+	if err != nil {
+		return c.String(http.StatusInternalServerError, err.Error())
+	}
+
+	if formData.ConnString != "" {
+		err := h.servs.DatabasesService.TestDatabase(
+			ctx, execution.DatabasePgVersion, formData.ConnString,
+		)
+		if err != nil {
+			return htmx.RespondToastError(c, err.Error())
+		}
+	}
+
+	go func() {
+		ctx := context.Background()
+		_ = h.servs.RestorationsService.RunRestoration(
+			ctx,
+			formData.ExecutionID,
+			uuid.NullUUID{
+				Valid: formData.DatabaseID != uuid.Nil,
+				UUID:  formData.DatabaseID,
+			},
+			formData.ConnString,
+		)
+	}()
+
+	return htmx.RespondToastSuccess(
+		c, "Process started, check the restorations page for more details",
+	)
+}
+
+func (h *handlers) restoreExecutionFormHandler(c echo.Context) error {
+	ctx := c.Request().Context()
+
+	executionID, err := uuid.Parse(c.Param("executionID"))
+	if err != nil {
+		return c.String(http.StatusBadRequest, err.Error())
+	}
+
+	execution, err := h.servs.ExecutionsService.GetExecution(ctx, executionID)
+	if err != nil {
+		return c.String(http.StatusInternalServerError, err.Error())
+	}
+
+	databases, err := h.servs.DatabasesService.GetAllDatabases(ctx)
+	if err != nil {
+		return c.String(http.StatusInternalServerError, err.Error())
+	}
+
+	return echoutil.RenderGomponent(c, http.StatusOK, restoreExecutionForm(
+		execution, databases,
+	))
+}
+
+func restoreExecutionForm(
+	execution dbgen.ExecutionsServiceGetExecutionRow,
+	databases []dbgen.DatabasesServiceGetAllDatabasesRow,
+) gomponents.Node {
+	return html.Form(
+		htmx.HxPost("/dashboard/executions/"+execution.ID.String()+"/restore"),
+		htmx.HxConfirm("Are you sure you want to restore this backup?"),
+		htmx.HxDisabledELT("find button"),
+
+		alpine.XData(`{ backup_to: "database" }`),
+
+		html.Input(
+			html.Type("hidden"),
+			html.Name("execution_id"),
+			html.Value(execution.ID.String()),
+		),
+
+		html.Div(
+			html.Class("space-y-2 text-base"),
+
+			component.SelectControl(component.SelectControlParams{
+				Name:     "backup_to",
+				Label:    "Backup to",
+				Required: true,
+				HelpText: "You can restore the backup to an existing database or any other database using a connection string",
+				Children: []gomponents.Node{
+					alpine.XModel("backup_to"),
+					html.Option(
+						html.Value("database"),
+						gomponents.Text("Existing database"),
+						html.Selected(),
+					),
+					html.Option(
+						html.Value("conn_string"),
+						gomponents.Text("Other database"),
+					),
+				},
+			}),
+
+			alpine.Template(
+				alpine.XIf("backup_to === 'database'"),
+				component.SelectControl(component.SelectControlParams{
+					Name:        "database_id",
+					Label:       "Database",
+					Placeholder: "Select a database",
+					Required:    true,
+					Children: []gomponents.Node{
+						component.GMap(
+							databases,
+							func(db dbgen.DatabasesServiceGetAllDatabasesRow) gomponents.Node {
+								return html.Option(
+									html.Value(db.ID.String()),
+									gomponents.Text(db.Name),
+									gomponents.If(
+										db.ID == execution.DatabaseID,
+										html.Selected(),
+									),
+								)
+							},
+						),
+					},
+				}),
+			),
+
+			alpine.Template(
+				alpine.XIf("backup_to === 'conn_string'"),
+				component.InputControl(component.InputControlParams{
+					Name:        "conn_string",
+					Label:       "Connection string",
+					Placeholder: "postgresql://user:password@localhost:5432/mydb",
+					Type:        component.InputTypeText,
+					Required:    true,
+				}),
+			),
+
+			html.Div(
+				html.Class("pt-2"),
+				html.Div(
+					html.Role("alert"),
+					html.Class("alert alert-warning"),
+					lucide.TriangleAlert(),
+					html.Div(
+						html.P(
+							component.BText(fmt.Sprintf(
+								"This restoration uses psql v%s", execution.DatabasePgVersion,
+							)),
+						),
+						component.PText(`
+							Please make sure the database you are restoring to is compatible
+							with this version of psql and double-check that the picked
+							database is the one you want to restore to.
+						`),
+					),
+				),
+			),
+
+			html.Div(
+				html.Class("flex justify-end items-center space-x-2 pt-2"),
+				component.HxLoadingMd(),
+				html.Button(
+					html.Class("btn btn-primary"),
+					html.Type("submit"),
+					component.SpanText("Start restoration"),
+					lucide.Zap(),
+				),
+			),
+		),
+	)
+}
+
+func restoreExecutionButton(execution dbgen.ExecutionsServicePaginateExecutionsRow) gomponents.Node {
+	if execution.Status != "success" || !execution.Path.Valid {
+		return nil
+	}
+
+	mo := component.Modal(component.ModalParams{
+		Size:  component.SizeMd,
+		Title: "Restore backup execution",
+		Content: []gomponents.Node{
+			html.Div(
+				htmx.HxGet("/dashboard/executions/"+execution.ID.String()+"/restore-form"),
+				htmx.HxSwap("outerHTML"),
+				htmx.HxTrigger("intersect once"),
+				html.Class("p-10 flex justify-center"),
+				component.HxLoadingMd(),
+			),
+		},
+	})
+
+	button := html.Button(
+		mo.OpenerAttr,
+		html.Class("btn btn-square btn-sm btn-ghost"),
+		lucide.ArchiveRestore(),
+	)
+
+	return html.Div(
+		html.Class("inline-block tooltip tooltip-right"),
+		html.Data("tip", "Restore backup execution"),
+		mo.HTML,
+		button,
+	)
+}

internal/view/web/dashboard/executions/router.go

@@ -23,4 +23,6 @@ func MountRouter(
 	parent.GET("/list", h.listExecutionsHandler)
 	parent.GET("/:executionID/download", h.downloadExecutionHandler)
 	parent.DELETE("/:executionID", h.deleteExecutionHandler)
+	parent.GET("/:executionID/restore-form", h.restoreExecutionFormHandler)
+	parent.POST("/:executionID/restore", h.restoreExecutionHandler)
 }