Function getSchemaOrTableTreeOptions

frontend/src/utils/v1/databaseResource.ts:119–177

({
  database,
  filterValueList,
  includeColumn,
}: {
  database: Database;
  filterValueList?: string[];
  includeColumn: boolean;
})

};

export const getSchemaOrTableTreeOptions = ({
  database,
  filterValueList,
  includeColumn,
}: {
  database: Database;
  filterValueList?: string[];
  includeColumn: boolean;
}) => {
  if (getInstanceResource(database).engine === Engine.MONGODB) {
    // do not support table level select for MongoDB.
    return [];
  }
  const databaseMetadata = useAppStore
    .getState()
    .getCachedDatabaseMetadata(database.name);
  if (!databaseMetadata) {
    return undefined;
  }
  if (hasSchemaProperty(getInstanceResource(database).engine)) {
    const schemaNodes = databaseMetadata.schemas.map(
      (schema): DatabaseTreeOption<"schemas"> => {
        const value = `${database.name}/schemas/${schema.name}`;
        const schemaNode: DatabaseTreeOption<"schemas"> = {
          level: "schemas",
          value,
          label: schema.name,
          children: getTableTreeOptions({
            prefix: value,
            tableList: schema.tables,
            filterValueList,
            includeColumn,
          }),
          isLeaf: true,
        };
        if (schemaNode.children && schemaNode.children.length > 0) {
          schemaNode.isLeaf = false;
        }
        return schemaNode;
      }
    );
    return filterValueList
      ? schemaNodes.filter((node) =>
          filterValueList.some(
            (value) => value.split("/tables/")[0] === node.value
          )
        )
      : schemaNodes;
  } else {
    return getTableTreeOptions({
      prefix: `${database.name}/schemas/`,
      tableList: flatten(
        databaseMetadata.schemas.map((schema) => schema.tables)
      ),
      filterValueList,
      includeColumn,
    });
  }