add channel table.

cmd/rpdata-restore/main.go

@@ -24,6 +24,7 @@ var flagPassword = flag.String("password", "", "")
 var flagMechanism = flag.String("mechanism", "", "")
 var flagInputFile = flag.String("infile", "dump.zip", "The file to read from.")
 var flagIncludeKeys = flag.Bool("include-keys", false, "Whether to include the keys.")
+var flagReplace = flag.Bool("replace", false, "Replace existing content")
 
 func main() {
 	flag.Parse()
@@ -82,6 +83,8 @@ func main() {
 
 		hideList := make(map[string]bool)
 
+		log.Println("Loaded", parts[1], parts[2], "from archive.")
+
 		switch parts[1] {
 		case "character":
 			{
@@ -91,6 +94,10 @@ func main() {
 					log.Fatalln("Could not parse character:", parts[2], err)
 				}
 
+				if *flagReplace {
+					_ = db.Characters().Delete(ctx, character)
+				}
+
 				_, err = db.Characters().Insert(ctx, character)
 				if err != nil {
 					log.Fatalln("Could not insert character:", parts[2], err)
@@ -106,6 +113,10 @@ func main() {
 					log.Fatalln("Could not parse channel:", parts[2], err)
 				}
 
+				if *flagReplace {
+					_ = db.Channels().Remove(ctx, channel)
+				}
+
 				_, err = db.Channels().Insert(ctx, channel)
 				if err != nil {
 					log.Fatalln("Could not insert channel:", parts[2], err)
@@ -121,6 +132,10 @@ func main() {
 					log.Fatalln("Could not parse character:", parts[2], err)
 				}
 
+				if *flagReplace {
+					_ = db.Changes().Remove(ctx, change)
+				}
+
 				_, err = db.Changes().Insert(ctx, change)
 				if err != nil {
 					log.Fatalln("Could not insert character:", parts[2], err)