From f22d39ce2932cdcd4d38358e10770f48a53be6f0 Mon Sep 17 00:00:00 2001
From: Shariq Ansari <sharique.rik@gmail.com>
Date: Fri, 19 Apr 2024 13:19:59 +0530
Subject: [PATCH] feat: allow sending image, doc & video

---
 crm/api/whatsapp.py                     |  5 +-
 frontend/src/components/WhatsAppBox.vue | 62 ++++++++++++++++++++++++-
 2 files changed, 63 insertions(+), 4 deletions(-)

diff --git a/crm/api/whatsapp.py b/crm/api/whatsapp.py
index 1f7fd657..2b8a5def 100644
--- a/crm/api/whatsapp.py
+++ b/crm/api/whatsapp.py
@@ -41,13 +41,14 @@ def parse_mobile_no(mobile_no: str):
 	return ''.join([c for c in mobile_no if c.isdigit() or c == '+'])
 
 @frappe.whitelist()
-def create_whatsapp_message(reference_doctype, reference_name, to, message, content_type="text"):
+def create_whatsapp_message(reference_doctype, reference_name, message, to, attach, content_type="text"):
 	doc = frappe.new_doc("WhatsApp Message")
 	doc.update({
 		"reference_doctype": reference_doctype,
 		"reference_name": reference_name,
+		"message": message or attach,
 		"to": to,
-		"message": message,
+		"attach": attach,
 		"content_type": content_type,
 	})
 	doc.insert(ignore_permissions=True)
diff --git a/frontend/src/components/WhatsAppBox.vue b/frontend/src/components/WhatsAppBox.vue
index e3e3c5b1..5b920f69 100644
--- a/frontend/src/components/WhatsAppBox.vue
+++ b/frontend/src/components/WhatsAppBox.vue
@@ -1,5 +1,22 @@
 <template>
   <div class="flex items-end gap-2 px-10 py-2.5">
+    <div class="flex h-8 items-center">
+      <FileUploader @success="(file) => uploadFile(file)">
+        <template v-slot="{ file, progress, uploading, openFileSelector }">
+          <div class="flex items-center space-x-2">
+            <Dropdown
+              v-bind="{ open }"
+              :options="uploadOptions(openFileSelector)"
+            >
+              <FeatherIcon
+                name="plus"
+                class="size-4.5 cursor-pointer text-gray-600"
+              />
+            </Dropdown>
+          </div>
+        </template>
+      </FileUploader>
+    </div>
     <Textarea
       ref="textarea"
       type="textarea"
@@ -24,7 +41,8 @@
 </template>
 
 <script setup>
-import { createResource, Textarea } from 'frappe-ui'
+import { createResource, Textarea, FileUploader, Dropdown } from 'frappe-ui'
+import FeatherIcon from 'frappe-ui/src/components/FeatherIcon.vue'
 import { ref, computed, nextTick } from 'vue'
 
 const props = defineProps({
@@ -38,6 +56,7 @@ const textarea = ref(null)
 
 const content = ref('')
 const placeholder = ref(__('Type your message here...'))
+const fileType = ref('')
 
 const isEmpty = computed(() => {
   return !content.value || content.value === '<p></p>'
@@ -47,15 +66,25 @@ function show() {
   nextTick(() => textarea.value.$el.focus())
 }
 
+function uploadFile(file) {
+  whatsapp.value.attach = file.file_url
+  whatsapp.value.content_type = fileType.value
+  sendWhatsAppMessage()
+}
+
 async function sendWhatsAppMessage() {
   let args = {
     reference_doctype: props.doctype,
     reference_name: doc.value.data.name,
     message: content.value,
     to: doc.value.data.mobile_no,
-    content_type: 'text',
+    attach: whatsapp.value.attach || '',
+    content_type: whatsapp.value.content_type,
   }
   content.value = ''
+  fileType.value = ''
+  whatsapp.value.attach = ''
+  whatsapp.value.content_type = 'text'
   createResource({
     url: 'crm.api.whatsapp.create_whatsapp_message',
     params: args,
@@ -64,5 +93,34 @@ async function sendWhatsAppMessage() {
   })
 }
 
+function uploadOptions(openFileSelector) {
+  return [
+    {
+      label: __('Upload Document'),
+      icon: 'file',
+      onClick: () => {
+        fileType.value = 'document'
+        openFileSelector()
+      },
+    },
+    {
+      label: __('Upload Image'),
+      icon: 'image',
+      onClick: () => {
+        fileType.value = 'image'
+        openFileSelector('image/*')
+      },
+    },
+    {
+      label: __('Upload Video'),
+      icon: 'video',
+      onClick: () => {
+        fileType.value = 'video'
+        openFileSelector('video/*')
+      },
+    },
+  ]
+}
+
 defineExpose({ show })
 </script>