package bao

import (
	"context"

	"github.com/docker/go-units"

	"git.lumeweb.com/LumeWeb/portal/bao/proto"
	"github.com/google/uuid"
	"github.com/hashicorp/go-plugin"
	"google.golang.org/grpc"
)

var _ Bao = (*BaoGRPC)(nil)

const VERIFY_CHUNK_SIZE = 256 * units.KiB

type Bao interface {
	NewHasher() uuid.UUID
	Hash(id uuid.UUID, data []byte) bool
	Finish(id uuid.UUID) Result
	Verify(data []byte, offset uint64, proof []byte, hash []byte) bool
}

type BaoPlugin struct {
	plugin.Plugin
}

func (p *BaoPlugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
	return nil
}

func (p *BaoPlugin) GRPCClient(ctx context.Context, broker *plugin.GRPCBroker, c *grpc.ClientConn) (interface{}, error) {
	return &BaoGRPC{client: proto.NewBaoClient(c)}, nil
}

type Result struct {
	Hash   []byte
	Proof  []byte
	Length uint
}
type BaoGRPC struct {
	client proto.BaoClient
}

func (b *BaoGRPC) NewHasher() uuid.UUID {
	ret, err := b.client.NewHasher(context.Background(), &proto.NewHasherRequest{})

	if err != nil {
		panic(err)
	}

	id, err := uuid.Parse(ret.Id)

	if err != nil {
		panic(err)
	}

	return id
}

func (b *BaoGRPC) Hash(id uuid.UUID, data []byte) bool {
	ret, err := b.client.Hash(context.Background(), &proto.HashRequest{Id: id.String(), Data: data})

	if err != nil {
		panic(err)
	}

	return ret.Status
}

func (b *BaoGRPC) Finish(id uuid.UUID) Result {
	ret, err := b.client.Finish(context.Background(), &proto.FinishRequest{Id: id.String()})

	if err != nil {
		panic(err)
	}

	return Result{Hash: ret.Hash, Proof: ret.Proof}
}

func (b *BaoGRPC) Verify(data []byte, offset uint64, proof []byte, hash []byte) bool {
	ret, err := b.client.Verify(context.Background(), &proto.VerifyRequest{Data: data, Offset: offset, Proof: proof, Hash: hash})

	if err != nil {
		panic(err)
	}

	return ret.Status
}