1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
8. 8
9. 9
10. 10
11. 11
12. 12
13. 13
14. 14
15. 15
16. 16
17. 17
18. 18
19. 19
20. 20
21. 21
22. 22
23. 23
24. 24
25. 25
26. 26
27. 27
28. 28
29. 29
30. 30
31. 31
32. 32
33. 33
34. 34
35. 35
36. 36
37. 37
38. 38
39. 39
40. 40
41. 41
42. 42
43. 43
44. 44
45. 45
46. 46
47. 47
48. 48
49. 49
50. 50
51. 51
52. 52
53. 53
54. 54
55. 55
56. 56
57. 57
58. 58
59. 59
60. 60
61. 61
62. 62
63. 63
64. 64
65. 65
66. 66
67. 67
68. 68
69. 69
70. 70
71. 71
72. 72
73. 73
74. 74
75. 75
76. 76
77. 77
78. 78
79. 79
80. 80
81. 81
82. 82
83. 83
84. 84
85. 85
86. 86
87. 87
88. 88
89. 89
90. 90
91. 91
92. 92
93. 93
94. 94
95. 95
96. 96
97. 97

package main

import (
	"database/sql"
	"flag"
	"fmt"
	"math/rand"
	"os"
	"os/exec"
	"time"

	_ "modernc.org/sqlite"
)

var (
	verbose = flag.Bool("v", false, "debug output")
	file    = flag.String("f", "cards", "cards file")
)

func main() {
	if !*verbose {
		// Seed the RNG
		rand.Seed(time.Now().UnixNano())
	}
	
	flag.Parse()

	db, err := sql.Open("sqlite", *file)
	if err != nil {
		panic(err)
	}

	// Get number of cards
	var N int
	_ = db.QueryRow("SELECT COUNT(*) FROM cards").Scan(&N)
	s := segmentTree{N, make([]int, 4*N)}

	// Build segment tree
	rows, err := db.Query("SELECT weight FROM cards")
	if err != nil {
		panic(err)
	}
	s.build(rows, 0, N-1, 1)
	rows.Close()

	if *verbose {
		fmt.Println(s)
	}

	// https://stackoverflow.com/questions/14094190/function-similar-to-getchar
	// disable input buffering
	exec.Command("stty", "-F", "/dev/tty", "cbreak", "min", "1").Run()
	// do not display entered characters on the screen
	exec.Command("stty", "-F", "/dev/tty", "-echo").Run()
	// restore the echoing state when exiting
	defer exec.Command("stty", "-F", "/dev/tty", "echo").Run()

	for {
		// Choose a random card
		x := rand.Intn(s.seg[1])
		w, i := s.query(x, 0, N-1, 1)

		if *verbose {
			fmt.Println(s.seg[1])
			fmt.Println(x)
			fmt.Println(w)
			fmt.Println(i)
		}

		// Get card contents from database
		var key, val string
		db.QueryRow("SELECT key, val FROM cards WHERE idx=?", i).Scan(&key, &val)
		fmt.Println(key)

		// Wait for confirmation
		var b []byte = make([]byte, 1)
		os.Stdin.Read(b)
		fmt.Println(val)

		// Read user input
		os.Stdin.Read(b)
		if b[0] == byte('y') {
			w >>= 1
		} else if b[0] == byte('n') {
			w <<= 1
		} else {
			os.Exit(0)
		}

		// Update segment tree and database
		s.update(i, w, 0, N-1, 1)
		_, err = db.Exec("UPDATE cards SET weight=? WHERE idx=?", w, i)
		if err != nil {
			panic(err)
		}
	}
}