package main

import (
	"bytes"
	"flag"
	"fmt"
	"os"

	"github.com/harlow/kinesis-connectors"
	"github.com/harlow/kinesis-connectors/emitter/s3"
)

var (
	app    = flag.String("a", "", "App name")
	bucket = flag.String("b", "", "Bucket name")
	stream = flag.String("s", "", "Stream name")
)

func main() {
	flag.Parse()

	emitter := &s3.Emitter{
		Bucket: *bucket,
		Region: "us-west-1",
	}

	cfg := connector.Config{
		MaxBatchCount: 500,
	}

	c := connector.NewConsumer(*app, *stream, cfg)

	c.Start(connector.HandlerFunc(func(b connector.Buffer) {
		body := new(bytes.Buffer)

		for _, r := range b.GetRecords() {
			body.Write(r.Data)
		}

		err := emitter.Emit(
			s3.Key("", b.FirstSeq(), b.LastSeq()),
			bytes.NewReader(body.Bytes()),
		)

		if err != nil {
			fmt.Printf("error %s\n", err)
			os.Exit(1)
		}
	}))

	select {} // run forever
}