// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.

package statdb

import (
	"context"
	"flag"

	"go.uber.org/zap"

	"storj.io/storj/pkg/provider"
	pb "storj.io/storj/pkg/statdb/proto"
)

//CtxKey Used as statdb key
type CtxKey int

const (
	ctxKeyStats CtxKey = iota
)

var (
	apiKey = flag.String("stat-db.auth.api-key", "", "statdb api key")
)

// Config is a configuration struct that is everything you need to start a
// StatDB responsibility
type Config struct {
	DatabaseURL    string `help:"the database connection string to use" default:"$CONFDIR/stats.db"`
	DatabaseDriver string `help:"the database driver to use" default:"sqlite3"`
}

// Run implements the provider.Responsibility interface
func (c Config) Run(ctx context.Context, server *provider.Provider) error {
	ns, err := NewServer(c.DatabaseDriver, c.DatabaseURL, *apiKey, zap.L())
	if err != nil {
		return err
	}

	pb.RegisterStatDBServer(server.GRPC(), ns)

	return server.Run(context.WithValue(ctx, ctxKeyStats, ns))
}

// LoadFromContext loads an existing StatDB from the Provider context
// stack if one exists.
func LoadFromContext(ctx context.Context) *Server {
	if v, ok := ctx.Value(ctxKeyStats).(*Server); ok {
		return v
	}
	return nil
}