export.go

package main

import (
	"archive/zip"
	"bytes"
	"encoding/csv"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"regexp"
	"time"

	"github.com/pkg/errors"
	"github.com/xuri/excelize/v2"

	"github.com/fanaticscripter/EggLedger/ei"
)

type mission struct {
	Id               string
	Ship             ei.MissionInfo_Spaceship
	ShipName         string
	DurationType     ei.MissionInfo_DurationType
	DurationTypeName string
	Level            uint32
	LaunchedAt       time.Time
	LaunchedAtStr    string
	ReturnedAt       time.Time
	ReturnedAtStr    string
	Duration         time.Duration
	DurationDays     float64
	Capacity         uint32
	Artifacts        []*ei.ArtifactSpec
	ArtifactNames    []string
}

func newMission(r *ei.CompleteMissionResponse) *mission {
	info := r.GetInfo()
	ship := info.GetShip()
	durationType := info.GetDurationType()
	launchedAt := unixToTime(info.GetStartTimeDerived()).Truncate(time.Second)
	durationSeconds := info.GetDurationSeconds()
	duration := time.Duration(durationSeconds) * time.Second
	returnedAt := launchedAt.Add(duration)
	var artifacts []*ei.ArtifactSpec
	var artifactNames []string
	for _, a := range r.Artifacts {
		artifacts = append(artifacts, a.Spec)
		artifactNames = append(artifactNames, a.Spec.Display())
	}
	return &mission{
		Id:               info.GetIdentifier(),
		Ship:             ship,
		ShipName:         ship.Name(),
		DurationType:     durationType,
		DurationTypeName: durationType.Display(),
		Level:            info.GetLevel(),
		LaunchedAt:       launchedAt,
		LaunchedAtStr:    launchedAt.Format(time.RFC3339),
		ReturnedAt:       returnedAt,
		ReturnedAtStr:    returnedAt.Format(time.RFC3339),
		Duration:         duration,
		DurationDays:     durationSeconds / 86400,
		Capacity:         info.GetCapacity(),
		Artifacts:        artifacts,
		ArtifactNames:    artifactNames,
	}
}

func exportMissionsToCsv(missions []*mission, path string) error {
	action := fmt.Sprintf("exporting missions to %s", path)
	wrap := func(err error) error {
		return errors.Wrap(err, "error "+action)
	}

	var maxArtifactCount int
	for _, m := range missions {
		count := len(m.ArtifactNames)
		if count > maxArtifactCount {
			maxArtifactCount = count
		}
	}
	header := []string{"ID", "Ship", "Type", "Level", "Launched at", "Returned at", "Duration days", "Capacity"}
	for i := 1; i <= maxArtifactCount; i++ {
		header = append(header, fmt.Sprintf("Artifact %d", i))
	}
	records := [][]string{header}
	for _, m := range missions {
		record := []string{
			m.Id,
			m.ShipName,
			m.DurationTypeName,
			fmt.Sprint(m.Level),
			m.LaunchedAtStr,
			m.ReturnedAtStr,
			fmt.Sprint(m.DurationDays),
			fmt.Sprint(m.Capacity),
		}
		count := len(m.ArtifactNames)
		for i := 0; i < maxArtifactCount; i++ {
			if i < count {
				record = append(record, m.ArtifactNames[i])
			} else {
				record = append(record, "")
			}
		}
		records = append(records, record)
	}

	temp, err := writeCsvToTempfile(records, filepath.Dir(path), tempfilePattern(path))
	if err != nil {
		return wrap(err)
	}
	if err := os.Rename(temp, path); err != nil {
		return wrap(err)
	}

	return nil
}

func writeCsvToTempfile(records [][]string, dir, pattern string) (temp string, err error) {
	f, err := os.CreateTemp(dir, pattern)
	if err != nil {
		return
	}
	_ = os.Chmod(f.Name(), 0644)
	temp = f.Name()
	defer func() { err = f.Close() }()
	w := csv.NewWriter(f)
	for _, record := range records {
		err = w.Write(record)
		if err != nil {
			return
		}
	}
	w.Flush()
	err = w.Error()
	if err != nil {
		return
	}
	return
}

