restockbot/config.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"path/filepath"
)

// Config to store JSON configuration
type Config struct {
	DatabaseConfig `json:"database"`
	TwitterConfig  `json:"twitter"`
	TelegramConfig `json:"telegram"`
	APIConfig      `json:"api"`
	AmazonConfig   `json:"amazon"`
	NvidiaFEConfig `json:"nvidia_fe"`
	URLs           []string     `json:"urls"`
	IncludeRegex   string       `json:"include_regex"`
	ExcludeRegex   string       `json:"exclude_regex"`
	PriceRanges    []PriceRange `json:"price_ranges"`
	BrowserAddress string       `json:"browser_address"`
}

// DatabaseConfig to store database configuration
type DatabaseConfig struct {
	Type string `json:"type"`
	DSN  string `json:"dsn"`
}

// TwitterConfig to store Twitter API secrets
type TwitterConfig struct {
	ConsumerKey       string              `json:"consumer_key"`
	ConsumerSecret    string              `json:"consumer_secret"`
	AccessToken       string              `json:"access_token"`
	AccessTokenSecret string              `json:"access_token_secret"`
	Hashtags          []map[string]string `json:"hashtags"`
	EnableReplies     bool                `json:"enable_replies"`
	Retention         int                 `json:"retention"`
}

// TelegramConfig to store Telegram API key
type TelegramConfig struct {
	Token         string `json:"token"`
	ChatID        int64  `json:"chat_id"`
	ChannelName   string `json:"channel_name"`
	EnableReplies bool   `json:"enable_replies"`
}

// APIConfig to store HTTP API configuration
type APIConfig struct {
	Address  string `json:"address"`
	Certfile string `json:"cert_file"`
	Keyfile  string `json:"key_file"`
}

// AmazonConfig to store Amazon API secrets
type AmazonConfig struct {
	Searches     []string `json:"searches"`
	AccessKey    string   `json:"access_key"`
	SecretKey    string   `json:"secret_key"`
	Marketplaces []struct {
		Name       string `json:"name"`
		PartnerTag string `json:"partner_tag"`
	} `json:"marketplaces"`
	AmazonFulfilled bool `json:"amazon_fulfilled"`
	AmazonMerchant  bool `json:"amazon_merchant"`
	AffiliateLinks  bool `json:"affiliate_links"`
}

// NvidiaFEConfig to store NVIDIA Founders Edition configuration
type NvidiaFEConfig struct {
	Locations []string `json:"locations"`
	GPUs      []string `json:"gpus"`
	UserAgent string   `json:"user_agent"`
	Timeout   int      `json:"timeout"`
}

// PriceRange to store rules to filter products with price outside of the range
type PriceRange struct {
	Model    string  `json:"model"`
	Min      float64 `json:"min"`
	Max      float64 `json:"max"`
	Currency string  `json:"currency"`
}

// NewConfig creates a Config struct
func NewConfig() *Config {
	return &Config{}
}

// Read Config from configuration file
func (c *Config) Read(file string) error {
	file, err := filepath.Abs(file)
	if err != nil {
		return err
	}

	jsonFile, err := ioutil.ReadFile(file)
	if err != nil {
		return err
	}

	err = json.Unmarshal(jsonFile, &c)
	if err != nil {
		return err
	}
	return nil
}

// HasTwitter returns true when Twitter has been configured
func (c *Config) HasTwitter() bool {
	return (c.TwitterConfig.AccessToken != "" && c.TwitterConfig.AccessTokenSecret != "" && c.TwitterConfig.ConsumerKey != "" && c.TwitterConfig.ConsumerSecret != "")
}

// HasTelegram returns true when Telegram has been configured
func (c *Config) HasTelegram() bool {
	return c.TelegramConfig.Token != "" && (c.TelegramConfig.ChatID != 0 || c.TelegramConfig.ChannelName != "")
}

// HasURLs returns true when list of URLS has been configured
func (c *Config) HasURLs() bool {
	return len(c.URLs) > 0
}

// HasDatabase returns true when database has been configured
func (c *Config) HasDatabase() bool {
	return c.DatabaseConfig.Type != "" && c.DatabaseConfig.DSN != ""
}

// HasNvidiaFE returns true when NVIDIA FE has been configured
func (c *Config) HasNvidiaFE() bool {
	return len(c.NvidiaFEConfig.Locations) > 0 && len(c.NvidiaFEConfig.GPUs) > 0
}

// HasAmazon returns true when Amazon has been configured
func (c *Config) HasAmazon() bool {
	var hasKeys, hasSearches, hasMarketplaces bool
	hasKeys = c.AmazonConfig.AccessKey != "" && c.AmazonConfig.SecretKey != ""
	hasSearches = len(c.AmazonConfig.Searches) > 0
	for _, marketplace := range c.AmazonConfig.Marketplaces {
		if marketplace.PartnerTag != "" && marketplace.Name != "" {
			hasMarketplaces = true
			break
		}
	}
	return hasKeys && hasSearches && hasMarketplaces
}