From c31f004a9fd9f884f0b3365b7954e270bb5953b3 Mon Sep 17 00:00:00 2001
From: Taylor Bantle <taylor@dolthub.com>
Date: Mon, 27 Nov 2023 16:09:39 -0800
Subject: [PATCH 1/2] web: If one database automatically connect

---
 .../pageComponents/DatabasesPage/index.tsx    | 64 ++++++++++++-------
 1 file changed, 40 insertions(+), 24 deletions(-)

diff --git a/web/components/pageComponents/DatabasesPage/index.tsx b/web/components/pageComponents/DatabasesPage/index.tsx
index d459ed22..5719ed9e 100644
--- a/web/components/pageComponents/DatabasesPage/index.tsx
+++ b/web/components/pageComponents/DatabasesPage/index.tsx
@@ -1,42 +1,58 @@
 import CreateDatabase from "@components/CreateDatabase";
+import ErrorMsg from "@components/ErrorMsg";
+import Loader from "@components/Loader";
 import MainLayout from "@components/layouts/MainLayout";
 import Link from "@components/links/Link";
 import { useDatabasesQuery } from "@gen/graphql-types";
 import { database } from "@lib/urls";
 import { FaChevronRight } from "@react-icons/all-files/fa/FaChevronRight";
 import cx from "classnames";
+import { useRouter } from "next/router";
+import { useEffect } from "react";
 import css from "./index.module.css";
 
 export default function DatabasesPage() {
   const res = useDatabasesQuery();
+  const router = useRouter();
+
+  useEffect(() => {
+    if (!res.data) return;
+    if (res.data.databases.length === 1) {
+      const { href, as } = database({ databaseName: res.data.databases[0] });
+      router.push(href, as).catch(console.error);
+    }
+  }, [res.data?.databases]);
 
   return (
     <MainLayout>
-      <h1>Choose a database</h1>
-      <p className={css.desc}>
-        Choose an existing database or create a new database to get started.
-      </p>
-      {res.data?.databases.length ? (
-        <ul>
-          {res.data.databases.map(db => (
-            <li key={db}>
-              <Link {...database({ databaseName: db })}>
-                <div className={css.database}>
-                  <span>{db}</span>
-                  <span className={css.go}>
-                    Go <FaChevronRight />
-                  </span>
-                </div>
-              </Link>
-            </li>
-          ))}
-        </ul>
-      ) : (
-        <p className={cx(css.noDbs, css.database)}>
-          No databases found. Create a database to get started.
+      <Loader loaded={!res.loading}>
+        <h1>Choose a database</h1>
+        <p className={css.desc}>
+          Choose an existing database or create a new database to get started.
         </p>
-      )}
-      <CreateDatabase buttonClassName={css.button} />
+        {res.data?.databases.length ? (
+          <ul>
+            {res.data.databases.map(db => (
+              <li key={db}>
+                <Link {...database({ databaseName: db })}>
+                  <div className={css.database}>
+                    <span>{db}</span>
+                    <span className={css.go}>
+                      Go <FaChevronRight />
+                    </span>
+                  </div>
+                </Link>
+              </li>
+            ))}
+          </ul>
+        ) : (
+          <p className={cx(css.noDbs, css.database)}>
+            No databases found. Create a database to get started.
+          </p>
+        )}
+        <CreateDatabase buttonClassName={css.button} />
+        <ErrorMsg err={res.error} />
+      </Loader>
     </MainLayout>
   );
 }

From d447806762329a13ee0545f73384091997c60a59 Mon Sep 17 00:00:00 2001
From: Taylor Bantle <taylor@dolthub.com>
Date: Mon, 27 Nov 2023 16:21:37 -0800
Subject: [PATCH 2/2] web: Improve create/drop database behavior

---
 web/components/SqlDataTable/index.tsx      | 25 ++--------
 web/components/SqlDataTable/useSqlQuery.ts | 54 ++++++++++++++++++++++
 web/lib/refetchQueries.ts                  |  1 +
 3 files changed, 58 insertions(+), 22 deletions(-)
 create mode 100644 web/components/SqlDataTable/useSqlQuery.ts

