evan/pi-statusline
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: type ModuleContext and index.ts functions with SDK types

Browse Source
This commit is contained in:
Evan Reichard
2026-05-03 21:28:15 -04:00
parent 26ba42f2b9
commit cf7ad047d3
2 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
index.ts
View File

@@ -1,4 +1,4 @@
import { AuthStorage, type AuthCredential, type ExtensionAPI, type ExtensionContext, type ExtensionCommandContext, type SessionStartEvent, type SessionShutdownEvent, type AgentEndEvent } from "@mariozechner/pi-coding-agent"; import { AuthStorage, type AuthCredential, type ExtensionAPI, type ExtensionContext, type ExtensionCommandContext, type ReadonlyFooterDataProvider, type SessionStartEvent, type SessionShutdownEvent, type AgentEndEvent } from "@mariozechner/pi-coding-agent";
import { statusbarConfig } from "./config"; import { statusbarConfig } from "./config";
import { contextModule, costModule, directoryModule, modelModule, thinkingModule } from "./modules/basic"; import { contextModule, costModule, directoryModule, modelModule, thinkingModule } from "./modules/basic";
import { usageModule } from "./modules/usage"; import { usageModule } from "./modules/usage";
@@ -21,7 +21,7 @@ function isRecord(value: unknown): value is Record<string, unknown> {
return typeof value === "object" && value !== null && !Array.isArray(value); return typeof value === "object" && value !== null && !Array.isArray(value);
} }
function activeProvider(ctx: any): Provider | undefined { function activeProvider(ctx: ExtensionContext): Provider | undefined {
const provider = ctx.model?.provider as string | undefined; const provider = ctx.model?.provider as string | undefined;
if (provider === "openai-codex" || provider === "zai" || provider === "anthropic") return provider; if (provider === "openai-codex" || provider === "zai" || provider === "anthropic") return provider;
return undefined; return undefined;
@@ -91,7 +91,7 @@ function renderModule(moduleCtx: ModuleContext, spec: ModuleSpec): RenderedModul
} }
} }
function renderFooter(ctx: any, footerData: any, state: StatusbarState, width: number, theme?: any): string[] { function renderFooter(ctx: ExtensionContext, footerData: ReadonlyFooterDataProvider, state: StatusbarState, width: number, theme?: any): string[] {
const moduleCtx: ModuleContext = { ctx, footerData, state, config: statusbarConfig }; const moduleCtx: ModuleContext = { ctx, footerData, state, config: statusbarConfig };
return statusbarConfig.rows.map(row => renderRow({ return statusbarConfig.rows.map(row => renderRow({
left: (row.left ?? []).map(spec => renderModule(moduleCtx, spec)), left: (row.left ?? []).map(spec => renderModule(moduleCtx, spec)),
@@ -110,7 +110,7 @@ export default function piStatusbarExtension(pi: ExtensionAPI) {
let timer: ReturnType<typeof setInterval> | undefined; let timer: ReturnType<typeof setInterval> | undefined;
let inFlight: AbortController | undefined; let inFlight: AbortController | undefined;
let lastProvider: Provider | undefined; let lastProvider: Provider | undefined;
let latestCtx: any; let latestCtx: ExtensionContext | undefined;
let requestRender: (() => void) | undefined; let requestRender: (() => void) | undefined;
const statusbarState: StatusbarState = {}; const statusbarState: StatusbarState = {};
const authStorage = AuthStorage.create(); const authStorage = AuthStorage.create();
@@ -124,7 +124,7 @@ export default function piStatusbarExtension(pi: ExtensionAPI) {
} }
let rerenderScheduled = false; let rerenderScheduled = false;
function rerender(ctx: any) { function rerender(ctx: ExtensionContext) {
latestCtx = ctx; latestCtx = ctx;
if (rerenderScheduled) return; if (rerenderScheduled) return;
rerenderScheduled = true; rerenderScheduled = true;
@@ -138,7 +138,7 @@ export default function piStatusbarExtension(pi: ExtensionAPI) {
}); });
} }
function installFooter(ctx: any) { function installFooter(ctx: ExtensionContext) {
if (!ctx.hasUI) return; if (!ctx.hasUI) return;
latestCtx = ctx; latestCtx = ctx;
ctx.ui.setFooter((tui: any, theme: any, footerData: any) => { ctx.ui.setFooter((tui: any, theme: any, footerData: any) => {
@@ -155,7 +155,7 @@ export default function piStatusbarExtension(pi: ExtensionAPI) {
}); });
} }
async function refresh(ctx: any, force = false) { async function refresh(ctx: ExtensionContext, force = false) {
latestCtx = ctx; latestCtx = ctx;
const provider = activeProvider(ctx); const provider = activeProvider(ctx);
lastProvider = provider; lastProvider = provider;

5
types.ts
View File

@@ -1,3 +1,4 @@
import type { ExtensionContext, ReadonlyFooterDataProvider } from "@mariozechner/pi-coding-agent";
import type { UsageLimit, UsageReport } from "./usage"; import type { UsageLimit, UsageReport } from "./usage";
export type ModuleName = "directory" | "context" | "model" | "thinking" | "cost" | "usage"; export type ModuleName = "directory" | "context" | "model" | "thinking" | "cost" | "usage";
@@ -40,8 +41,8 @@ export interface StatusbarState {
} }
export interface ModuleContext { export interface ModuleContext {
ctx: any; ctx: ExtensionContext;
footerData: any; footerData: ReadonlyFooterDataProvider;
state: StatusbarState; state: StatusbarState;
config: StatusbarConfig; config: StatusbarConfig;
} }