func exportMissionsToXlsx(missions []*mission, path string) error {
	action := fmt.Sprintf("exporting missions to %s", path)
	wrap := func(err error) error {
		return errors.Wrap(err, "error "+action)
	}

	var maxArtifactCount int
	var maxArtifactNameLength int
	for _, m := range missions {
		count := len(m.ArtifactNames)
		if count > maxArtifactCount {
			maxArtifactCount = count
		}
		for _, name := range m.ArtifactNames {
			if len(name) > maxArtifactNameLength {
				maxArtifactNameLength = len(name)
			}
		}
	}

	f := excelize.NewFile()
	f.SetDefaultFont("Consolas")

	datetimeStyle, err := f.NewStyle(&excelize.Style{CustomNumFmt: sptr("yyyy-mm-dd hh:MM:ss")})
	if err != nil {
		return wrap(err)
	}
	durationStyle, err := f.NewStyle(&excelize.Style{CustomNumFmt: sptr("d\\dh\\hm\\m")})
	if err != nil {
		return wrap(err)
	}

	sw, err := f.NewStreamWriter("Sheet1")
	if err != nil {
		return wrap(err)
	}
	// Width of each column is set to max number of characters plus 5.
	colWidths := []float64{56, 25, 13, 8, 24, 24, 13, 8}
	for i := 1; i <= maxArtifactCount; i++ {
		colWidths = append(colWidths, float64(maxArtifactNameLength+5))
	}
	for i, width := range colWidths {
		if err := sw.SetColWidth(i+1, i+1, width); err != nil {
			return wrap(err)
		}
	}

	header := []interface{}{"ID", "Ship", "Type", "Level", "Launched at", "Returned at", "Duration", "Capacity"}
	for i := 1; i <= maxArtifactCount; i++ {
		header = append(header, fmt.Sprintf("Artifact %d", i))
	}
	if err = sw.SetRow("A1", header); err != nil {
		return wrap(err)
	}
	rowId := 1
	for _, m := range missions {
		rowId++
		row := []interface{}{
			m.Id,
			m.ShipName,
			m.DurationTypeName,
			m.Level,
			&excelize.Cell{Value: m.LaunchedAt, StyleID: datetimeStyle},
			&excelize.Cell{Value: m.ReturnedAt, StyleID: datetimeStyle},
			&excelize.Cell{Value: m.DurationDays, StyleID: durationStyle},
			m.Capacity,
		}
		for _, name := range m.ArtifactNames {
			row = append(row, name)
		}
		cell, err := excelize.CoordinatesToCellName(1, rowId)
		if err != nil {
			return wrap(err)
		}
		if err := sw.SetRow(cell, row); err != nil {
			return wrap(err)
		}
	}
	if err := sw.Flush(); err != nil {
		return wrap(err)
	}

	temp, err := os.CreateTemp(filepath.Dir(path), tempfilePattern(path))
	if err != nil {
		return wrap(err)
	}
	if err := temp.Close(); err != nil {
		return wrap(err)
	}
	_ = os.Chmod(temp.Name(), 0644)
	if err := f.SaveAs(temp.Name()); err != nil {
		return wrap(err)
	}
	if err := f.Close(); err != nil {
		return wrap(err)
	}

	if err := os.Rename(temp.Name(), path); err != nil {
		return wrap(err)
	}

	return nil
}

// findLastMatchingFile returns the path of the alphabetically last file in
// directory matching the regexp pattern. Empty string is returned if there's no
// file matching the pattern.
func findLastMatchingFile(directory, pattern string) (string, error) {
	re, err := regexp.Compile(pattern)
	if err != nil {
		return "", err
	}
	entries, err := os.ReadDir(directory)
	if err != nil {
		return "", err
	}
	var last string
	for _, entry := range entries {
		if entry.IsDir() {
			continue
		}
		name := entry.Name()
		if re.MatchString(name) {
			last = name
		}
	}
	if last == "" {
		return "", nil
	}
	return filepath.Join(directory, last), nil
}

// cmpFiles compares two files and returns true if their contents are equal.
// Both files can be fully read into memory; only suitable for small files.
func cmpFiles(path1, path2 string) (bool, error) {
	wrap := func(err error) error {
		return errors.Wrapf(err, "error comparing files %s and %s", path1, path2)
	}
	s1, err := os.Stat(path1)
	if err != nil {
		return false, wrap(err)
	}
	s2, err := os.Stat(path2)
	if err != nil {
		return false, wrap(err)
	}
	if s1.Size() != s2.Size() {
		return false, nil
	}
	b1, err := os.ReadFile(path1)
	if err != nil {
		return false, wrap(err)
	}
	b2, err := os.ReadFile(path2)
	if err != nil {
		return false, wrap(err)
	}
	return bytes.Equal(b1, b2), nil
}

// cmpZipFiles compares two zip files and returns true if their contents are
// equal. This is needed since zip is not deterministic given the exact same
// source files.
func cmpZipFiles(path1, path2 string) (bool, error) {
	wrap := func(err error) error {
		return errors.Wrapf(err, "error comparing zip files %s and %s", path1, path2)
	}
	files1, err := readZipContent(path1)
	if err != nil {
		return false, wrap(err)
	}
	files2, err := readZipContent(path2)
	if err != nil {
		return false, wrap(err)
	}
	if len(files1) != len(files2) {
		return false, nil
	}
	for name, content1 := range files1 {
		content2, ok := files2[name]
		if !ok {
			return false, nil
		}
		if !bytes.Equal(content1, content2) {
			return false, nil
		}
	}
	return true, nil
}

// readZipContent returns a map from file names to file contents for all files
// in the zip.
func readZipContent(path string) (map[string][]byte, error) {
	r, err := zip.OpenReader(path)
	if err != nil {
		return nil, err
	}
	defer r.Close()
	files := make(map[string][]byte)
	for _, f := range r.File {
		rc, err := f.Open()
		if err != nil {
			return files, err
		}
		body, err := io.ReadAll(rc)
		if err != nil {
			return files, err
		}
		if err := rc.Close(); err != nil {
			return files, err
		}
		files[f.Name] = body
	}
	return files, nil
}

func filenameWithoutExt(f string) string {
	f = filepath.Base(f)
	ext := filepath.Ext(f)
	return f[:len(f)-len(ext)]
}

func tempfilePattern(f string) string {
	f = filepath.Base(f)
	ext := filepath.Ext(f)
	return f[:len(f)-len(ext)] + ".*" + ext
}

func sptr(s string) *string {
	return &s
}