@jackwener/opencli 1.5.5 → 1.5.7

package/src/clis/notebooklm/source-get.ts

@@ -0,0 +1,60 @@
+import { cli, Strategy } from '../../registry.js';
+import type { IPage } from '../../types.js';
+import { EmptyResultError } from '../../errors.js';
+import { NOTEBOOKLM_DOMAIN, NOTEBOOKLM_SITE } from './shared.js';
+import {
+  ensureNotebooklmNotebookBinding,
+  findNotebooklmSourceRow,
+  getNotebooklmPageState,
+  listNotebooklmSourcesFromPage,
+  listNotebooklmSourcesViaRpc,
+  requireNotebooklmSession,
+} from './utils.js';
+
+cli({
+  site: NOTEBOOKLM_SITE,
+  name: 'source-get',
+  description: 'Get one source from the currently opened NotebookLM notebook by id or title',
+  domain: NOTEBOOKLM_DOMAIN,
+  strategy: Strategy.COOKIE,
+  browser: true,
+  navigateBefore: false,
+  args: [
+    {
+      name: 'source',
+      positional: true,
+      required: true,
+      help: 'Source id or title from the current notebook',
+    },
+  ],
+  columns: ['title', 'id', 'type', 'size', 'created_at', 'updated_at', 'url', 'source'],
+  func: async (page: IPage, kwargs) => {
+    await ensureNotebooklmNotebookBinding(page);
+    await requireNotebooklmSession(page);
+    const state = await getNotebooklmPageState(page);
+    if (state.kind !== 'notebook') {
+      throw new EmptyResultError(
+        'opencli notebooklm source-get',
+        'Open a specific NotebookLM notebook tab first, then retry.',
+      );
+    }
+
+    const rpcRows = await listNotebooklmSourcesViaRpc(page).catch(() => []);
+    const rows = rpcRows.length > 0 ? rpcRows : await listNotebooklmSourcesFromPage(page);
+    if (rows.length === 0) {
+      throw new EmptyResultError(
+        'opencli notebooklm source-get',
+        'No NotebookLM sources were found on the current page.',
+      );
+    }
+
+    const query = typeof kwargs.source === 'string' ? kwargs.source : String(kwargs.source ?? '');
+    const matched = findNotebooklmSourceRow(rows, query);
+    if (matched) return [matched];
+
+    throw new EmptyResultError(
+      'opencli notebooklm source-get',
+      `Source "${query}" was not found in the current notebook.`,
+    );
+  },
+});

package/src/clis/notebooklm/source-guide.test.ts

