158 lines
3.4 KiB
Go
158 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
|
|
toml "github.com/pelletier/go-toml/v2"
|
|
)
|
|
|
|
type Config struct {
|
|
Address string
|
|
Port int
|
|
ScanDirectory string
|
|
//Age int
|
|
//Cats []string
|
|
//Pi float64
|
|
//Perfection []int
|
|
//DOB time.Time // requires `import time`
|
|
}
|
|
|
|
func get_config_dir() string {
|
|
// Find config directory
|
|
|
|
// Linux (XDG base directory specification compliant)
|
|
xdg_config_home := os.Getenv("XDG_CONFIG_HOME")
|
|
|
|
if xdg_config_home != "" {
|
|
return filepath.Join(xdg_config_home, "ikinuki")
|
|
} else {
|
|
home := os.Getenv("HOME")
|
|
return filepath.Join(home, ".config", "ikinuki")
|
|
}
|
|
}
|
|
|
|
func get_config(config_dir string) (*Config, error) {
|
|
config := Config{
|
|
Address: "127.0.0.1",
|
|
Port: 32520,
|
|
ScanDirectory: "FILL IN",
|
|
}
|
|
|
|
err := os.MkdirAll(config_dir, 0755)
|
|
if err != nil {
|
|
return &config, fmt.Errorf("ERROR: Could not create config directory: %v", err)
|
|
}
|
|
|
|
config_path := filepath.Join(config_dir, "server.toml")
|
|
|
|
fmt.Printf("Config File Dir: %v\n", config_path)
|
|
|
|
// open file, creating it if empty
|
|
config_file, err := os.OpenFile(config_path, os.O_RDWR|os.O_CREATE, 0755)
|
|
defer config_file.Close()
|
|
|
|
if err != nil {
|
|
return &config, fmt.Errorf("ERROR: Could not open config file: %v", err)
|
|
}
|
|
|
|
fi, err := config_file.Stat()
|
|
if err != nil {
|
|
return &config, fmt.Errorf("ERROR: Could not get config file size: %v", err)
|
|
}
|
|
|
|
if fi.Size() == 0 {
|
|
// if file is empty, write default config
|
|
data, err := toml.Marshal(config)
|
|
if err != nil {
|
|
return &config, fmt.Errorf("ERROR: Could not write default config to file: %v", err)
|
|
}
|
|
config_file.Write(data)
|
|
// return nil as the user must edit the config file
|
|
return nil, nil
|
|
} else {
|
|
// try to parse config file
|
|
data, err := ioutil.ReadAll(config_file)
|
|
if err != nil {
|
|
return &config, fmt.Errorf("ERROR: Could not read data from config file: %v", err)
|
|
}
|
|
err = toml.Unmarshal(data, &config)
|
|
if err != nil {
|
|
return &config, fmt.Errorf("ERROR: Colud not parse config file contents: %v", err)
|
|
}
|
|
return &config, nil
|
|
}
|
|
|
|
}
|
|
|
|
func get_data_dir() string {
|
|
// Find data directory
|
|
|
|
// Linux (XDG base directory specification compliant)
|
|
xdg_data_home := os.Getenv("XDG_DATA_HOME")
|
|
|
|
if xdg_data_home != "" {
|
|
return filepath.Join(xdg_data_home, "ikinuki")
|
|
} else {
|
|
home := os.Getenv("HOME")
|
|
return filepath.Join(home, ".local", "share", "ikinuki")
|
|
}
|
|
}
|
|
|
|
func open_database(data_dir string) (*sql.DB, error) {
|
|
err := os.MkdirAll(data_dir, 0755)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("ERROR: could not create data directory: %v", err)
|
|
}
|
|
|
|
db_path := filepath.Join(data_dir, "ikinuki-server.db")
|
|
|
|
db, err := sql.Open("sqlite3", db_path)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("ERROR: could not open database: %v", err)
|
|
}
|
|
|
|
return db, nil
|
|
}
|
|
|
|
func scan_directory(directory string) {
|
|
files, err := ioutil.ReadDir(directory)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for _, file := range files {
|
|
if file.IsDir() {
|
|
fmt.Printf("%v\n", file.Name())
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func main() {
|
|
|
|
config_dir := get_config_dir()
|
|
|
|
config, err := get_config(config_dir)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if config == nil {
|
|
fmt.Printf("Created config file at %v, please edit and launch again\n", config_dir)
|
|
os.Exit(0)
|
|
}
|
|
fmt.Printf("Config: %v\n", config)
|
|
|
|
data_dir := get_data_dir()
|
|
|
|
open_database(data_dir)
|
|
|
|
scan_directory(config.ScanDirectory)
|
|
|
|
}
|