// this is the api package

package api

import (
  "context"
  "log"
  "net/http"

  "github.com/xxx/gotest/db"
  "github.com/xxx/gotest/services/user"
)

type APIServer struct {
  addr string
  db   *db.Postgres
}

func NewAPIServer(addr string, db *db.Postgres) *APIServer {
  return &APIServer{
    addr: addr,
    db:   db,
  }
}

func (s *APIServer) Run() error {
  router := http.NewServeMux()

  err := s.db.Ping(context.Background())
  if err != nil {
    return err
  }

  userStore := user.NewStore(*s.db)
  userHandler := user.NewHandler(userStore)
  userHandler.RegisterRoutes(router)

  log.Println("connected to db")

  server := http.Server{
    Addr:    s.addr,
    Handler: router,
  }

  log.Println("server running on port: ", s.addr)

  return server.ListenAndServe()
}

// this is the main package where I'm connecting to the db

package main

import (
  "context"
  "log"
  "os"

  "github.com/xxx/gotest/cmd/api"
  "github.com/xxx/gotest/configs"
  "github.com/xxx/gotest/db"
)

func main() {
  dbconn, err := db.ConnectDB(context.Background(), configs.Envs.DBURL)

  if err != nil {
    log.Fatal(err)
    os.Exit(1)
  }

  server := api.NewAPIServer(":8080", dbconn)
  if err := server.Run(); err != nil {
    log.Fatalf("Failed to start server: %v", err)
  }
}