From 535686ff4e2413e67fd1efbf6e3deb2e1591aa4f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Yhoan=20Alejandro=20Guzm=C3=A1n=20Garc=C3=ADa?=
 <41337901+yaguzmang@users.noreply.github.com>
Date: Fri, 3 Jan 2025 01:11:42 -0500
Subject: [PATCH] 4001 - Datatable use Datagrid - ButtonCopyValues (#4194)

* 4001 - Copy values from grid

* 4001 - Move and refactor Copy logic to useCopyValues hook

* 4001 - Minor rtl fix for disabled input number
---
 .../ButtonCopyValues/ButtonCopyValues.tsx     | 77 ++--------------
 .../ButtonCopyValues/hooks/useCopyValues.ts   | 92 +++++++++++++++++++
 .../DataTable/Table/ButtonCopyValues/types.ts |  8 ++
 .../Table/RowData/Cell/Number/Number.scss     | 12 +++
 .../pages/Section/DataTable/Table/Table.tsx   |  2 +-
 5 files changed, 119 insertions(+), 72 deletions(-)
 create mode 100644 src/client/pages/Section/DataTable/Table/ButtonCopyValues/hooks/useCopyValues.ts
 create mode 100644 src/client/pages/Section/DataTable/Table/ButtonCopyValues/types.ts

diff --git a/src/client/pages/Section/DataTable/Table/ButtonCopyValues/ButtonCopyValues.tsx b/src/client/pages/Section/DataTable/Table/ButtonCopyValues/ButtonCopyValues.tsx
index 70c5f11cb6..30957c09c7 100644
--- a/src/client/pages/Section/DataTable/Table/ButtonCopyValues/ButtonCopyValues.tsx
+++ b/src/client/pages/Section/DataTable/Table/ButtonCopyValues/ButtonCopyValues.tsx
@@ -1,83 +1,18 @@
-import React, { MutableRefObject, useCallback } from 'react'
+import React from 'react'
 import { useTranslation } from 'react-i18next'
 
-import { Table, TableNames } from 'meta/assessment'
-
-import { useCycle } from 'client/store/assessment'
-import { useUser } from 'client/store/user'
 import Button from 'client/components/Buttons/Button'
-import { getData } from 'client/components/ButtonTableExport/utils'
-
-type CopyValuesProps = {
-  tableRef: MutableRefObject<HTMLTableElement>
-  table: Table
-}
 
-// tableMappings is a mapping of table names to the variables that should be copied to clipboard
-const tableMappings2020: Record<string, Array<string>> = {
-  [TableNames.forestCharacteristics]: [
-    'forestCharacteristics.naturalForestArea',
-    'forestCharacteristics.plantationForestArea',
-    'forestCharacteristics.plantationForestIntroducedArea',
-    'forestCharacteristics.otherPlantedForestArea',
-  ],
-  [TableNames.growingStockAvg]: [
-    'growingStock.naturallyRegeneratingForest',
-    'growingStock.plantedForest',
-    'growingStock.plantationForest',
-    'growingStock.otherPlantedForest',
-  ],
-}
-
-const tableMappings2025: Record<string, Array<string>> = {
-  [TableNames.forestCharacteristics]: [
-    'fra.forestCharacteristics.naturalForestArea2025',
-    'fra.growingStock.plantationForest2025',
-    'fra.forestCharacteristics.plantationForestIntroducedArea2025',
-    'fra.forestCharacteristics.ofWhichOtherPlantedForest',
-  ],
-  [TableNames.growingStockAvg]: [
-    'fra.growingStock.naturallyRegeneratingForest2025',
-    'fra.growingStock.plantedForest2025',
-    'fra.growingStock.plantationForest2025',
-    'fra.growingStock.otherPlantedForest2025',
-  ],
-}
-
-const colMapping = [1990, 2000, 2010, 2015, 2020, 2025]
+import { useCopyValues } from './hooks/useCopyValues'
+import { CopyValuesProps } from './types'
 
 const ButtonCopyValues: React.FC<CopyValuesProps> = (props: CopyValuesProps) => {
-  const { tableRef, table } = props
-  const cycle = useCycle()
-  const user = useUser()
   const { t } = useTranslation()
+  const { onClick, showButton } = useCopyValues(props)
 
-  const tableMappings = cycle.name === '2025' ? tableMappings2025 : tableMappings2020
-
-  const showButton = Object.keys(tableMappings).includes(table.props.name)
-
-  const _onClick = useCallback(() => {
-    const _table = tableRef.current
-    if (!_table) return
-    const csv = getData(_table)
-    const include = tableMappings[table.props.name].map((variableLabel) => t(variableLabel))
-    // A list of indexes of the table columns that should be copied to clipboard
-    const correctIndexes = colMapping.map((year) => csv[1].indexOf(year.toString()))
-    const z = csv
-      .filter((row) => {
-        return include.some((translatedVariable) => row[0].includes(translatedVariable))
-      })
-      .map((row: string[]) => {
-        return row.filter((_, i) => correctIndexes.includes(i))
-      })
-
-    navigator.clipboard.writeText(z.map((row: Array<string>) => row.join('\t')).join('\n'))
-  }, [t, table.props.name, tableMappings, tableRef])
-
-  // Hide button if incorrect table or user is not logged in
-  if (!user || !showButton) return null
+  if (!showButton) return null
 
-  return <Button iconName="content_copy" label={t('tableWithOdp.copyToClipboard')} onClick={_onClick} />
+  return <Button iconName="content_copy" label={t('tableWithOdp.copyToClipboard')} onClick={onClick} />
 }
 
 export default ButtonCopyValues
diff --git a/src/client/pages/Section/DataTable/Table/ButtonCopyValues/hooks/useCopyValues.ts b/src/client/pages/Section/DataTable/Table/ButtonCopyValues/hooks/useCopyValues.ts
new file mode 100644
index 0000000000..a72d0e4343
--- /dev/null
+++ b/src/client/pages/Section/DataTable/Table/ButtonCopyValues/hooks/useCopyValues.ts
@@ -0,0 +1,92 @@
+import { useCallback, useMemo } from 'react'
+import { useTranslation } from 'react-i18next'
+
+import { Objects } from 'utils/objects'
+
+import { TableNames } from 'meta/assessment'
+
+import { useUser } from 'client/store/user'
+import { useCycleRouteParams } from 'client/hooks/useRouteParams'
+import { getDataGridData } from 'client/components/DataGrid/ButtonGridExport/utils'
+import { CopyValuesProps } from 'client/pages/Section/DataTable/Table/ButtonCopyValues/types'
+
+// Cycle -> Table name -> Variables to be coppied to clipboard
+const tableMappings: Record<string, Record<string, Array<string>>> = {
+  '2020': {
+    [TableNames.forestCharacteristics]: [
+      'forestCharacteristics.naturalForestArea',
+      'forestCharacteristics.plantationForestArea',
+      'forestCharacteristics.plantationForestIntroducedArea',
+      'forestCharacteristics.otherPlantedForestArea',
+    ],
+    [TableNames.growingStockAvg]: [
+      'growingStock.naturallyRegeneratingForest',
+      'growingStock.plantedForest',
+      'growingStock.plantationForest',
+      'growingStock.otherPlantedForest',
+    ],
+  },
+  '2025': {
+    [TableNames.forestCharacteristics]: [
+      'fra.forestCharacteristics.naturalForestArea2025',
+      'fra.growingStock.plantationForest2025',
+      'fra.forestCharacteristics.plantationForestIntroducedArea2025',
+      'fra.forestCharacteristics.ofWhichOtherPlantedForest',
+    ],
+    [TableNames.growingStockAvg]: [
+      'fra.growingStock.naturallyRegeneratingForest2025',
+      'fra.growingStock.plantedForest2025',
+      'fra.growingStock.plantationForest2025',
+      'fra.growingStock.otherPlantedForest2025',
+    ],
+  },
+}
+
+const yearsToCopy = [1990, 2000, 2010, 2015, 2020, 2025]
+
+type Returned = {
+  onClick: () => void
+  showButton: boolean
+}
+
+export const useCopyValues = (props: CopyValuesProps): Returned => {
+  const { gridRef, table } = props
+
+  const { t } = useTranslation()
+  const { cycleName } = useCycleRouteParams()
+  const user = useUser()
+
+  const cycleTableMappings = tableMappings[cycleName]
+
+  const isTableIncluded = useMemo<boolean>(
+    () => Object.keys(cycleTableMappings ?? {}).includes(table.props.name),
+    [cycleTableMappings, table.props.name]
+  )
+
+  const onClick = useCallback<Returned['onClick']>(() => {
+    if (!isTableIncluded) return
+
+    const grid = gridRef.current
+    if (!grid) return
+
+    const csv = getDataGridData(grid)
+    const variablesToCopy = cycleTableMappings[table.props.name].map((variableLabel) => t(variableLabel))
+    const columnIndexesToCopy = yearsToCopy.map((year) => csv[1].indexOf(year.toString()))
+
+    const z = csv.reduce((acc: Array<Array<string>>, row: Array<string>) => {
+      const shouldIncludeRow = variablesToCopy.some((translatedVariable) => row[0].includes(translatedVariable))
+      if (shouldIncludeRow) {
+        const filteredRow = row.filter((_, i) => columnIndexesToCopy.includes(i))
+        acc.push(filteredRow)
+      }
+      return acc
+    }, [])
+
+    navigator.clipboard.writeText(z.map((row: Array<string>) => row.join('\t')).join('\n'))
+  }, [cycleTableMappings, gridRef, isTableIncluded, t, table.props.name])
+
+  return {
+    onClick,
+    showButton: isTableIncluded && !Objects.isEmpty(user),
+  }
+}
diff --git a/src/client/pages/Section/DataTable/Table/ButtonCopyValues/types.ts b/src/client/pages/Section/DataTable/Table/ButtonCopyValues/types.ts
new file mode 100644
index 0000000000..f79ae74729
--- /dev/null
+++ b/src/client/pages/Section/DataTable/Table/ButtonCopyValues/types.ts
@@ -0,0 +1,8 @@
+import { MutableRefObject } from 'react'
+
+import { Table } from 'meta/assessment'
+
+export type CopyValuesProps = {
+  gridRef: MutableRefObject<HTMLDivElement>
+  table: Table
+}
diff --git a/src/client/pages/Section/DataTable/Table/RowData/Cell/Number/Number.scss b/src/client/pages/Section/DataTable/Table/RowData/Cell/Number/Number.scss
index fcd4770566..fd805678a9 100644
--- a/src/client/pages/Section/DataTable/Table/RowData/Cell/Number/Number.scss
+++ b/src/client/pages/Section/DataTable/Table/RowData/Cell/Number/Number.scss
@@ -1,3 +1,5 @@
+@import 'src/client/style/partials';
+
 .table-grid__data-cell-input-number {
   text-align: right;
 
@@ -9,3 +11,13 @@
     }
   }
 }
+
+@include rtl {
+  .table-grid__data-cell-input-number {
+    &.input-text {
+      &.disabled {
+        justify-content: start;
+      }
+    }
+  }
+}
diff --git a/src/client/pages/Section/DataTable/Table/Table.tsx b/src/client/pages/Section/DataTable/Table/Table.tsx
index 6bd0ecc96b..6374561eb9 100644
--- a/src/client/pages/Section/DataTable/Table/Table.tsx
+++ b/src/client/pages/Section/DataTable/Table/Table.tsx
@@ -64,7 +64,7 @@ const Table: React.FC<Props> = (props) => {
       <div className="fra-table__scroll-wrapper">
         <div className="fra-table__editor">
           {!print && <ButtonGridExport filename={fileName} gridRef={gridRef} />}
-          <ButtonCopyValues table={table} tableRef={tableRef} />
+          <ButtonCopyValues gridRef={gridRef} table={table} />
           {canClearData && <ButtonTableClear disabled={disabled} sectionName={sectionName} table={table} />}
         </div>