package main

   import (
       "net/http"
       "strconv"

       "github.com/labstack/echo/v4"
       "gorm.io/driver/mysql"
       "gorm.io/gorm"
   )

   type User struct {
       ID   int    `gorm:"primaryKey" json:"id"`
       Name string `json:"name"`
       Age  int    `json:"age"`
   }

   var DB *gorm.DB

   func CreateUser(c echo.Context) error {
       user := &User{}
       if err := c.Bind(user); err != nil {
           return c.JSON(http.StatusBadRequest, err.Error())
       }

       if err := DB.Create(&user).Error; err != nil {
           return c.JSON(http.StatusInternalServerError, err.Error())
       }
       return c.JSON(http.StatusCreated, "User created successfully")
   }

   func GetAllUsers(c echo.Context) error {
       var users []User
       if err := DB.Find(&users).Error; err != nil {
           return c.JSON(http.StatusInternalServerError, err.Error())
       }
       return c.JSON(http.StatusOK, users)
   }


   func DeleteUser(c echo.Context) error {
       var user User
       id := c.Param("id")
       ID, err := strconv.ParseInt(id, 0, 0)
       if err != nil {
           return c.JSON(http.StatusInternalServerError, err.Error())
       }


       if err := DB.Where("id = ?", ID).Delete(&user).Error; err != nil {
           return c.JSON(http.StatusInternalServerError, err.Error())
       }
       return c.JSON(http.StatusOK, "User deleted successfully")
   }


   func main() {
       e := echo.New()
       // Initiate DB
       dsn := "root:191491@tcp(127.0.0.1:3306)/basiccrud?charset=utf8mb4&parseTime=True&loc=Local"
       d, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
       if err != nil {
           panic(err)
       }
       DB = d
       // Create and migrate db
       DB.AutoMigrate(&User{}
       // Routes
       e.GET("/users", GetAllUsers)
       e.POST("/users", CreateUser)
       e.DELETE("/users/:id", DeleteUser)
       // Start server
       e.Logger.Fatal(e.Start(":1111"))
   }