Add purge configuration and scheduled purger functionality

config.go

@@ -44,6 +44,7 @@ type Config struct {
 	ECS       ECSCfg     `yaml:"ecs"`
 	Link      Link       `yaml:"link"`
 	Auth      *Auth      `yaml:"auth"`
+	Purge     *Purge     `yaml:"purge"`
 
 	compatV1  bool
 	localMode bool
@@ -234,7 +235,8 @@ func NewConfig(ctx context.Context, p *ConfigParams) (*Config, error) {
 		ECS: ECSCfg{
 			Region: os.Getenv("AWS_REGION"),
 		},
-		Auth: nil,
+		Auth:  nil,
+		Purge: nil,
 
 		localMode: p.LocalMode,
 		compatV1:  p.CompatV1,
@@ -318,6 +320,12 @@ func NewConfig(ctx context.Context, p *ConfigParams) (*Config, error) {
 	if err := cfg.fillECSDefaults(ctx); err != nil {
 		slog.Warn(f("failed to fill ECS defaults: %s", err))
 	}
+
+	if cfg.Purge != nil {
+		if err := cfg.Purge.Validate(); err != nil {
+			return nil, fmt.Errorf("invalid purge config: %w", err)
+		}
+	}
 	return cfg, nil
 }
 

mirage.go

@@ -74,9 +74,10 @@ func (m *Mirage) Run(ctx context.Context) error {
 		}(v.ListenPort)
 	}
 
-	wg.Add(2)
+	wg.Add(3)
 	go m.syncECSToMirage(ctx, &wg)
 	go m.RunAccessCountCollector(ctx, &wg)
+	go m.RunScheduledPurger(ctx, &wg)
 	wg.Wait()
 	slog.Info("shutdown mirage-ecs")
 	select {
@@ -147,6 +148,32 @@ func (m *Mirage) RunAccessCountCollector(ctx context.Context, wg *sync.WaitGroup
 	}
 }
 
+func (m *Mirage) RunScheduledPurger(ctx context.Context, wg *sync.WaitGroup) {
+	defer wg.Done()
+	p := m.Config.Purge
+	if p == nil {
+		slog.Debug("Purge is not configured")
+		return
+	}
+	slog.Info(f("starting up RunScheduledPurger() schedule: %s", p.cron.String()))
+	for {
+		now := time.Now().Add(time.Minute)
+		next := p.cron.Next(now)
+		slog.Info(f("next purge invocation at: %s", next))
+		duration := time.Until(next)
+		select {
+		case <-ctx.Done():
+			slog.Info("RunScheduledPurger() is done")
+			return
+		case <-time.After(duration):
+			slog.Info("scheduled purge invoked")
+			if err := m.WebApi.purge(ctx, p.purgeParams); err != nil {
+				slog.Warn(err.Error())
+			}
+		}
+	}
+}
+
 const (
 	CloudWatchMetricNameSpace = "mirage-ecs"
 	CloudWatchMetricName      = "RequestCount"

purge.go

@@ -0,0 +1,34 @@
+package mirageecs
+
+import (
+	"fmt"
+
+	"github.com/winebarrel/cronplan"
+)
+
+type Purge struct {
+	Schedule string           `json:"schedule"`
+	Request  *APIPurgeRequest `json:"request"`
+
+	purgeParams *PurgeParams
+	cron        *cronplan.Expression
+}
+
+func (p *Purge) Validate() error {
+	cron, err := cronplan.Parse(p.Schedule)
+	if err != nil {
+		return fmt.Errorf("invalid schedule expression %s: %w", p.Schedule, err)
+	}
+	p.cron = cron
+
+	if p.Request == nil {
+		return fmt.Errorf("purge request is required")
+	}
+	purgeParams, err := p.Request.Validate()
+	if err != nil {
+		return fmt.Errorf("invalid purge request: %w", err)
+	}
+	p.purgeParams = purgeParams
+
+	return nil
+}

types.go

@@ -59,10 +59,10 @@ func (r *APILaunchRequest) MergeForm(form url.Values) {
 }
 
 type APIPurgeRequest struct {
-	Duration      json.Number `json:"duration" form:"duration"`
-	Excludes      []string    `json:"excludes" form:"excludes"`
-	ExcludeTags   []string    `json:"exclude_tags" form:"exclude_tags"`
-	ExcludeRegexp string      `json:"exclude_regexp" form:"exclude_regexp"`
+	Duration      json.Number `json:"duration" form:"duration" yaml:"duration"`
+	Excludes      []string    `json:"excludes" form:"excludes" yaml:"excludes"`
+	ExcludeTags   []string    `json:"exclude_tags" form:"exclude_tags" yaml:"exclude_tags"`
+	ExcludeRegexp string      `json:"exclude_regexp" form:"exclude_regexp" yaml:"exclude_regexp"`
 }
 
 type PurgeParams struct {

webapi.go

@@ -240,11 +240,22 @@ func (api *WebApi) ApiAccess(c echo.Context) error {
 }
 
 func (api *WebApi) ApiPurge(c echo.Context) error {
-	code, err := api.purge(c)
+	r := APIPurgeRequest{}
+	if err := c.Bind(&r); err != nil {
+		return c.JSON(http.StatusBadRequest, APICommonResponse{Result: err.Error()})
+	}
+
+	params, err := r.Validate()
 	if err != nil {
-		return c.JSON(code, APICommonResponse{Result: err.Error()})
+		slog.Error(f("purge failed: %s", err))
+		return c.JSON(http.StatusBadRequest, APICommonResponse{Result: err.Error()})
 	}
-	return c.JSON(code, APICommonResponse{Result: "accepted"})
+
+	ctx := c.Request().Context()
+	if err := api.purge(ctx, params); err != nil {
+		return c.JSON(http.StatusInternalServerError, APICommonResponse{Result: err.Error()})
+	}
+	return c.JSON(http.StatusOK, APICommonResponse{Result: "accepted"})
 }
 
 func (api *WebApi) logs(c echo.Context) (int, []string, error) {
@@ -371,22 +382,11 @@ func validateSubdomain(s string) error {
 	return nil
 }
 
-func (api *WebApi) purge(c echo.Context) (int, error) {
-	r := APIPurgeRequest{}
-	if err := c.Bind(&r); err != nil {
-		return http.StatusBadRequest, err
-	}
-
-	p, err := r.Validate()
-	if err != nil {
-		slog.Error(f("purge failed: %s", err))
-		return http.StatusBadRequest, err
-	}
-
-	infos, err := api.runner.List(c.Request().Context(), statusRunning)
+func (api *WebApi) purge(ctx context.Context, p *PurgeParams) error {
+	infos, err := api.runner.List(ctx, statusRunning)
 	if err != nil {
 		slog.Error(f("list ecs failed: %s", err))
-		return http.StatusInternalServerError, err
+		return fmt.Errorf("list tasks failed: %w", err)
 	}
 	slog.Info("purge subdomains",
 		"duration", p.Duration,
@@ -408,7 +408,7 @@ func (api *WebApi) purge(c echo.Context) (int, error) {
 	}
 
 	slog.Info("no subdomains to purge")
-	return http.StatusOK, nil
+	return nil
 }
 
 func (api *WebApi) purgeSubdomains(ctx context.Context, subdomains []string, duration time.Duration) {