server.go

package main

import (
	"database/sql"
	"encoding/json"
	"fmt"
	"github.com/gofiber/fiber/v2"
	"github.com/gofiber/template/html"
	_ "github.com/lib/pq"
	"log"
	"os"
)

type todo struct {
	Item string
}

type TfOutput struct {
	Sensitive bool
	Type      string
	Value     string
}

func indexHandler(c *fiber.Ctx, db *sql.DB) error {
	var res string
	var todos []string
	rows, err := db.Query("SELECT * FROM todos")
	defer rows.Close()
	if err != nil {
		log.Fatalln(err)
		c.JSON("An error occured")
	}
	for rows.Next() {
		rows.Scan(&res)
		todos = append(todos, res)
	}
	return c.Render("index", fiber.Map{
		"Todos": todos,
	})
}

func postHandler(c *fiber.Ctx, db *sql.DB) error {
	newTodo := todo{}
	if err := c.BodyParser(&newTodo); err != nil {
		log.Printf("An error occured: %v", err)
		return c.SendString(err.Error())
	}
	fmt.Printf("%v", newTodo)
	if newTodo.Item != "" {
		_, err := db.Exec("INSERT into todos VALUES ($1)", newTodo.Item)
		if err != nil {
			log.Fatalf("An error occured while executing query: %v", err)
		}
	}
	return c.Redirect("/")
}

func putHandler(c *fiber.Ctx, db *sql.DB) error {
	olditem := c.Query("olditem")
	newitem := c.Query("newitem")
	db.Exec("UPDATE todos SET item=$1 WHERE item=$2", newitem, olditem)
	return c.Redirect("/")
}

func deleteHandler(c *fiber.Ctx, db *sql.DB) error {
	todoToDelete := c.Query("item")
	db.Exec("DELETE from todos WHERE item=$1", todoToDelete)
	return c.SendString("deleted")
}

func isJSON(s string) bool {
	var js map[string]interface{}
	return json.Unmarshal([]byte(s), &js) == nil
}

func main() {
	var host TfOutput
	var name TfOutput
	var username TfOutput
	var password TfOutput

	var dbHost string
	var dbName string
	var dbUsername string
	var dbPassword string

	if isJSON(os.Getenv("DB_HOST")) {
		json.Unmarshal([]byte(os.Getenv("DB_HOST")), &host)
		dbHost = host.Value
	} else {
		dbHost = os.Getenv("DB_HOST")
	}

	if isJSON(os.Getenv("DB_NAME")) {
		json.Unmarshal([]byte(os.Getenv("DB_NAME")), &name)
		dbName = name.Value
	} else {
		dbName = os.Getenv("DB_NAME")
	}

	if isJSON(os.Getenv("DB_USERNAME")) {
		json.Unmarshal([]byte(os.Getenv("DB_USERNAME")), &username)
		dbUsername = username.Value
	} else {
		dbUsername = os.Getenv("DB_USERNAME")
	}

	if isJSON(os.Getenv("DB_PASSWORD")) {
		json.Unmarshal([]byte(os.Getenv("DB_PASSWORD")), &password)
		dbPassword = password.Value
	} else {
		dbPassword = os.Getenv("DB_PASSWORD")
	}

	connStr := "postgresql://" + dbUsername + ":" + dbPassword + "@" + dbHost + "/" + dbName + "?sslmode=disable"
	db, err := sql.Open("postgres", connStr)
	if err != nil {
		log.Fatal(err)
	}
	engine := html.New("./views", ".html")
	app := fiber.New(fiber.Config{
		Views: engine,
	})

	app.Get("/", func(c *fiber.Ctx) error {
		return indexHandler(c, db)
	})

	app.Post("/", func(c *fiber.Ctx) error {
		return postHandler(c, db)
	})

	app.Put("/update", func(c *fiber.Ctx) error {
		return putHandler(c, db)
	})

	app.Delete("/delete", func(c *fiber.Ctx) error {
		return deleteHandler(c, db)
	})

	port := os.Getenv("SERVER_PORT")
	if port == "" {
		port = "8080"
	}
	app.Static("/", "./public")
	log.Fatalln(app.Listen(fmt.Sprintf(":%v", port)))
}