lotw.go

package tasks

import (
	"bytes"
	"fmt"
	"log"
	"math"
	"math/rand"
	"os/exec"
	"path/filepath"
	"sync"
	"time"

	"github.com/bbathe/golog/adif"
	"github.com/bbathe/golog/config"
	"github.com/bbathe/golog/models/qso"
	"github.com/bbathe/golog/util"
)

var muxLotwUpload sync.Mutex

// QSLLotw uploads QSOs to LoTW
func QSLLotw() error {
	muxLotwUpload.Lock()
	defer muxLotwUpload.Unlock()

	qsos, err := qso.FindQSLsToSend(qso.QSLLotw, config.LogbookServices.QSLDelay)
	if err != nil {
		log.Printf("%+v", err)
		return err
	}

	if len(qsos) > 0 {
		// get oldest loaded time of all these qsos
		minLoadedAt := int64(math.MaxInt64)
		for _, qso := range qsos {
			if minLoadedAt > qso.LoadedAt {
				minLoadedAt = qso.LoadedAt
			}
		}

		// pick a number between 11 & 17
		r := rand.Intn(17-11) + 11

		// if batch is greater than r size or oldest is older than 3 hours, then push to lotw
		if len(qsos) > r || time.Now().Add(-3*time.Hour).Unix() > minLoadedAt {
			err = uploadQSOsToLoTW(qsos, false)
			if err != nil {
				log.Printf("%+v", err)
				return err
			}
		}
	}

	return nil
}

// QSLLotwFinal uploads all QSOs to LoTW regardless of how old they are
func QSLLotwFinal() {
	muxLotwUpload.Lock()
	defer muxLotwUpload.Unlock()

	qsos, err := qso.FindQSLsToSend(qso.QSLLotw, 0)
	if err != nil {
		log.Printf("%+v", err)
		return
	}

	if len(qsos) > 0 {
		err = uploadQSOsToLoTW(qsos, true)
		if err != nil {
			log.Printf("%+v", err)
			return
		}
	}
}

// uploadQSOsToLoTW leverages tqsl to upload qsos to LoTW
func uploadQSOsToLoTW(qsos []qso.QSO, force bool) error {
	// form working file name
	fname := filepath.Join(config.WorkingDirectory, "LoTW-"+time.Now().UTC().Format("2006-Jan-02_15-04-05")+".adif")

	// write qsos as adif to file
	err := adif.WriteToFile(qsos, fname)
	if err != nil {
		log.Printf("%+v", err)
		return err
	}

	var stdout bytes.Buffer
	var stderr bytes.Buffer

	// setup command execution, capturing stdout & stderr
	// #nosec G204
	cmd := exec.Command(
		config.LogbookServices.TQSL.ExeLocation,
		"--quiet",
		"--batch",
		"--nodate",
		"--upload",
		fmt.Sprintf("--location=%s", config.LogbookServices.TQSL.StationLocationName),
		fname,
	)
	cmd.Stdout = &stdout
	cmd.Stderr = &stderr

	// doit!
	err = cmd.Run()
	if err != nil {
		log.Printf("error: %+v", err)
		log.Printf("stdout: %s", stdout.String())
		log.Printf("stderr: %s", stderr.String())
		return err
	}

	// set as sent in db
	err = qso.UpdateQSLsToSent(qsos, qso.QSLLotw)
	if err != nil {
		log.Printf("%+v", err)
		return err
	}

	// only keep the last 5 files of ours in the working directory
	err = util.DeleteHistoricalFiles(5, config.WorkingDirectory, "LoTW-", ".adif")
	if err != nil {
		log.Printf("%+v", err)
		return err
	}

	return nil
}