|
1 | 1 | package main |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "flag" |
4 | 5 | "fmt" |
5 | | - "math/rand" |
| 6 | + "log" |
| 7 | + "strings" |
| 8 | + "time" |
| 9 | + |
| 10 | + "utils" |
| 11 | + |
| 12 | + "github.com/fatih/color" |
6 | 13 | ) |
7 | 14 |
|
8 | | -func generateUniqueRandomNumbers(n, max int) []int { |
9 | | - set := make(map[int]bool) |
10 | | - var result []int |
11 | | - for len(set) < n { |
12 | | - value := rand.Intn(max) |
13 | | - if !set[value] { |
14 | | - set[value] = true |
15 | | - result = append(result, value) |
16 | | - } |
| 15 | +// recall func |
| 16 | +func checkRecall(rememberLogfile string, recallLogfile string, recallLog bool, showhint int, dbfile string) { |
| 17 | + |
| 18 | + fmt.Println("What do you remember?") |
| 19 | + recallString := utils.ReadAndFormat() |
| 20 | + fmt.Printf("\nYou have entered: %s\n", recallString) |
| 21 | + |
| 22 | + // toCheck := fmt.Sprintf("[%s]", recallString) |
| 23 | + // correctStr := utils.FindContentInFile(rememberLogfile, recallString) |
| 24 | + correctStr := utils.FindContentInDB(dbfile, recallString) |
| 25 | + recallResult := utils.CompareHint(dbfile, recallString, correctStr, showhint, true) |
| 26 | + |
| 27 | + datetime := time.Now() |
| 28 | + datetimeFormatted := datetime.Format("2006-01-02 15:04:05") |
| 29 | + utils.AddRecalls(dbfile, []string{datetimeFormatted}, []string{recallString}, []string{fmt.Sprintf("%t", recallResult)}) |
| 30 | + |
| 31 | + if recallLog { |
| 32 | + datetime := time.Now() |
| 33 | + datetimeFormatted := datetime.Format("2006-01-02 15:04:05") |
| 34 | + info := fmt.Sprintf("%s Recall: %s, Result: %v\n", datetimeFormatted, recallString, recallResult) |
| 35 | + utils.WriteInfo(recallLogfile, info) |
17 | 36 | } |
18 | | - return result |
19 | 37 | } |
20 | 38 |
|
21 | 39 | func main() { |
22 | | - uniqueRandomNumbers := generateUniqueRandomNumbers(10, 100) |
23 | | - fmt.Printf("Random Number is %d\n", uniqueRandomNumbers) |
| 40 | + log.SetFlags(log.Ldate | log.Ltime | log.Lmicroseconds | log.Llongfile) |
| 41 | + |
| 42 | + num := flag.Int("n", 30, "Number of Random Numbers.") |
| 43 | + maxium := flag.Int("m", 100, "Generated number wont't be bigger than this number.") |
| 44 | + unique := flag.Bool("u", true, "If set, all generated numbers will be unique.") |
| 45 | + remember := flag.Bool("r", false, "If set, generated numbers will be logged into `rememberLogfile`.\nYou should set this if you want to do recall test later!") |
| 46 | + rememberLogfile := flag.String("r_file", "log.txt", "Filename of remember log") |
| 47 | + recall := flag.Bool("recall", false, "If set, run a recall test, instead of generating random numbers.") |
| 48 | + recallLog := flag.Bool("recallLog", true, "If set, recall info will be logged into `recallLogfile`.") |
| 49 | + recallLogfile := flag.String("recall_file", "recall_log.txt", "Filename of recall log") |
| 50 | + recallShowHint := flag.Int("hint", 0, "If set, when recall test failes, hint will be given.\n0 for no hint, 1 for diff hint, 2 for full hint") |
| 51 | + dbFilename := flag.String("db", "data.db", "Filename of database") |
| 52 | + migratedb := flag.Bool("migratedb", true, "If set, recall info will be logged into `recallLogfile`.") |
| 53 | + |
| 54 | + flag.Parse() |
| 55 | + |
| 56 | + if *migratedb { |
| 57 | + utils.InitDB(*rememberLogfile, *recallLogfile, *dbFilename) |
| 58 | + } |
| 59 | + |
| 60 | + if *recall { |
| 61 | + defer utils.Timer("checkRecall")() |
| 62 | + checkRecall(*rememberLogfile, *recallLogfile, *recallLog, *recallShowHint, *dbFilename) |
| 63 | + return |
| 64 | + } |
| 65 | + |
| 66 | + datetime := time.Now() |
| 67 | + datetimeFormatted := datetime.Format("2006-01-02 15:04:05") |
| 68 | + |
| 69 | + uniqueRandomNumbers := utils.GenerateRandomNumbers(*num, *maxium, *unique) |
| 70 | + |
| 71 | + // format output string, making it easier to remember |
| 72 | + var outputStr string |
| 73 | + boldBlue := color.New(color.Bold, color.FgHiBlue).SprintFunc() |
| 74 | + for idx, val := range uniqueRandomNumbers { |
| 75 | + info := "" |
| 76 | + if (idx+1)%5 == 0 { |
| 77 | + info = fmt.Sprintf(" %-2s", boldBlue(fmt.Sprintf("%-2d", val))) |
| 78 | + if (idx+1)%10 == 0 { |
| 79 | + info += "\n" |
| 80 | + } |
| 81 | + } else { |
| 82 | + if (idx+1)%10 == 1 { |
| 83 | + info = fmt.Sprintf("%-2d", val) |
| 84 | + } else { |
| 85 | + info = fmt.Sprintf(" %-2d", val) |
| 86 | + } |
| 87 | + } |
| 88 | + outputStr += info |
| 89 | + } |
| 90 | + |
| 91 | + fmt.Printf("%s Random Number Generated:\n%s\n", datetimeFormatted, outputStr) |
| 92 | + |
| 93 | + // type of remember is a pointer, add `*` prefix to get its value |
| 94 | + if *remember { |
| 95 | + info := fmt.Sprintf("%s %d\n", datetimeFormatted, uniqueRandomNumbers) |
| 96 | + filename := *rememberLogfile |
| 97 | + utils.WriteInfo(filename, info) |
| 98 | + numbersStr := fmt.Sprintf("%v", uniqueRandomNumbers) |
| 99 | + numbersStr = strings.TrimPrefix(numbersStr, "[") |
| 100 | + numbersStr = strings.TrimSuffix(numbersStr, "]") |
| 101 | + utils.AddNumbers(*dbFilename, []string{datetimeFormatted}, []string{numbersStr}) |
| 102 | + } |
24 | 103 | } |
0 commit comments