main.go

package main

import (
	"database/sql"
	"encoding/json"
	"fmt"
	"log"
	"net/http"

	_ "github.com/lib/pq" // Importa o driver PostgreSQL
)

var db *sql.DB

func main() {
	// String de conexão para usar o proxy local (localhost) ao invés do hostname Fly.io
	//connStr := "postgres://postgres:4BSaG5SsurRao7G@localhost:5433/postgres?sslmode=disable"
	connStr := "postgres://postgres:4BSaG5SsurRao7G@counterwebsitedb.flycast:5432/postgres?sslmode=disable"

	var err error
	db, err = sql.Open("postgres", connStr) // Nome do driver é "postgres"
	if err != nil {
		log.Fatalf("Erro ao abrir banco de dados: %v", err)
	}
	defer db.Close()

	// Cria a tabela se não existir
	_, err = db.Exec(`CREATE TABLE IF NOT EXISTS contador (
        id SERIAL PRIMARY KEY,
        visitas INTEGER DEFAULT 0
    )`)
	if err != nil {
		log.Fatalf("Erro ao criar tabela: %v", err)
	}

	// Insere a primeira linha caso a tabela esteja vazia
	_, err = db.Exec(`INSERT INTO contador (id, visitas) VALUES (1, 0) ON CONFLICT DO NOTHING`)
	if err != nil {
		log.Fatalf("Erro ao inserir dados iniciais: %v", err)
	}

	http.HandleFunc("/api/contador-visitas", contadorHandler)

	fmt.Println("Servidor rodando na porta 8080")
	log.Fatal(http.ListenAndServe(":8080", nil))
}

func contadorHandler(w http.ResponseWriter, r *http.Request) {
	// Definir os cabeçalhos CORS para permitir o acesso de diferentes origens
	w.Header().Set("Access-Control-Allow-Origin", "*")
	w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
	w.Header().Set("Access-Control-Allow-Headers", "Content-Type")

	// Verificar se é uma requisição OPTIONS, que é usada pelo navegador para o preflight
	if r.Method == http.MethodOptions {
		w.WriteHeader(http.StatusOK)
		return
	}

	// Incrementa o contador no banco de dados
	_, err := db.Exec(`UPDATE contador SET visitas = visitas + 1 WHERE id = 1`)
	if err != nil {
		http.Error(w, "Erro ao atualizar contador", http.StatusInternalServerError)
		return
	}

	// Lê o valor atualizado
	row := db.QueryRow(`SELECT visitas FROM contador WHERE id = 1`)
	var visitas int
	err = row.Scan(&visitas)
	if err != nil {
		http.Error(w, "Erro ao ler contador", http.StatusInternalServerError)
		return
	}

	// Retorna o valor em JSON
	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(map[string]int{"visitas": visitas})
}