main.go

package main

import (
	"flag"
	"fmt"
	"os"
	"time"

	"github.com/jpxd/papertrans/papercut"
)

const defaultConfigPath = "config.store"
const comment = "Have some pages..."

func main() {
	createConfigFlag := flag.Bool("create-config", false, "create a new configuration file")
	checkFlag := flag.Bool("check", false, "only check page count")
	flag.Parse()

	modeFlags := []*bool{createConfigFlag, checkFlag}
	activeModeFlags := 0

	for _, flag := range modeFlags {
		if *flag {
			activeModeFlags++
		}
	}

	if activeModeFlags > 1 {
		fmt.Fprintf(os.Stderr, "Please specify at most one of these flags:\n")
		fmt.Fprintf(os.Stderr, "  -create-config, -check\n")
		os.Exit(2)
	}

	if *createConfigFlag {
		config := papercut.CreateConfig()
		err := config.Save(defaultConfigPath)
		if err != nil {
			fmt.Println("Failed to save config file:", err)
			return
		}
		fmt.Println("Done")
		return
	}

	// get credentials
	fmt.Println("Reading credentials from config")
	config := papercut.LoadOrCreateConfig(defaultConfigPath)

	// check time window
	if !(*checkFlag) && config.TimeSlotMinutes > 0 {
		fmt.Println("Checking time window")

		now := time.Now()
		currentYear, currentMonth, _ := now.Date()
		location := now.Location()

		firstOfThisMonth := time.Date(currentYear, currentMonth, 1, 0, 0, 0, 0, location)
		firstOfNextMonth := firstOfThisMonth.AddDate(0, 1, 0)

		windowDuration := time.Duration(config.TimeSlotMinutes) * time.Minute
		beginningOfWindow := firstOfNextMonth.Add(-windowDuration)

		if now.Before(beginningOfWindow) {
			fmt.Println("Not in time window")
			fmt.Println("Done")
			return
		}
	}

	// connect webclient to ssh tunnel
	fmt.Println("Connecting tunnel via SSH")
	ssh, err := papercut.CreateSSHClient(papercut.DefaultSSHHost, papercut.DefaultSSHHostKey, config.SSHUser, config.SSHKeyFile)
	if err != nil {
		fmt.Println("Failed to connect to SSH server")
		fmt.Println(err)
		return
	}
	defer ssh.Close()

	client := papercut.CreateTunneledWebClient(ssh)

	// create papercut api
	fmt.Println("Logging into PaperCut")
	pc := papercut.CreatePapercutAPIWithWebClient(config.PaperCutUsername, config.PaperCutPassword, client)
	defer pc.Close()

	// get page count
	count := pc.GetPagesLeft()
	fmt.Println("Pages left:", count)

	// if we just wanted to check the page count, we are done now
	if *checkFlag {
		fmt.Println("Done")
		return
	}

	// make sure we have enough pages
	if count <= config.MinPagesLeft {
		fmt.Println("Not enough pages left, aborting...")
		fmt.Println("Done")
		return
	}

	// lets transfer some pages
	amountToTransfer := count - config.MinPagesLeft
	fmt.Println("Transferring", amountToTransfer, "pages to", config.Receiver)
	if pc.TransferPages(config.Receiver, amountToTransfer, comment) {
		fmt.Println("Transfer was successful")
	} else {
		fmt.Println("Transfer has failed")
	}

	count = pc.GetPagesLeft()
	fmt.Println("Pages left:", count)

	fmt.Println("Done")
}