feat(client): Add SQLite file extension registration

src/filetypes.ts

@@ -67,3 +67,20 @@ export function addOrcFileType(
   });
   return docRegistry.getFileType(name)!;
 }
+
+export function addSqliteFileType(
+  docRegistry: DocumentRegistry,
+  options: Partial<DocumentRegistry.IFileType> = {},
+): DocumentRegistry.IFileType {
+  const name = "sqlite";
+  docRegistry.addFileType({
+    ...options,
+    name,
+    displayName: "SQLite",
+    mimeTypes: ["application/vnd.sqlite3"],
+    extensions: [".sqlite", ".sqlite3", ".db", ".db3", ".s3db", ".sl3"],
+    contentType: "file",
+    fileFormat: "base64",
+  });
+  return docRegistry.getFileType(name)!;
+}

src/index.ts

@@ -8,8 +8,20 @@ import type * as services from "@jupyterlab/services";
 import type { Contents } from "@jupyterlab/services";
 import type { DataGrid } from "@lumino/datagrid";
 
-import { addAvroFileType, addIpcFileType, addOrcFileType, addParquetFileType } from "./filetypes";
-import { getArrowIPCIcon, getAvroIcon, getORCIcon, getParquetIcon } from "./labicons";
+import {
+  addAvroFileType,
+  addIpcFileType,
+  addOrcFileType,
+  addParquetFileType,
+  addSqliteFileType,
+} from "./filetypes";
+import {
+  getArrowIPCIcon,
+  getAvroIcon,
+  getORCIcon,
+  getParquetIcon,
+  getSqliteIcon,
+} from "./labicons";
 import { ArrowGridViewerFactory } from "./widget";
 import type { ArrowGridViewer, ITextRenderConfig } from "./widget";
 
@@ -114,12 +126,13 @@ function activateArrowGrid(
   let avo_ft = addAvroFileType(app.docRegistry, { icon: getAvroIcon(isLight) });
   let ipc_ft = addIpcFileType(app.docRegistry, { icon: getArrowIPCIcon(isLight) });
   let orc_ft = addOrcFileType(app.docRegistry, { icon: getORCIcon(isLight) });
+  let sqlite_ft = addSqliteFileType(app.docRegistry, { icon: getSqliteIcon(isLight) });
 
   const factory = new ArrowGridViewerFactory({
     name: factory_arrow,
     label: trans.__("Arrow Dataframe Viewer"),
-    fileTypes: [csv_ft.name, avo_ft.name, prq_ft.name, ipc_ft.name, orc_ft.name],
-    defaultFor: [csv_ft.name, avo_ft.name, prq_ft.name, ipc_ft.name, orc_ft.name],
+    fileTypes: [csv_ft.name, avo_ft.name, prq_ft.name, ipc_ft.name, orc_ft.name, sqlite_ft.name],
+    defaultFor: [csv_ft.name, avo_ft.name, prq_ft.name, ipc_ft.name, orc_ft.name, sqlite_ft.name],
     readOnly: true,
     translator,
     contentProviderId: NOOP_CONTENT_PROVIDER_ID,
@@ -175,6 +188,7 @@ function activateArrowGrid(
     avo_ft = addAvroFileType(app.docRegistry, { icon: getAvroIcon(isLightNew) });
     ipc_ft = addIpcFileType(app.docRegistry, { icon: getArrowIPCIcon(isLightNew) });
     orc_ft = addOrcFileType(app.docRegistry, { icon: getORCIcon(isLightNew) });
+    sqlite_ft = addSqliteFileType(app.docRegistry, { icon: getSqliteIcon(isLightNew) });
   };
   if (themeManager) {
     themeManager.themeChanged.connect((_, args) => {

src/labicons.ts

@@ -7,6 +7,8 @@ import orcLightSvg from "../style/icons/orc.svg";
 import orcDarkSvg from "../style/icons/orc_dark.svg";
 import parquetSvgLight from "../style/icons/parquet.svg";
 import parquetSvgDark from "../style/icons/parquet_dark.svg";
+import sqliteSvgLight from "../style/icons/sqlite.svg";
+import sqliteSvgDark from "../style/icons/sqlite_dark.svg";
 
 export const getLabIcon = (labIconName: string, iconSvg: string) => {
   return new LabIcon({
@@ -28,3 +30,6 @@ export const getORCIcon = (isLight: boolean) => {
 export const getAvroIcon = (isLight: boolean) => {
   return getLabIcon("avro", isLight ? avroSvg : avroSvg);
 };
+export const getSqliteIcon = (isLight: boolean) => {
+  return getLabIcon("sqlite", isLight ? sqliteSvgLight : sqliteSvgDark);
+};