add open stats for buckets

api/open_status_check.go

@@ -0,0 +1,131 @@
+package api
+
+import (
+	"context"
+	"github.com/application-research/edge-ur/core"
+	"github.com/application-research/edge-ur/jobs"
+	"github.com/ipfs/go-cid"
+	"github.com/labstack/echo/v4"
+)
+
+func ConfigureOpenStatusCheckRouter(e *echo.Group, node *core.LightNode) {
+	e.GET("/status/content/:id", func(c echo.Context) error {
+
+		var content core.Content
+		node.DB.Raw("select * from contents as c where id = ?", c.Param("id")).Scan(&content)
+		content.RequestingApiKey = ""
+
+		if content.ID == 0 {
+			return c.JSON(404, map[string]interface{}{
+				"message": "Content not found. Please check if you have the proper API key or if the content id is valid",
+			})
+		}
+
+		// trigger status check
+		job := jobs.CreateNewDispatcher()
+		job.AddJob(jobs.NewDealItemChecker(node, content))
+		job.Start(1)
+
+		return c.JSON(200, map[string]interface{}{
+			"content": content,
+		})
+	})
+	e.GET("/status/bucket/contents/:uuid", func(c echo.Context) error {
+		var bucket core.Bucket
+		node.DB.Model(&core.Bucket{}).Where("uuid = ?", c.Param("uuid")).Scan(&bucket)
+		if bucket.ID == 0 {
+			return c.JSON(404, map[string]interface{}{
+				"message": "Bucket not found. Please check if you have the proper API key or if the bucket uuid is valid",
+			})
+		}
+
+		var contents []core.Content
+		node.DB.Model(&core.Content{}).Where("bucket_uuid = ?", c.Param("uuid")).Scan(&contents)
+
+		if len(contents) == 0 {
+			return c.JSON(404, map[string]interface{}{
+				"message": "Bucket has no contents",
+			})
+		}
+
+		var contentResponse []core.Content
+		for _, content := range contents {
+			content.RequestingApiKey = ""
+			contentResponse = append(contentResponse, content)
+			job := jobs.CreateNewDispatcher()
+			job.AddJob(jobs.NewDealItemChecker(node, content))
+
+		}
+
+		// trigger status check of the bucket
+		job := jobs.CreateNewDispatcher()
+		job.AddJob(jobs.NewCarDealItemChecker(node, bucket))
+		job.Start(len(contents) + 1)
+
+		bucket.RequestingApiKey = ""
+		return c.JSON(200, map[string]interface{}{
+			"bucket":          bucket,
+			"content_entries": contentResponse,
+		})
+	})
+	e.GET("/status/bucket/dag/:uuid", func(c echo.Context) error {
+
+		var bucket core.Bucket
+		node.DB.Model(&core.Bucket{}).Where("uuid = ?", c.Param("uuid")).Scan(&bucket)
+
+		if bucket.ID == 0 {
+			return c.JSON(404, map[string]interface{}{
+				"message": "Bucket not found or has no DAGs. Please check if you have the proper API key or if the bucket uuid is valid",
+			})
+		}
+
+		// get the cid
+		bucketCid, err := cid.Decode(bucket.Cid)
+		if err != nil {
+			return c.JSON(404, map[string]interface{}{
+				"message": "Bucket not found. Please check if you have the proper API key or if the bucket uuid is valid",
+			})
+		}
+		dirNdRaw, err := node.Node.DAGService.Get(context.Background(), bucketCid)
+		if err != nil {
+			return c.JSON(404, map[string]interface{}{
+				"message": "Bucket not found. Please check if you have the proper API key or if the bucket id is valid",
+			})
+		}
+
+		var contents []core.Content
+		node.DB.Model(&core.Content{}).Where("bucket_uuid = ?", c.Param("uuid")).Scan(&contents)
+
+		var contentResponse []core.Content
+		for _, content := range contents {
+			content.RequestingApiKey = ""
+
+			for _, link := range dirNdRaw.Links() {
+				cidFromDb, err := cid.Decode(content.Cid)
+				if err != nil {
+					continue
+				}
+				if link.Cid == cidFromDb {
+					content.Cid = link.Cid.String()
+					content.RequestingApiKey = ""
+					contentResponse = append(contentResponse, content)
+				}
+
+			}
+			job := jobs.CreateNewDispatcher()
+			job.AddJob(jobs.NewDealItemChecker(node, content))
+
+		}
+		// trigger status check
+		job := jobs.CreateNewDispatcher()
+		job.AddJob(jobs.NewCarDealItemChecker(node, bucket))
+		job.Start(len(contents) + 1)
+
+		bucket.RequestingApiKey = ""
+		return c.JSON(200, map[string]interface{}{
+			"bucket":        bucket,
+			"content_links": contentResponse,
+		})
+	})
+
+}

api/router.go

@@ -68,6 +68,7 @@ func InitializeEchoRouterConfig(ln *core.LightNode) {
 	ConfigureNodeInfoRouter(defaultOpenRoute, ln)
 
 	apiGroup := e.Group("/api/v1")
+	openApiGroup := e.Group("/open")
 	apiGroup.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
 		return func(c echo.Context) error {
 			authorizationString := c.Request().Header.Get("Authorization")
@@ -128,7 +129,9 @@ func InitializeEchoRouterConfig(ln *core.LightNode) {
 	})
 	ConfigureRetrieveRouter(apiGroup, ln)
 	ConfigurePinningRouter(apiGroup, ln)
+
 	ConfigureStatusCheckRouter(apiGroup, ln)
+	ConfigureOpenStatusCheckRouter(openApiGroup, ln)
 
 	// Start server