dnd/cmd/dndimport/main.go

package main

import (
	"archive/zip"
	"flag"
	"io"
	"os"
	"path/filepath"
	"strings"

	"github.com/fatih/camelcase"
	"github.com/firestuff/dnd/internal"
	"github.com/samber/lo"
	"golang.org/x/exp/slog"
)

var removeWords = map[string]bool{
	"+":             true,
	"1":             true,
	"2":             true,
	"diamond":       true,
	"gridless":      true,
	"high":          true,
	"psd":           true,
	"res":           true,
	"roll20":        true,
	"roll20+tokens": true,
	"tokens":        true,
}

var actions = map[string]string{
	"./*.psd":                            "Maps/{MAPNAME}",
	"*/gridless/*.jpg":                   "Maps/{MAPNAME}",
	"*/high res/gridless/*.jpg":          "Maps/{MAPNAME}",
	"*/high res/gridless/*.png":          "Maps/{MAPNAME}",
	"*/high res/gridless/attic/*.jpg":    "Maps/{MAPNAME}",
	"*/high res/gridless/attic/*.png":    "Maps/{MAPNAME}",
	"*/high res/gridless/basement/*.jpg": "Maps/{MAPNAME}",
	"*/high res/gridless/basement/*.png": "Maps/{MAPNAME}",
	"*/high res/gridless/floor 1/*.jpg":  "Maps/{MAPNAME}",
	"*/high res/gridless/floor 1/*.png":  "Maps/{MAPNAME}",
	"*/high res/gridless/floor 2/*.jpg":  "Maps/{MAPNAME}",
	"*/high res/gridless/floor 2/*.png":  "Maps/{MAPNAME}",
	"*/high resolution/gridless/*.jpg":   "Maps/{MAPNAME}",
	"*/grid/*.jpg":                       "{SKIP}",
	"*/high res/grid/*.jpg":              "{SKIP}",
	"*/high resolution/grid/*.jpg":       "{SKIP}",
	"*/creature tokens/*.png":            "Creatures",
	"*/creature tokens/variants/*.png":   "Creatures",
	"*/map tokens/*.png":                 "Maps/{MAPNAME}/Objects",
	"*/tokens/*.png":                     "Maps/{MAPNAME}/Objects",
	"*/roll20/grid/*.jpg":                "{SKIP}",
	"*/roll20/grid/*.png":                "{SKIP}",
	"*/roll20/grid/attic/*.jpg":          "{SKIP}",
	"*/roll20/grid/attic/*.png":          "{SKIP}",
	"*/roll20/grid/basement/*.jpg":       "{SKIP}",
	"*/roll20/grid/basement/*.png":       "{SKIP}",
	"*/roll20/grid/floor 1/*.jpg":        "{SKIP}",
	"*/roll20/grid/floor 1/*.png":        "{SKIP}",
	"*/roll20/grid/floor 2/*.jpg":        "{SKIP}",
	"*/roll20/grid/floor 2/*.png":        "{SKIP}",
	"*/roll20/gridless/*.jpg":            "{SKIP}",
	"*/roll20/gridless/*.png":            "{SKIP}",
	"*/roll20/gridless/attic/*.jpg":      "{SKIP}",
	"*/roll20/gridless/attic/*.png":      "{SKIP}",
	"*/roll20/gridless/basement/*.jpg":   "{SKIP}",
	"*/roll20/gridless/basement/*.png":   "{SKIP}",
	"*/roll20/gridless/floor 1/*.jpg":    "{SKIP}",
	"*/roll20/gridless/floor 1/*.png":    "{SKIP}",
	"*/roll20/gridless/floor 2/*.jpg":    "{SKIP}",
	"*/roll20/gridless/floor 2/*.png":    "{SKIP}",
}

var root = flag.String("root", "", "root directory to write to")

func main() {
	l := slog.New(slog.NewTextHandler(os.Stderr, nil))

	flag.Parse()

	if *root == "" {
		l.Error("please specify --root")
		os.Exit(1)
	}

	for _, path := range flag.Args() {
		zipL := l.With("zipPath", path)

		s := lo.Must(os.Stat(path))
		if s.Size() == 0 {
			zipL.Warn("SKIP")
			continue
		}

		ok := importZIP(zipL, path)
		if !ok {
			os.Exit(1)
		}

		lo.Must0(os.Remove(path))
	}
}

func importZIP(l *slog.Logger, path string) bool {
	name := mapName(path)
	l = l.With("mapName", name)

	z := lo.Must(zip.OpenReader(path))
	defer z.Close()

	for _, file := range z.File {
		if strings.HasSuffix(file.Name, "/") {
			continue
		}

		sig := strings.ToLower(internal.PathSig(file.Name))

		dst := actions[sig]

		if dst == "" {
			parts := strings.Split(sig, "/")
			dst = actions["*/"+filepath.Join(parts[1:]...)]
		}

		if dst == "" {
			l.Error("unknown file signature",
				"filePath", file.Name,
				"signature", sig,
			)
			return false
		}

		if dst == "{SKIP}" {
			l.Info("SKIP",
				"src", file.Name,
			)
			continue
		}

		dst = strings.ReplaceAll(dst, "{MAPNAME}", name)
		dst = filepath.Join(*root, dst, filepath.Base(file.Name))

		l.Info("COPY",
			"src", file.Name,
			"dst", dst,
		)

		lo.Must0(os.MkdirAll(filepath.Dir(dst), 0755))
		dstFile := lo.Must(os.OpenFile(dst, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644))
		srcFile := lo.Must(file.Open())

		lo.Must(io.Copy(dstFile, srcFile))

		dstFile.Close()
		srcFile.Close()
	}

	return true
}

func mapName(path string) string {
	withoutZIP := strings.TrimSuffix(filepath.Base(path), ".zip")
	parts := strings.Split(withoutZIP, " ")

	if len(parts) == 1 {
		parts = camelcase.Split(parts[0])
	}

	i := len(parts) - 1
	for i >= 0 && removeWords[strings.ToLower(parts[i])] {
		i--
	}

	return strings.Join(parts[:i+1], " ")
}