pasty/cmd/transfer/main.go

package main

import (
	"log"
	"os"

	"github.com/lus/pasty/internal/config"
	"github.com/lus/pasty/internal/shared"
	"github.com/lus/pasty/internal/storage"
)

func main() {
	// Validate the command line arguments
	if len(os.Args) != 3 {
		panic("Invalid command line arguments")
	}

	// Load the configuration
	log.Println("Loading the application configuration...")
	config.Load()

	// Create and initialize the first (from) driver
	from, err := storage.GetDriver(shared.StorageType(os.Args[1]))
	if err != nil {
		panic(err)
	}
	err = from.Initialize()
	if err != nil {
		panic(err)
	}

	// Create and initialize the second (to) driver
	to, err := storage.GetDriver(shared.StorageType(os.Args[2]))
	if err != nil {
		panic(err)
	}
	err = to.Initialize()
	if err != nil {
		panic(err)
	}

	// Retrieve a list of IDs from the first (from) driver
	ids, err := from.ListIDs()
	if err != nil {
		panic(err)
	}

	// Transfer every paste to the second (to) driver
	for _, id := range ids {
		log.Println("Transferring ID " + id + "...")

		// Retrieve the paste
		paste, err := from.Get(id)
		if err != nil {
			log.Println("[ERR]", err.Error())
			continue
		}

		// Move the content of the deletion token field to the modification field
		if paste.DeletionToken != "" {
			if paste.ModificationToken == "" {
				paste.ModificationToken = paste.DeletionToken
			}
			paste.DeletionToken = ""
			log.Println("[INFO] Paste " + id + " was a legacy one.")
		}

		// Initialize a new metadata map if the old one is null
		if paste.Metadata == nil {
			paste.Metadata = make(map[string]interface{})
		}

		// Save the paste
		err = to.Save(paste)
		if err != nil {
			log.Println("[ERR]", err.Error())
			continue
		}

		log.Println("Transferred ID " + id + ".")
	}
}