@@ -0,0 +1,121 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+const {
+  mockListNotebooklmSourcesViaRpc,
+  mockListNotebooklmSourcesFromPage,
+  mockGetNotebooklmSourceGuideViaRpc,
+  mockGetNotebooklmPageState,
+  mockRequireNotebooklmSession,
+} = vi.hoisted(() => ({
+  mockListNotebooklmSourcesViaRpc: vi.fn(),
+  mockListNotebooklmSourcesFromPage: vi.fn(),
+  mockGetNotebooklmSourceGuideViaRpc: vi.fn(),
+  mockGetNotebooklmPageState: vi.fn(),
+  mockRequireNotebooklmSession: vi.fn(),
+}));
+
+vi.mock('./utils.js', async () => {
+  const actual = await vi.importActual<typeof import('./utils.js')>('./utils.js');
+  return {
+    ...actual,
+    listNotebooklmSourcesViaRpc: mockListNotebooklmSourcesViaRpc,
+    listNotebooklmSourcesFromPage: mockListNotebooklmSourcesFromPage,
+    getNotebooklmSourceGuideViaRpc: mockGetNotebooklmSourceGuideViaRpc,
+    getNotebooklmPageState: mockGetNotebooklmPageState,
+    requireNotebooklmSession: mockRequireNotebooklmSession,
+  };
+});
+
+import { getRegistry } from '../../registry.js';
+import './source-guide.js';
+
+describe('notebooklm source-guide', () => {
+  const command = getRegistry().get('notebooklm/source-guide');
+
+  beforeEach(() => {
+    mockListNotebooklmSourcesViaRpc.mockReset();
+    mockListNotebooklmSourcesFromPage.mockReset();
+    mockGetNotebooklmSourceGuideViaRpc.mockReset();
+    mockGetNotebooklmPageState.mockReset();
+    mockRequireNotebooklmSession.mockReset();
+    mockRequireNotebooklmSession.mockResolvedValue(undefined);
+    mockGetNotebooklmPageState.mockResolvedValue({
+      url: 'https://notebooklm.google.com/notebook/nb-demo',
+      title: 'Browser Automation',
+      hostname: 'notebooklm.google.com',
+      kind: 'notebook',
+      notebookId: 'nb-demo',
+      loginRequired: false,
+      notebookCount: 1,
+    });
+  });
+
+  it('returns source guide for a source matched from rpc source rows', async () => {
+    mockListNotebooklmSourcesViaRpc.mockResolvedValue([
+      {
+        id: 'src-yt',
+        notebook_id: 'nb-demo',
+        title: 'Video Source',
+        url: 'https://notebooklm.google.com/notebook/nb-demo',
+        source: 'rpc',
+        type: 'youtube',
+        type_code: 9,
+      },
+    ]);
+    mockGetNotebooklmSourceGuideViaRpc.mockResolvedValue({
+      source_id: 'src-yt',
+      notebook_id: 'nb-demo',
+      title: 'Video Source',
+      type: 'youtube',
+      summary: 'Guide summary.',
+      keywords: ['AI', 'agents'],
+      source: 'rpc',
+    });
+
+    const result = await command!.func!({} as any, { source: 'src-yt' });
+
+    expect(result).toEqual([
+      {
+        source_id: 'src-yt',
+        notebook_id: 'nb-demo',
+        title: 'Video Source',
+        type: 'youtube',
+        summary: 'Guide summary.',
+        keywords: ['AI', 'agents'],
+        source: 'rpc',
+      },
+    ]);
+  });
+
+  it('matches by title from dom rows when rpc source list is unavailable', async () => {
+    mockListNotebooklmSourcesViaRpc.mockResolvedValue([]);
+    mockListNotebooklmSourcesFromPage.mockResolvedValue([
+      {
+        id: 'src-1',
+        notebook_id: 'nb-demo',
+        title: 'Example Source',
+        url: 'https://notebooklm.google.com/notebook/nb-demo',
+        source: 'current-page',
+      },
+    ]);
+    mockGetNotebooklmSourceGuideViaRpc.mockResolvedValue({
+      source_id: 'src-1',
+      notebook_id: 'nb-demo',
+      title: 'Example Source',
+      type: null,
+      summary: 'Guide summary.',
+      keywords: ['topic'],
+      source: 'rpc',
+    });
+
+    const result = await command!.func!({} as any, { source: 'example source' });
+
+    expect(result).toEqual([
+      expect.objectContaining({
+        source_id: 'src-1',
+        title: 'Example Source',
+        summary: 'Guide summary.',
+      }),
+    ]);
+  });
+});

package/src/clis/notebooklm/source-guide.ts

@@ -0,0 +1,69 @@
+import { cli, Strategy } from '../../registry.js';
+import type { IPage } from '../../types.js';
+import { EmptyResultError } from '../../errors.js';
+import { NOTEBOOKLM_DOMAIN, NOTEBOOKLM_SITE } from './shared.js';
+import {
+  ensureNotebooklmNotebookBinding,
+  findNotebooklmSourceRow,
+  getNotebooklmPageState,
+  getNotebooklmSourceGuideViaRpc,
+  listNotebooklmSourcesFromPage,
+  listNotebooklmSourcesViaRpc,
+  requireNotebooklmSession,
+} from './utils.js';
+
+cli({
+  site: NOTEBOOKLM_SITE,
+  name: 'source-guide',
+  description: 'Get the guide summary and keywords for one source in the currently opened NotebookLM notebook',
+  domain: NOTEBOOKLM_DOMAIN,
+  strategy: Strategy.COOKIE,
+  browser: true,
+  navigateBefore: false,
+  args: [
+    {
+      name: 'source',
+      positional: true,
+      required: true,
+      help: 'Source id or title from the current notebook',
+    },
+  ],
+  columns: ['source_id', 'notebook_id', 'title', 'type', 'summary', 'keywords', 'source'],
+  func: async (page: IPage, kwargs) => {
+    await ensureNotebooklmNotebookBinding(page);
+    await requireNotebooklmSession(page);
+    const state = await getNotebooklmPageState(page);
+    if (state.kind !== 'notebook') {
+      throw new EmptyResultError(
+        'opencli notebooklm source-guide',
+        'Open a specific NotebookLM notebook tab first, then retry.',
+      );
+    }
+
+    const rpcRows = await listNotebooklmSourcesViaRpc(page).catch(() => []);
+    const rows = rpcRows.length > 0 ? rpcRows : await listNotebooklmSourcesFromPage(page);
+    if (rows.length === 0) {
+      throw new EmptyResultError(
+        'opencli notebooklm source-guide',
+        'No NotebookLM sources were found on the current page.',
+      );
+    }
+
+    const query = typeof kwargs.source === 'string' ? kwargs.source : String(kwargs.source ?? '');
+    const matched = findNotebooklmSourceRow(rows, query);
+    if (!matched) {
+      throw new EmptyResultError(
+        'opencli notebooklm source-guide',
+        `Source "${query}" was not found in the current notebook.`,
+      );
+    }
+
+    const guide = await getNotebooklmSourceGuideViaRpc(page, matched).catch(() => null);
+    if (guide) return [guide];
+
+    throw new EmptyResultError(
+      'opencli notebooklm source-guide',
+      `NotebookLM guide was not available for source "${matched.title}".`,
+    );
+  },
+});

