import { Config, getConfig } from '../config.js';
import { getOverridableConfig, OverridableConfig } from '../overridableConfig.js';
import { RestApiArgs, useRestApi } from '../restApiInstance.js';
import { McpSiteSettings } from '../sdks/tableau/types/mcpSiteSettings.js';
import { ExpiringMap } from './expiringMap.js';
import { getSiteLuidFromAccessToken } from './getSiteLuidFromAccessToken.js';
import { DistributiveOmit } from './types.js';
type SiteNameOrSiteId = string;
let mcpSiteSettingsCache: ExpiringMap<SiteNameOrSiteId, McpSiteSettings>;
async function getMcpSiteSettings({
restApiArgs,
}: {
restApiArgs: RestApiArgs;
}): Promise<McpSiteSettings | undefined> {
const { config, tableauAuthInfo } = restApiArgs;
if (!config.enableMcpSiteSettings) {
return;
}
if (!mcpSiteSettingsCache) {
mcpSiteSettingsCache = new ExpiringMap<SiteNameOrSiteId, McpSiteSettings>({
defaultExpirationTimeMs: config.mcpSiteSettingsCheckIntervalInMinutes * 60 * 1000,
});
}
const cacheKey = config.siteName || getSiteLuidFromAccessToken(tableauAuthInfo);
if (!cacheKey) {
throw new Error('Could not determine site ID/name');
}
const cachedSettings = mcpSiteSettingsCache.get(cacheKey);
if (cachedSettings) {
return cachedSettings;
}
const settings = await useRestApi({
...restApiArgs,
jwtScopes: ['tableau:mcp_site_settings:read'],
callback: async (restApi) => await restApi.siteMethods.getMcpSettings(),
});
mcpSiteSettingsCache.set(cacheKey, settings);
return settings;
}
// Make "config" and "signal" optional
type GetConfigWithOverridesArgs = DistributiveOmit<RestApiArgs, 'config' | 'signal'> &
Partial<{ config: Config; signal: AbortSignal }>;
export async function getConfigWithOverrides({
restApiArgs,
}: {
restApiArgs: GetConfigWithOverridesArgs;
}): Promise<OverridableConfig> {
const config = restApiArgs.config ?? getConfig();
const signal = restApiArgs.signal ?? AbortSignal.timeout(config.maxRequestTimeoutMs);
const overrides = await getMcpSiteSettings({
restApiArgs: { ...restApiArgs, config, signal },
});
return getOverridableConfig(overrides);
}
