mirror of https://github.com/lus/pasty.git
68 lines
1.2 KiB
Go
68 lines
1.2 KiB
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"github.com/Lukaesebrot/pasty/internal/env"
|
||
|
"github.com/Lukaesebrot/pasty/internal/storage"
|
||
|
"log"
|
||
|
"os"
|
||
|
)
|
||
|
|
||
|
func main() {
|
||
|
// Validate the command line arguments
|
||
|
if len(os.Args) != 3 {
|
||
|
panic("Invalid command line arguments")
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// Load the optional .env file
|
||
|
log.Println("Loading the optional .env file...")
|
||
|
env.Load()
|
||
|
|
||
|
// Create and initialize the first (from) driver
|
||
|
from, err := storage.GetDriver(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(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
|
||
|
}
|
||
|
|
||
|
// Save the paste
|
||
|
err = to.Save(paste)
|
||
|
if err != nil {
|
||
|
log.Println("[ERR]", err.Error())
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
log.Println("Transferred ID " + id + ".")
|
||
|
}
|
||
|
}
|