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

package satellitedb

import (
	"context"

	"github.com/skyrings/skyring-common/tools/uuid"

	"storj.io/storj/satellite/console"
	dbx "storj.io/storj/satellite/satellitedb/dbx"
)

// projectpayments is the an implementation of console.ProjectPayments.
// Allows to work with project payment info storage
type projectpayments struct {
	db dbx.Methods
}

// Create stores new project payment info into db
func (infos *projectpayments) Create(ctx context.Context, info console.ProjectPayment) (*console.ProjectPayment, error) {
	dbxInfo, err := infos.db.Create_ProjectPayment(ctx,
		dbx.ProjectPayment_ProjectId(info.ProjectID[:]),
		dbx.ProjectPayment_PayerId(info.PayerID[:]),
		dbx.ProjectPayment_PaymentMethodId(info.PaymentMethodID))

	if err != nil {
		return nil, err
	}

	return fromDBXProjectPayment(dbxInfo)
}

// GetByProjectID retrieves project payment info from db by projectID
func (infos *projectpayments) GetByProjectID(ctx context.Context, projectID uuid.UUID) (*console.ProjectPayment, error) {
	dbxInfo, err := infos.db.Get_ProjectPayment_By_ProjectId(ctx, dbx.ProjectPayment_ProjectId(projectID[:]))
	if err != nil {
		return nil, err
	}

	return fromDBXProjectPayment(dbxInfo)
}

// GetByPayerID retrieves project payment info from db by payerID(userID)
func (infos *projectpayments) GetByPayerID(ctx context.Context, payerID uuid.UUID) (*console.ProjectPayment, error) {
	dbxInfo, err := infos.db.Get_ProjectPayment_By_PayerId(ctx, dbx.ProjectPayment_PayerId(payerID[:]))
	if err != nil {
		return nil, err
	}

	return fromDBXProjectPayment(dbxInfo)
}

// fromDBXProjectPayment is a helper method to convert from *dbx.ProjectPayment to *console.ProjectPayment
func fromDBXProjectPayment(dbxInfo *dbx.ProjectPayment) (*console.ProjectPayment, error) {
	projectID, err := bytesToUUID(dbxInfo.ProjectId)
	if err != nil {
		return nil, err
	}

	payerID, err := bytesToUUID(dbxInfo.PayerId)
	if err != nil {
		return nil, err
	}

	return &console.ProjectPayment{
		ProjectID:       projectID,
		PayerID:         payerID,
		PaymentMethodID: dbxInfo.PaymentMethodId,
		CreatedAt:       dbxInfo.CreatedAt,
	}, nil
}