main.go

package main

import (
	"context"
	"fmt"
	"github.com/akamensky/argparse"
	"github.com/sirupsen/logrus"
	"os"
	"personal-feed/pkg/config"
	_ "personal-feed/pkg/crawlers/registry"
	"personal-feed/pkg/repo"
	_ "personal-feed/pkg/repo/registry"
	"personal-feed/pkg/server"
	"time"
)

func main() {
	var logger = logrus.New()

	parser := argparse.NewParser("personal-feed", "daemon, collect useful info from the internet and to structure it")
	configPath := parser.String("c", "config", &argparse.Options{Required: true, Help: "path to config file"})
	isOnce := parser.Flag("o", "once", &argparse.Options{Required: false, Help: "if specified, exits after one cycle"})
	skipHandlingSources := parser.Flag("s", "skip-handling-sources", &argparse.Options{Required: false, Help: "if specified, updating cycle is not turned-on"})
	generateLiquibaseCfg := parser.Flag("l", "generate-liquibase", &argparse.Options{Required: false, Help: "generate liquibase.properties file from config"})
	allowCORS := parser.Flag("", "allow-cors", &argparse.Options{Required: false, Help: "add header Access-Control-Allow-Origin:*"})
	err := parser.Parse(os.Args)
	if err != nil {
		logger.Errorf("unable to parse arguments: %s" + err.Error())
		os.Exit(1)
	}

	configFile, err := os.Open(*configPath)
	if err != nil {
		logger.Fatalf("unable to open file: %s", *configPath)
	}
	currConfig, err := config.Load(configFile)
	if err != nil {
		logger.Errorf("unable to load config: %s" + err.Error())
		os.Exit(1)
	}

	currConfig.AllowCORS = false
	if allowCORS != nil && *allowCORS {
		currConfig.AllowCORS = true
	}

	ctx := context.Background()

	if generateLiquibaseCfg != nil && *generateLiquibaseCfg {
		repoClient, err := repo.NewRepo(ctx, currConfig.Repo, logger)
		if err != nil {
			panic(err)
		}
		propertiesFileContent, err := repoClient.GenerateLiquibaseProperties()
		if err != nil {
			panic(err)
		}
		fmt.Println(propertiesFileContent)
		os.Exit(0)
	}

	currServer := server.NewServer(currConfig, logger)
	defer currServer.Close()

	if skipHandlingSources != nil && *skipHandlingSources {
		for {
			fmt.Printf("time:%s, just do nothing\n", time.Now())
			time.Sleep(time.Second)
		}
	}

	for {
		err := currServer.HandleAllSources(context.TODO())
		if err != nil {
			logger.Errorf("server returned error: %s" + err.Error())
		}
		if isOnce != nil && *isOnce {
			break
		}
		time.Sleep(time.Minute)
	}
}