map.go

// Copyright (c) 2024 Michael D Henderson. All rights reserved.

package main

import (
	"encoding/json"
	"fmt"
	"github.com/mdhender/tcfna/internal/model"
	"github.com/mdhender/tcfna/internal/store/csvdb"
	"github.com/mdhender/tcfna/internal/store/memory"
	"github.com/spf13/cobra"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
)

var argsMap struct {
	Import struct {
		Format string // csv, json
		Name   string // leave blank to avoid import
	}
	Export struct {
		Format string // html, json, png, svg
		Name   string // leave blank to avoid export
	}
}

var cmdMap = &cobra.Command{
	Use:   "map",
	Short: "map commands",
	Long:  `Commands to import, export, and create board maps.`,
	Run: func(cmd *cobra.Command, args []string) {
		//rootDir := viper.Get("files.path").(string)
		//log.Printf("[map] rootDir %q\n", rootDir)

		var board *model.MAP
		var err error

		log.Println(argsMap)
		if argsMap.Import.Name != "" {
			switch argsMap.Import.Format {
			case "csv":
				board, err = csvdb.Convert(argsMap.Import.Name)
				cobra.CheckErr(err)
			case "json":
				//board, err = jsondb.Convert(argsMap.Import.Name)
				//cobra.CheckErr(err)
			default:
				log.Fatalf("[map] unsupported import format %q\n", argsMap.Import.Format)
			}
		}
		if board == nil {
			cobra.CheckErr(fmt.Errorf("missing input file name"))
		}

		ds := memory.New(board)

		if argsMap.Export.Name != "" {
			switch argsMap.Export.Format {
			case "html":
				cobra.CheckErr(ioutil.WriteFile("svg-test.html", ds.BoardAsHTML(), 0644))
			case "json":
				if argsMap.Import.Name == argsMap.Export.Name {
					log.Fatal("[map] cowardly refusing to overwrite input file\n")
				}
				data := struct {
					Data model.HEXES `json:"data"`
				}{Data: board.Sorted}
				b, err := json.MarshalIndent(data, "", "  ")
				if err != nil {
					log.Fatalf("[map] encoding json: %+v\n", err)
				}
				cobra.CheckErr(os.WriteFile(filepath.Join("..", "data", "board.json"), b, 0644))
				ds.BoardAsImage(true)
			case "png":
				ds.BoardAsImage(true)
			case "svg":
				cobra.CheckErr(os.WriteFile(filepath.Join("..", "data", "board.svg"), []byte(ds.BoardAsSVG().String()), 0644))
			default:
				log.Fatalf("[map] unsupported export format %q\n", argsMap.Export.Format)
			}
		}
	},
}