feat: add pure helpers for list excel row and column copy

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
wuzilongandCursor committed 2026-10-19 19:06:54 +08:00
1 parent 9177e1b7f1
commit aae4465f6c
2 files changed
+184

No files matched your search

+120
View File
@@ -0,0 +1,120 @@
import assert from 'node:assert/strict'
import { describe, it } from 'node:test'
import {
buildExcelTsv,
excelCell,
isCopyableColumn,
resolveCopyAxes,
toggleAxis
} from './list-excel-copy'
describe('isCopyableColumn', () => {
it('keeps ordinary data columns', () => {
assert.equal(isCopyableColumn({ id: 'name', label: '联系人' }), true)
})
it('drops selection, index, expand, label 操作, and className containing col-actions', () => {
assert.equal(isCopyableColumn({ id: 's', label: '', type: 'selection' }), false)
assert.equal(isCopyableColumn({ id: 'i', label: '', type: 'index' }), false)
assert.equal(isCopyableColumn({ id: 'e', label: '', type: 'expand' }), false)
assert.equal(isCopyableColumn({ id: 'a', label: '操作' }), false)
assert.equal(
isCopyableColumn({ id: 'act', label: '编辑', className: 'col-actions fixed' }),
false
)
})
})
describe('excelCell', () => {
it('trims and turns tabs and line breaks into spaces', () => {
assert.equal(excelCell(' 合作中 \n'), '合作中')
assert.equal(excelCell('重庆\t长寿\r\n区'), '重庆 长寿 区')
})
})
describe('toggleAxis', () => {
it('toggles one index and remembers anchor; second toggle of same index removes it but keeps anchor', () => {
let selected = new Set<number>()
let anchor: number | null = null
const first = toggleAxis(selected, 2, false, anchor)
selected = first.selected
anchor = first.anchor
assert.deepEqual([...selected].sort(), [2])
assert.equal(anchor, 2)
const second = toggleAxis(selected, 2, false, anchor)
selected = second.selected
anchor = second.anchor
assert.deepEqual([...selected], [])
assert.equal(anchor, 2)
})
it('shift-selects inclusive span and keeps earlier picks: start with set containing 0 and anchor 0, shift-click 3 yields 0,1,2,3 and anchor 3', () => {
const selected = new Set([0])
const result = toggleAxis(selected, 3, true, 0)
assert.deepEqual([...result.selected].sort((a, b) => a - b), [0, 1, 2, 3])
assert.equal(result.anchor, 3)
})
it('shift without an anchor toggles the clicked index', () => {
const selected = new Set<number>()
const result = toggleAxis(selected, 1, true, null)
assert.deepEqual([...result.selected], [1])
assert.equal(result.anchor, 1)
})
})
describe('resolveCopyAxes', () => {
it('returns null when nothing selected', () => {
assert.equal(
resolveCopyAxes({
selectedCols: [],
selectedRows: [],
columnCount: 3,
rowCount: 2
}),
null
)
})
it('expands columns to every current row, sorting cols: selectedCols [2,0], rowCount 2, columnCount 3 => cols [0,2] rows [0,1]', () => {
const result = resolveCopyAxes({
selectedCols: [2, 0],
selectedRows: [],
columnCount: 3,
rowCount: 2
})
assert.deepEqual(result, { cols: [0, 2], rows: [0, 1] })
})
it('expands rows to every copyable column: selectedRows [3,1], columnCount 2, rowCount 4 => cols [0,1] rows [1,3]', () => {
const result = resolveCopyAxes({
selectedCols: [],
selectedRows: [3, 1],
columnCount: 2,
rowCount: 4
})
assert.deepEqual(result, { cols: [0, 1], rows: [1, 3] })
})
it('intersection when both selected: selectedCols [1], selectedRows [0,2], columnCount 3, rowCount 4 => cols [1] rows [0,2]', () => {
const result = resolveCopyAxes({
selectedCols: [1],
selectedRows: [0, 2],
columnCount: 3,
rowCount: 4
})
assert.deepEqual(result, { cols: [1], rows: [0, 2] })
})
})
describe('buildExcelTsv', () => {
it('builds CRLF-separated TSV with header row', () => {
const tsv = buildExcelTsv(['联系人', '状态'], [
['张三', '合作中'],
['李四', '跟进中']
])
assert.equal(tsv, '联系人\t状态\r\n张三\t合作中\r\n李四\t跟进中')
})
})
+64
View File
@@ -0,0 +1,64 @@
export type ListExcelColumn = {
id: string
label: string
type?: string
className?: string
}
export function isCopyableColumn(col: ListExcelColumn): boolean {
if (col.type === 'selection' || col.type === 'index' || col.type === 'expand') return false
if (col.label === '操作') return false
const classes = (col.className || '').split(/\s+/).filter(Boolean)
if (classes.includes('col-actions')) return false
return true
}
export function excelCell(value: string): string {
return String(value ?? '')
.replace(/\t/g, ' ')
.replace(/\r?\n/g, ' ')
.trim()
}
export function toggleAxis(
selected: Set<number>,
index: number,
shift: boolean,
anchor: number | null
): { selected: Set<number>; anchor: number } {
const next = new Set(selected)
if (shift && anchor !== null) {
const from = Math.min(anchor, index)
const to = Math.max(anchor, index)
for (let i = from; i <= to; i += 1) next.add(i)
return { selected: next, anchor: index }
}
if (next.has(index)) next.delete(index)
else next.add(index)
return { selected: next, anchor: index }
}
export function resolveCopyAxes(input: {
selectedCols: number[]
selectedRows: number[]
columnCount: number
rowCount: number
}): { cols: number[]; rows: number[] } | null {
const { selectedCols, selectedRows, columnCount, rowCount } = input
if (!selectedCols.length && !selectedRows.length) return null
const cols = (selectedCols.length ? selectedCols : range(columnCount)).slice().sort((a, b) => a - b)
const rows = (selectedRows.length ? selectedRows : range(rowCount)).slice().sort((a, b) => a - b)
return { cols, rows }
}
export function buildExcelTsv(headers: string[], rows: string[][]): string {
const lines = [headers.map(excelCell).join('\t')]
rows.forEach((row) => {
lines.push(row.map(excelCell).join('\t'))
})
return lines.join('\r\n')
}
function range(count: number): number[] {
return Array.from({ length: count }, (_, index) => index)
}