Make output dir configurable

Author: StopMotionCuber

internal/config/config.go

@@ -4,13 +4,16 @@ import (
 	"errors"
 	"os"
 	"strconv"
+	"strings"
 	"sync"
 )
 
 // Config is a struct that holds the configuration for the application
 type Config struct {
 	// DatabasePath is the path to the sqlite database
 	DatabasePath string
+	// OutputDirectory is the directory where the output files are stored
+	OutputDirectory string
 	// DiscordToken is the token for the discord bot
 	DiscordToken string
 	// MailServer is the address of the mail server
@@ -34,12 +37,16 @@ func GetConfig() (*Config, error) {
 	if instance == nil {
 		once.Do(func() {
 			instance = &Config{
-				DatabasePath: "./kindExport.sqlite",
-				DiscordToken: "",
-				MailServer:   "",
-				MailPort:     587,
-				MailUser:     "",
-				MailPassword: "",
+				DatabasePath:    "./kindExport.sqlite",
+				OutputDirectory: "./output",
+				DiscordToken:    "",
+				MailServer:      "",
+				MailPort:        587,
+				MailUser:        "",
+				MailPassword:    "",
+			}
+			if os.Getenv("OUTPUT_DIRECTORY") != "" {
+				instance.OutputDirectory = strings.TrimRight(os.Getenv("OUTPUT_DIRECTORY"), "/")
 			}
 			if os.Getenv("DATABASE_PATH") != "" {
 				instance.DatabasePath = os.Getenv("DATABASE_PATH")

internal/scrape/substack_scraper.go

@@ -11,6 +11,7 @@ import (
 	"golang.org/x/text/transform"
 	"golang.org/x/text/unicode/norm"
 	"google.golang.org/appengine/log"
+	"kindExport/internal/config"
 	"net/http"
 	"net/url"
 	"os"
@@ -244,15 +245,15 @@ func (s SubstackScraper) Scrape(url *string) (*Book, error) {
 	}
 
 	// Create output dir if not already existing
-
-	err = os.MkdirAll("output", os.ModePerm)
+	conf, _ := config.GetConfig()
+	err = os.MkdirAll(fmt.Sprintf("%s/%s", conf.OutputDirectory, book.Title()), os.ModePerm)
 	if err != nil {
 		log.Errorf(nil, "Failed to create output directory: %s", err.Error())
 		return nil, err
 	}
 
 	// We can now create an EPUB from the parsed HTML content
-	epubPath := fmt.Sprintf("output/%s.epub", book.Title())
+	epubPath := fmt.Sprintf("%s/%s.epub", conf.OutputDirectory, book.Title())
 	err = book.Write(epubPath)
 
 	if err != nil {