258 lines
7.2 KiB
Go
258 lines
7.2 KiB
Go
package controllers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gitlab.com/pactual1/backend/database/contract"
|
|
"gitlab.com/pactual1/backend/models"
|
|
"gitlab.com/pactual1/backend/shared"
|
|
)
|
|
|
|
func GetLatestContracts(c *gin.Context) {
|
|
// Existing parameters
|
|
limitStr := c.DefaultQuery("limit", "50")
|
|
offsetStr := c.DefaultQuery("offset", "0")
|
|
status := strings.Split(c.DefaultQuery("status", models.ContractStatusActive), ",")
|
|
|
|
// New/Updated optional parameters
|
|
companyName := c.Query("company_name")
|
|
companyAddress := c.Query("company_address")
|
|
companyEmail := c.Query("company_email")
|
|
companyPhone := c.Query("company_phone")
|
|
contractName := c.Query("contract_name")
|
|
startTimeStr := c.Query("start_time")
|
|
endTimeStr := c.Query("end_time")
|
|
deviceIDsStr := c.QueryArray("deviceIDs[]")
|
|
|
|
// Convert limit and offset to int
|
|
limit, err := strconv.Atoi(limitStr)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid limit value"})
|
|
return
|
|
}
|
|
|
|
offset, err := strconv.Atoi(offsetStr)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid offset value"})
|
|
return
|
|
}
|
|
|
|
// Convert startTime to time.Time
|
|
var startTime time.Time
|
|
if startTimeStr != "" {
|
|
startTimeUnix, err := strconv.ParseInt(startTimeStr, 10, 64)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid start_time value"})
|
|
return
|
|
}
|
|
startTime = time.Unix(startTimeUnix, 0)
|
|
}
|
|
|
|
// Convert endTime to time.Time
|
|
var endTime time.Time
|
|
if endTimeStr != "" {
|
|
endTimeUnix, err := strconv.ParseInt(endTimeStr, 10, 64)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid end_time value"})
|
|
return
|
|
}
|
|
endTime = time.Unix(endTimeUnix, 0)
|
|
}
|
|
|
|
// Convert deviceIDs to []int64
|
|
var deviceIDs []int64
|
|
for _, idStr := range deviceIDsStr {
|
|
id, err := strconv.ParseInt(idStr, 10, 64)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid deviceID value"})
|
|
return
|
|
}
|
|
deviceIDs = append(deviceIDs, id)
|
|
}
|
|
|
|
// Fetch contracts
|
|
contracts, total, st, err := contract.GetContracts(status, companyName, companyAddress, companyEmail, companyPhone, &startTime, &endTime, contractName, deviceIDs, 0, nil, limit, offset)
|
|
|
|
if err != nil {
|
|
c.JSON(st, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
// Respond with the contracts and the total count
|
|
c.JSON(http.StatusOK, gin.H{"total": total, "data": models.ConvertContractToResponse(contracts)})
|
|
}
|
|
|
|
|
|
func GetBuyerContracts(c *gin.Context) {
|
|
// Existing parameters
|
|
limitStr := c.DefaultQuery("limit", "50")
|
|
offsetStr := c.DefaultQuery("offset", "0")
|
|
status := strings.Split(c.DefaultQuery("status", models.ContractStatusActive), ",")
|
|
|
|
contractIDStr := c.DefaultQuery("contract_id","0")
|
|
dateCreatedStr := c.Query("date_created")
|
|
|
|
|
|
// Convert limit and offset to int
|
|
limit, err := strconv.Atoi(limitStr)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid limit value"})
|
|
return
|
|
}
|
|
|
|
offset, err := strconv.Atoi(offsetStr)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid offset value"})
|
|
return
|
|
}
|
|
|
|
// Convert limit and offset to int
|
|
contractID, err := strconv.Atoi(contractIDStr)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid contractID value"})
|
|
return
|
|
}
|
|
|
|
// Convert startTime to time.Time
|
|
var dateCreated time.Time
|
|
if dateCreatedStr != "" {
|
|
startTimeUnix, err := strconv.ParseInt(dateCreatedStr, 10, 64)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid start_time value"})
|
|
return
|
|
}
|
|
dateCreated = time.Unix(startTimeUnix, 0)
|
|
}
|
|
|
|
|
|
|
|
// Fetch contracts
|
|
contracts, total, st, err := contract.GetContracts(status, "", "", "", "", nil, nil, "", nil, contractID, &dateCreated, limit, offset)
|
|
|
|
if err != nil {
|
|
c.JSON(st, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
// Respond with the contracts and the total count
|
|
c.JSON(http.StatusOK, gin.H{"total": total, "data": models.ConvertContractToResponseModel(contracts)})
|
|
}
|
|
|
|
func GetContractStatuses(c *gin.Context) {
|
|
|
|
// Respond with the contract statuses
|
|
c.JSON(http.StatusOK, gin.H{"data": models.GetContractStatuses()})
|
|
}
|
|
|
|
func CreateContract(c *gin.Context) {
|
|
|
|
var payload models.CreateContractRequestPayload
|
|
|
|
if err := c.ShouldBindJSON(&payload); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid JSON payload"})
|
|
log.Printf("Invalid JSON payload: %v", err)
|
|
return
|
|
}
|
|
|
|
db := shared.GetDb()
|
|
|
|
newContract := models.Contract{
|
|
BuyerID: payload.BuyerID,
|
|
SellerID: payload.SellerID,
|
|
Description: payload.Description,
|
|
ProductID: payload.ProductID,
|
|
MinTemp: payload.MinTemp,
|
|
MaxTemp: payload.MaxTemp,
|
|
ArrivalDate: time.Unix(payload.ArrivalDate, 0),
|
|
PenaltyType: payload.PenaltyRec,
|
|
PenaltyValue : payload.PenaltyValue,
|
|
PenaltyRec : payload.PenaltyRec,
|
|
}
|
|
|
|
if err := db.Create(&newContract).Error; err != nil {
|
|
log.Printf("SaveContractInfo CREATE - Contract DB Error: %v", err)
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Could not create new contract"})
|
|
return
|
|
}
|
|
|
|
log.Printf("Successfully received and saved contract: %v", newContract)
|
|
c.JSON(http.StatusOK, gin.H{"message": "Successfully received and saved contract", "id": newContract.ID})
|
|
}
|
|
|
|
func GetContractByID(c *gin.Context) {
|
|
// Get the contract ID from url parameters
|
|
contractIDStr := c.Param("contract_id")
|
|
|
|
if contractIDStr == "" {
|
|
log.Printf("GetContractByID Error: ID is required")
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "ID is required"})
|
|
return
|
|
}
|
|
|
|
contractID, err := strconv.ParseUint(contractIDStr, 10, 32)
|
|
if err != nil {
|
|
log.Printf("GetContractByID Error: Invalid ID: %v", err)
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid ID"})
|
|
return
|
|
}
|
|
|
|
// Fetch contract
|
|
contract, st, err := contract.GetContractByID(contractID)
|
|
if err != nil {
|
|
c.JSON(st, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
// Respond with the contracts and the total count
|
|
c.JSON(http.StatusOK, gin.H{"data": contract})
|
|
}
|
|
|
|
func UpdateContract(c *gin.Context) {
|
|
|
|
var contractModel models.Contract
|
|
rawData, _ := c.GetRawData()
|
|
|
|
// Unmarshal to the important info structure
|
|
err := json.Unmarshal(rawData, &contractModel)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid JSON payload"})
|
|
log.Printf("Invalid json pyload : %v", err)
|
|
return
|
|
}
|
|
|
|
// Get the contract ID from url parameters
|
|
contractIDStr := c.Param("contract_id")
|
|
|
|
if contractIDStr == "" {
|
|
log.Printf("UpdateContract Error: ID is required")
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "ID is required"})
|
|
return
|
|
}
|
|
|
|
contractID, err := strconv.ParseUint(contractIDStr, 10, 32)
|
|
if err != nil {
|
|
log.Printf("UpdateContract Error: Invalid ID: %v", err)
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid ID"})
|
|
return
|
|
}
|
|
|
|
contractModel.ID = uint(contractID)
|
|
|
|
// update contract
|
|
contractModel, st, err := contract.UpdateContract(contractModel)
|
|
if err != nil {
|
|
c.JSON(st, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
// Respond with the contracts and the total count
|
|
c.JSON(http.StatusOK, gin.H{"data": contractModel})
|
|
}
|
|
|