1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
// SPDX-License-Identifier: ISC
// Copyright © 2021 siddharth <s@ricketyspace.net>
package config
import (
"encoding/json"
"fmt"
"os"
"path"
"strings"
"ricketyspace.net/fern/feed"
"ricketyspace.net/fern/file"
)
// Represents the fern config
type FernConfig struct {
YDLPath string `json:"ydl-path"` // Path to the youtube-dl program.
DumpDir string `json:"dump-dir"` // Path where media needs to be downloaded to.
Feeds []feed.Feed `json:"feeds"` // Feeds to download.
}
// Tries to reads the fern config at `$HOME/.config/fern/fern.json`
// and unmarshals it into `FernConfig`
//
// Returns point to `FernConfig` on success. On error, the returned
// config is `nil` and the returned error is non-nil.
func Read() (*FernConfig, error) {
// Construct config file path.
h, err := os.UserHomeDir()
if err != nil {
return nil, err
}
c := path.Join(h, ".config", "fern", "fern.json")
// Open config file.
f, err := os.Open(c)
if err != nil {
return nil, err
}
// Read config file.
bs, err := file.Read(f)
if err != nil {
return nil, err
}
// Unmarshal config into an object.
config := new(FernConfig)
err = json.Unmarshal(bs, config)
if err != nil {
return nil, err
}
// Validate config.
err = config.validate()
if err != nil {
return nil, err
}
return config, nil
}
// Validates the FernConfig.
//
// Returns nil if validation succeeds; error otherwise.
func (config *FernConfig) validate() error {
// Validate 'ydl-path' in config.
if len(config.YDLPath) == 0 {
return fmt.Errorf("'ydl-path' not set in config")
}
_, err := os.Stat(config.YDLPath)
if err != nil {
return err
}
// Validate 'dump-dir' in config.
if len(config.DumpDir) == 0 {
return fmt.Errorf("'dump-dir' not set in config")
}
// Replace "~" with user's home directory in the dump
// directory path.
h, err := os.UserHomeDir()
if err != nil {
return err
}
config.DumpDir = strings.Replace(config.DumpDir, "~", h, 1)
// Ensure dump directory exists.
err = os.MkdirAll(config.DumpDir, 0755)
if err != nil {
return err
}
// Validate 'feeds' in config.
if len(config.Feeds) == 0 {
return fmt.Errorf("'feeds' not set in config")
}
for i, feed := range config.Feeds {
err = feed.Validate(config.DumpDir)
if err != nil {
return err
}
config.Feeds[i].YDLPath = config.YDLPath
config.Feeds[i].DumpDir = feed.DumpDir
}
return nil
}
|
