Implement integration metadata retrieval

Update messages.IntegrationMetadata struct as mention in ddvk#241 (comment)
nemunaire · Oct 17, 2023 · cb37f62 · cb37f62
1 parent 37b4729
commit cb37f62
Show file tree

Hide file tree

Showing 8 changed files with 94 additions and 7 deletions.
diff --git a/internal/app/handlers.go b/internal/app/handlers.go
@@ -750,9 +750,25 @@ func (app *App) syncGetRootV3(c *gin.Context) {
 }
 
 func (app *App) integrationsGetMetadata(c *gin.Context) {
-	var metadata messages.IntegrationMetadata
-	metadata.Thumbnail = ""
-	c.JSON(http.StatusOK, &metadata)
+	uid := c.GetString(userIDKey)
+	integrationID := common.ParamS(integrationKey, c)
+	fileID := common.ParamS(fileKey, c)
+
+	integrationProvider, err := integrations.GetIntegrationProvider(app.userStorer, uid, integrationID)
+	if err != nil {
+		log.Error(fmt.Errorf("can't get integration, %v", err))
+		c.AbortWithStatus(http.StatusInternalServerError)
+		return
+	}
+
+	metadata, err := integrationProvider.GetMetadata(fileID)
+	if err != nil {
+		log.Error(err)
+		c.AbortWithStatus(http.StatusInternalServerError)
+		return
+	}
+
+	c.JSON(http.StatusOK, metadata)
 }
 
 func (app *App) integrationsUpload(c *gin.Context) {

diff --git a/internal/integrations/dropbox.go b/internal/integrations/dropbox.go
@@ -65,6 +65,10 @@ func (d *DropBox) List(folderID string, depth int) (*messages.IntegrationFolder,
 	return response, nil
 }
 
+func (d *DropBox) GetMetadata(fileID string) (*messages.IntegrationMetadata, error) {
+	return nil, nil
+}
+
 func (d *DropBox) Download(fileID string) (io.ReadCloser, int64, error) {
 	return nil, 0, nil
 

diff --git a/internal/integrations/integrations.go b/internal/integrations/integrations.go
@@ -21,6 +21,7 @@ const (
 
 // IntegrationProvider abstracts 3rd party integrations
 type IntegrationProvider interface {
+	GetMetadata(fileID string) (result *messages.IntegrationMetadata, err error)
 	List(folderID string, depth int) (result *messages.IntegrationFolder, err error)
 	Download(fileID string) (io.ReadCloser, int64, error)
 	Upload(folderID, name, fileType string, reader io.ReadCloser) (string, error)

diff --git a/internal/integrations/localfs.go b/internal/integrations/localfs.go
@@ -54,6 +54,25 @@ func (d *localFS) List(folder string, depth int) (*messages.IntegrationFolder, e
 	return response, nil
 }
 
+func (d *localFS) GetMetadata(fileID string) (*messages.IntegrationMetadata, error) {
+	decoded, err := decodeName(fileID)
+	if err != nil {
+		return nil, err
+	}
+
+	ext := path.Ext(decoded)
+	contentType := contentTypeFromExt(ext)
+
+	return &messages.IntegrationMetadata{
+		ID:               fileID,
+		Name:             path.Base(decoded),
+		Thumbnail:        []byte{},
+		SourceFileType:   contentType,
+		ProvidedFileType: contentType,
+		FileType:         ext,
+	}, nil
+}
+
 func (d *localFS) Download(fileID string) (io.ReadCloser, int64, error) {
 	decoded, err := decodeName(fileID)
 	if err != nil {

diff --git a/internal/integrations/webdav.go b/internal/integrations/webdav.go
@@ -88,6 +88,25 @@ func (w *WebDavIntegration) Upload(folderID, name, fileType string, reader io.Re
 	return
 }
 
+func (w *WebDavIntegration) GetMetadata(fileID string) (*messages.IntegrationMetadata, error) {
+	decoded, err := decodeName(fileID)
+	if err != nil {
+		return nil, err
+	}
+
+	ext := path.Ext(decoded)
+	contentType := contentTypeFromExt(ext)
+
+	return &messages.IntegrationMetadata{
+		ID:               fileID,
+		Name:             path.Base(decoded),
+		Thumbnail:        []byte{},
+		SourceFileType:   contentType,
+		ProvidedFileType: contentType,
+		FileType:         ext,
+	}, nil
+}
+
 // Download downloads
 func (w *WebDavIntegration) Download(fileID string) (io.ReadCloser, int64, error) {
 	decoded, err := decodeName(fileID)

diff --git a/internal/messages/messages.go b/internal/messages/messages.go
@@ -190,8 +190,11 @@ type IntegrationFolder struct {
 }
 
 type IntegrationMetadata struct {
-	FileType  string `json:"fileType"`
-	ID        string `json:"id"`
-	Name      string `json:"name"`
-	Thumbnail string `json:"thumbnail"`
+	ID   string `json:"id"`
+	Name string `json:"name"`
+	// Thumbnail is base64 encoded string of an image/png
+	Thumbnail        []byte `json:"thumbnail"`
+	SourceFileType   string `json:"sourceFileType"`
+	ProvidedFileType string `json:"providedFileType"`
+	FileType         string `json:"fileType"`
 }
diff --git a/internal/ui/handlers.go b/internal/ui/handlers.go
@@ -676,6 +676,30 @@ func (app *ReactAppWrapper) exploreIntegration(c *gin.Context) {
 	c.JSON(http.StatusOK, response)
 }
 
+func (app *ReactAppWrapper) getMetadataIntegration(c *gin.Context) {
+	uid := c.GetString(userIDContextKey)
+
+	integrationID := common.ParamS(intIDParam, c)
+
+	integrationProvider, err := integrations.GetIntegrationProvider(app.userStorer, uid, integrationID)
+	if err != nil {
+		log.Error(err)
+		c.AbortWithStatus(http.StatusInternalServerError)
+		return
+	}
+
+	fileid := common.ParamS("path", c)
+
+	response, err := integrationProvider.GetMetadata(fileid)
+	if err != nil {
+		log.Error(err)
+		c.AbortWithStatus(http.StatusInternalServerError)
+		return
+	}
+
+	c.JSON(http.StatusOK, response)
+}
+
 func (app *ReactAppWrapper) downloadThroughIntegration(c *gin.Context) {
 	uid := c.GetString(userIDContextKey)
 

diff --git a/internal/ui/routes.go b/internal/ui/routes.go
@@ -80,6 +80,7 @@ func (app *ReactAppWrapper) RegisterRoutes(router *gin.Engine) {
 	auth.DELETE("integrations/:intid", app.deleteIntegration)
 
 	auth.GET("integrations/:intid/explore/*path", app.exploreIntegration)
+	auth.GET("integrations/:intid/metadata/*path", app.getMetadataIntegration)
 	auth.GET("integrations/:intid/download/*path", app.downloadThroughIntegration)
 
 	//admin