package main

import (
	"bufio"
	"fmt"
	"log"
	"os"

	"github.com/boltdb/bolt"
)

const usage = "bdb <(w)rite|(r)ead> <db>"

func main() {
	log.SetFlags(log.Lshortfile)
	if len(os.Args) != 3 {
		fmt.Fprintf(os.Stderr, "%s\n", usage)
		os.Exit(1)
	}

	switch os.Args[1] {
	case "write", "read", "w", "r":
	default:
		fmt.Fprintf(os.Stderr, "%s\n", usage)
		os.Exit(1)
	}

	db, err := bolt.Open(os.Args[2], 0600, nil)
	if err != nil {
		panic(err)
	}
	defer db.Close()
	err = db.Update(func(tx *bolt.Tx) error {
		_, err := tx.CreateBucketIfNotExists([]byte("queue"))
		if err != nil {
			return err
		}
		return nil
	})
	if err != nil {
		panic(err)
	}

	switch os.Args[1] {
	case "write", "w":
		err := db.Update(func(tx *bolt.Tx) error {
			s := bufio.NewScanner(os.Stdin)
			b := tx.Bucket([]byte("queue"))
			for s.Scan() {
				err := b.Put(s.Bytes(), s.Bytes())
				if err != nil {
					return err
				}
			}
			return s.Err()
		})
		if err != nil {
			panic(err)
		}
	case "read", "r":
		db.View(func(tx *bolt.Tx) error {
			b := tx.Bucket([]byte("queue"))
			c := b.Cursor()

			for k, v := c.First(); k != nil; k, v = c.Next() {
				fmt.Printf("key=%s, value=%s\n", k, v)
			}

			return nil
		})
	}
}