package main

import (
	"database/sql"
	"fmt"
	"log"
	"os"

	_ "github.com/jackc/pgx/v5/stdlib"
)

const dbURL = "postgres://postgres:123@localhost:55432/saveinmed?sslmode=disable"

func main() {
	if len(os.Args) < 2 {
		log.Fatal("Usage: go run main.go <uuid>")
	}
	id := os.Args[1]

	db, err := sql.Open("pgx", dbURL)
	if err != nil {
		log.Fatalf("Failed to connect: %v", err)
	}
	defer db.Close()

	var exists bool
	var name string

	// Check Companies
	err = db.QueryRow("SELECT EXISTS(SELECT 1 FROM companies WHERE id = $1)", id).Scan(&exists)
	if err != nil {
		log.Printf("Error checking companies: %v", err)
	} else if exists {
		db.QueryRow("SELECT corporate_name FROM companies WHERE id = $1", id).Scan(&name)
		fmt.Printf("ID %s FOUND in 'companies' table. Name: %s\n", id, name)
		return
	} else {
		fmt.Printf("ID %s NOT found in 'companies'.\n", id)
	}

	// Check Users
	err = db.QueryRow("SELECT EXISTS(SELECT 1 FROM users WHERE id = $1)", id).Scan(&exists)
	if err != nil {
		log.Printf("Error checking users: %v", err)
	} else if exists {
		db.QueryRow("SELECT name FROM users WHERE id = $1", id).Scan(&name)
		fmt.Printf("ID %s FOUND in 'users' table. Name: %s\n", id, name)
		return
	} else {
		fmt.Printf("ID %s NOT found in 'users'.\n", id)
	}
}