index.js

import express from "express";
import bodyParser from "body-parser";
import pg from "pg";

const app = express();
const port = 3000;

const db = new pg.Client({
  user: "postgres",
  host: "localhost",
  database: "*****",
  password: "*****",
  port: 5432,
});
db.connect();

app.use(bodyParser.urlencoded({ extended: true }));
app.use(express.static("public"));

let currentUserId = 1;

let users = [
  { id: 1, name: "Angela", color: "teal" },
  { id: 2, name: "Jack", color: "powderblue" },
];

async function checkVisisted() {
  const result = await db.query("SELECT country_code FROM visited_countries JOIN users ON users.id = user_id WHERE user_id = $1;", [currentUserId]);
  let countries = [];
  
  result.rows.forEach((country) => {
    if(!countries.includes(country.country_code)){
      countries.push(country.country_code);
    }
  });
  return countries;
}


async function getCurrentUser(){
  const result = await db.query("SELECT * FROM users");
  users = result.rows;
  const user = users.find(user => user.id == currentUserId);
  return user;
}

app.get("/", async (req, res) => {
  const countries = await checkVisisted();
  const currentUser = await getCurrentUser();
  res.render("index.ejs", {
    countries: countries,
    total: countries.length,
    users: users,
    color: currentUser.color,
  });
  console.log(countries);
  console.log(countries.length);
  console.log(users);
  console.log(currentUser.color);
});

app.post("/add", async (req, res) => {
  const input = req.body["country"];
  // const currentUser = await getCurrentUser();
  try {
    const result = await db.query(
      "SELECT country_code FROM countries WHERE LOWER(country_name) LIKE '%' || $1 || '%';",
      [input.toLowerCase()]
    );
    console.log(result.rows);
    const data = result.rows[0];
    
    let countryCode = data.country_code;

    if (countryCode === "IO") {

      countryCode = "IN";

    }

    try {
      await db.query(
        `INSERT INTO visited_countries (country_code, user_id) VALUES ($1, $2)`,
        [countryCode, currentUserId]
      );
      res.redirect("/");
    } catch (err) {
      console.log(err);
    }
  } catch (err) {
    console.log(err);
  }
});

app.post("/user", async (req, res) => {
  if(req.body.add === "new"){
    res.render("new.ejs")
  } else {
    currentUserId = req.body.user;
    res.redirect('/');
  }
});

app.post("/new", async (req, res) => {
  const name  = req.body.name;
  const color = req.body.color;
    const result = await db.query("INSERT INTO users (name, color) VALUES ($1, $2) RETURNING *;",[name, color]);
    currentUserId = result.rows[0].id;
    res.redirect("/");
});

app.listen(port, () => {
  console.log(`Server running on http://localhost:${port}`);
});