Files
lndhub.go/controllers_v2/invoice.ctrl.go
2023-12-07 16:44:47 +05:30

247 lines
8.8 KiB
Go

package v2controllers
import (
"net/http"
"time"
"github.com/getAlby/lndhub.go/common"
"github.com/getAlby/lndhub.go/lib/responses"
"github.com/getAlby/lndhub.go/lib/service"
"github.com/labstack/echo/v4"
"github.com/labstack/gommon/log"
)
// InvoiceController : Add invoice controller struct
type InvoiceController struct {
svc *service.LndhubService
}
func NewInvoiceController(svc *service.LndhubService) *InvoiceController {
return &InvoiceController{svc: svc}
}
type Invoice struct {
PaymentHash string `json:"payment_hash"`
PaymentRequest string `json:"payment_request"`
Description string `json:"description"`
DescriptionHash string `json:"description_hash,omitempty"`
PaymentPreimage string `json:"payment_preimage,omitempty"`
Destination string `json:"destination"`
Amount int64 `json:"amount"`
Fee int64 `json:"fee"`
Status string `json:"status"`
Type string `json:"type"`
ErrorMessage string `json:"error_message,omitempty"`
SettledAt time.Time `json:"settled_at"`
ExpiresAt time.Time `json:"expires_at"`
IsPaid bool `json:"is_paid"`
Keysend bool `json:"keysend"`
CustomRecords map[uint64][]byte `json:"custom_records,omitempty"`
}
// GetOutgoingInvoices godoc
// @Summary Retrieve outgoing payments
// @Description Returns a list of outgoing payments for a user
// @Accept json
// @Produce json
// @Tags Invoice
// @Success 200 {object} []Invoice
// @Failure 400 {object} responses.ErrorResponse
// @Failure 500 {object} responses.ErrorResponse
// @Router /v2/invoices/outgoing [get]
// @Security OAuth2Password
func (controller *InvoiceController) GetOutgoingInvoices(c echo.Context) error {
userId := c.Get("UserID").(int64)
invoices, err := controller.svc.InvoicesFor(c.Request().Context(), userId, common.InvoiceTypeOutgoing)
if err != nil {
c.Logger().Errorj(
log.JSON{
"message": "failed to get invoices",
"error": err,
"lndhub_user_id": userId,
},
)
return c.JSON(http.StatusBadRequest, responses.BadArgumentsError)
}
response := make([]Invoice, len(invoices))
for i, invoice := range invoices {
response[i] = Invoice{
PaymentHash: invoice.RHash,
PaymentRequest: invoice.PaymentRequest,
Description: invoice.Memo,
DescriptionHash: invoice.DescriptionHash,
PaymentPreimage: invoice.Preimage,
Destination: invoice.DestinationPubkeyHex,
Amount: invoice.Amount,
Fee: invoice.Fee,
Status: invoice.State,
Type: common.InvoiceTypePaid,
ErrorMessage: invoice.ErrorMessage,
SettledAt: invoice.SettledAt.Time,
ExpiresAt: invoice.ExpiresAt.Time,
IsPaid: invoice.State == common.InvoiceStateSettled,
Keysend: invoice.Keysend,
CustomRecords: invoice.DestinationCustomRecords,
}
}
return c.JSON(http.StatusOK, &response)
}
// GetIncomingInvoices godoc
// @Summary Retrieve incoming invoices
// @Description Returns a list of incoming invoices for a user
// @Accept json
// @Produce json
// @Tags Invoice
// @Success 200 {object} []Invoice
// @Failure 400 {object} responses.ErrorResponse
// @Failure 500 {object} responses.ErrorResponse
// @Router /v2/invoices/incoming [get]
// @Security OAuth2Password
func (controller *InvoiceController) GetIncomingInvoices(c echo.Context) error {
userId := c.Get("UserID").(int64)
invoices, err := controller.svc.InvoicesFor(c.Request().Context(), userId, common.InvoiceTypeIncoming)
if err != nil {
c.Logger().Errorj(
log.JSON{
"message": "failed to get invoices",
"error": err,
"lndhub_user_id": userId,
},
)
return c.JSON(http.StatusBadRequest, responses.BadArgumentsError)
}
response := make([]Invoice, len(invoices))
for i, invoice := range invoices {
response[i] = Invoice{
PaymentHash: invoice.RHash,
PaymentRequest: invoice.PaymentRequest,
Description: invoice.Memo,
DescriptionHash: invoice.DescriptionHash,
Destination: invoice.DestinationPubkeyHex,
Amount: invoice.Amount,
Fee: invoice.Fee,
Status: invoice.State,
Type: common.InvoiceTypeUser,
ErrorMessage: invoice.ErrorMessage,
SettledAt: invoice.SettledAt.Time,
ExpiresAt: invoice.ExpiresAt.Time,
IsPaid: invoice.State == common.InvoiceStateSettled,
Keysend: invoice.Keysend,
CustomRecords: invoice.DestinationCustomRecords,
}
}
return c.JSON(http.StatusOK, &response)
}
type AddInvoiceRequestBody struct {
Amount int64 `json:"amount" validate:"gte=0"`
Description string `json:"description"`
DescriptionHash string `json:"description_hash" validate:"omitempty,hexadecimal,len=64"`
}
type AddInvoiceResponseBody struct {
PaymentHash string `json:"payment_hash"`
PaymentRequest string `json:"payment_request"`
ExpiresAt time.Time `json:"expires_at"`
CreatedAt time.Time `json:"created_at"`
}
// AddInvoice godoc
// @Summary Generate a new invoice
// @Description Returns a new bolt11 invoice
// @Accept json
// @Produce json
// @Tags Invoice
// @Param invoice body AddInvoiceRequestBody True "Add Invoice"
// @Success 200 {object} AddInvoiceResponseBody
// @Failure 400 {object} responses.ErrorResponse
// @Failure 500 {object} responses.ErrorResponse
// @Router /v2/invoices [post]
// @Security OAuth2Password
func (controller *InvoiceController) AddInvoice(c echo.Context) error {
userID := c.Get("UserID").(int64)
var body AddInvoiceRequestBody
if err := c.Bind(&body); err != nil {
c.Logger().Errorf("Failed to load addinvoice request body: %v", err)
return c.JSON(http.StatusBadRequest, responses.BadArgumentsError)
}
if err := c.Validate(&body); err != nil {
c.Logger().Errorf("Invalid addinvoice request body: %v", err)
return c.JSON(http.StatusBadRequest, responses.BadArgumentsError)
}
resp, err := controller.svc.CheckIncomingPaymentAllowed(c, body.Amount, userID)
if err != nil {
return c.JSON(http.StatusInternalServerError, responses.GeneralServerError)
}
if resp != nil {
c.Logger().Errorf("Error: %v user_id:%v amount:%v", resp.Message, userID, body.Amount)
return c.JSON(resp.HttpStatusCode, resp)
}
c.Logger().Infof("Adding invoice: user_id:%v memo:%s value:%v description_hash:%s", userID, body.Description, body.Amount, body.DescriptionHash)
invoice, errResp := controller.svc.AddIncomingInvoice(c.Request().Context(), userID, body.Amount, body.Description, body.DescriptionHash)
if errResp != nil {
return c.JSON(errResp.HttpStatusCode, errResp)
}
responseBody := AddInvoiceResponseBody{
PaymentHash: invoice.RHash,
PaymentRequest: invoice.PaymentRequest,
ExpiresAt: invoice.ExpiresAt.Time,
CreatedAt: invoice.CreatedAt,
}
return c.JSON(http.StatusOK, &responseBody)
}
// GetInvoice godoc
// @Summary Get a specific invoice
// @Description Retrieve information about a specific invoice by payment hash
// @Accept json
// @Produce json
// @Tags Invoice
// @Param payment_hash path string true "Payment hash"
// @Success 200 {object} Invoice
// @Failure 400 {object} responses.ErrorResponse
// @Failure 500 {object} responses.ErrorResponse
// @Router /v2/invoices/{payment_hash} [get]
// @Security OAuth2Password
func (controller *InvoiceController) GetInvoice(c echo.Context) error {
userID := c.Get("UserID").(int64)
rHash := c.Param("payment_hash")
invoice, err := controller.svc.FindInvoiceByPaymentHash(c.Request().Context(), userID, rHash)
// Probably we did not find the invoice
if err != nil {
c.Logger().Errorf("Invalid checkpayment request user_id:%v payment_hash:%s", userID, rHash)
return c.JSON(http.StatusBadRequest, responses.BadArgumentsError)
}
responseBody := Invoice{
PaymentHash: invoice.RHash,
PaymentRequest: invoice.PaymentRequest,
Description: invoice.Memo,
DescriptionHash: invoice.DescriptionHash,
PaymentPreimage: invoice.Preimage,
Destination: invoice.DestinationPubkeyHex,
Amount: invoice.Amount,
Fee: invoice.Fee,
Status: invoice.State,
Type: invoice.Type,
ErrorMessage: invoice.ErrorMessage,
SettledAt: invoice.SettledAt.Time,
ExpiresAt: invoice.ExpiresAt.Time,
IsPaid: invoice.State == common.InvoiceStateSettled,
Keysend: invoice.Keysend,
CustomRecords: invoice.DestinationCustomRecords,
}
return c.JSON(http.StatusOK, &responseBody)
}