Archived
1
0
Fork 0
This repository has been archived on 2024-12-18. You can view files and clone it, but cannot push or open issues or pull requests.
restockbot/config.go
Julien Riou 45025def65
Configure headless browser address
Add `browser_address` configuration setting to define where is the headless
browser instead of relying on the default value.

Signed-off-by: Julien Riou <julien@riou.xyz>
2021-03-01 13:11:58 +01:00

54 lines
1.4 KiB
Go

package main
import (
"encoding/json"
"io/ioutil"
"path/filepath"
)
// Config to store JSON configuration
type Config struct {
TwitterConfig `json:"twitter"`
URLs []string `json:"urls"`
IncludeRegex string `json:"include_regex"`
ExcludeRegex string `json:"exclude_regex"`
BrowserAddress string `json:"browser_address"`
}
// 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"`
}
// 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 != "")
}