package/src/clis/notebooklm/source-list.ts

@@ -0,0 +1,45 @@
+import { cli, Strategy } from '../../registry.js';
+import type { IPage } from '../../types.js';
+import { EmptyResultError } from '../../errors.js';
+import { NOTEBOOKLM_DOMAIN, NOTEBOOKLM_SITE } from './shared.js';
+import {
+  ensureNotebooklmNotebookBinding,
+  getNotebooklmPageState,
+  listNotebooklmSourcesFromPage,
+  listNotebooklmSourcesViaRpc,
+  requireNotebooklmSession,
+} from './utils.js';
+
+cli({
+  site: NOTEBOOKLM_SITE,
+  name: 'source-list',
+  description: 'List sources for the currently opened NotebookLM notebook',
+  domain: NOTEBOOKLM_DOMAIN,
+  strategy: Strategy.COOKIE,
+  browser: true,
+  navigateBefore: false,
+  args: [],
+  columns: ['title', 'id', 'type', 'size', 'created_at', 'updated_at', 'url', 'source'],
+  func: async (page: IPage) => {
+    await ensureNotebooklmNotebookBinding(page);
+    await requireNotebooklmSession(page);
+    const state = await getNotebooklmPageState(page);
+    if (state.kind !== 'notebook') {
+      throw new EmptyResultError(
+        'opencli notebooklm source-list',
+        'Open a specific NotebookLM notebook tab first, then retry.',
+      );
+    }
+
+    const rpcRows = await listNotebooklmSourcesViaRpc(page).catch(() => []);
+    if (rpcRows.length > 0) return rpcRows;
+
+    const domRows = await listNotebooklmSourcesFromPage(page);
+    if (domRows.length > 0) return domRows;
+
+    throw new EmptyResultError(
+      'opencli notebooklm source-list',
+      'No NotebookLM sources were found on the current page.',
+    );
+  },
+});

package/src/clis/notebooklm/status.ts

@@ -0,0 +1,34 @@
+import { cli, Strategy } from '../../registry.js';
+import type { IPage } from '../../types.js';
+import { NOTEBOOKLM_DOMAIN, NOTEBOOKLM_HOME_URL, NOTEBOOKLM_SITE } from './shared.js';
+import { ensureNotebooklmNotebookBinding, getNotebooklmPageState } from './utils.js';
+
+cli({
+  site: NOTEBOOKLM_SITE,
+  name: 'status',
+  description: 'Check NotebookLM page availability and login state in the current Chrome session',
+  domain: NOTEBOOKLM_DOMAIN,
+  strategy: Strategy.COOKIE,
+  browser: true,
+  navigateBefore: false,
+  args: [],
+  columns: ['status', 'login', 'page', 'url', 'title', 'notebooks'],
+  func: async (page: IPage) => {
+    await ensureNotebooklmNotebookBinding(page);
+    const currentUrl = await page.getCurrentUrl?.().catch(() => null);
+    if (!currentUrl || !currentUrl.includes(NOTEBOOKLM_DOMAIN)) {
+      await page.goto(NOTEBOOKLM_HOME_URL);
+      await page.wait(2);
+    }
+
+    const state = await getNotebooklmPageState(page);
+    return [{
+      status: state.hostname === NOTEBOOKLM_DOMAIN ? 'Connected' : 'Unavailable',
+      login: state.loginRequired ? 'Required' : 'OK',
+      page: state.kind,
+      url: state.url,
+      title: state.title,
+      notebooks: state.notebookCount,
+    }];
+  },
+});

package/src/clis/notebooklm/summary.test.ts

