server/main.go

package main

import (
	"database/sql"
	"encoding/xml"
	"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("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("Could not open config file: %v", err)
	}

	fi, err := config_file.Stat()
	if err != nil {
		return &config, fmt.Errorf("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("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("Could not read data from config file: %v", err)
		}
		err = toml.Unmarshal(data, &config)
		if err != nil {
			return &config, fmt.Errorf("Could 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) {
	// create data directory
	err := os.MkdirAll(data_dir, 0755)
	if err != nil {
		return nil, fmt.Errorf("Could not create data directory: %v", err)
	}

	// open database
	db_path := filepath.Join(data_dir, "ikinuki-server.db")

	db, err := sql.Open("sqlite3", db_path)
	if err != nil {
		return nil, fmt.Errorf("Could not open database: %v", err)
	}

	// initalize database
	sqlStmt := `
CREATE TABLE tvshows(
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    title TEXT,
    original_title TEXT,
    show_title TEXT,
    year INTEGER
);
CREATE TABLE episodes(
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    number INTEGER,
    season INTEGER
);
`

	_, err = db.Exec(sqlStmt)
	if err != nil {
		return nil, fmt.Errorf("Could not initalize database: %v", err)
	}

	return db, nil
}

type TVShow struct {
	XMLName      xml.Name `xml:"tvshow"`
	Title        string   `xml:"title"`
	OriginalTile string   `xml:"originaltitle"`
	ShowTitle    string   `xml:"showtitle"`
	Year         int      `xml:"year"`
}

func scan_tvshow_root(root string, db *sql.DB) error {
	files, err := ioutil.ReadDir(root)
	if err != nil {
		return fmt.Errorf("Could not get files in directory \"%v\": %v", root, err)
	}

	tx, err := db.Begin()
	if err != nil {
		return fmt.Errorf("Could not begin db transaction: %v", err)
	}
	defer tx.Commit()

	stmt, err := tx.Prepare("insert into tvshows(title, original_title, show_title, year) values(?, ?, ?, ?)")
	if err != nil {
		return fmt.Errorf("Could not prepare statement: %v", err)
	}
	defer stmt.Close()

	for _, file := range files {
		if file.IsDir() {
			show_path := filepath.Join(root, file.Name())
			if insert_tvshow_nfo(show_path, stmt) {
				err = scan_tvshow_episodes(show_path)
				if err != nil {
					fmt.Fprintf(os.Stderr, "Could not scan for episodes in %v: %v", show_path, err)
					continue
				}
			}
		}
	}
	return nil
}

func insert_tvshow_nfo(show_dir string, stmt *sql.Stmt) bool {
	tvshow_nfo_path := filepath.Join(show_dir, "tvshow.nfo")

	tvshow_nfo_file, err := os.OpenFile(tvshow_nfo_path, os.O_RDONLY, 0755)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Could not open tvshow.nfo file in \"%v\"\n", show_dir)
		return false
	}
	defer tvshow_nfo_file.Close()

	// try to parse nfo file
	data, err := ioutil.ReadAll(tvshow_nfo_file)
	if err != nil {
		fmt.Fprintf(os.Stderr, "ERROR: Could not read data from tvshow.nfo file: %v\n", err)
		return false
	}
	var tvshow TVShow

	err = xml.Unmarshal(data, &tvshow)
	if err != nil {
		fmt.Fprintf(os.Stderr, "ERROR: Could not parse tvshow.nfo file contents: %v\n", err)
		return false
	}

	// TODO: Replace with print formatting logic
	fmt.Printf("%v (%v):\n", tvshow.Title, tvshow.Year)
	fmt.Printf("    Original Title: %v\n", tvshow.OriginalTile)
	fmt.Printf("    Show Title: %v\n", tvshow.ShowTitle)

	_, err = stmt.Exec(tvshow.Title, tvshow.OriginalTile, tvshow.ShowTitle, tvshow.Year)
	if err != nil {
		fmt.Fprintf(os.Stderr, "ERROR: Could not insert tvshow into database: %v\n", err)
		return false
	}
	return true
}

type Episode struct {
	XMLName   xml.Name `xml:"episodedetails"`
	Title     string   `xml:"title"`
	ShowTitle string   `xml:"showtitle"`
	Season    int      `xml:"season"`
	Episode   int      `xml:"episode"`
}

func scan_tvshow_episodes(show_dir string) error {

	fmt.Printf("Scanning episodes for show %v\n", show_dir)

	files, err := ioutil.ReadDir(show_dir)
	if err != nil {
		return fmt.Errorf("Could not enumerate files in folder: %v", err)
	}

	num_episodes := 0

	for _, file := range files {
		ext := filepath.Ext(file.Name())
		nfo_path := filepath.Join(show_dir, file.Name())

		if file.Name() == "tvshow.nfo" {
			continue
		} else if ext == ".nfo" {
			fmt.Printf("Found nfo file: \"%v\"\n", file.Name())
		} else {
			fmt.Printf("Skipping file \"%v\", not an nfo file\n", file.Name())
			continue
		}

		fmt.Printf("%v\n", file.Name())

		nfo_file, err := os.OpenFile(nfo_path, os.O_RDONLY, 0755)
		if err != nil {
			fmt.Fprintf(os.Stderr, "Could not open nfo file \"%v\": %v", nfo_path, err)
			continue
		}
		defer nfo_file.Close()

		// try to parse nfo file
		data, err := ioutil.ReadAll(nfo_file)
		if err != nil {
			fmt.Fprintf(os.Stderr, "ERROR: Could not read data from nfo file \"%v\": %v\n", nfo_path, err)
			continue
		}
		var episode Episode

		err = xml.Unmarshal(data, &episode)
		if err != nil {
			fmt.Fprintf(os.Stderr, "ERROR: Could not parse contents of nfo file \"%v\": %v\n", nfo_path, err)
			continue
		}

		fmt.Printf("    Title: %v\n", episode.Title)
		fmt.Printf("    Season %v\n", episode.Season)
		fmt.Printf("    Episode %v\n", episode.Episode)
		fmt.Printf("    Show: %v\n", episode.ShowTitle)

		num_episodes++

	}

	fmt.Printf("Found %v episodes\n", num_episodes)

	return nil
}

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()

	db, err := open_database(data_dir)
	if err != nil {
		fmt.Fprintf(os.Stderr, "ERROR: Could not open database: %v", err)
		os.Exit(1)
	}
	defer db.Close()

	err = scan_tvshow_root(config.ScanDirectory, db)
	if err != nil {
		fmt.Fprintf(os.Stderr, "ERROR: Could not scan directory \"%v\": %v\n", config.ScanDirectory, err)
		os.Exit(1)
	}

}