add endpoint for c4gh header size

Co-authored-by: Kostas Koumpouras <[email protected]> Co-authored-by: Panos Chatzopoulos <[email protected]>"
neicnordic · Nov 1, 2023 · 6883b26 · 6883b26
1 parent e49dac6
commit 6883b26
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/api/api.go b/api/api.go
@@ -44,6 +44,7 @@ func Setup() *http.Server {
 	router.GET("/s3-encrypted/*path", SelectedMiddleware(), s3.Download)
 	router.GET("/health", healthResponse)
 	router.GET("/header/*path", SelectedMiddleware(), s3.Download)
+	router.GET("/c4gheadsize/*path", SelectedMiddleware(), s3.Download)
 
 	// Configure TLS settings
 	log.Info("(3/5) Configuring TLS")

diff --git a/api/s3/s3.go b/api/s3/s3.go
@@ -224,6 +224,8 @@ func GetObject(c *gin.Context) {
 	c.Params = append(c.Params, gin.Param{Key: "fileid", Value: fileInfo.FileID})
 	if strings.Contains(c.Request.URL.String(), "header") {
 		c.Params = append(c.Params, gin.Param{Key: "type", Value: "header"})
+	} else if strings.Contains(c.Request.URL.String(), "c4gheadsize") {
+		c.Params = append(c.Params, gin.Param{Key: "type", Value: "headersize"})
 	}
 
 	// Download the file

diff --git a/api/sda/sda.go b/api/sda/sda.go
@@ -221,6 +221,7 @@ func Download(c *gin.Context) {
 			return
 		}
 	}
+	log.Debug("- - - - - - - test 1 - - - - - - - ")
 	if c.Param("type") == "header" {
 		bamReader, err := bam.NewReader(fileStream, 0)
 		if err != nil {
@@ -237,7 +238,21 @@ func Download(c *gin.Context) {
 
 		return
 
+	} else if c.Param("type") == "headersize" {
+		log.Debug("- - - - - - - test 2 - - - - - - - ")
+		head := fileDetails.Header
+		log.Debug("head: ", head)
+		//headlength := len(head)
+		headlength := bytes.NewReader(head)
+		log.Debug("headlength size: ", headlength.Size())
+		buf := new(bytes.Buffer)
+		len := strconv.Itoa(int(headlength.Size()))
+		fmt.Fprint(buf, len)
+		sendStream(c.Writer, buf)
+
+		return
 	}
+	log.Debug("- - - - - - - test 3 - - - - - - - ")
 	sendStream(c.Writer, fileStream)
 }