refactor(ModelMappingDiagram): restructure component to utilize new modular columns and improve type definitions

This commit is contained in:
thanhtunguet
2026-01-31 21:47:49 +07:00
parent ce47d6d985
commit 0d40eecbe7
5 changed files with 757 additions and 456 deletions

View File

@@ -1,23 +1,24 @@
import { forwardRef, useCallback, useEffect, useImperativeHandle, useLayoutEffect, useMemo, useRef, useState, type DragEvent, type MouseEvent as ReactMouseEvent } from 'react';
import { createPortal } from 'react-dom';
import { useTranslation } from 'react-i18next';
import type { OAuthModelAliasEntry } from '@/types';
import { useThemeStore } from '@/stores';
import { Modal } from '@/components/ui/Modal';
import { Input } from '@/components/ui/Input';
import { Button } from '@/components/ui/Button';
import { ToggleSwitch } from '@/components/ui/ToggleSwitch';
import { IconTrash2 } from '@/components/ui/icons';
import { AliasColumn, ProviderColumn, SourceColumn } from './ModelMappingDiagramColumns';
import { DiagramContextMenu } from './ModelMappingDiagramContextMenu';
import {
AddAliasModal,
RenameAliasModal,
SettingsAliasModal,
SettingsSourceModal
} from './ModelMappingDiagramModals';
import type {
AliasNode,
AuthFileModelItem,
ContextMenuState,
DiagramLine,
SourceNode
} from './ModelMappingDiagramTypes';
import styles from './ModelMappingDiagram.module.scss';
// Type definition for models from API
export interface AuthFileModelItem {
id: string;
display_name?: string;
type?: string;
owned_by?: string;
}
export interface ModelMappingDiagramProps {
modelAlias: Record<string, OAuthModelAliasEntry[]>;
allProviderModels?: Record<string, AuthFileModelItem[]>;
@@ -31,7 +32,6 @@ export interface ModelMappingDiagramProps {
className?: string;
}
// Helper to generate consistent colors
const PROVIDER_COLORS = [
'#3b82f6', '#10b981', '#f59e0b', '#ef4444',
'#8b5cf6', '#ec4899', '#06b6d4', '#84cc16'
@@ -42,26 +42,6 @@ function getProviderColor(provider: string): string {
return PROVIDER_COLORS[hash % PROVIDER_COLORS.length];
}
interface SourceNode {
id: string; // unique: provider::name
provider: string;
name: string;
aliases: { alias: string; fork: boolean }[]; // all aliases this source maps to
}
interface AliasNode {
id: string; // alias
alias: string;
sources: SourceNode[];
}
interface ContextMenuState {
x: number;
y: number;
type: 'alias' | 'background' | 'provider' | 'source';
data?: string;
}
export interface ModelMappingDiagramRef {
collapseAll: () => void;
refreshLayout: () => void;
@@ -84,7 +64,7 @@ export const ModelMappingDiagram = forwardRef<ModelMappingDiagramRef, ModelMappi
const isDark = resolvedTheme === 'dark';
const containerRef = useRef<HTMLDivElement>(null);
const [lines, setLines] = useState<{ path: string; color: string; id: string }[]>([]);
const [lines, setLines] = useState<DiagramLine[]>([]);
const [draggedSource, setDraggedSource] = useState<SourceNode | null>(null);
const [draggedAlias, setDraggedAlias] = useState<string | null>(null);
const [dropTargetAlias, setDropTargetAlias] = useState<string | null>(null);
@@ -100,18 +80,6 @@ export const ModelMappingDiagram = forwardRef<ModelMappingDiagramRef, ModelMappi
const [addAliasError, setAddAliasError] = useState('');
const [settingsAlias, setSettingsAlias] = useState<string | null>(null);
const [settingsSourceId, setSettingsSourceId] = useState<string | null>(null);
const contextMenuRef = useRef<HTMLDivElement | null>(null);
// Close context menu on click outside
useEffect(() => {
const handleClick = (event: globalThis.MouseEvent) => {
if (!contextMenuRef.current?.contains(event.target as Node)) {
setContextMenu(null);
}
};
document.addEventListener('mousedown', handleClick);
return () => document.removeEventListener('mousedown', handleClick);
}, []);
// Parse data: each source model (provider+name) and each alias is distinct by id; 1 source -> many aliases.
const { aliasNodes, providerNodes } = useMemo(() => {
@@ -458,77 +426,6 @@ export const ModelMappingDiagram = forwardRef<ModelMappingDiagramRef, ModelMappi
}
};
const renderAliasMenu = () => {
const aliasData = contextMenu?.data;
if (contextMenu?.type !== 'alias' || aliasData == null) return null;
return (
<>
<div className={styles.menuItem} onClick={() => handleRenameClick(aliasData)}>
<span>{t('oauth_model_alias.diagram_rename')}</span>
</div>
<div
className={styles.menuItem}
onClick={() => {
closeContextMenu();
setSettingsAlias(aliasData);
}}
>
<span>{t('oauth_model_alias.diagram_settings')}</span>
</div>
<div className={styles.menuDivider} />
<div className={`${styles.menuItem} ${styles.danger}`} onClick={() => handleDeleteClick(aliasData)}>
<span>{t('oauth_model_alias.diagram_delete_alias')}</span>
</div>
</>
);
};
const renderProviderMenu = () => {
const provider = contextMenu?.data;
if (contextMenu?.type !== 'provider' || !provider) return null;
return (
<>
<div
className={styles.menuItem}
onClick={() => {
closeContextMenu();
onEditProvider?.(provider);
}}
>
<span>{t('common.edit')}</span>
</div>
<div className={styles.menuDivider} />
<div
className={`${styles.menuItem} ${styles.danger}`}
onClick={() => {
closeContextMenu();
onDeleteProvider?.(provider);
}}
>
<span>{t('oauth_model_alias.delete')}</span>
</div>
</>
);
};
const renderSourceMenu = () => {
const sourceId = contextMenu?.data;
const source = resolveSourceById(sourceId ?? null);
if (contextMenu?.type !== 'source' || !source) return null;
return (
<>
<div
className={styles.menuItem}
onClick={() => {
closeContextMenu();
setSettingsSourceId(source.id);
}}
>
<span>{t('oauth_model_alias.diagram_settings')}</span>
</div>
</>
);
};
return (
<div
@@ -551,356 +448,120 @@ export const ModelMappingDiagram = forwardRef<ModelMappingDiagramRef, ModelMappi
))}
</svg>
{/* Column 1: Providers (mindmap root branch) */}
<div
className={`${styles.column} ${styles.providers}`}
onContextMenu={(e) => {
e.preventDefault();
e.stopPropagation();
handleContextMenu(e, 'background');
<ProviderColumn
providerNodes={providerNodes}
collapsedProviders={collapsedProviders}
getProviderColor={getProviderColor}
providerRefs={providerRefs}
onToggleCollapse={toggleProviderCollapse}
onContextMenu={(e, type, data) => handleContextMenu(e, type, data)}
label={t('oauth_model_alias.diagram_providers')}
expandLabel={t('oauth_model_alias.diagram_expand')}
collapseLabel={t('oauth_model_alias.diagram_collapse')}
/>
<SourceColumn
providerNodes={providerNodes}
collapsedProviders={collapsedProviders}
sourceRefs={sourceRefs}
getProviderColor={getProviderColor}
draggedSource={draggedSource}
dropTargetSource={dropTargetSource}
draggable={!!onUpdate}
onDragStart={handleDragStart}
onDragEnd={() => {
setDraggedSource(null);
setDropTargetAlias(null);
}}
>
<div className={styles.columnHeader}>{t('oauth_model_alias.diagram_providers')}</div>
{providerNodes.map(({ provider, sources }) => {
const collapsed = collapsedProviders.has(provider);
return (
<div
key={provider}
ref={(el) => {
if (el) providerRefs.current.set(provider, el);
else providerRefs.current.delete(provider);
}}
className={`${styles.item} ${styles.providerItem}`}
style={{ borderLeftColor: getProviderColor(provider) }}
onContextMenu={(e) => {
e.preventDefault();
e.stopPropagation();
handleContextMenu(e, 'provider', provider);
}}
>
<button
type="button"
className={styles.collapseBtn}
onClick={() => toggleProviderCollapse(provider)}
aria-label={collapsed ? t('oauth_model_alias.diagram_expand') : t('oauth_model_alias.diagram_collapse')}
title={collapsed ? t('oauth_model_alias.diagram_expand') : t('oauth_model_alias.diagram_collapse')}
>
<span className={collapsed ? styles.chevronRight : styles.chevronDown} />
</button>
<span
className={styles.providerLabel}
style={{ color: getProviderColor(provider) }}
>
{provider}
</span>
<span className={styles.itemCount}>{sources.length}</span>
</div>
);
})}
</div>
{/* Column 2: Source Models (children per provider; hidden when provider collapsed) */}
<div
className={`${styles.column} ${styles.sources}`}
onContextMenu={(e) => {
e.preventDefault();
e.stopPropagation();
handleContextMenu(e, 'background');
onDragOver={handleDragOverSource}
onDragLeave={handleDragLeaveSource}
onDrop={handleDropOnSource}
onContextMenu={(e, type, data) => handleContextMenu(e, type, data)}
label={t('oauth_model_alias.diagram_source_models')}
/>
<AliasColumn
aliasNodes={aliasNodes}
aliasRefs={aliasRefs}
dropTargetAlias={dropTargetAlias}
draggedAlias={draggedAlias}
draggable={!!onUpdate}
onDragStart={handleDragStartAlias}
onDragEnd={() => {
setDraggedAlias(null);
setDropTargetSource(null);
}}
>
<div className={styles.columnHeader}>{t('oauth_model_alias.diagram_source_models')}</div>
{providerNodes.flatMap(({ provider, sources }) => {
if (collapsedProviders.has(provider)) return [];
return sources.map((source) => (
<div
key={source.id}
ref={(el) => {
if (el) sourceRefs.current.set(source.id, el);
else sourceRefs.current.delete(source.id);
}}
className={`${styles.item} ${styles.sourceItem} ${
draggedSource?.id === source.id ? styles.dragging : ''
} ${dropTargetSource === source.id ? styles.dropTarget : ''}`}
draggable={!!onUpdate}
onDragStart={(e) => handleDragStart(e, source)}
onDragEnd={() => {
setDraggedSource(null);
setDropTargetAlias(null);
}}
onDragOver={(e) => handleDragOverSource(e, source)}
onDragLeave={handleDragLeaveSource}
onDrop={(e) => handleDropOnSource(e, source)}
onContextMenu={(e) => {
e.preventDefault();
e.stopPropagation();
handleContextMenu(e, 'source', source.id);
}}
>
<span className={styles.itemName} title={source.name}>
{source.name}
</span>
<div
className={styles.dot}
style={{
background: getProviderColor(source.provider),
opacity: source.aliases.length > 0 ? 1 : 0.3
}}
/>
</div>
));
})}
</div>
onDragOver={handleDragOver}
onDragLeave={handleDragLeave}
onDrop={handleDrop}
onContextMenu={(e, type, data) => handleContextMenu(e, type, data)}
label={t('oauth_model_alias.diagram_aliases')}
/>
<div
className={`${styles.column} ${styles.aliases}`}
onContextMenu={(e) => {
e.preventDefault();
e.stopPropagation();
handleContextMenu(e, 'background');
<DiagramContextMenu
contextMenu={contextMenu}
t={t}
onRequestClose={() => setContextMenu(null)}
onAddAlias={handleAddAlias}
onRenameAlias={handleRenameClick}
onOpenAliasSettings={(alias) => {
setContextMenu(null);
setSettingsAlias(alias);
}}
>
<div className={styles.columnHeader}>{t('oauth_model_alias.diagram_aliases')}</div>
{aliasNodes.map((node) => (
<div
key={node.id}
ref={(el) => {
if (el) aliasRefs.current.set(node.id, el);
else aliasRefs.current.delete(node.id);
}}
className={`${styles.item} ${styles.aliasItem} ${
dropTargetAlias === node.alias ? styles.dropTarget : ''
} ${draggedAlias === node.alias ? styles.dragging : ''}`}
draggable={!!onUpdate}
onDragStart={(e) => handleDragStartAlias(e, node.alias)}
onDragEnd={() => {
setDraggedAlias(null);
setDropTargetSource(null);
}}
onDragOver={(e) => handleDragOver(e, node.alias)}
onDragLeave={handleDragLeave}
onDrop={(e) => handleDrop(e, node.alias)}
onContextMenu={(e) => {
e.preventDefault();
e.stopPropagation();
handleContextMenu(e, 'alias', node.alias);
}}
>
<div className={`${styles.dot} ${styles.dotLeft}`} />
<span className={styles.itemName} title={node.alias}>
{node.alias}
</span>
<span className={styles.itemCount}>{node.sources.length}</span>
</div>
))}
</div>
onDeleteAlias={handleDeleteClick}
onEditProvider={(provider) => {
setContextMenu(null);
onEditProvider?.(provider);
}}
onDeleteProvider={(provider) => {
setContextMenu(null);
onDeleteProvider?.(provider);
}}
onOpenSourceSettings={(sourceId) => {
setContextMenu(null);
setSettingsSourceId(sourceId);
}}
/>
{contextMenu &&
createPortal(
<div
ref={contextMenuRef}
className={styles.contextMenu}
style={{ top: contextMenu.y, left: contextMenu.x }}
onClick={(e) => e.stopPropagation()}
>
{contextMenu.type === 'background' && (
<div className={styles.menuItem} onClick={handleAddAlias}>
<span>{t('oauth_model_alias.diagram_add_alias')}</span>
</div>
)}
{contextMenu.type === 'alias' && renderAliasMenu()}
{contextMenu.type === 'provider' && renderProviderMenu()}
{contextMenu.type === 'source' && renderSourceMenu()}
</div>,
document.body
)}
<Modal
<RenameAliasModal
open={!!renameState}
t={t}
value={renameValue}
error={renameError}
onChange={(value) => {
setRenameValue(value);
setRenameError('');
}}
onClose={() => setRenameState(null)}
title={t('oauth_model_alias.diagram_rename_alias_title')}
width={400}
footer={
<>
<Button variant="secondary" onClick={() => setRenameState(null)}>
{t('common.cancel')}
</Button>
<Button onClick={handleRenameSubmit}>
{t('oauth_model_alias.diagram_rename_btn')}
</Button>
</>
}
>
<Input
label={t('oauth_model_alias.diagram_rename_alias_label')}
value={renameValue}
onChange={(e) => {
setRenameValue(e.target.value);
setRenameError('');
}}
onKeyDown={(e) => {
if (e.key === 'Enter') handleRenameSubmit();
}}
error={renameError}
placeholder={t('oauth_model_alias.diagram_rename_placeholder')}
autoFocus
/>
</Modal>
<Modal
onSubmit={handleRenameSubmit}
/>
<AddAliasModal
open={addAliasOpen}
t={t}
value={addAliasValue}
error={addAliasError}
onChange={(value) => {
setAddAliasValue(value);
setAddAliasError('');
}}
onClose={() => setAddAliasOpen(false)}
title={t('oauth_model_alias.diagram_add_alias_title')}
width={400}
footer={
<>
<Button variant="secondary" onClick={() => setAddAliasOpen(false)}>
{t('common.cancel')}
</Button>
<Button onClick={handleAddAliasSubmit}>
{t('oauth_model_alias.diagram_add_btn')}
</Button>
</>
}
>
<Input
label={t('oauth_model_alias.diagram_add_alias_label')}
value={addAliasValue}
onChange={(e) => {
setAddAliasValue(e.target.value);
setAddAliasError('');
}}
onKeyDown={(e) => {
if (e.key === 'Enter') handleAddAliasSubmit();
}}
error={addAliasError}
placeholder={t('oauth_model_alias.diagram_add_placeholder')}
autoFocus
/>
</Modal>
<Modal
onSubmit={handleAddAliasSubmit}
/>
<SettingsAliasModal
open={Boolean(settingsAlias)}
t={t}
alias={settingsAlias}
aliasNodes={aliasNodes}
onClose={() => setSettingsAlias(null)}
title={t('oauth_model_alias.diagram_settings_title', { alias: settingsAlias ?? '' })}
width={720}
footer={
<Button variant="secondary" onClick={() => setSettingsAlias(null)}>
{t('common.close')}
</Button>
}
>
{settingsAlias ? (
(() => {
const node = aliasNodes.find((n) => n.alias === settingsAlias);
if (!node || node.sources.length === 0) {
return <div className={styles.settingsEmpty}>{t('oauth_model_alias.diagram_settings_empty')}</div>;
}
return (
<div className={styles.settingsList}>
{node.sources.map((source) => {
const entry = source.aliases.find((item) => item.alias === settingsAlias);
const forkEnabled = entry?.fork === true;
return (
<div key={source.id} className={styles.settingsRow}>
<div className={styles.settingsNames}>
<span className={styles.settingsSource}>{source.name}</span>
<span className={styles.settingsArrow}></span>
<span className={styles.settingsAlias}>{settingsAlias}</span>
</div>
<div className={styles.settingsActions}>
<span className={styles.settingsLabel}>
{t('oauth_model_alias.alias_fork_label')}
</span>
<ToggleSwitch
checked={forkEnabled}
onChange={(value) => handleToggleFork(source.provider, source.name, settingsAlias, value)}
ariaLabel={t('oauth_model_alias.alias_fork_label')}
/>
<button
type="button"
className={styles.settingsDelete}
onClick={() => handleUnlinkSource(source.provider, source.name, settingsAlias)}
aria-label={t('oauth_model_alias.diagram_delete_link', {
provider: source.provider,
name: source.name
})}
title={t('oauth_model_alias.diagram_delete_link', {
provider: source.provider,
name: source.name
})}
>
<IconTrash2 size={14} />
</button>
</div>
</div>
);
})}
</div>
);
})()
) : null}
</Modal>
<Modal
onToggleFork={handleToggleFork}
onUnlink={handleUnlinkSource}
/>
<SettingsSourceModal
open={Boolean(settingsSourceId)}
t={t}
source={resolveSourceById(settingsSourceId)}
onClose={() => setSettingsSourceId(null)}
title={t('oauth_model_alias.diagram_settings_source_title')}
width={720}
footer={
<Button variant="secondary" onClick={() => setSettingsSourceId(null)}>
{t('common.close')}
</Button>
}
>
{settingsSourceId ? (
(() => {
const source = resolveSourceById(settingsSourceId);
if (!source || source.aliases.length === 0) {
return <div className={styles.settingsEmpty}>{t('oauth_model_alias.diagram_settings_empty')}</div>;
}
return (
<div className={styles.settingsList}>
{source.aliases.map((entry) => (
<div key={`${source.id}-${entry.alias}`} className={styles.settingsRow}>
<div className={styles.settingsNames}>
<span className={styles.settingsSource}>{source.name}</span>
<span className={styles.settingsArrow}></span>
<span className={styles.settingsAlias}>{entry.alias}</span>
</div>
<div className={styles.settingsActions}>
<span className={styles.settingsLabel}>
{t('oauth_model_alias.alias_fork_label')}
</span>
<ToggleSwitch
checked={entry.fork === true}
onChange={(value) =>
handleToggleFork(source.provider, source.name, entry.alias, value)
}
ariaLabel={t('oauth_model_alias.alias_fork_label')}
/>
<button
type="button"
className={styles.settingsDelete}
onClick={() => handleUnlinkSource(source.provider, source.name, entry.alias)}
aria-label={t('oauth_model_alias.diagram_delete_link', {
provider: source.provider,
name: source.name
})}
title={t('oauth_model_alias.diagram_delete_link', {
provider: source.provider,
name: source.name
})}
>
<IconTrash2 size={14} />
</button>
</div>
</div>
))}
</div>
);
})()
) : null}
</Modal>
onToggleFork={handleToggleFork}
onUnlink={handleUnlinkSource}
/>
</div>
);
});