Pay method
This commit is contained in:
218
frontend/style/app/pay/pay.go
Normal file
218
frontend/style/app/pay/pay.go
Normal file
@@ -0,0 +1,218 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/gorilla/handlers"
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
||||
// Структура для тела запроса на localhost:5000
|
||||
type PaymentRequest struct {
|
||||
ID string `json:"id"`
|
||||
Sum float64 `json:"sum"`
|
||||
}
|
||||
|
||||
// Структура для ответа от localhost:5000
|
||||
type PaymentResponse struct {
|
||||
Message string `json:"message"`
|
||||
URL string `json:"redir_url"`
|
||||
}
|
||||
|
||||
var db *sql.DB
|
||||
|
||||
func initDB() {
|
||||
var err error
|
||||
dbFile := "urls_pay.db"
|
||||
db, err = sql.Open("sqlite3", dbFile)
|
||||
if err != nil {
|
||||
log.Fatal("Ошибка подключения к базе данных:", err)
|
||||
}
|
||||
|
||||
// Создаем таблицу pay_urls, если она не существует
|
||||
createTableSQL := `
|
||||
CREATE TABLE IF NOT EXISTS pay_urls (
|
||||
id TEXT PRIMARY KEY,
|
||||
url TEXT NOT NULL
|
||||
);`
|
||||
_, err = db.Exec(createTableSQL)
|
||||
if err != nil {
|
||||
log.Fatal("Ошибка при создании таблицы pay_urls:", err)
|
||||
}
|
||||
|
||||
fmt.Println("База данных готова к работе.")
|
||||
}
|
||||
|
||||
func payHandler(w http.ResponseWriter, r *http.Request) {
|
||||
|
||||
cookie, err := r.Cookie("totalPrice")
|
||||
if err != nil {
|
||||
http.Error(w, "Не удалось получить amount из cookies", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
// Преобразуем amount из строки в float64
|
||||
amount, err := strconv.ParseFloat(cookie.Value, 64)
|
||||
if err != nil {
|
||||
http.Error(w, "Некорректное значение amount в cookies", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
// Генерируем новый UUID
|
||||
newUUID := uuid.New().String()
|
||||
|
||||
// Создаем тело запроса для localhost:5000
|
||||
paymentRequest := PaymentRequest{
|
||||
ID: newUUID,
|
||||
Sum: amount,
|
||||
}
|
||||
|
||||
// Преобразуем тело запроса в JSON
|
||||
requestBody, err := json.Marshal(paymentRequest)
|
||||
if err != nil {
|
||||
http.Error(w, "Ошибка при создании запроса", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Отправляем POST-запрос на localhost:5000
|
||||
resp, err := http.Post("http://localhost:5000/pay", "application/json", bytes.NewBuffer(requestBody))
|
||||
if err != nil {
|
||||
http.Error(w, "Ошибка при отправке запроса на сервер", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// Читаем ответ от сервера
|
||||
var paymentResponse PaymentResponse
|
||||
if err := json.NewDecoder(resp.Body).Decode(&paymentResponse); err != nil {
|
||||
http.Error(w, "Ошибка при чтении ответа от сервера", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Сохраняем данные в базу данных
|
||||
insertSQL := `INSERT INTO pay_urls (uuid, url) VALUES (?, ?)`
|
||||
_, err = db.Exec(insertSQL, newUUID, paymentResponse.URL)
|
||||
if err != nil {
|
||||
http.Error(w, "Ошибка при сохранении данных в базу данных", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Формируем ответ клиенту в виде простого текста
|
||||
response := fmt.Sprintf("UUID: %s\nURL: %s", newUUID, paymentResponse.URL)
|
||||
|
||||
// Set uuid cookie
|
||||
cookie = &http.Cookie{
|
||||
Name: "uuid",
|
||||
Value: newUUID,
|
||||
Expires: time.Now().Add(30 * time.Minute),
|
||||
}
|
||||
http.SetCookie(w, cookie)
|
||||
// Отправляем ответ в виде простого текста
|
||||
w.Header().Set("Content-Type", "text/plain")
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(response))
|
||||
}
|
||||
|
||||
func checkPayHandler(w http.ResponseWriter, r *http.Request) {
|
||||
// Извлекаем uuid из cookie
|
||||
cookie, err := r.Cookie("uuid")
|
||||
if err != nil {
|
||||
http.Error(w, "Не найден cookie с uuid", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
uuid := cookie.Value
|
||||
if uuid == "" {
|
||||
http.Error(w, "Пустое значение uuid", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
// Проверяем существование uuid в базе данных
|
||||
var exists bool
|
||||
querySQL := `SELECT EXISTS(SELECT 1 FROM pay_urls WHERE uuid = ?)`
|
||||
err = db.QueryRow(querySQL, uuid).Scan(&exists)
|
||||
if err != nil {
|
||||
http.Error(w, "Ошибка при проверке uuid в базе данных", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
if !exists {
|
||||
http.Error(w, "UUID не найден", http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
|
||||
// Создаем тело запроса для localhost:5000/check_pay
|
||||
checkRequest := struct {
|
||||
UUID string `json:"uuid"`
|
||||
}{
|
||||
UUID: uuid,
|
||||
}
|
||||
|
||||
requestBody, err := json.Marshal(checkRequest)
|
||||
if err != nil {
|
||||
http.Error(w, "Ошибка при создании запроса", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Отправляем POST-запрос на localhost:5000/check_pay
|
||||
resp, err := http.Post("http://localhost:5000/check_pay", "application/json", bytes.NewBuffer(requestBody))
|
||||
if err != nil {
|
||||
http.Error(w, "Ошибка при отправке запроса на сервер", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// Читаем ответ от сервера
|
||||
var checkResponse struct {
|
||||
Mes string `json:"mes"`
|
||||
}
|
||||
if err := json.NewDecoder(resp.Body).Decode(&checkResponse); err != nil {
|
||||
http.Error(w, "Ошибка при чтении ответа от сервера", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Формируем ответ клиенту в зависимости от полученного результата
|
||||
var response string
|
||||
if checkResponse.Mes == "OK" {
|
||||
response = "OK"
|
||||
} else {
|
||||
response = "GTHO"
|
||||
}
|
||||
|
||||
// Отправляем ответ в виде простого текста
|
||||
w.Header().Set("Content-Type", "text/plain")
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(response))
|
||||
}
|
||||
|
||||
func main() {
|
||||
|
||||
initDB()
|
||||
defer db.Close()
|
||||
// Setup CORS
|
||||
cors := handlers.CORS(
|
||||
handlers.AllowedOrigins([]string{"*"}),
|
||||
handlers.AllowedMethods([]string{"GET", "POST"}),
|
||||
handlers.AllowedHeaders([]string{"Content-Type"}),
|
||||
)
|
||||
|
||||
// Register handlers
|
||||
http.HandleFunc("/api/pay", payHandler)
|
||||
http.HandleFunc("/api/check_pay", checkPayHandler)
|
||||
|
||||
// Start server with CORS middleware
|
||||
port := os.Getenv("PORT")
|
||||
if port == "" {
|
||||
port = "8081"
|
||||
}
|
||||
|
||||
fmt.Println("Go-сервер запущен на порту", port)
|
||||
log.Fatal(http.ListenAndServe(":"+port, cors(http.DefaultServeMux)))
|
||||
}
|
||||
BIN
frontend/style/app/pay/urls_pay.db
Normal file
BIN
frontend/style/app/pay/urls_pay.db
Normal file
Binary file not shown.
Reference in New Issue
Block a user