ODLC Endpoints

controllers/image_controller.go

@@ -0,0 +1,72 @@
+package controllers
+
+import (
+	"fmt"
+	"gcom-backend/models"
+	"github.com/labstack/echo/v4"
+	"gorm.io/gorm"
+	"net/http"
+	"os"
+	"regexp"
+	"strconv"
+)
+
+var imgDirectory = "./imgs/"
+
+func UploadImage(c echo.Context) error {
+	file, err := c.FormFile("file")
+	if err != nil {
+		return c.JSON(http.StatusBadRequest, err.Error())
+	}
+
+	src, err := file.Open()
+	if err != nil {
+		return c.JSON(http.StatusBadRequest, err.Error())
+	}
+	defer src.Close()
+
+	match, err := regexp.MatchString("(\\d{10})", file.Filename)
+	if err != nil || !match {
+		return c.JSON(http.StatusBadRequest, "Invalid image name, use UNIX timestamp")
+	}
+	dst, err := os.Create(imgDirectory + file.Filename)
+	if err != nil {
+		return c.JSON(http.StatusBadRequest, "Error saving image")
+	}
+	defer dst.Close()
+
+	timestamp, _ := strconv.Atoi(file.Filename[:10])
+	fmt.Println(file.Filename[:13])
+	fmt.Println(timestamp)
+	image := &models.Image{
+		Timestamp: int64(timestamp),
+		Filename:  file.Filename,
+	}
+
+	db, _ := c.Get("db").(*gorm.DB)
+	if createErr := db.Create(&image).Error; createErr != nil {
+		return c.JSON(http.StatusInternalServerError, createErr.Error())
+	}
+
+	return c.JSON(http.StatusAccepted, "Upload sucessful")
+}
+
+func ListImages(c echo.Context) error {
+	var images []models.Image
+	db, _ := c.Get("db").(*gorm.DB)
+	if err := db.Find(&images).Error; err != nil {
+		return c.JSON(http.StatusInternalServerError, err.Error())
+	}
+
+	return c.JSON(http.StatusOK, images)
+}
+
+func GetImage(c echo.Context) error {
+	imgPath, err := os.Stat(imgDirectory + c.Param("filename"))
+	fmt.Println(imgPath.Name())
+	if err != nil {
+		return c.JSON(http.StatusBadRequest, err.Error())
+	} else {
+		return c.Attachment(imgDirectory+c.Param("filename"), c.Param("filename"))
+	}
+}

controllers/waypoint_controller.go

@@ -74,8 +74,8 @@ func CreateWaypoint(c echo.Context) error {
 	}
 
 	return c.JSON(http.StatusOK, responses.SingleResponse[models.Waypoint]{
-		Message:  "Waypoint created!",
-		Model: waypoint})
+		Message: "Waypoint created!",
+		Model:   waypoint})
 }
 
 // CreateWaypointBatch creates multiple waypoints
@@ -121,8 +121,8 @@ func CreateWaypointBatch(c echo.Context) error {
 	}
 
 	return c.JSON(http.StatusOK, responses.MultipleResponse[models.Waypoint]{
-		Message:   "Waypoints created!",
-		Models: waypoints})
+		Message: "Waypoints created!",
+		Models:  waypoints})
 }
 
 // EditWaypoint edits a waypoint
@@ -199,8 +199,8 @@ func EditWaypoint(c echo.Context) error {
 	db.First(&updatedWaypoint, waypointId)
 
 	return c.JSON(http.StatusOK, responses.SingleResponse[models.Waypoint]{
-		Message:  "Waypoint updated!",
-		Model: updatedWaypoint,
+		Message: "Waypoint updated!",
+		Model:   updatedWaypoint,
 	})
 }
 
@@ -230,8 +230,8 @@ func GetWaypoint(c echo.Context) error {
 	}
 
 	return c.JSON(http.StatusOK, responses.SingleResponse[models.Waypoint]{
-		Message:  "Waypoint found!",
-		Model: waypoint,
+		Message: "Waypoint found!",
+		Model:   waypoint,
 	})
 }
 
@@ -267,8 +267,8 @@ func DeleteWaypoint(c echo.Context) error {
 	}
 
 	return c.JSON(http.StatusOK, responses.SingleResponse[models.Waypoint]{
-		Message:  "Waypoint deleted!",
-		Model: models.Waypoint{},
+		Message: "Waypoint deleted!",
+		Model:   models.Waypoint{},
 	})
 }
 
@@ -329,8 +329,8 @@ func DeleteWaypointBatch(c echo.Context) error {
 	}
 
 	return c.JSON(http.StatusOK, responses.SingleResponse[models.Waypoint]{
-		Message:  "Waypoints deleted!",
-		Model: models.Waypoint{},
+		Message: "Waypoints deleted!",
+		Model:   models.Waypoint{},
 	})
 }
 
@@ -360,7 +360,7 @@ func GetAllWaypoints(c echo.Context) error {
 	}
 
 	return c.JSON(http.StatusOK, responses.MultipleResponse[models.Waypoint]{
-		Message:   "Waypoint found!",
-		Models: waypoints,
+		Message: "Images found!",
+		Models:  waypoints,
 	})
 }

main.go

@@ -1,11 +1,13 @@
 package main
 
 import (
+	"fmt"
 	"gcom-backend/configs"
 	"gcom-backend/controllers"
 	_ "gcom-backend/docs"
 	"gcom-backend/util"
 	"log"
+	"os"
 
 	"github.com/labstack/echo/v4"
 	"github.com/labstack/echo/v4/middleware"
@@ -28,6 +30,12 @@ import (
 
 func main() {
 	db := configs.Connect(false)
+	err := os.MkdirAll("db", 0755)  //Create db dir
+	err = os.MkdirAll("imgs", 0755) //Create images dir
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
 
 	mp, err := configs.ConnectMissionPlanner("http://host.docker.internal:9000")
 	if err != nil {
@@ -80,6 +88,11 @@ func main() {
 	e.DELETE("/groundobjects", controllers.DeleteGroundObjectBatch)
 	e.GET("/groundobjects", controllers.GetAllGroundObjects)
 
+	//Image Handling
+	e.POST("/image", controllers.UploadImage)
+	e.GET("/image/list", controllers.ListImages)
+	e.GET("/image/:filename", controllers.GetImage)
+
 	//Websockets
 	e.Any("/socket.io/", controllers.WebsocketHandler())
 

models/image_model.go

@@ -0,0 +1,6 @@
+package models
+
+type Image struct {
+	Timestamp int64  `json:"timestamp" gorm:"primaryKey" validate:"required" example:"1698544781" extensions:"x-order=1"`
+	Filename  string `json:"filename" example:"1714898050.png" extensions:"x-order=2"`
+}

models/migrate.go

@@ -9,7 +9,7 @@ import "gorm.io/gorm"
 */
 
 func Migrate(db *gorm.DB) {
-	err := db.AutoMigrate(&Waypoint{}, &Drone{}, &GroundObject{})
+	err := db.AutoMigrate(&Waypoint{}, &Drone{}, &GroundObject{}, &Image{})
 	if err != nil {
 		panic(err)
 	}

responses/multple_response.go

@@ -4,6 +4,6 @@ package responses
 //
 // @Description Describes a response with multiple waypoints
 type MultipleResponse[T any] struct {
-	Message   string            `json:"message" example:"Sample success message"`
-	Models []T `json:"waypoints"`
+	Message string `json:"message" example:"Sample success message"`
+	Models  []T    `json:"models"`
 }