package api

import (
	"net/http"
	"time"

	"github.com/pkg/errors"

	"github.com/CovenantSQL/CovenantSQL/api/models"
	"github.com/CovenantSQL/CovenantSQL/rpc/jsonrpc"
)

var (
	rpc    = jsonrpc.NewHandler()
	server *jsonrpc.WebsocketServer
)

func init() {
	server = &jsonrpc.WebsocketServer{
		Server: http.Server{
			ReadTimeout:  30 * time.Second,
			WriteTimeout: 60 * time.Second,
		},
	}
}

// Serve runs an API server on the specified address and database file.
func Serve(addr, dbFile string) error {
	// setup database
	if err := models.InitModels(dbFile); err != nil {
		return errors.WithMessage(err, "api: init models failed")
	}
	server.Addr = addr
	server.RPCHandler = rpc
	return server.Serve()
}

// StopService stops the API server.
func StopService() {
	server.Stop()
}