package main import ( "database/sql" "fmt" "log" "time" _ "github.com/mattn/go-sqlite3" ) // Init database func init_limit_db(DBPath string) *sql.DB { db, err := sql.Open("sqlite3", DBPath) if err != nil { log.Fatal(err) } cmd := `CREATE TABLE IF NOT EXISTS Limits (id INTEGER PRIMARY KEY AUTOINCREMENT, acct TEXT, ticket INTEGER, time TEXT)` stat, err := db.Prepare(cmd) if err != nil { log.Fatal(err) } stat.Exec() return db } // Add account to database func add_to_db(acct string, limit uint16, DBPath string) { db := init_limit_db(DBPath) cmd := `INSERT INTO Limits (acct, ticket) VALUES (?, ?)` stat, err := db.Prepare(cmd) if err != nil { log.Fatal(err) } stat.Exec(acct, limit) } // Take ticket for tooting func take_ticket(acct string, DBPath string) { db := init_limit_db(DBPath) cmd1 := `SELECT ticket FROM Limits WHERE acct = ?` cmd2 := `UPDATE Limits SET ticket = ?, time = ? WHERE acct = ?` var ticket uint16 db.QueryRow(cmd1, acct).Scan(&ticket) if ticket > 0 { ticket = ticket - 1 } stat, err := db.Prepare(cmd2) if err != nil { log.Fatal(err) } now := time.Now() last_toot_at := time.Date(now.Year(), now.Month(), now.Day(), now.Hour(), now.Minute(), now.Second(), 0, time.Local).Format("2006/01/02 15:04:05 MST") stat.Exec(ticket, last_toot_at, acct) } // Check followed once func followed(acct string, DBPath string) bool { db := init_limit_db(DBPath) cmd := `SELECT acct FROM Limits WHERE acct = ?` err := db.QueryRow(cmd, acct).Scan(&acct) if err != nil { if err != sql.ErrNoRows { log.Fatal(err) } return false } return true } // Check ticket availability func check_ticket(acct string, ticket uint16, toots_interval uint16, DBPath string) uint16 { db := init_limit_db(DBPath) cmd1 := `SELECT ticket FROM Limits WHERE acct = ?` cmd2 := `SELECT time FROM Limits WHERE acct = ?` var tickets uint16 var lastS string db.QueryRow(cmd1, acct).Scan(&tickets) db.QueryRow(cmd2, acct).Scan(&lastS) lastT, _ := time.Parse("2006/01/02 15:04:05 MST", lastS) since := time.Since(lastT) limit := fmt.Sprintf("%dh", toots_interval) interval, _ := time.ParseDuration(limit) if since >= interval { cmd := `UPDATE Limits SET ticket = ? WHERE acct = ?` stat, err := db.Prepare(cmd) if err != nil { log.Fatal(err) } stat.Exec(ticket, acct) return ticket } return tickets }