mirror of https://github.com/lus/pasty.git
83 lines
1.7 KiB
Go
83 lines
1.7 KiB
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 + ".")
|
|
}
|
|
}
|