Function SchemaProvider

src/context/schema-provider.tsx:66–170

({ children }: Readonly<PropsWithChildren>)

}

export function SchemaProvider({ children }: Readonly<PropsWithChildren>) {
  const { updateTableList, updateTableSchema } = useAutoComplete();
  const [error, setError] = useState<string>();
  const [loading, setLoading] = useState(true);
  const { databaseDriver, extensions } = useStudioContext();

  const [schema, setSchema] = useState<DatabaseSchemas>({});
  const [currentSchema, setCurrentSchema] = useState<DatabaseSchemaItem[]>([]);
  const [currentSchemaName, setCurrentSchemaName] = useState(
    () => databaseDriver.getFlags().defaultSchema
  );

  const fetchSchema = useCallback(
    (refresh?: boolean) => {
      if (refresh) {
        setLoading(true);
      }

      const getSchema = async () => {
        const schemaResult = await databaseDriver.schemas();
        let selectedSchema = null;

        // If databasse driver support get current schema,
        // We will use it to override the default schema
        if (databaseDriver.getFlags().supportUseStatement) {
          selectedSchema = await databaseDriver.getCurrentSchema();
        }

        return { schemaResult, selectedSchema };
      };

      getSchema()
        .then((result) => {
          setSchema(result.schemaResult);

          if (result.selectedSchema) {
            setCurrentSchemaName(result.selectedSchema);
          }

          setError(undefined);
          setLoading(false);
        })
        .catch((e) => {
          setError(e.message);
          setLoading(false);
        });
    },
    [databaseDriver, setError]
  );

  useEffect(() => {
    if (schema[currentSchemaName]) {
      setCurrentSchema(schema[currentSchemaName]);
    }
  }, [currentSchemaName, schema, setCurrentSchema]);

  /**
   * Triggered when re-fetching the database schema.