#251 - Add default content type for Google Docs

formkiq · Jul 30, 2024 · b711ead · b711ead
1 parent 10f5416
commit b711ead
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 3 deletions.
diff --git a/dynamodb-documents/src/main/java/com/formkiq/stacks/dynamodb/DocumentServiceImpl.java b/dynamodb-documents/src/main/java/com/formkiq/stacks/dynamodb/DocumentServiceImpl.java
@@ -2134,8 +2134,15 @@ private void updateCurrentChecksumFromPrevious(final DocumentItem document,
 
   private void updateDeepLinkPath(final Map<String, AttributeValue> pkvalues,
       final DocumentItem document) {
+
     if (!isEmpty(document.getDeepLinkPath())) {
       addS(pkvalues, "deepLinkPath", document.getDeepLinkPath());
+
+      if (document.getDeepLinkPath().startsWith("https://docs.google.com/document")
+          && isEmpty(document.getContentType())) {
+        document.setContentType("application/vnd.google-apps.document");
+      }
+
     } else {
       addS(pkvalues, "deepLinkPath", "");
     }

diff --git a/lambda-api/src/test/java/com/formkiq/stacks/api/handler/DocumentsRequestTest.java b/lambda-api/src/test/java/com/formkiq/stacks/api/handler/DocumentsRequestTest.java
@@ -420,14 +420,13 @@ public void testPost09() throws ApiException {
   /**
    * Save new File with invaliddocumentId.
    *
-   * @throws ApiException ApiException
    */
   @Test
-  public void testPost10() throws ApiException {
+  public void testPost10() {
     // given
     for (String siteId : Arrays.asList(null, UUID.randomUUID().toString())) {
 
-      String documentId = "casaljdalsdjat" + UUID.randomUUID().toString();
+      String documentId = "casaljdalsdjat" + UUID.randomUUID();
       setBearerToken(siteId);
 
       AddDocumentRequest req = new AddDocumentRequest().documentId(documentId).content("dummy data")
@@ -446,6 +445,40 @@ public void testPost10() throws ApiException {
     }
   }
 
+  /**
+   * Save google drive deep link.
+   *
+   */
+  @Test
+  public void testPost11() throws ApiException {
+    // given
+    for (String siteId : Arrays.asList("default", UUID.randomUUID().toString())) {
+
+      setBearerToken(siteId);
+
+      String deepLink = "https://docs.google.com/document/d/1tyOQ3yUL9dtpbuMOt7s/edit?usp=sharing";
+
+      AddDocumentRequest req = new AddDocumentRequest().deepLinkPath(deepLink);
+
+      // when
+      String documentId = this.documentsApi.addDocument(req, null, null).getDocumentId();
+
+      // then
+      GetDocumentResponse doc = this.documentsApi.getDocument(documentId, siteId, null);
+      assertEquals("application/vnd.google-apps.document", doc.getContentType());
+
+      // given
+      req.setContentType("application/pdf");
+
+      // when
+      documentId = this.documentsApi.addDocument(req, null, null).getDocumentId();
+
+      // then
+      doc = this.documentsApi.getDocument(documentId, siteId, null);
+      assertEquals("application/pdf", doc.getContentType());
+    }
+  }
+
   /**
    * Test Publish no published document.
    *