config.go

package main

import (
	"log"
	"os"
	"sync"
	"time"

	"github.com/gorhill/cronexpr"
	"golang.org/x/sys/unix"
	"gopkg.in/yaml.v3"
)

const (
	DefaultAddress       = ""
	DefaultPort          = 9129
	DefaultCacheDir      = "/var/cache/pacoloco"
	DefaultTTLUnaccessed = 30
	DefaultTTLUnupdated  = 200
	DefaultDBName        = "sqlite-pkg-cache.db"
)

type Repo struct {
	URL                  string     `yaml:"url"`
	URLs                 []string   `yaml:"urls"`
	Mirrorlist           string     `yaml:"mirrorlist"`
	HttpProxy            string     `yaml:"http_proxy"`
	LastMirrorlistCheck  time.Time  `yaml:"-"`
	MirrorlistMutex      sync.Mutex `yaml:"-"`
	LastModificationTime time.Time  `yaml:"-"`
}

type RefreshPeriod struct {
	Cron          string `yaml:"cron"`
	TTLUnaccessed int    `yaml:"ttl_unaccessed_in_days"`
	TTLUnupdated  int    `yaml:"ttl_unupdated_in_days"`
}

type Config struct {
	CacheDir        string           `yaml:"cache_dir"`
	Address         string           `yaml:"address"`
	Port            int              `yaml:"port"`
	Repos           map[string]*Repo `yaml:"repos,omitempty"`
	PurgeFilesAfter int              `yaml:"purge_files_after"`
	DownloadTimeout int              `yaml:"download_timeout"`
	Prefetch        *RefreshPeriod   `yaml:"prefetch"`
	HttpProxy       string           `yaml:"http_proxy"`
	UserAgent       string           `yaml:"user_agent"`
	LogTimestamp    bool             `yaml:"set_timestamp_to_logs"`
}

var config *Config

func parseConfig(raw []byte) *Config {
	result := Config{
		CacheDir: DefaultCacheDir,
		Address:  DefaultAddress,
		Port:     DefaultPort,
		Prefetch: nil,
	}

	if err := yaml.Unmarshal(raw, &result); err != nil {
		log.Fatal(err)
	}

	// validate config
	for name, repo := range result.Repos {
		if repo.URL != "" && len(repo.URLs) > 0 {
			log.Fatalf("repo '%v' specifies both url and urls parameters, please use only one of them", name)
		}
		if repo.URL != "" && repo.Mirrorlist != "" {
			log.Fatalf("repo '%v' specifies both url and mirrorlist parameter, please use only one of them", name)
		}
		if len(repo.URLs) > 0 && repo.Mirrorlist != "" {
			log.Fatalf("repo '%v' specifies both urls and mirrorlist parameter, please use only one of them", name)
		}
		if repo.URL == "" && len(repo.URLs) == 0 && repo.Mirrorlist == "" {
			log.Fatalf("please specify url(s) or mirrorlist for repo '%v'", name)
		}
		// validate Mirrorlist config
		if repo.Mirrorlist != "" && unix.Access(repo.Mirrorlist, unix.R_OK) != nil {
			log.Fatalf("mirrorlist file %v for repo %v does not exist or isn't readable for userid %v", repo.Mirrorlist, name, os.Getuid())
		}
	}

	if result.PurgeFilesAfter < 10*60 && result.PurgeFilesAfter != 0 {
		log.Fatalf("'purge_files_after' period is too low (%v) please specify at least 10 minutes", result.PurgeFilesAfter)
	}

	if unix.Access(result.CacheDir, unix.R_OK|unix.W_OK) != nil {
		log.Fatalf("directory %v does not exist or isn't writable for userid %v", result.CacheDir, os.Getuid())
	}
	// validate Prefetch config

	if result.Prefetch != nil {

		// set default values
		if result.Prefetch.TTLUnaccessed == 0 {
			result.Prefetch.TTLUnaccessed = DefaultTTLUnaccessed
		}
		if result.Prefetch.TTLUnupdated == 0 {
			result.Prefetch.TTLUnupdated = DefaultTTLUnupdated
		}
		// check Prefetch config
		if result.Prefetch.TTLUnaccessed < 0 {
			log.Fatal("'ttl_unaccessed_in_days' value is too low. Please set it to a value greater than 0")
		}
		if result.Prefetch.TTLUnupdated < 0 {
			log.Fatal("'ttl_unupdated_in_days' value is too low. Please set it to a value greater than 0")
		}
		if _, err := cronexpr.Parse(result.Prefetch.Cron); err != nil {
			log.Fatal("Invalid cron string (if you don't know how to compose them, there are many online utilities for doing so). Please check https://github.com/gorhill/cronexpr#implementation for documentation.")
		}
	}
	return &result
}