@@ -0,0 +1,94 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+const {
+  mockReadNotebooklmSummaryFromPage,
+  mockGetNotebooklmSummaryViaRpc,
+  mockGetNotebooklmPageState,
+  mockRequireNotebooklmSession,
+} = vi.hoisted(() => ({
+  mockReadNotebooklmSummaryFromPage: vi.fn(),
+  mockGetNotebooklmSummaryViaRpc: vi.fn(),
+  mockGetNotebooklmPageState: vi.fn(),
+  mockRequireNotebooklmSession: vi.fn(),
+}));
+
+vi.mock('./utils.js', async () => {
+  const actual = await vi.importActual<typeof import('./utils.js')>('./utils.js');
+  return {
+    ...actual,
+    readNotebooklmSummaryFromPage: mockReadNotebooklmSummaryFromPage,
+    getNotebooklmSummaryViaRpc: mockGetNotebooklmSummaryViaRpc,
+    getNotebooklmPageState: mockGetNotebooklmPageState,
+    requireNotebooklmSession: mockRequireNotebooklmSession,
+  };
+});
+
+import { getRegistry } from '../../registry.js';
+import './summary.js';
+
+describe('notebooklm summary', () => {
+  const command = getRegistry().get('notebooklm/summary');
+
+  beforeEach(() => {
+    mockReadNotebooklmSummaryFromPage.mockReset();
+    mockGetNotebooklmSummaryViaRpc.mockReset();
+    mockGetNotebooklmPageState.mockReset();
+    mockRequireNotebooklmSession.mockReset();
+    mockRequireNotebooklmSession.mockResolvedValue(undefined);
+    mockGetNotebooklmPageState.mockResolvedValue({
+      url: 'https://notebooklm.google.com/notebook/nb-demo',
+      title: 'Browser Automation',
+      hostname: 'notebooklm.google.com',
+      kind: 'notebook',
+      notebookId: 'nb-demo',
+      loginRequired: false,
+      notebookCount: 1,
+    });
+  });
+
+  it('returns the current notebook summary from the visible page first', async () => {
+    mockReadNotebooklmSummaryFromPage.mockResolvedValue({
+      notebook_id: 'nb-demo',
+      title: 'Browser Automation',
+      summary: 'A concise notebook summary.',
+      url: 'https://notebooklm.google.com/notebook/nb-demo',
+      source: 'summary-dom',
+    });
+
+    const result = await command!.func!({} as any, {});
+
+    expect(result).toEqual([
+      {
+        notebook_id: 'nb-demo',
+        title: 'Browser Automation',
+        summary: 'A concise notebook summary.',
+        url: 'https://notebooklm.google.com/notebook/nb-demo',
+        source: 'summary-dom',
+      },
+    ]);
+    expect(mockGetNotebooklmSummaryViaRpc).not.toHaveBeenCalled();
+  });
+
+  it('falls back to rpc summary extraction when no visible summary block is found', async () => {
+    mockReadNotebooklmSummaryFromPage.mockResolvedValue(null);
+    mockGetNotebooklmSummaryViaRpc.mockResolvedValue({
+      notebook_id: 'nb-demo',
+      title: 'Browser Automation',
+      summary: 'Summary recovered from rpc.',
+      url: 'https://notebooklm.google.com/notebook/nb-demo',
+      source: 'rpc',
+    });
+
+    const result = await command!.func!({} as any, {});
+
+    expect(result).toEqual([
+      {
+        notebook_id: 'nb-demo',
+        title: 'Browser Automation',
+        summary: 'Summary recovered from rpc.',
+        url: 'https://notebooklm.google.com/notebook/nb-demo',
+        source: 'rpc',
+      },
+    ]);
+  });
+});

package/src/clis/notebooklm/summary.ts

@@ -0,0 +1,45 @@
+import { cli, Strategy } from '../../registry.js';
+import type { IPage } from '../../types.js';
+import { EmptyResultError } from '../../errors.js';
+import { NOTEBOOKLM_DOMAIN, NOTEBOOKLM_SITE } from './shared.js';
+import {
+  ensureNotebooklmNotebookBinding,
+  getNotebooklmPageState,
+  getNotebooklmSummaryViaRpc,
+  readNotebooklmSummaryFromPage,
+  requireNotebooklmSession,
+} from './utils.js';
+
+cli({
+  site: NOTEBOOKLM_SITE,
+  name: 'summary',
+  description: 'Get the summary block from the currently opened NotebookLM notebook',
+  domain: NOTEBOOKLM_DOMAIN,
+  strategy: Strategy.COOKIE,
+  browser: true,
+  navigateBefore: false,
+  args: [],
+  columns: ['title', 'summary', 'source', 'url'],
+  func: async (page: IPage) => {
+    await ensureNotebooklmNotebookBinding(page);
+    await requireNotebooklmSession(page);
+    const state = await getNotebooklmPageState(page);
+    if (state.kind !== 'notebook') {
+      throw new EmptyResultError(
+        'opencli notebooklm summary',
+        'Open a specific NotebookLM notebook tab first, then retry.',
+      );
+    }
+
+    const domSummary = await readNotebooklmSummaryFromPage(page);
+    if (domSummary) return [domSummary];
+
+    const rpcSummary = await getNotebooklmSummaryViaRpc(page).catch(() => null);
+    if (rpcSummary) return [rpcSummary];
+
+    throw new EmptyResultError(
+      'opencli notebooklm summary',
+      'NotebookLM summary was not found on the current page.',
+    );
+  },
+});