refactor

packages/datalib/src/PerspectiveTreeNode.ts

@@ -0,0 +1,102 @@
+import { ColumnInfo, DatabaseInfo, ForeignKeyInfo, TableInfo } from 'dbgate-types';
+import { clearConfigCache } from 'prettier';
+import { ChangePerspectiveConfigFunc, PerspectiveConfig } from './PerspectiveConfig';
+
+export abstract class PerspectiveTreeNode {
+  constructor(
+    public config: PerspectiveConfig,
+    public setConfig: ChangePerspectiveConfigFunc,
+    public parentNode: PerspectiveTreeNode
+  ) {}
+  abstract get title();
+  abstract get codeName();
+  abstract get props();
+  abstract get isExpandable();
+  abstract get childNodes(): PerspectiveTreeNode[];
+  get uniqueName() {
+    if (this.parentNode) return `${this.parentNode.uniqueName}.${this.codeName}`;
+    return this.codeName;
+  }
+  get level() {
+    if (this.parentNode) return this.parentNode.level + 1;
+    return 0;
+  }
+  get isExpanded() {
+    return this.config.expandedColumns.includes(this.uniqueName);
+  }
+
+  toggleExpanded(value?: boolean) {
+    this.includeInColumnSet('expandedColumns', this.uniqueName, value == null ? !this.isExpanded : value);
+  }
+
+  includeInColumnSet(field: keyof PerspectiveConfig, uniqueName: string, isIncluded: boolean) {
+    if (isIncluded) {
+      this.setConfig(cfg => ({
+        ...cfg,
+        [field]: [...(cfg[field] || []), uniqueName],
+      }));
+    } else {
+      this.setConfig(cfg => ({
+        ...cfg,
+        [field]: (cfg[field] || []).filter(x => x != uniqueName),
+      }));
+    }
+  }
+}
+
+export class PerspectiveTableColumnNode extends PerspectiveTreeNode {
+  foreignKey: ForeignKeyInfo;
+  constructor(
+    public column: ColumnInfo,
+    public table: TableInfo,
+    public db: DatabaseInfo,
+    config: PerspectiveConfig,
+    setConfig: ChangePerspectiveConfigFunc,
+    parentColumn: PerspectiveTreeNode
+  ) {
+    super(config, setConfig, parentColumn);
+
+    this.foreignKey =
+      table.foreignKeys &&
+      table.foreignKeys.find(fk => fk.columns.length == 1 && fk.columns[0].columnName == column.columnName);
+  }
+
+  get codeName() {
+    return this.column.columnName;
+  }
+
+  get title() {
+    return this.column.columnName;
+  }
+
+  get props() {
+    return this.column;
+  }
+
+  get isExpandable() {
+    return !!this.foreignKey;
+  }
+
+  get childNodes(): PerspectiveTreeNode[] {
+    if (!this.foreignKey) return [];
+    const tbl = this?.db?.tables?.find(
+      x => x.pureName == this.foreignKey?.refTableName && x.schemaName == this.foreignKey?.refSchemaName
+    );
+    return getTableChildPerspectiveNodes(tbl, this.db, this.config, this.setConfig, this);
+    // return (
+    //   tbl?.columns?.map(col => new PerspectiveTableColumnNode(col, tbl, this.db, this.config, this.setConfig, this)) ||
+    //   []
+    // );
+  }
+}
+
+export function getTableChildPerspectiveNodes(
+  table: TableInfo,
+  db: DatabaseInfo,
+  config: PerspectiveConfig,
+  setConfig: ChangePerspectiveConfigFunc,
+  parentColumn: PerspectiveTreeNode
+) {
+  if (!table) return [];
+  return table.columns.map(col => new PerspectiveTableColumnNode(col, table, db, config, setConfig, parentColumn));
+}