main.go

package main

import (
	"math/rand"
	"net/http"
	"time"

	"github.com/rwlist/coub/pkg/coubs"
	"github.com/rwlist/coub/pkg/local"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/credentials"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/s3"
	"github.com/davecgh/go-spew/spew"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"

	"github.com/rwlist/coub/pkg/conf"

	"github.com/prometheus/client_golang/prometheus/promhttp"
	log "github.com/sirupsen/logrus"
)

func main() {
	rand.Seed(time.Now().UnixNano())

	log.SetFormatter(&log.JSONFormatter{})
	log.SetReportCaller(true)
	log.SetLevel(log.DebugLevel)

	cfg, err := conf.ParseEnv()
	if err != nil {
		log.WithError(err).Fatal("failed to parse config from env")
	}

	go func() {
		mux := http.NewServeMux()
		mux.Handle("/metrics", promhttp.Handler())
		err := http.ListenAndServe(cfg.PrometheusBind, mux) //nolint:govet
		if err != nil && err != http.ErrServerClosed {
			log.WithError(err).Fatal("prometheus server error")
		}
	}()

	db, err := gorm.Open(postgres.Open(cfg.PostgresDSN), &gorm.Config{})
	if err != nil {
		log.WithError(err).Fatal("failed to connect to postgres")
	}
	db = db.Debug()

	err = local.AutoMigrate(db)
	if err != nil {
		log.WithError(err).Fatal("failed to migrate tables")
	}

	cookies := coubs.NewCookies(db)
	err = cookies.AutoMigrate()
	if err != nil {
		log.WithError(err).Fatal("failed to migrate kv table")
	}

	c, err := cookies.Get()
	spew.Dump(c, err)

	// Configure to use MinIO Server
	s3Config := &aws.Config{
		Credentials:      credentials.NewStaticCredentials(cfg.S3AccessKey, cfg.S3SecretKey, ""),
		Endpoint:         aws.String(cfg.S3Endpoint),
		Region:           aws.String(cfg.S3Region),
		DisableSSL:       aws.Bool(false),
		S3ForcePathStyle: aws.Bool(true),
	}
	newSession, err := session.NewSession(s3Config)
	if err != nil {
		log.WithError(err).Fatal("failed to create new session")
	}

	s3Client := s3.New(newSession)

	state := local.NewSharedState()
	cli := coubs.NewClient(cookies)
	downloader := local.NewDownloader(cli, s3Client, db, cfg)
	backup := local.NewBackup(downloader, cli, db, state)

	if cfg.EnableBackup {
		go func() {
			usernames := cfg.BackupProfiles

			for _, username := range usernames {
				log.WithField("username", username).Info("backup started")
				err = backup.Profile(username)
				if err != nil {
					log.WithError(err).Fatal("failed to download profile")
				}
			}
		}()
	}

	server := local.NewServer(s3Client, db, cfg, state)
	r := server.Router()
	err = http.ListenAndServe(cfg.BindHTTP, r)
	if err != nil {
		log.WithError(err).Fatal("http server error")
	}
}