Lifei Zhou
98 lines
2.8 KiB
TypeScript
98 lines
2.8 KiB
TypeScript
import type { Session } from './types/session';
|
|
import type { ExtensionConfig } from './types/extensions';
|
|
import type { setViewType } from './hooks/useNavigation';
|
|
import type { FixedExtensionEntry } from './components/ConfigContext';
|
|
import { AppEvents } from './constants/events';
|
|
import { acpChatSessionController } from './acp/chatSessionController';
|
|
import { getConfiguredGooseExtensions, gooseExtensionName } from './acp/extensions';
|
|
|
|
export function getSessionDisplayName(session: Session): string {
|
|
if (session.user_set_name) {
|
|
return session.name;
|
|
}
|
|
if (session.recipe?.title) {
|
|
return session.recipe.title;
|
|
}
|
|
return session.name;
|
|
}
|
|
|
|
interface CreateSessionOptions {
|
|
recipeDeeplink?: string;
|
|
recipeId?: string;
|
|
extensionConfigs?: ExtensionConfig[];
|
|
allExtensions?: FixedExtensionEntry[];
|
|
}
|
|
|
|
function selectedExtensionConfigs(options?: CreateSessionOptions): ExtensionConfig[] {
|
|
if (options?.extensionConfigs && options.extensionConfigs.length > 0) {
|
|
return options.extensionConfigs;
|
|
}
|
|
if (options?.allExtensions) {
|
|
return options.allExtensions
|
|
.filter((extension) => extension.enabled)
|
|
.map((extension) => {
|
|
const { enabled: _enabled, ...config } = extension;
|
|
return config as ExtensionConfig;
|
|
});
|
|
}
|
|
return [];
|
|
}
|
|
|
|
async function createAcpSession(
|
|
workingDir: string,
|
|
options?: CreateSessionOptions
|
|
): Promise<Session> {
|
|
const selectedNames = new Set(selectedExtensionConfigs(options).map((config) => config.name));
|
|
const gooseExtensions =
|
|
selectedNames.size > 0
|
|
? (await getConfiguredGooseExtensions())
|
|
.filter((entry) => selectedNames.has(gooseExtensionName(entry.extension)))
|
|
.map((entry) => entry.extension)
|
|
: [];
|
|
return acpChatSessionController.createSession(workingDir, gooseExtensions, {
|
|
recipeId: options?.recipeId,
|
|
recipeDeeplink: options?.recipeDeeplink,
|
|
});
|
|
}
|
|
|
|
export async function createSession(
|
|
workingDir: string,
|
|
options?: CreateSessionOptions
|
|
): Promise<Session> {
|
|
return createAcpSession(workingDir, options);
|
|
}
|
|
|
|
export async function startNewSession(
|
|
initialText: string | undefined,
|
|
setView: setViewType,
|
|
workingDir: string,
|
|
options?: {
|
|
recipeDeeplink?: string;
|
|
recipeId?: string;
|
|
allExtensions?: FixedExtensionEntry[];
|
|
}
|
|
): Promise<Session> {
|
|
const session = await createSession(workingDir, options);
|
|
window.dispatchEvent(new CustomEvent(AppEvents.SESSION_CREATED, { detail: { session } }));
|
|
|
|
const initialMessage = initialText ? { msg: initialText, images: [] } : undefined;
|
|
|
|
const eventDetail = {
|
|
sessionId: session.id,
|
|
initialMessage,
|
|
};
|
|
|
|
window.dispatchEvent(
|
|
new CustomEvent(AppEvents.ADD_ACTIVE_SESSION, {
|
|
detail: eventDetail,
|
|
})
|
|
);
|
|
|
|
setView('pair', {
|
|
disableAnimation: true,
|
|
initialMessage,
|
|
resumeSessionId: session.id,
|
|
});
|
|
return session;
|
|
}
|