dischord/config/config.go

241 lines
6.8 KiB
Go
Raw Normal View History

2022-09-20 00:54:22 +02:00
package config
import (
"github.com/BurntSushi/toml"
"git.nobrain.org/r4/dischord/extractor"
"errors"
"fmt"
"os"
"os/exec"
"runtime"
)
var (
ErrTokenNotSet = errors.New("bot token not set")
ErrInvalidYoutubeDlPath = errors.New("invalid youtube-dl path")
ErrInvalidFfmpegPath = errors.New("invalid FFmpeg path")
ErrYoutubeDlNotFound = errors.New("youtube-dl not found, please install it from https://youtube-dl.org/ first")
ErrFfmpegNotFound = errors.New("FFmpeg not found, please install it from https://ffmpeg.org first")
ErrPythonNotInstalled = errors.New("python not installed")
)
type Config struct {
2022-09-20 17:43:47 +02:00
Token string `toml:"bot-token"`
FfmpegPath string `toml:"ffmpeg-path"`
2022-09-20 00:54:22 +02:00
Extractors extractor.Config `toml:"extractors"`
}
const (
defaultToken = "insert your Discord bot token here"
)
var (
ffmpegPaths = []string{"ffmpeg", "./ffmpeg"}
youtubeDlPaths = []string{"youtube-dl", "./youtube-dl", "yt-dlp", "./yt-dlp", "youtube-dlc", "./youtube-dlc"}
)
// Returns a valid path if one exists. Returns "" if none found.
func searchExecPaths(paths ...string) string {
for _, pathbase := range paths {
path := pathbase
if runtime.GOOS == "windows" {
path += ".exe"
}
if _, err := exec.LookPath(path); err == nil {
return path
}
}
return ""
}
func write(filename string, cfg *Config) error {
file, err := os.Create(filename)
if err != nil {
return err
}
defer file.Close()
if _, err := file.WriteString(`# Insert your Discord bot token here.
# It can be found at https://discord.com/developers/applications -> <your application> -> Bot -> Reset Token.
# Make sure to keep the "" around your token text.
`); err != nil {
return err
}
enc := toml.NewEncoder(file)
enc.Indent = ""
if err := enc.Encode(cfg); err != nil {
return err
}
return nil
}
func macosEnableExecutable(filename string) error {
if runtime.GOOS != "darwin" {
return nil
}
// Commands from http://www.osxexperts.net/
if runtime.GOARCH == "arm64" {
cmd := exec.Command("xattr", "-cr", filename)
if err := cmd.Run(); err != nil {
return err
}
cmd = exec.Command("codesign", "-s", "-", filename)
if err := cmd.Run(); err != nil {
return err
}
} else if runtime.GOARCH == "amd64" {
cmd := exec.Command("xattr", "-dr", "com.apple.quarantine", filename)
if err := cmd.Run(); err != nil {
return err
}
}
return nil
}
// Tries to load the given TOML config file. Returns an error if the
// configuration file does not exist or is invalid.
func Load(filename string) (*Config, error) {
cfg := &Config{}
meta, err := toml.DecodeFile(filename, cfg)
2022-09-20 00:54:22 +02:00
if err != nil {
if pe, ok := err.(toml.ParseError); ok {
fmt.Println(pe.ErrorWithUsage())
}
2022-09-20 00:54:22 +02:00
return nil, err
}
if undec := meta.Undecoded(); len(undec) > 0 {
return nil, fmt.Errorf("%v: field '%v' could not be decoded", filename, undec[0])
}
2022-09-20 00:54:22 +02:00
if cfg.Token == defaultToken || cfg.Token == "" {
return nil, ErrTokenNotSet
}
if err := cfg.Extractors.CheckValidity(); err != nil {
2022-09-20 00:54:22 +02:00
return nil, err
}
if _, err := exec.LookPath(cfg.Extractors["youtube-dl"]["youtube-dl-path"].(string)); err != nil {
return nil, ErrInvalidYoutubeDlPath
}
if _, err := exec.LookPath(cfg.FfmpegPath); err != nil {
return nil, ErrInvalidFfmpegPath
}
return cfg, nil
}
// Automatically creates a TOML configuration file with the default values and
// prints information and instructions for the user to stdout.
func Autoconf(filename string) (*Config, error) {
cfg := &Config{
Token: defaultToken,
Extractors: extractor.DefaultConfig(),
}
download := func(executable bool, urlsByOS map[string]map[string]string) (filename string, err error) {
2022-09-20 17:43:47 +02:00
filename, err = download(executable, urlsByOS, func(progress float32) {
2022-09-20 00:54:22 +02:00
fmt.Printf("Progress: %.1f%%\r", progress*100.0)
})
if err != nil {
fmt.Println()
return "", err
} else {
fmt.Println("Progress: Finished downloading")
}
return filename, nil
}
python3IsPython := false
if runtime.GOOS != "windows" {
if _, err := exec.LookPath("python"); err != nil {
if _, err := exec.LookPath("python3"); err == nil {
python3IsPython = true
} else {
return nil, ErrPythonNotInstalled
}
}
}
youtubeDlPath := searchExecPaths(youtubeDlPaths...)
if youtubeDlPath == "" {
fmt.Println("Downloading youtube-dl")
filename, err := download(true, map[string]map[string]string{
2022-09-20 17:43:47 +02:00
"windows": {
"amd64": "https://yt-dl.org/downloads/latest/youtube-dl.exe",
"386": "https://yt-dl.org/downloads/latest/youtube-dl.exe",
},
"any": {
"any": "https://yt-dl.org/downloads/latest/youtube-dl",
},
})
2022-09-20 00:54:22 +02:00
if err != nil {
return nil, err
}
2022-09-20 17:43:47 +02:00
youtubeDlPath = "./" + filename
2022-09-20 00:54:22 +02:00
macosEnableExecutable(youtubeDlPath)
if python3IsPython {
// Replace first line with `replacement`
data, err := os.ReadFile(youtubeDlPath)
if err != nil {
return nil, err
}
replacement := []byte("#!/usr/bin/env python3")
for i, c := range data {
if c == '\n' {
data = append(replacement, data[i:]...)
break
}
}
if err := os.WriteFile(youtubeDlPath, data, 0777); err != nil {
return nil, err
}
}
} else {
fmt.Println("Using youtube-dl executable found at", youtubeDlPath)
}
cfg.Extractors["youtube-dl"]["youtube-dl-path"] = youtubeDlPath
cfg.FfmpegPath = searchExecPaths(ffmpegPaths...)
if cfg.FfmpegPath == "" {
targetFile := "ffmpeg"
if runtime.GOOS == "windows" {
targetFile += ".exe"
}
fmt.Println("Downloading FFmpeg")
filename, err := download(false, map[string]map[string]string{
2022-09-20 17:43:47 +02:00
"linux": {
"amd64": "https://johnvansickle.com/ffmpeg/releases/ffmpeg-release-amd64-static.tar.xz",
"386": "https://johnvansickle.com/ffmpeg/releases/ffmpeg-release-i686-static.tar.xz",
"arm64": "https://johnvansickle.com/ffmpeg/releases/ffmpeg-release-arm64-static.tar.xz",
"arm": "https://johnvansickle.com/ffmpeg/releases/ffmpeg-release-armhf-static.tar.xz",
},
"windows": {
"amd64": "https://www.gyan.dev/ffmpeg/builds/ffmpeg-release-essentials.zip",
"386": "https://github.com/sudo-nautilus/FFmpeg-Builds-Win32/releases/download/latest/ffmpeg-n5.1-latest-win32-gpl-5.1.zip",
},
"darwin": {
"amd64": "https://evermeet.cx/ffmpeg/getrelease/zip",
"arm64": "https://www.osxexperts.net/FFmpeg511ARM.zip",
},
})
2022-09-20 00:54:22 +02:00
if err != nil {
return nil, err
}
fmt.Println("Unpacking", targetFile, "from", filename)
if err := unarchiveSingleFile(filename, targetFile); err != nil {
return nil, err
}
if err := os.Remove(filename); err != nil {
return nil, err
}
2022-09-20 17:43:47 +02:00
cfg.FfmpegPath = "./" + targetFile
2022-09-20 00:54:22 +02:00
macosEnableExecutable(cfg.FfmpegPath)
} else {
fmt.Println("Using FFmpeg executable found at", cfg.FfmpegPath)
}
fmt.Println("Writing configuration to", filename)
write(filename, cfg)
fmt.Println("Almost done. Now just edit", filename, "and set your bot token.")
return cfg, nil
}