feat: read raw file endpoint

thegalactiks · Jan 17, 2024 · 515f0d4 · 515f0d4
1 parent 003d336
commit 515f0d4
Show file tree

Hide file tree

Showing 5 changed files with 62 additions and 11 deletions.
diff --git a/api/files.go b/api/files.go
@@ -28,18 +28,33 @@ func (h *Handler) GetFiles(c *gin.Context) {
 	}
 
 	repo := hosting.Repository{Owner: uri.Owner, Name: uri.Repo}
-	file, files, err := hsting.(hosting.Hosting).GetFiles(c.Request.Context(), &repo, uri.Path)
-	if err != nil {
-		c.AbortWithStatusJSON(http.StatusBadGateway, gin.H{
-			"error": err.Error(),
-		})
+	switch c.NegotiateFormat(gin.MIMEJSON, RawMimeTypes) {
+	case RawMimeTypes:
+		file, err := hsting.(hosting.Hosting).GetRawFile(c.Request.Context(), &repo, uri.Path)
+		if err != nil {
+			c.AbortWithStatusJSON(http.StatusBadGateway, gin.H{
+				"error": err.Error(),
+			})
+			return
+		}
+
+		c.Data(200, RawMimeTypes, file)
 		return
-	}
 
-	if file != nil {
-		c.JSON(200, file)
-		return
+	default:
+		file, files, err := hsting.(hosting.Hosting).GetFiles(c.Request.Context(), &repo, uri.Path)
+		if err != nil {
+			c.AbortWithStatusJSON(http.StatusBadGateway, gin.H{
+				"error": err.Error(),
+			})
+			return
+		}
+
+		if file != nil {
+			c.JSON(200, file)
+			return
+		}
+
+		c.JSON(200, files)
 	}
-
-	c.JSON(200, files)
 }
diff --git a/api/handler.go b/api/handler.go
@@ -6,6 +6,10 @@ import (
 	"github.com/thegalactiks/giteway/internal/config"
 )
 
+const (
+	RawMimeTypes string = "application/vnd.giteway.raw"
+)
+
 type Handler struct {
 	c *config.Config
 	e *gin.Engine

diff --git a/hosting/hosting.go b/hosting/hosting.go
@@ -49,4 +49,5 @@ type Hosting interface {
 	GetBranches(ctx context.Context, repo *Repository) ([]Branch, error)
 	GetCommits(ctx context.Context, repo *Repository) ([]Commit, error)
 	GetFiles(ctx context.Context, repo *Repository, path string) (*File, []File, error)
+	GetRawFile(ctx context.Context, repo *Repository, path string) ([]byte, error)
 }
diff --git a/internal/hosting/github/file.go b/internal/hosting/github/file.go
@@ -2,6 +2,7 @@ package github
 
 import (
 	"context"
+	"errors"
 
 	"github.com/google/go-github/v58/github"
 	"github.com/thegalactiks/giteway/hosting"
@@ -42,3 +43,21 @@ func (h *HostingGithub) GetFiles(ctx context.Context, repo *hosting.Repository,
 
 	return nil, files, nil
 }
+
+func (h *HostingGithub) GetRawFile(ctx context.Context, repo *hosting.Repository, path string) ([]byte, error) {
+	fileContent, _, _, err := h.client.Repositories.GetContents(ctx, repo.Owner, repo.Name, path, &github.RepositoryContentGetOptions{})
+	if err != nil {
+		return nil, err
+	}
+
+	if fileContent == nil {
+		return nil, errors.New("the path should be a valid file path")
+	}
+
+	c, err := fileContent.GetContent()
+	if err != nil {
+		return nil, err
+	}
+
+	return []byte(c), nil
+}
diff --git a/internal/hosting/gitlab/file.go b/internal/hosting/gitlab/file.go
@@ -67,3 +67,15 @@ func (h *HostingGitlab) GetFiles(ctx context.Context, repo *hosting.Repository,
 
 	return nil, files, nil
 }
+
+func (h *HostingGitlab) GetRawFile(ctx context.Context, repo *hosting.Repository, path string) ([]byte, error) {
+	pid := createPid(repo)
+	pathWithoutSlash := strings.TrimLeft(path, "/")
+	ref := "master"
+
+	file, _, err := h.client.RepositoryFiles.GetRawFile(pid, pathWithoutSlash, &gitlab.GetRawFileOptions{
+		Ref: &ref,
+	})
+
+	return file, err
+}