Fix nested problem adding database starting with db-*

Author: aeisenberg

extensions/ql-vscode/src/databases-ui.ts

@@ -594,7 +594,8 @@ export class DatabaseUI extends DisposableObject {
     if ((await fs.stat(dbPath)).isFile()) {
       dbPath = path.dirname(dbPath);
     }
-    if (path.basename(dbPath).startsWith('db-')) {
+
+    if (isLikelyDbFolder(dbPath)) {
       dbPath = path.dirname(dbPath);
     }
     return Uri.file(dbPath);
@@ -609,3 +610,8 @@ export class DatabaseUI extends DisposableObject {
     }
   }
 }
+
+const dbRegeEx = /^db-(javascript|go|cpp|java|python)$/;
+function isLikelyDbFolder(dbPath: string) {
+  return path.basename(dbPath).match(dbRegeEx);
+}

extensions/ql-vscode/src/vscode-tests/no-workspace/databases-ui.test.ts

@@ -25,7 +25,7 @@ describe('databases-ui', () => {
 
     it('should choose parent direcory when db-* is selected', async () => {
       const dir = tmp.dirSync().name;
-      const dbDir = path.join(dir, 'db-hucairz');
+      const dbDir = path.join(dir, 'db-javascript');
       await fs.mkdirs(dbDir);
 
       const uri = await fixDbUri(Uri.file(dbDir));
@@ -34,14 +34,26 @@ describe('databases-ui', () => {
 
     it('should choose parent\'s parent direcory when file selected is in db-*', async () => {
       const dir = tmp.dirSync().name;
-      const dbDir = path.join(dir, 'db-hucairz');
+      const dbDir = path.join(dir, 'db-javascript');
       const file = path.join(dbDir, 'nested');
       await fs.mkdirs(dbDir);
       await fs.createFile(file);
 
       const uri = await fixDbUri(Uri.file(file));
       expect(uri.toString()).to.eq(Uri.file(dir).toString());
     });
-  });
 
+    it('should handle a parent whose name is db-*', async () => {
+      // fixes https://github.com/github/vscode-codeql/issues/482
+      const dir = tmp.dirSync().name;
+      const parentDir = path.join(dir, 'db-hucairz');
+      const dbDir = path.join(parentDir, 'db-javascript');
+      const file = path.join(dbDir, 'nested');
+      await fs.mkdirs(dbDir);
+      await fs.createFile(file);
+
+      const uri = await fixDbUri(Uri.file(file));
+      expect(uri.toString()).to.eq(Uri.file(parentDir).toString());
+    });
+  });
 });