main.go

package main

import (
	"encoding/json"
	"log"
	"math/rand"
	"os"
	"strings"
	"time"

	"github.com/insomnimus/libman/history"

	"github.com/insomnimus/libman/auth"
	"github.com/insomnimus/libman/control"
	// "github.com/vrischmann/userdir"
)

const VERSION = "0.22.4"

func init() {
	log.SetFlags(0)
	log.SetPrefix("")

	rand.Seed(time.Now().UnixNano())
}

func main() {
	// read config
	c, err := configFromArgs()
	if err != nil {
		log.Fatal(err)
	}

	control.DataHome = c.DataHome

	// load hist file, if any
	if c.HistFile != "" {
		h, err := history.Load(c.HistFile)
		if err != nil {
			log.Printf("Warning: could not load the history file: %s.\n", err)
		} else {
			*control.Hist = *h
		}
	}
	if c.HistSize > 0 {
		history.HistorySize = uint32(c.HistSize)
	}
	defer func() {
		if c.HistFile != "" && control.Hist.Modified() {
			err := control.Hist.Save(c.HistFile)
			if err != nil {
				log.Printf("Warning: could not save the history file: %s.\n", err)
			}
		}
	}()

	creds, err := auth.Login(c)
	if err != nil {
		log.Fatal(err)
	}

	// load the commands from the rc file, if it exists
	var commands []string

	if c.RCFile != "" {
		if _, err := os.Stat(c.RCFile); err == nil {
			data, err := os.ReadFile(c.RCFile)
			if err != nil {
				log.Fatalf("error reading from the libman startup script file at %s:\n%s\n", c.RCFile, err)
			}
			commands = strings.Split(string(data), "\n")
		}
	}

	go control.Start(creds.Client, creds.User, c.Prompt, commands)
	<-control.Terminator
	// save the token if there's a cache file specified
	if c.CacheFile != "" {
		token, err := creds.Client.Token()
		if err != nil {
			log.Fatalf("error retreiving token: %s", err)
		}
		data, err := json.MarshalIndent(token, "", "\t")
		if err != nil {
			log.Fatalf("error serializing token as json: %s", err)
		}
		err = os.WriteFile(c.CacheFile, data, 0o600)
		if err != nil {
			log.Fatalf("error saving the token to the cache file: %s", err)
		}
	} else {
		log.Printf("Warning: the access token is not saved because no cache file is specified.\nYou can configure the cache file from the libman config file located at %s.\n", c.ConfigPath)
	}
}