main.go

package main

import (
	"flag"
	"fmt"
	"nft/functional/combiner"
	"nft/functional/domain"
	"nft/functional/generator"
	"nft/functional/helpers"
	"nft/functional/inmemory"
	"nft/functional/pinata"
	"nft/functional/trait"
	"os"
)

func main() {
	// Flags.
	appContext := domain.NewAppContext()

	// Generator flags.
	flag.IntVar(&appContext.GeneratorParams.Length, "generated-image-length", 2048, "generated image length")
	flag.IntVar(&appContext.GeneratorParams.Width, "generated-image-width", 2048, "generated image width")
	flag.StringVar(&appContext.GeneratorParams.InputDirectory, "generated-image-input", "input-dir", "generated image input directory")
	flag.StringVar(&appContext.GeneratorParams.OutputDirectory, "generated-image-output", "output-dir", "generated image output directory")
	flag.IntVar(&appContext.GeneratorParams.Number, "generated-image-number", 100, "generated image number")
	generate := flag.Bool("generate", false, "Generate images")

	// Ipfs flags.
	flag.StringVar(&appContext.IpfsParams.InputDirectory, "ipfs-input", "output-dir", "ipfs input directory")
	flag.StringVar(&appContext.IpfsParams.OutputDirectory, "ipfs-output", "ipfs-metadata", "ipfs input directory")
	flag.StringVar(&appContext.IpfsParams.APIKey, "ipfs-api-key", "", "ipfs api key")
	flag.StringVar(&appContext.IpfsParams.SecretKey, "ipfs-secret-key", "", "ipfs secret key")
	ipfsUpload := flag.Bool("ipfs-upload", false, "Upload files to ipfs")

	// Other.
	printInfo := flag.Bool("info", false, "Print info")
	flag.CommandLine.SetOutput(os.Stdout)
	flag.Parse()

	if *generate {
		fmt.Println("Start generating images ...")
		traitService := trait.NewBasicTraitService(
			inmemory.NewGroupRepository(),
			inmemory.NewTraitRepository(),
		)
		_, err := traitService.Import(appContext.GeneratorParams.InputDirectory)
		if err != nil {
			panic(err)
		}
		generatorService := generator.NewBasicImageGenerator(
			appContext.GeneratorParams,
			traitService,
			combiner.NewBasicImageCombiner(),
		)
		err = generatorService.GenerateImages()
		if err != nil {
			panic(err)
		}
		return
	}
	if *ipfsUpload {
		fmt.Println("Uploading files to ipfs ...")
		ipfsService := pinata.NewIpfsService(appContext.IpfsParams)
		if err := ipfsService.UploadImages(); err != nil {
			panic(err)
		}
	}
	if *printInfo {
		helpers.PrintInfoV2(
			appContext.IpfsParams.OutputDirectory,
			appContext.GeneratorParams.OutputDirectory,
		)
		helpers.PrintInfo(appContext.GeneratorParams.OutputDirectory)
	}
}