main.go

package main

import (
	"strings"

	"github.com/kelseyhightower/envconfig"
	log "github.com/sirupsen/logrus"
	flag "github.com/spf13/pflag"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

type Config struct {
	DatabaseFile       string `required:"false" split_words:"true"`
	LogFormat          string `required:"false" split_words:"true"`
	NotifySlack        bool   `required:"false" split_words:"true"`
	RapidApiKey        string `required:"false" split_words:"true"`
	Site               string `required:"false" split_words:"true"`
	SlackChannelID     string `required:"true" split_words:"true"`
	SlackToken         string `required:"true" split_words:"true"`
	Tag                string `required:"true" split_words:"true"`
	TwitterBearerToken string `required:"false" split_words:"true"`
}

func LoadConfig() *Config {
	var config Config
	err := envconfig.Process("", &config)
	if err != nil {
		panic(err)
	}

	return &config
}
func CreateDB(databaseFile string) (*gorm.DB, error) {
	if databaseFile == "" {
		databaseFile = "stackoverflow-notifications.db"
	}

	db, err := gorm.Open(sqlite.Open(databaseFile), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Silent),
	})
	if err != nil {
		return db, err
	}

	return db, nil
}

type processor func(*Config, *gorm.DB) error

func main() {
	services := flag.String("services", "", "comma-separated list of services to process")
	notifySlack := flag.Bool("notify-slack", true, "whether to notify slack or not")
	flag.Parse()

	config := LoadConfig()
	if config.LogFormat == "json" {
		log.SetFormatter(&log.JSONFormatter{})
	}

	if !*notifySlack {
		config.NotifySlack = false
	}

	if config.Tag == "" {
		log.Fatal("No TAG environment variable specified")
	}

	db, err := CreateDB(config.DatabaseFile)
	if err != nil {
		log.WithError(err).Fatal("error creating db")
	}

	processorMap := map[string]processor{
		"devto":              processDevtoArticles,
		"github":             processGithubRepositories,
		"hackernews_comment": processHackernewsComments,
		"hackernews_story":   processHackernewsStories,
		"mastodon":           processMastodon,
		"medium":             processMediumArticles,
		"reddit":             processRedditPosts,
		"stackoverflow":      processStackoverflow,
		"twitter":            processTwitter,
	}

	// allow disabling services
	if len(*services) > 0 {
		enabledServices := map[string]bool{}
		for _, service := range strings.Split(*services, ",") {
			enabledServices[service] = true
		}

		disabledServices := []string{}
		for service := range processorMap {
			if !enabledServices[service] {
				disabledServices = append(disabledServices, service)
			}
		}

		for _, service := range disabledServices {
			log.WithField("service", service).Info("Disabling service")
			delete(processorMap, service)
		}
	}

	for service, processor := range processorMap {
		log.WithField("service", service).Info("Processing service")
		if err := processor(config, db); err != nil {
			log.WithError(err).WithField("service", service).Fatal("error processing")
		}
	}
}