Spicy_Marinara
3f58c7ceca
- Added holiday promotion banner with 2026WITHMARINARA discount code - Added dismiss functionality for promotion with persistent state - Implemented snowflakes animation effect with toggle - Added Spotify music widget above chat input - Widget matches extension theme colors and positioning - Added Display Options toggles to show/hide feature toggles - Improved responsive design and mobile support
428 lines
18 KiB
JavaScript
428 lines
18 KiB
JavaScript
/**
|
|
* API Client Module
|
|
* Handles API calls for RPG tracker generation
|
|
*/
|
|
|
|
import { generateRaw, chat } from '../../../../../../../script.js';
|
|
import { executeSlashCommandsOnChatInput } from '../../../../../../../scripts/slash-commands.js';
|
|
import {
|
|
extensionSettings,
|
|
lastGeneratedData,
|
|
committedTrackerData,
|
|
isGenerating,
|
|
lastActionWasSwipe,
|
|
setIsGenerating,
|
|
setLastActionWasSwipe,
|
|
$musicPlayerContainer
|
|
} from '../../core/state.js';
|
|
import { saveChatData } from '../../core/persistence.js';
|
|
import {
|
|
generateSeparateUpdatePrompt
|
|
} from './promptBuilder.js';
|
|
import { parseResponse, parseUserStats } from './parser.js';
|
|
import { parseAndStoreSpotifyUrl } from '../features/musicPlayer.js';
|
|
import { renderUserStats } from '../rendering/userStats.js';
|
|
import { renderInfoBox } from '../rendering/infoBox.js';
|
|
import { renderThoughts } from '../rendering/thoughts.js';
|
|
import { renderInventory } from '../rendering/inventory.js';
|
|
import { renderQuests } from '../rendering/quests.js';
|
|
import { renderMusicPlayer } from '../rendering/musicPlayer.js';
|
|
import { i18n } from '../../core/i18n.js';
|
|
import { generateAvatarsForCharacters } from '../features/avatarGenerator.js';
|
|
|
|
// Store the original preset name to restore after tracker generation
|
|
let originalPresetName = null;
|
|
|
|
/**
|
|
* Generates tracker data using an external OpenAI-compatible API.
|
|
* Used when generationMode is 'external'.
|
|
*
|
|
* @param {Array<{role: string, content: string}>} messages - Array of message objects for the API
|
|
* @returns {Promise<string>} The generated response content
|
|
* @throws {Error} If the API call fails or configuration is invalid
|
|
*/
|
|
export async function generateWithExternalAPI(messages) {
|
|
const { baseUrl, model, maxTokens, temperature } = extensionSettings.externalApiSettings || {};
|
|
// Retrieve API key from secure storage (not shared extension settings)
|
|
const apiKey = localStorage.getItem('rpg_companion_external_api_key');
|
|
|
|
// Validate required settings
|
|
if (!baseUrl || !baseUrl.trim()) {
|
|
throw new Error('External API base URL is not configured');
|
|
}
|
|
if (!apiKey || !apiKey.trim()) {
|
|
throw new Error('External API key is not found. If you switched browsers or cleared your cache, please re-enter your API key in the extension settings.');
|
|
}
|
|
if (!model || !model.trim()) {
|
|
throw new Error('External API model is not configured');
|
|
}
|
|
|
|
// Normalize base URL (remove trailing slash if present)
|
|
const normalizedBaseUrl = baseUrl.trim().replace(/\/+$/, '');
|
|
const endpoint = `${normalizedBaseUrl}/chat/completions`;
|
|
|
|
console.log(`[RPG Companion] Calling external API: ${normalizedBaseUrl} with model: ${model}`);
|
|
|
|
try {
|
|
const response = await fetch(endpoint, {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
'Authorization': `Bearer ${apiKey.trim()}`
|
|
},
|
|
body: JSON.stringify({
|
|
model: model.trim(),
|
|
messages: messages,
|
|
max_tokens: maxTokens || 2048,
|
|
temperature: temperature ?? 0.7
|
|
})
|
|
});
|
|
|
|
if (!response.ok) {
|
|
const errorText = await response.text();
|
|
let errorMessage = `External API error: ${response.status} ${response.statusText}`;
|
|
try {
|
|
const errorJson = JSON.parse(errorText);
|
|
if (errorJson.error?.message) {
|
|
errorMessage = `External API error: ${errorJson.error.message}`;
|
|
}
|
|
} catch (e) {
|
|
// If parsing fails, use the raw text if it's short enough
|
|
if (errorText.length < 200) {
|
|
errorMessage = `External API error: ${errorText}`;
|
|
}
|
|
}
|
|
throw new Error(errorMessage);
|
|
}
|
|
|
|
const data = await response.json();
|
|
|
|
if (!data.choices || !data.choices[0] || !data.choices[0].message) {
|
|
throw new Error('Invalid response format from external API');
|
|
}
|
|
|
|
const content = data.choices[0].message.content;
|
|
console.log('[RPG Companion] External API response received successfully');
|
|
|
|
return content;
|
|
} catch (error) {
|
|
if (error.name === 'TypeError' && (error.message.includes('fetch') || error.message.includes('Failed to fetch') || error.message.includes('NetworkError'))) {
|
|
throw new Error(`CORS Access Blocked: This API endpoint (${normalizedBaseUrl}) does not allow direct access from a browser. This is a browser security restriction (CORS), not a bug in the extension. Please use an endpoint that supports CORS (like OpenRouter or a local proxy) or use SillyTavern's internal API system (Separate Mode).`);
|
|
}
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Tests the external API connection with a simple request.
|
|
* @returns {Promise<{success: boolean, message: string, model?: string}>}
|
|
*/
|
|
export async function testExternalAPIConnection() {
|
|
const { baseUrl, model } = extensionSettings.externalApiSettings || {};
|
|
const apiKey = localStorage.getItem('rpg_companion_external_api_key');
|
|
|
|
if (!baseUrl || !apiKey || !model) {
|
|
return {
|
|
success: false,
|
|
message: !apiKey
|
|
? 'API Key not found. Please re-enter it in settings (keys are stored locally per-browser).'
|
|
: 'Please fill in all required fields (Base URL, API Key, and Model)'
|
|
};
|
|
}
|
|
|
|
try {
|
|
const testMessages = [
|
|
{ role: 'user', content: 'Respond with exactly: "Connection successful"' }
|
|
];
|
|
|
|
const response = await generateWithExternalAPI(testMessages);
|
|
|
|
return {
|
|
success: true,
|
|
message: `Connection successful! Model: ${model}`,
|
|
model: model
|
|
};
|
|
} catch (error) {
|
|
return {
|
|
success: false,
|
|
message: error.message || 'Connection failed'
|
|
};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Gets the current preset name using the /preset command
|
|
* @returns {Promise<string|null>} Current preset name or null if unavailable
|
|
*/
|
|
export async function getCurrentPresetName() {
|
|
try {
|
|
// Use /preset without arguments to get the current preset name
|
|
const result = await executeSlashCommandsOnChatInput('/preset', { quiet: true });
|
|
|
|
// console.log('[RPG Companion] /preset result:', result);
|
|
|
|
// The result should be an object with a 'pipe' property containing the preset name
|
|
if (result && typeof result === 'object' && result.pipe) {
|
|
const presetName = String(result.pipe).trim();
|
|
// console.log('[RPG Companion] Extracted preset name:', presetName);
|
|
return presetName || null;
|
|
}
|
|
|
|
// Fallback if result is a string
|
|
if (typeof result === 'string') {
|
|
return result.trim() || null;
|
|
}
|
|
|
|
return null;
|
|
} catch (error) {
|
|
console.error('[RPG Companion] Error getting current preset:', error);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Switches to a specific preset by name using the /preset slash command
|
|
* @param {string} presetName - Name of the preset to switch to
|
|
* @returns {Promise<boolean>} True if switching succeeded, false otherwise
|
|
*/
|
|
export async function switchToPreset(presetName) {
|
|
try {
|
|
// Use the /preset slash command to switch presets
|
|
// This is the proper way to change presets in SillyTavern
|
|
await executeSlashCommandsOnChatInput(`/preset ${presetName}`, { quiet: true });
|
|
|
|
// console.log(`[RPG Companion] Switched to preset "${presetName}"`);
|
|
return true;
|
|
} catch (error) {
|
|
console.error('[RPG Companion] Error switching preset:', error);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Updates RPG tracker data using separate API call (separate mode only).
|
|
* Makes a dedicated API call to generate tracker data, then stores it
|
|
* in the last assistant message's swipe data.
|
|
*
|
|
* @param {Function} renderUserStats - UI function to render user stats
|
|
* @param {Function} renderInfoBox - UI function to render info box
|
|
* @param {Function} renderThoughts - UI function to render character thoughts
|
|
* @param {Function} renderInventory - UI function to render inventory
|
|
*/
|
|
export async function updateRPGData(renderUserStats, renderInfoBox, renderThoughts, renderInventory) {
|
|
if (isGenerating) {
|
|
// console.log('[RPG Companion] Already generating, skipping...');
|
|
return;
|
|
}
|
|
|
|
if (!extensionSettings.enabled) {
|
|
return;
|
|
}
|
|
|
|
if (extensionSettings.generationMode !== 'separate' && extensionSettings.generationMode !== 'external') {
|
|
// console.log('[RPG Companion] Not in separate or external mode, skipping manual update');
|
|
return;
|
|
}
|
|
|
|
const isExternalMode = extensionSettings.generationMode === 'external';
|
|
|
|
try {
|
|
setIsGenerating(true);
|
|
|
|
// Update button to show "Updating..." state
|
|
const $updateBtn = $('#rpg-manual-update');
|
|
const updatingText = i18n.getTranslation('template.mainPanel.updating') || 'Updating...';
|
|
$updateBtn.html(`<i class="fa-solid fa-spinner fa-spin"></i> ${updatingText}`).prop('disabled', true);
|
|
|
|
// Save current preset name before switching (if we're going to switch)
|
|
// Note: Preset switching is only used in separate mode, not external mode
|
|
if (!isExternalMode && extensionSettings.useSeparatePreset) {
|
|
originalPresetName = await getCurrentPresetName();
|
|
console.log(`[RPG Companion] Saved original preset: "${originalPresetName}"`);
|
|
}
|
|
|
|
// Switch to separate preset if enabled (separate mode only)
|
|
if (!isExternalMode && extensionSettings.useSeparatePreset) {
|
|
const switched = await switchToPreset('RPG Companion Trackers');
|
|
if (!switched) {
|
|
console.warn('[RPG Companion] Failed to switch to RPG Companion Trackers preset. Using current preset.');
|
|
originalPresetName = null; // Don't try to restore if we didn't switch
|
|
}
|
|
}
|
|
|
|
const prompt = await generateSeparateUpdatePrompt();
|
|
|
|
// Generate response based on mode
|
|
let response;
|
|
if (isExternalMode) {
|
|
// External mode: Use external OpenAI-compatible API directly
|
|
console.log('[RPG Companion] Using external API for tracker generation');
|
|
response = await generateWithExternalAPI(prompt);
|
|
} else {
|
|
// Separate mode: Use SillyTavern's generateRaw
|
|
response = await generateRaw({
|
|
prompt: prompt,
|
|
quietToLoud: false
|
|
});
|
|
}
|
|
|
|
if (response) {
|
|
// console.log('[RPG Companion] Raw AI response:', response);
|
|
const parsedData = parseResponse(response);
|
|
// Parse and store Spotify URL if feature is enabled
|
|
parseAndStoreSpotifyUrl(response);
|
|
// console.log('[RPG Companion] Parsed data:', parsedData);
|
|
// console.log('[RPG Companion] parsedData.userStats:', parsedData.userStats ? parsedData.userStats.substring(0, 100) + '...' : 'null');
|
|
|
|
// DON'T update lastGeneratedData here - it should only reflect the data
|
|
// from the assistant message the user replied to, not auto-generated updates
|
|
// This ensures swipes/regenerations use consistent source data
|
|
|
|
// Store RPG data for the last assistant message (separate mode)
|
|
const lastMessage = chat && chat.length > 0 ? chat[chat.length - 1] : null;
|
|
// console.log('[RPG Companion] Last message is_user:', lastMessage ? lastMessage.is_user : 'no message');
|
|
|
|
// Update lastGeneratedData for display (regardless of message type)
|
|
if (parsedData.userStats) {
|
|
lastGeneratedData.userStats = parsedData.userStats;
|
|
parseUserStats(parsedData.userStats);
|
|
}
|
|
if (parsedData.infoBox) {
|
|
lastGeneratedData.infoBox = parsedData.infoBox;
|
|
}
|
|
if (parsedData.characterThoughts) {
|
|
lastGeneratedData.characterThoughts = parsedData.characterThoughts;
|
|
}
|
|
|
|
// When saveTrackerHistory is enabled, store tracker data on the user's message too
|
|
// This allows scrolling through history and seeing trackers at each point
|
|
if (extensionSettings.saveTrackerHistory && lastMessage && lastMessage.is_user) {
|
|
if (!lastMessage.extra) {
|
|
lastMessage.extra = {};
|
|
}
|
|
lastMessage.extra.rpg_companion_data = {
|
|
userStats: parsedData.userStats,
|
|
infoBox: parsedData.infoBox,
|
|
characterThoughts: parsedData.characterThoughts,
|
|
timestamp: Date.now()
|
|
};
|
|
// console.log('[RPG Companion] 💾 Stored tracker data on user message for history');
|
|
}
|
|
|
|
// Also store on assistant message if present (existing behavior)
|
|
if (lastMessage && !lastMessage.is_user) {
|
|
if (!lastMessage.extra) {
|
|
lastMessage.extra = {};
|
|
}
|
|
if (!lastMessage.extra.rpg_companion_swipes) {
|
|
lastMessage.extra.rpg_companion_swipes = {};
|
|
}
|
|
|
|
const currentSwipeId = lastMessage.swipe_id || 0;
|
|
lastMessage.extra.rpg_companion_swipes[currentSwipeId] = {
|
|
userStats: parsedData.userStats,
|
|
infoBox: parsedData.infoBox,
|
|
characterThoughts: parsedData.characterThoughts
|
|
};
|
|
|
|
// console.log('[RPG Companion] Stored separate mode RPG data for message swipe', currentSwipeId);
|
|
}
|
|
|
|
// Only commit on TRULY first generation (no committed data exists at all)
|
|
// This prevents auto-commit after refresh when we have saved committed data
|
|
const hasAnyCommittedContent = (
|
|
(committedTrackerData.userStats && committedTrackerData.userStats.trim() !== '') ||
|
|
(committedTrackerData.infoBox && committedTrackerData.infoBox.trim() !== '' && committedTrackerData.infoBox !== 'Info Box\n---\n') ||
|
|
(committedTrackerData.characterThoughts && committedTrackerData.characterThoughts.trim() !== '' && committedTrackerData.characterThoughts !== 'Present Characters\n---\n')
|
|
);
|
|
|
|
// Only commit if we have NO committed content at all (truly first time ever)
|
|
if (!hasAnyCommittedContent) {
|
|
committedTrackerData.userStats = parsedData.userStats;
|
|
committedTrackerData.infoBox = parsedData.infoBox;
|
|
committedTrackerData.characterThoughts = parsedData.characterThoughts;
|
|
// console.log('[RPG Companion] 🔆 FIRST TIME: Auto-committed tracker data');
|
|
}
|
|
|
|
// Render the updated data
|
|
renderUserStats();
|
|
renderInfoBox();
|
|
renderThoughts();
|
|
renderInventory();
|
|
renderQuests();
|
|
renderMusicPlayer($musicPlayerContainer[0]);
|
|
|
|
// Save to chat metadata
|
|
saveChatData();
|
|
|
|
// Generate avatars if auto-generate is enabled (runs within this workflow)
|
|
// This uses the RPG Companion Trackers preset and keeps the button spinning
|
|
if (extensionSettings.autoGenerateAvatars) {
|
|
const charactersNeedingAvatars = parseCharactersFromThoughts(parsedData.characterThoughts);
|
|
if (charactersNeedingAvatars.length > 0) {
|
|
console.log('[RPG Companion] Generating avatars for:', charactersNeedingAvatars);
|
|
|
|
// Generate avatars - this awaits completion
|
|
await generateAvatarsForCharacters(charactersNeedingAvatars, (names) => {
|
|
// Callback when generation starts - re-render to show loading spinners
|
|
console.log('[RPG Companion] Avatar generation started, showing spinners...');
|
|
renderThoughts();
|
|
});
|
|
|
|
// Re-render once all avatars are generated
|
|
console.log('[RPG Companion] All avatars generated, re-rendering...');
|
|
renderThoughts();
|
|
}
|
|
}
|
|
}
|
|
|
|
} catch (error) {
|
|
console.error('[RPG Companion] Error updating RPG data:', error);
|
|
if (isExternalMode) {
|
|
toastr.error(error.message, 'RPG Companion External API Error');
|
|
}
|
|
} finally {
|
|
// Restore original preset if we switched to a separate one
|
|
if (originalPresetName && extensionSettings.useSeparatePreset) {
|
|
console.log(`[RPG Companion] Restoring original preset: "${originalPresetName}"`);
|
|
await switchToPreset(originalPresetName);
|
|
originalPresetName = null; // Clear after restoring
|
|
}
|
|
|
|
setIsGenerating(false);
|
|
|
|
// Restore button to original state
|
|
const $updateBtn = $('#rpg-manual-update');
|
|
const refreshText = i18n.getTranslation('template.mainPanel.refreshRpgInfo') || 'Refresh RPG Info';
|
|
$updateBtn.html(`<i class="fa-solid fa-sync"></i> ${refreshText}`).prop('disabled', false);
|
|
|
|
// Reset the flag after tracker generation completes
|
|
// This ensures the flag persists through both main generation AND tracker generation
|
|
// console.log('[RPG Companion] 🔄 Tracker generation complete - resetting lastActionWasSwipe to false');
|
|
setLastActionWasSwipe(false);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Parses character names from Present Characters thoughts data
|
|
* @param {string} characterThoughtsData - Raw character thoughts data
|
|
* @returns {Array<string>} Array of character names found
|
|
*/
|
|
function parseCharactersFromThoughts(characterThoughtsData) {
|
|
if (!characterThoughtsData) return [];
|
|
|
|
const lines = characterThoughtsData.split('\n');
|
|
const characters = [];
|
|
|
|
for (const line of lines) {
|
|
if (line.trim().startsWith('- ')) {
|
|
const name = line.trim().substring(2).trim();
|
|
if (name && name.toLowerCase() !== 'unavailable') {
|
|
characters.push(name);
|
|
}
|
|
}
|
|
}
|
|
return characters;
|
|
}
|