conex-builder/server/db.go

package main

import (
	"database/sql"
	"encoding/json"
	"fmt"
	"time"

	_ "github.com/lib/pq"
)

const (
	errDBRegisterSite       = "db.go (sites): Register site"
	errDBUpdateDue          = "db.go (sites): Update due date"
	errDBRegisterPayment    = "db.go (payments): Register payment"
	errDBTXBeginUpdateSite  = "db.go: Begin transaction"
	errDBTXCommitUpdateSite = "db.go: Commit transaction"
	errDBGetPrevRaw         = "db.go (sites): Query old raw json"
	errDBUpdateRaw          = "db.go (sites): Update raw json"
	errDBChangesRaw         = "db.go (changes): Register raw json change"
	errDBUpdateSiteAuth     = "db.go (sites): Auth"
)

func RegisterSitePayment(
	capture Capture, directory string, editorData json.RawMessage,
) error {
	var (
		// Payment
		id       string
		amount   string
		currency string
		pstatus  string
		date     time.Time
		// Website
		wstatus string
		due     time.Time
		name    string
		surname string
		email   string
		phone   string
		country string
	)

	captureData := capture.PurchaseUnits[0].Payments.Captures[0]

	id = captureData.ID
	amount = captureData.Amount.Value
	currency = captureData.Amount.CurrencyCode
	pstatus = captureData.Status
	date = captureData.CreateTime
	wstatus = "down"
	due = date.AddDate(1, 0, 0)
	name = capture.Payer.Name.GivenName
	surname = capture.Payer.Name.Surname
	email = capture.Payer.EmailAddress
	phone = capture.Payer.Phone.PhoneNumber.NationalNumber
	country = capture.Payer.Address.CountryCode

	var pkey int
	newSite := db.QueryRow(`
		SELECT id FROM sites WHERE folder = $1
		`, directory).Scan(&pkey)

	if newSite == sql.ErrNoRows {
		if err := db.QueryRow(`
			INSERT INTO sites
			(folder, status, due, name, sur, email, phone, code, raw)
			VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)
			RETURNING id
			`, directory, wstatus, due,
			name, surname, email, phone, country,
			editorData).Scan(&pkey); err != nil {
			return fmt.Errorf("%s: %v", errDBRegisterSite, err)
		}
	} else {
		if err := db.QueryRow(`
			UPDATE sites SET due = due + INTERVAL '1 year'
			WHERE id = $1
			RETURNING id
			`, pkey).Scan(&pkey); err != nil {
			return fmt.Errorf("%s: %v", errDBUpdateDue, err)
		}
	}

	if _, err := db.Exec(`
		INSERT INTO payments
		(capture, site, amount, currency, date, status)
		VALUES ($1, $2, $3, $4, $5, $6)
		`, id, pkey, amount, currency, date, pstatus); err != nil {
		return fmt.Errorf("%s: %v", errDBRegisterPayment, err)
	}

	return nil
}

func UpdateSite(pkey int, editorData json.RawMessage) error {
	tx, err := db.Begin()
	if err != nil {
		return fmt.Errorf("%s: %v", errDBTXBeginUpdateSite, err)
	}

	defer func() {
		if err != nil {
			tx.Rollback()
		}
	}()

	var prev json.RawMessage
	if err := tx.QueryRow(`
		SELECT raw FROM sites WHERE id = $1
		`, pkey).Scan(&prev); err != nil {
		return fmt.Errorf("%s: %v", errDBGetPrevRaw, err)
	}

	if _, err = tx.Exec(`
		UPDATE sites SET raw = $1 WHERE id = $2
		`, editorData, pkey); err != nil {
		return fmt.Errorf("%s: %v", errDBUpdateRaw, err)
	}

	if _, err = tx.Exec(`
		INSERT INTO changes
		(by, site, payment, col, descrip, date)
		VALUES ($1, $2, NULL, 'raw', $3, CURRENT_DATE);
		`, "server", pkey, "Cambio automatizado de sitio"); err != nil {
		return fmt.Errorf("%s: %v", errDBChangesRaw, err)
	}

	if err := tx.Commit(); err != nil {
		return fmt.Errorf("%s: %v", errDBTXCommitUpdateSite, err)
	}

	return nil
}

func UpdateSiteAuth(folder string, code string) (string, error) {
	valid := time.Now().Add(5 * time.Minute)

	var email string
	if err := db.QueryRow(`
		UPDATE sites
		SET auth = $1, valid = $2
		WHERE folder = $3
		RETURNING email;
		`, code, valid, folder).Scan(&email); err != nil {
		return "", fmt.Errorf("%s: %v", errDBUpdateSiteAuth, err)
	}

	return email, nil
}

func ValidateSiteAuth(folder string, code string) (int, error) {
	var dbCode string
	var validTime time.Time

	err := db.QueryRow(`
		SELECT auth, valid FROM sites
		WHERE folder = $1;
		`, folder).Scan(&dbCode, &validTime)
	if err != nil {
		if err == sql.ErrNoRows {
			return 0, fmt.Errorf("%s: %v", "No such directory", err)
		}
		return 0, fmt.Errorf("%s: %v", "Failed to query DB", err)
	}

	if code != dbCode {
		return 0, fmt.Errorf("%s", "Incorrect code")
	}

	if time.Now().After(validTime) {
		return 0, fmt.Errorf("%s", "Auth expired")
	}

	var pkey int
	if err := db.QueryRow(`
		UPDATE sites
		SET valid = $1
		WHERE folder = $2
		RETURNING id;
		`, time.Now(), folder).Scan(&pkey); err != nil {
		return 0, fmt.Errorf("%s: %v", "Void used code", err)
	}

	return pkey, nil
}