udemy-go-web-2/cmd/web/handlers.go

292 lines
7.1 KiB
Go
Raw Normal View History

package main
2024-08-03 20:26:40 +00:00
import (
"myapp/internal/cards"
2024-08-10 09:10:27 +00:00
"myapp/internal/models"
2024-08-03 20:26:40 +00:00
"net/http"
2024-08-07 09:56:53 +00:00
"strconv"
"github.com/go-chi/chi/v5"
2024-08-03 20:26:40 +00:00
)
2024-08-10 09:10:27 +00:00
func (app *application) Home(w http.ResponseWriter, r *http.Request) {
if err := app.renderTemplate(w, r, "home", &templateData{}); err != nil {
app.errorLog.Println(err)
}
}
func (app *application) VirtualTerminal(w http.ResponseWriter, r *http.Request) {
if err := app.renderTemplate(w, r, "terminal", &templateData{}); err != nil {
2024-08-03 20:26:40 +00:00
app.errorLog.Println(err)
}
}
2024-08-04 16:17:56 +00:00
type TransactionData struct {
FirstName string
LastName string
Email string
PaymentIntentID string
PaymentMethodID string
PaymentAmount int
PaymentCurrency string
LastFour string
ExpiryMonth int
ExpiryYear int
BankReturnCode string
}
func (app *application) GetTransactionData(r *http.Request) (TransactionData, error) {
var txnData TransactionData
2024-08-04 16:17:56 +00:00
err := r.ParseForm()
if err != nil {
app.errorLog.Println(err)
return txnData, err
2024-08-04 16:17:56 +00:00
}
2024-08-10 09:10:27 +00:00
firstName := r.Form.Get("first_name")
lastName := r.Form.Get("last_name")
2024-08-04 16:17:56 +00:00
email := r.Form.Get("cardholder_email")
paymentIntent := r.Form.Get("payment_intent")
paymentMethod := r.Form.Get("payment_method")
paymentAmount := r.Form.Get("payment_amount")
paymentCurrency := r.Form.Get("payment_currency")
amount, _ := strconv.Atoi(paymentAmount)
// TODO: validation of the data
2024-08-04 16:17:56 +00:00
card := cards.Card{
Secret: app.config.stripe.secret,
Key: app.config.stripe.key,
}
pi, err := card.RetrievePaymentIntent(paymentIntent)
if err != nil {
app.errorLog.Println(err)
return txnData, err
}
pm, err := card.GetPaymentMethod(paymentMethod)
if err != nil {
app.errorLog.Println(err)
return txnData, err
}
lastFour := pm.Card.Last4
expiryMonth := pm.Card.ExpMonth
expiryYear := pm.Card.ExpYear
txnData = TransactionData{
FirstName: firstName,
LastName: lastName,
Email: email,
PaymentIntentID: paymentIntent,
PaymentMethodID: paymentMethod,
PaymentAmount: amount,
PaymentCurrency: paymentCurrency,
LastFour: lastFour,
ExpiryMonth: int(expiryMonth),
ExpiryYear: int(expiryYear),
BankReturnCode: pi.LatestCharge.ID,
}
return txnData, nil
}
func (app *application) PaymentSucceeded(w http.ResponseWriter, r *http.Request) {
// read posted data
err := r.ParseForm()
if err != nil {
app.errorLog.Println(err)
return
}
widgetID, _ := strconv.Atoi(r.Form.Get("product_id"))
txnData, err := app.GetTransactionData(r)
if err != nil {
app.errorLog.Println(err)
return
}
customerID, err := app.SaveCustomer(txnData.FirstName, txnData.LastName, txnData.Email)
2024-08-10 09:10:27 +00:00
if err != nil {
app.errorLog.Println(err)
return
}
app.infoLog.Printf("custumer id: %d", customerID)
transaction := models.Transaction{
Amount: txnData.PaymentAmount,
Currency: txnData.PaymentCurrency,
LastFour: txnData.LastFour,
ExpiryMonth: txnData.ExpiryMonth,
ExpiryYear: txnData.ExpiryYear,
PaymentIntent: txnData.PaymentIntentID,
PaymentMethod: txnData.PaymentMethodID,
BankReturnCode: txnData.BankReturnCode,
2024-08-10 09:10:27 +00:00
TransactionStatusID: 2, // TODO: use an enum
}
txnID, err := app.SaveTransaction(transaction)
if err != nil {
app.errorLog.Println(err)
return
}
app.infoLog.Printf("transaction id: %d", txnID)
order := models.Order{
WidgetID: widgetID,
TransactionID: txnID,
CustomerID: customerID,
StatusID: 1,
Quantity: 1,
Amount: txnData.PaymentAmount,
2024-08-10 09:10:27 +00:00
}
orderID, err := app.SaveOrder(order)
if err != nil {
app.errorLog.Println(err)
return
}
app.infoLog.Printf("order id: %d", orderID)
app.Session.Put(r.Context(), "txn", txnData)
http.Redirect(w, r, "/receipt", http.StatusSeeOther)
}
2024-08-11 10:30:57 +00:00
func (app *application) VirtualTerminalPaymentSucceeded(w http.ResponseWriter, r *http.Request) {
txnData, err := app.GetTransactionData(r)
if err != nil {
app.errorLog.Println(err)
return
}
transaction := models.Transaction{
Amount: txnData.PaymentAmount,
Currency: txnData.PaymentCurrency,
LastFour: txnData.LastFour,
ExpiryMonth: txnData.ExpiryMonth,
ExpiryYear: txnData.ExpiryYear,
PaymentIntent: txnData.PaymentIntentID,
PaymentMethod: txnData.PaymentMethodID,
BankReturnCode: txnData.BankReturnCode,
TransactionStatusID: 2, // TODO: use an enum
}
_, err = app.SaveTransaction(transaction)
if err != nil {
app.errorLog.Println(err)
return
}
app.Session.Put(r.Context(), "txn", txnData)
http.Redirect(w, r, "/virtual-terminal-receipt", http.StatusSeeOther)
}
func (app *application) Receipt(w http.ResponseWriter, r *http.Request) {
txn := app.Session.Pop(r.Context(), "txn").(TransactionData)
2024-08-04 16:17:56 +00:00
data := make(map[string]interface{})
data["txn"] = txn
if err := app.renderTemplate(w, r, "receipt", &templateData{
2024-08-04 16:17:56 +00:00
Data: data,
}); err != nil {
app.errorLog.Println(err)
}
}
2024-08-06 11:29:32 +00:00
2024-08-11 10:30:57 +00:00
func (app *application) VirtualTerminalReceipt(w http.ResponseWriter, r *http.Request) {
txn := app.Session.Pop(r.Context(), "txn").(TransactionData)
data := make(map[string]interface{})
data["txn"] = txn
if err := app.renderTemplate(w, r, "virtual-terminal-receipt", &templateData{
Data: data,
}); err != nil {
app.errorLog.Println(err)
}
}
2024-08-10 09:10:27 +00:00
func (app *application) SaveCustomer(firstName, lastName, email string) (int, error) {
customer := models.Customer{
FirstName: firstName,
LastName: lastName,
Email: email,
}
id, err := app.DB.InsertCustomer(customer)
if err != nil {
return 0, err
}
return id, nil
}
func (app *application) SaveTransaction(txn models.Transaction) (int, error) {
txnID, err := app.DB.InsertTransaction(txn)
if err != nil {
app.errorLog.Println(err)
return 0, err
}
return txnID, nil
}
func (app *application) SaveOrder(order models.Order) (int, error) {
id, err := app.DB.InsertOrder(order)
if err != nil {
app.errorLog.Println(err)
return 0, err
}
return id, nil
}
2024-08-06 11:29:32 +00:00
// ChargeOnce displays the page to buy one widget
func (app *application) ChargeOnce(w http.ResponseWriter, r *http.Request) {
2024-08-07 09:56:53 +00:00
id := chi.URLParam(r, "id")
widgetID, _ := strconv.Atoi(id)
widget, err := app.DB.GetWidget(widgetID)
if err != nil {
app.errorLog.Println(err)
}
data := make(map[string]interface{})
data["widget"] = widget
if err := app.renderTemplate(w, r, "buy-once", &templateData{
Data: data,
}); err != nil {
2024-08-06 11:29:32 +00:00
app.errorLog.Println(err)
}
}
2024-08-11 20:08:55 +00:00
func (app *application) BronzePlan(w http.ResponseWriter, r *http.Request) {
widget, err := app.DB.GetWidget(2)
if err != nil {
app.errorLog.Println(err)
return
}
data := make(map[string]interface{})
data["widget"] = widget
if err := app.renderTemplate(w, r, "bronze-plan", &templateData{
Data: data,
}); err != nil {
app.errorLog.Println(err)
}
}
2024-08-12 17:23:44 +00:00
func (app *application) BronzePlanReceipt(w http.ResponseWriter, r *http.Request) {
if err := app.renderTemplate(w, r, "receipt-plan", &templateData{}); err != nil {
app.errorLog.Println(err)
}
}
2024-08-12 19:38:12 +00:00
func (app *application) LoginPage(w http.ResponseWriter, r *http.Request) {
if err := app.renderTemplate(w, r, "login", &templateData{}); err != nil {
app.errorLog.Println(err)
}
}