// Copyright 2013 The Chihaya Authors. All rights reserved.
// Use of this source code is governed by the BSD 2-Clause license,
// which can be found in the LICENSE file.

// Package tracker provides a generic interface for manipulating a
// BitTorrent tracker's fast-moving data.
package tracker

import (
	"errors"
	"fmt"

	"github.com/chihaya/chihaya/config"
	"github.com/chihaya/chihaya/storage"
)

var (
	ErrMissingResource = errors.New("tracker: resource missing")
	drivers            = make(map[string]Driver)
)

type Driver interface {
	New(*config.DataStore) Pool
}

// Register makes a database driver available by the provided name.
// If Register is called twice with the same name or if driver is nil,
// it panics.
func Register(name string, driver Driver) {
	if driver == nil {
		panic("tracker: Register driver is nil")
	}
	if _, dup := drivers[name]; dup {
		panic("tracker: Register called twice for driver " + name)
	}
	drivers[name] = driver
}

// Open creates a pool of data store connections specified by a storage configuration.
func Open(conf *config.DataStore) (Pool, error) {
	driver, ok := drivers[conf.Driver]
	if !ok {
		return nil, fmt.Errorf(
			"tracker: unknown driver %q (forgotten import?)",
			conf.Driver,
		)
	}
	pool := driver.New(conf)
	return pool, nil
}

// Pool represents a thread-safe pool of connections to the data store
// that can be used to safely within concurrent goroutines.
type Pool interface {
	Close() error
	Get() (Conn, error)
}

// Conn represents a connection to the data store that can be used
// to make reads/writes.
type Conn interface {
	// Reads
	FindUser(passkey string) (*storage.User, bool, error)
	FindTorrent(infohash string) (*storage.Torrent, bool, error)
	ClientWhitelisted(peerID string) (bool, error)

	// Writes
	RecordSnatch(u *storage.User, t *storage.Torrent) error
	MarkActive(t *storage.Torrent) error
	AddLeecher(t *storage.Torrent, p *storage.Peer) error
	AddSeeder(t *storage.Torrent, p *storage.Peer) error
	RemoveLeecher(t *storage.Torrent, p *storage.Peer) error
	RemoveSeeder(t *storage.Torrent, p *storage.Peer) error
	SetLeecher(t *storage.Torrent, p *storage.Peer) error
	SetSeeder(t *storage.Torrent, p *storage.Peer) error
	LeecherFinished(t *storage.Torrent, p *storage.Peer) error

	// Priming / Testing
	AddTorrent(t *storage.Torrent) error
	RemoveTorrent(t *storage.Torrent) error
	AddUser(u *storage.User) error
	RemoveUser(u *storage.User) error
	WhitelistClient(peerID string) error
	UnWhitelistClient(peerID string) error
}