From d85aab14b8d20a0618d517f3d4d0c94adc724443 Mon Sep 17 00:00:00 2001
From: Pranav C <pranavxc@gmail.com>
Date: Mon, 19 Jun 2023 23:45:55 +0530
Subject: [PATCH] refactor: update drag and drop behaviour

Signed-off-by: Pranav C <pranavxc@gmail.com>
---
 .../nc-gui/components/tabs/Smartsheet.vue     | 38 ++++++++++---------
 1 file changed, 21 insertions(+), 17 deletions(-)

diff --git a/packages/nc-gui/components/tabs/Smartsheet.vue b/packages/nc-gui/components/tabs/Smartsheet.vue
index f248d5f04d..3549950afd 100644
--- a/packages/nc-gui/components/tabs/Smartsheet.vue
+++ b/packages/nc-gui/components/tabs/Smartsheet.vue
@@ -1,6 +1,6 @@
 <script setup lang="ts">
 import type { ColumnType, LinkToAnotherRecordType, TableType } from 'nocodb-sdk'
-import { UITypes, isLinksOrLTAR } from 'nocodb-sdk'
+import { UITypes, isLinksOrLTAR, RelationTypes } from 'nocodb-sdk'
 import {
   ActiveViewInj,
   FieldsInj,
@@ -95,24 +95,28 @@ const onDrop = async (event: DragEvent) => {
 
       const ltarOptions = c.colOptions as LinkToAnotherRecordType
 
-      if (ltarOptions.type !== 'mm') {
-        return false
-      }
-
-      if (ltarOptions.fk_related_model_id === childMeta.id) {
-        return true
-      }
-
-      return false
+      return ltarOptions.fk_related_model_id === childMeta.id
     })
     if (relationCol) {
-      const lookupCol = childMeta.columns?.find((c) => c.pv) ?? childMeta.columns?.[0]
-      grid.value?.openColumnCreate({
-        uidt: UITypes.Lookup,
-        title: `${data.title}Lookup`,
-        fk_relation_column_id: relationCol.id,
-        fk_lookup_column_id: lookupCol?.id,
-      })
+      if (relationCol.colOptions.type === RelationTypes.BELONGS_TO) {
+        const lookupCol = childMeta.columns?.find((c) => c.pv) ?? childMeta.columns?.[0]
+
+        grid.value?.openColumnCreate({
+          uidt: UITypes.Lookup,
+          title: `${relationCol.title}Lookup`,
+          fk_relation_column_id: relationCol.id,
+          fk_lookup_column_id: lookupCol?.id,
+        })
+      } else {
+        const rollupCol = childMeta.columns?.find((c) => c.pk) ?? childMeta.columns?.[0]
+        grid.value?.openColumnCreate({
+          uidt: UITypes.Rollup,
+          title: `${relationCol.title}Rollup`,
+          fk_relation_column_id: relationCol.id,
+          fk_rollup_column_id: rollupCol?.id,
+          rollup_function: 'max',
+        })
+      }
     } else {
       grid.value?.openColumnCreate({
         uidt: UITypes.Links,