Add custom go-cron

Dockerfile

@@ -1,14 +1,27 @@
-FROM alpine:3.19
+FROM alpine:3.20 AS build
+
+WORKDIR /app
+
+RUN apk update \
+	&& apk upgrade \
+	&& apk add go
+
+COPY main.go /app/main.go
+
+RUN go mod init github.com/itbm/postgresql-backup-s3 \
+	&& go get github.com/robfig/cron/v3 \
+	&& go build -o out/go-cron
+
+FROM alpine:3.20
 LABEL maintainer="ITBM"
 
 RUN apk update \
 	&& apk upgrade \
-	&& apk add coreutils postgresql16-client aws-cli openssl curl \
-	&& curl -L --insecure https://github.com/odise/go-cron/releases/download/v0.0.7/go-cron-linux.gz | zcat > /usr/local/bin/go-cron \
- 	&& chmod u+x /usr/local/bin/go-cron \
-	&& apk del curl \
+	&& apk add coreutils postgresql16-client aws-cli openssl \
 	&& rm -rf /var/cache/apk/*
 
+COPY --from=build /app/out/go-cron /usr/local/bin/go-cron
+
 ENV POSTGRES_DATABASE **None**
 ENV POSTGRES_HOST **None**
 ENV POSTGRES_PORT 5432

main.go

@@ -0,0 +1,120 @@
+package main
+
+import (
+	"bufio"
+	"context"
+	"fmt"
+	"io"
+	"os"
+	"os/exec"
+	"strings"
+	"time"
+
+	"github.com/robfig/cron/v3"
+)
+
+func timestampedPrint(prefix, message string) {
+	timestamp := time.Now().Format("2006-01-02 15:04:05")
+	fmt.Printf("[%s] %s: %s", timestamp, prefix, message)
+}
+
+func streamOutput(prefix string, reader io.Reader) {
+	scanner := bufio.NewScanner(reader)
+	for scanner.Scan() {
+		timestampedPrint(prefix, scanner.Text()+"\n")
+	}
+	if err := scanner.Err(); err != nil {
+		timestampedPrint("ERROR", fmt.Sprintf("Error reading output: %v\n", err))
+	}
+}
+
+func validateSchedule(schedule string) error {
+	// Handle @every syntax
+	if strings.HasPrefix(schedule, "@every ") {
+		duration := strings.TrimPrefix(schedule, "@every ")
+		_, err := time.ParseDuration(duration)
+		return err
+	}
+
+	// Handle standard cron syntax and descriptors
+	parser := cron.NewParser(cron.Minute | cron.Hour | cron.Dom | cron.Month | cron.Dow | cron.Descriptor)
+
+	_, err := parser.Parse(schedule)
+	return err
+}
+
+func main() {
+	if len(os.Args) < 3 {
+		fmt.Println("Usage: go-cron <schedule> <command> [args...]")
+		os.Exit(1)
+	}
+
+	schedule := os.Args[1]
+	command := os.Args[2]
+	args := os.Args[3:]
+
+	// Validate schedule
+	if err := validateSchedule(schedule); err != nil {
+		timestampedPrint("ERROR", fmt.Sprintf("Invalid schedule format: %v\n", err))
+		os.Exit(1)
+	}
+
+	// Validate command
+	if _, err := exec.LookPath(command); err != nil {
+		timestampedPrint("ERROR", fmt.Sprintf("Command not found: %s\n", command))
+		os.Exit(1)
+	}
+
+	c := cron.New()
+
+	_, err := c.AddFunc(schedule, func() {
+		timestampedPrint("INFO", fmt.Sprintf("Executing command: %s %v\n", command, args))
+
+		ctx, cancel := context.WithTimeout(context.Background(), 1*time.Hour)
+		defer cancel()
+
+		cmd := exec.CommandContext(ctx, command, args...)
+
+		stdout, err := cmd.StdoutPipe()
+		if err != nil {
+			timestampedPrint("ERROR", fmt.Sprintf("Error creating stdout pipe: %v\n", err))
+			return
+		}
+
+		stderr, err := cmd.StderrPipe()
+		if err != nil {
+			timestampedPrint("ERROR", fmt.Sprintf("Error creating stderr pipe: %v\n", err))
+			return
+		}
+
+		err = cmd.Start()
+		if err != nil {
+			timestampedPrint("ERROR", fmt.Sprintf("Error starting command: %v\n", err))
+			return
+		}
+
+		go streamOutput("STDOUT", stdout)
+		go streamOutput("STDERR", stderr)
+
+		err = cmd.Wait()
+		if err != nil {
+			if ctx.Err() == context.DeadlineExceeded {
+				timestampedPrint("ERROR", "Command timed out after 1 hour\n")
+			} else {
+				timestampedPrint("ERROR", fmt.Sprintf("Command finished with error: %v\n", err))
+			}
+		} else {
+			timestampedPrint("INFO", "Command finished successfully\n")
+		}
+	})
+
+	if err != nil {
+		timestampedPrint("ERROR", fmt.Sprintf("Error adding cron job: %v\n", err))
+		os.Exit(1)
+	}
+
+	timestampedPrint("INFO", fmt.Sprintf("Cron job scheduled: %s\n", schedule))
+	timestampedPrint("INFO", fmt.Sprintf("Command to run: %s %v\n", command, strings.Join(args, " ")))
+
+	c.Run()
+}