package main

import (
	"fmt"
	"os"

	"inou/lib"
)

func main() {
	if len(os.Args) < 2 {
		fmt.Println("Usage: migrate <dob|category>")
		os.Exit(1)
	}

	if err := lib.Init(); err != nil {
		fmt.Printf("Error initializing: %v\n", err)
		os.Exit(1)
	}
	defer lib.DBClose()

	switch os.Args[1] {
	case "dob":
		if err := lib.MigrateDOB(); err != nil {
			fmt.Printf("Error: %v\n", err)
			os.Exit(1)
		}
	case "category":
		if err := lib.MigrateCategory(); err != nil {
			fmt.Printf("Error: %v\n", err)
			os.Exit(1)
		}
	default:
		fmt.Printf("Unknown migration: %s\n", os.Args[1])
		os.Exit(1)
	}
}