rdb.go

package main

import (
	"fmt"
	"github.com/cameronpm/rdb-cli/decoder"
	"github.com/cupcake/rdb"
	"github.com/jessevdk/go-flags"
	"log"
	"os"
)

var Version = "0.2"

var formats = map[string]rdb.Decoder{
	"protocol": decoder.Protocol(os.Stdout),
	"diff":     decoder.Diff(),
	"size":     decoder.Size(),
}

type rdbOptions struct {
	Path string `required:"1" positional-arg-name:"RDBPATH" description:"Path to RDB file"`
}

var options struct {
	Version func()     `long:"version" description:"Print version and exit"`
	Format  string     `long:"format" choice:"diff" choice:"protocol" choice:"size" description:"Output format of RDB file"`
	Rdb     rdbOptions `positional-args:"1"`
}

func format(rdbPath string, d rdb.Decoder) {
	file, err := os.Open(rdbPath)
	if err != nil {
		log.Fatalf("Unable to open RDB file '%s': %s\n", options.Rdb.Path, err)
	}

	err = rdb.Decode(file, d)
	if err != nil {
		log.Fatalf("Failed to decode RDB file '%s': %s\n", options.Rdb.Path, err)
	}
}

func main() {
	options.Version = func() {
		fmt.Printf("rdb-cli v%s\n", Version)
		os.Exit(0)
	}

	if _, err := flags.Parse(&options); err != nil {
		os.Exit(1)
	}

	format(options.Rdb.Path, formats[options.Format])
}