改善

config/config.go

@@ -7,42 +7,79 @@ import (
 )
 
 func Load() {
-
+	// where to read config
 	viper.SetConfigName("config")
 	viper.SetConfigType("yaml")
 	viper.AddConfigPath(".")
 
-	if err := viper.ReadInConfig(); err != nil {
-		fmt.Printf("Error reading config file: %v\n", err)
-		os.Exit(1)
-	}
+	readconfig()
 
-	ListenPort = viper.GetInt("ListenPort")
-	File = viper.GetString("File")
-	DB.Path = viper.GetString("DB.Path")
-	//DB.Port = viper.GetString("DB.Port")
-	DB.Type = viper.GetString("DB.Type")
-
-	err := check()
-	if err != nil {
-		fmt.Println("check your config!")
-		os.Exit(1)
-	}
+	check()
 
 	debugprint()
 
+	if len(os.Args) != 1 {
+		handleArguments()
+	}
+
 }
 
-func check() error {
-
-	return nil
+func check() {
+	// 暂时只支持SQLite
+	if DB.Type != "SQLite" {
+		fmt.Println("sorry,we support SQLite only so far(At config/config.go : check())")
+		os.Exit(1)
+	}
 
 }
 func debugprint() {
 
 	fmt.Printf("ListenPort=%v\n", ListenPort)
-	fmt.Printf("File=%v\n", File)
-	fmt.Printf("database path : %s\n", DB.Path)
 	fmt.Printf("database type:%s\n", DB.Type)
+	fmt.Printf("database path : %s\n", DB.Path)
+	fmt.Printf("database port:%d\n", DB.Port)
+	fmt.Printf("database user:%s\n", DB.User)
+	fmt.Printf("database passowrd:%s\n", DB.Password)
+	fmt.Printf("database name:%s\n", DB.Name)
+	fmt.Printf("session:%s\n", Session)
+	fmt.Printf("semester:%d\n", Semester)
+	fmt.Printf("start time:%s\n", StartTime)
+	fmt.Printf("week:%d\n", Week)
+	fmt.Printf("File=%v\n", File)
 
 }
+
+func handleArguments() {
+	if len(os.Args) > 2 {
+		fmt.Println("Please enter only 1 argument")
+		os.Exit(1)
+	}
+	switch os.Args[1] {
+	case "newsemester":
+		if DB.Type == "SQLite" {
+			sqliteNewSemester()
+		}
+	default:
+		panic("invalid argument")
+	}
+
+}
+
+func readconfig() {
+	if err := viper.ReadInConfig(); err != nil {
+		fmt.Printf("Error reading config file: %v\n", err)
+		os.Exit(1)
+	}
+	ListenPort = viper.GetInt("ListenPort")
+	DB.Type = viper.GetString("DB.Type")
+	DB.Path = viper.GetString("DB.Path")
+	DB.Port = viper.GetInt("DB.Port")
+	DB.User = viper.GetString("DB.User")
+	DB.Password = viper.GetString("DB.Password")
+	DB.Name = viper.GetString("DB.Name")
+	Session = viper.GetString("Session")
+	Semester = viper.GetInt("Semester")
+	StartTime = viper.GetString("StartTime")
+	Week = viper.GetInt("Week")
+	File = viper.GetString("File")
+}