diff --git a/web/components/SqlDataTable/index.tsx b/web/components/SqlDataTable/index.tsx
index d8e81476..0f731182 100644
--- a/web/components/SqlDataTable/index.tsx
+++ b/web/components/SqlDataTable/index.tsx
@@ -15,13 +15,12 @@ import {
 } from "@gen/graphql-types";
 import useSessionQueryHistory from "@hooks/useSessionQueryHistory";
 import { SqlQueryParams } from "@lib/params";
-import { isMutation } from "@lib/parseSqlQuery";
-import { refetchSqlUpdateQueriesCacheEvict } from "@lib/refetchQueries";
-import { useEffect, useState } from "react";
+import { useState } from "react";
 import SqlMessage from "./SqlMessage";
 import { isReadOnlyDatabaseRevisionError } from "./SqlMessage/utils";
 import WorkingDiff from "./WorkingDiff";
 import css from "./index.module.css";
+import useSqlQuery from "./useSqlQuery";
 
 type Props = {
   params: SqlQueryParams;
@@ -37,25 +36,7 @@ type InnerProps = Props & {
 };
 
 function Inner(props: InnerProps) {
-  const { addMutation } = useSessionQueryHistory();
-  const isMut = isMutation(props.params.q);
-
-  useEffect(() => {
-    if (!isMut) return;
-    addMutation(props.params.q);
-  }, [props.params.q]);
-
-  useEffect(() => {
-    if (!isMut) return;
-    if (props.gqlError) return;
-    // Need timeout here so that queries are not refetched before sql query has
-    // time to finish
-    setTimeout(() => {
-      props.client
-        .refetchQueries(refetchSqlUpdateQueriesCacheEvict)
-        .catch(console.error);
-    }, 300);
-  }, [props.gqlError, isMut, props.client]);
+  const isMut = useSqlQuery(props.params, props.client, props.gqlError);
 
   const msg = <SqlMessage {...props} rowsLen={props.rows?.length ?? 0} />;
   return (
diff --git a/web/components/SqlDataTable/useSqlQuery.ts b/web/components/SqlDataTable/useSqlQuery.ts
new file mode 100644
index 00000000..e2747b54
--- /dev/null
+++ b/web/components/SqlDataTable/useSqlQuery.ts
@@ -0,0 +1,54 @@
+import {
+  ApolloClient,
+  ApolloError,
+  NormalizedCacheObject,
+} from "@apollo/client";
+import useSessionQueryHistory from "@hooks/useSessionQueryHistory";
+import { SqlQueryParams } from "@lib/params";
+import { isMutation } from "@lib/parseSqlQuery";
+import { refetchSqlUpdateQueriesCacheEvict } from "@lib/refetchQueries";
+import { databases } from "@lib/urls";
+import { useRouter } from "next/router";
+import { useEffect } from "react";
+
+export default function useSqlQuery(
+  params: SqlQueryParams,
+  client: ApolloClient<NormalizedCacheObject>,
+  gqlError?: ApolloError,
+): boolean {
+  const router = useRouter();
+  const { addMutation } = useSessionQueryHistory();
+  const isMut = isMutation(params.q);
+
+  useEffect(() => {
+    if (!isMut) return;
+    addMutation(params.q);
+  }, [params.q]);
+
+  useEffect(() => {
+    if (!isMut) return;
+    if (gqlError) return;
+    // Need timeout here so that queries are not refetched before sql query has
+    // time to finish
+    setTimeout(() => {
+      client
+        .refetchQueries(refetchSqlUpdateQueriesCacheEvict)
+        .catch(console.error);
+    }, 300);
+  }, [gqlError, isMut, client]);
+
+  useEffect(() => {
+    if (gqlError) {
+      return;
+    }
+    const lower = params.q.toLowerCase();
+    if (
+      lower.includes("drop database") &&
+      lower.includes(params.databaseName)
+    ) {
+      router.push(databases.hrefPathname()).catch(console.error);
+    }
+  }, [params.q, params.databaseName, gqlError]);
+
+  return isMut;
+}
diff --git a/web/lib/refetchQueries.ts b/web/lib/refetchQueries.ts
index bf870558..3d9aabde 100644
--- a/web/lib/refetchQueries.ts
+++ b/web/lib/refetchQueries.ts
@@ -122,6 +122,7 @@ export const refetchSqlUpdateQueriesCacheEvict: RefetchOptions = {
       "schemaDiff",
       "doltProcedures",
       "doltSchemas",
+      "databases",
     ].forEach(fieldName => {
       cache.evict({ fieldName });
     });