backup to s3, backup intervals

backup.go

@@ -0,0 +1,120 @@
+package main
+
+import (
+	"archive/zip"
+	"context"
+	"io"
+	"log"
+	"os"
+	"path/filepath"
+	"time"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	awsConfig "github.com/aws/aws-sdk-go-v2/config"
+	"github.com/aws/aws-sdk-go-v2/service/s3"
+)
+
+type AwsCredentials struct {
+	AccessKeyID     string
+	SecretAccessKey string
+	Region          string
+	Bucket          string
+}
+
+func backupDatabase() {
+	ticker := time.NewTicker(time.Duration(config.BackupIntervalHours) * time.Hour)
+	defer ticker.Stop()
+
+	for {
+		select {
+		case <-ticker.C:
+			ZipDirectory("db", "db.zip")
+			switch config.BackupProvider {
+			case "aws":
+				S3Upload()
+			default:
+				log.Println("🚫 we only support AWS at this time")
+			}
+		}
+	}
+}
+
+func S3Upload() {
+
+	bucket := getEnv("AWS_BUCKET_NAME")
+	cfg, err := awsConfig.LoadDefaultConfig(context.TODO())
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// Create an Amazon S3 service client
+	client := s3.NewFromConfig(cfg)
+
+	// Upload the file to S3
+	file, err := os.Open("db.zip")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	_, err = client.PutObject(context.TODO(), &s3.PutObjectInput{
+		Bucket: aws.String(bucket),
+		Key:    aws.String("db.zip"),
+		Body:   file,
+	})
+
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	log.Printf("✅ Successfully uploaded %q to %q\n", "db.zip", bucket)
+
+	// delete the file
+	err = os.Remove("db.zip")
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+func ZipDirectory(sourceDir, zipFileName string) error {
+	log.Println("📦 zipping up the database")
+	file, err := os.Create("db.zip")
+	if err != nil {
+		panic(err)
+	}
+	defer file.Close()
+
+	w := zip.NewWriter(file)
+	defer w.Close()
+
+	walker := func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if info.IsDir() {
+			return nil
+		}
+		file, err := os.Open(path)
+		if err != nil {
+			return err
+		}
+		defer file.Close()
+		f, err := w.Create(path)
+		if err != nil {
+			return err
+		}
+
+		_, err = io.Copy(f, file)
+		if err != nil {
+			return err
+		}
+
+		return nil
+	}
+	err = filepath.Walk("db", walker)
+	if err != nil {
+		//panic(err)
+	}
+
+	log.Println("📦 database zipped up!")
+	return nil
+}