chore: refine gorm schemas

src/controllers/companies_controller.go

@@ -24,17 +24,17 @@ func GetCompanyByID(c *gin.Context) {
 func CreateCompany(context *gin.Context) {
 	var input models.Company
 	if err := context.ShouldBindJSON(&input); err != nil {
-		context.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		context.JSON(http.StatusBadRequest, gin.H{"status": "failed", "message": err.Error(), "data": nil})
 		return
 	}
-	// savedCompany, err := input.
+	savedCompany, err := input.Save()
 
-	// if err != nil {
-	// 	context.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
-	// 	return
-	// }
+	if err != nil {
+		context.JSON(http.StatusBadRequest, gin.H{"status": "failed", "message": err.Error(), "data": nil})
+		return
+	}
 
-	// context.JSON(http.StatusCreated, gin.H{"data": savedEntry})
+	context.JSON(http.StatusCreated, gin.H{"status": "failed", "message": "Comnapy saved successfuly", "data": savedCompany})
 }
 
 func UpdateCompany(c *gin.Context) {

src/models/company.go

@@ -17,6 +17,7 @@ type Company struct {
 }
 
 type Office struct {
+	gorm.Model
 	Country string `json:"country"`
 	Town    string `json:"town"`
 	Venue   string `json:"venue"`
@@ -29,3 +30,11 @@ type SocialMedia struct {
 	LinkedIn  string `json:"linkedin"`
 	Instagram string `json:"instagram"`
 }
+
+func (company *Company) Save() (*Company, error) {
+	err := Database.Create(&company).Error
+	if err != nil {
+		return &Company{}, err
+	}
+	return company, nil
+}