feat: add proxy support for extensions

Author: raghavyuva

api/internal/features/extension/engine/proxy.go

@@ -0,0 +1,82 @@
+package engine
+
+import (
+	"fmt"
+	"strconv"
+
+	"github.com/raghavyuva/caddygo"
+	"github.com/raghavyuva/nixopus-api/internal/config"
+	"github.com/raghavyuva/nixopus-api/internal/features/deploy/tasks"
+	"github.com/raghavyuva/nixopus-api/internal/features/ssh"
+	"github.com/raghavyuva/nixopus-api/internal/types"
+)
+
+type proxyModule struct{}
+
+func (proxyModule) Type() string { return "proxy" }
+
+func AddDomainToProxy(domain string, port string) error {
+	client := tasks.GetCaddyClient()
+	p, err := strconv.Atoi(port)
+	if err != nil {
+		return fmt.Errorf("invalid port: %w", err)
+	}
+	upstreamHost := config.AppConfig.SSH.Host
+	if err := client.AddDomainWithAutoTLS(domain, upstreamHost, p, caddygo.DomainOptions{}); err != nil {
+		return err
+	}
+	client.Reload()
+	return nil
+}
+
+func UpdateDomainInProxy(domain string, port string) error {
+	return AddDomainToProxy(domain, port)
+}
+
+func RemoveDomainFromProxy(domain string) error {
+	client := tasks.GetCaddyClient()
+	if err := client.DeleteDomain(domain); err != nil {
+		return err
+	}
+	client.Reload()
+	return nil
+}
+
+func (proxyModule) Execute(sshClient *ssh.SSH, step types.SpecStep, vars map[string]interface{}) (string, func(), error) {
+	action, _ := step.Properties["action"].(string)
+	domain, _ := step.Properties["domain"].(string)
+	port, _ := step.Properties["port"].(string)
+
+	switch action {
+	case "add":
+		if domain == "" || port == "" {
+			return "", nil, fmt.Errorf("domain and port are required")
+		}
+		if err := AddDomainToProxy(domain, port); err != nil {
+			return "", nil, err
+		}
+		return fmt.Sprintf("proxy added for %s -> %s:%s", domain, config.AppConfig.SSH.Host, port), nil, nil
+	case "update":
+		if domain == "" || port == "" {
+			return "", nil, fmt.Errorf("domain and port are required")
+		}
+		if err := UpdateDomainInProxy(domain, port); err != nil {
+			return "", nil, err
+		}
+		return fmt.Sprintf("proxy updated for %s -> %s:%s", domain, config.AppConfig.SSH.Host, port), nil, nil
+	case "remove":
+		if domain == "" {
+			return "", nil, fmt.Errorf("domain is required")
+		}
+		if err := RemoveDomainFromProxy(domain); err != nil {
+			return "", nil, err
+		}
+		return fmt.Sprintf("proxy removed for %s", domain), nil, nil
+	default:
+		return "", nil, fmt.Errorf("unsupported proxy action: %s", action)
+	}
+}
+
+func init() {
+	RegisterModule(proxyModule{})
+}

api/internal/features/extension/parser/validate.go

@@ -81,6 +81,7 @@ func (p *Parser) validateStep(step ExecutionStep) error {
 		"user":           p.validateUserStep,
 		"docker":         p.validateDockerStep,
 		"docker_compose": p.validateDockerComposeStep,
+		"proxy":          p.validateProxyStep,
 	}
 	v, ok := validators[step.Type]
 	if !ok {
@@ -149,6 +150,21 @@ func (p *Parser) validateDockerComposeStep(step ExecutionStep) error {
 	}
 }
 
+func (p *Parser) validateProxyStep(step ExecutionStep) error {
+	if err := p.requireProps(step, map[string]bool{"action": true}); err != nil {
+		return err
+	}
+	action, _ := step.Properties["action"].(string)
+	switch action {
+	case "add", "update":
+		return p.requireProps(step, map[string]bool{"domain": true, "port": true})
+	case "remove":
+		return p.requireProps(step, map[string]bool{"domain": true})
+	default:
+		return fmt.Errorf("unsupported proxy action: %s", action)
+	}
+}
+
 func (p *Parser) requireProps(step ExecutionStep, required map[string]bool) error {
 	for key := range required {
 		if _, ok := step.Properties[key]; !ok {