// ==UserScript==
// @name NovelAI Prompt Preset / Wildcards Manager
// @name:ja NovelAI Prompt Preset / Wildcards Manager
// @namespace https://github.com/NeviumX/NovelAI-Prompt-Preset-Manager
// @version 1.0
// @description Script to replace __TOKEN__ with prompt anything you want before making a request to the API on the NovelAI.
// @description:ja NovelAIでAPIにリクエストする前に、__TOKEN__を任意のプロンプトに置き換えるスクリプトです。
// @author Gemini 2.5 Pro, ChatGPT o3, Nevium7
// @copyright Nevium7
// @license MIT
// @match https://novelai.net/*
// @icon https://novelai.net/icons/novelai-round.png
// @grant GM_getValue
// @grant GM_setValue
// @grant GM_addStyle
// @grant GM_listValues
// @grant GM_deleteValue
// @grant unsafeWindow
// @require https://update.greasyforks.org/scripts/473358/1237031/JSZip.js
// @run-at document-start
// ==/UserScript==
/* ---------- styles ---------- */
GM_addStyle(/* css */`
.nai-preset-panel {background:#1b1b1b;border:1px solid #444;padding:8px 10px;margin-top:8px;border-radius:6px;font-size:13px;color:#f0f0f0}
.nai-preset-title {font-weight:700;margin-bottom:4px}
.nai-preset-textarea {width:100%;min-height:60px;padding:6px;border-radius:4px;border:1px solid #555;background:#141414;color:#f8f8f8;resize:vertical}
.nai-preset-controls {display:flex;gap:6px;align-items:center;margin:6px 0}
.nai-preset-input {flex:1 1 0;padding:4px 6px;border-radius:4px;border:1px solid #555;background:#141414;color:#f8f8f8}
.nai-btn {padding:4px 10px;border:1px solid #666;background:#222;color:#f8f8f8;border-radius:4px;cursor:pointer;font-weight:600}
.nai-preset-list {display:flex;flex-wrap:wrap;gap:6px}
.nai-preset-item {background:#242424;border:1px solid #444;padding:2px 6px;border-radius:4px;display:inline-flex;width:fit-content;max-width:100%;white-space:nowrap;gap:4px;align-items:center;}
.nai-preset-item input {margin:0}
.nai-btn-remove {display:none;border:none;background:none;color:#dd6666;font-size:15px;cursor:pointer;line-height:1}
.nai-btn-toggle {width:26px;padding:4px 0;font-weight:700}
.nai-gear-wrap {position:absolute; top:6px; right:6px}
.nai-gear-btn {width:20px;height:20px;cursor:pointer;border:none;background:none;padding:0;
display:flex;align-items:center;justify-content:center;border-radius:2px;
color:#bbb;transition:background .15s}
.nai-gear-btn:hover {background:#333;color:#fff}
.nai-popup {position:absolute;
bottom:100%;
right:0;
margin-bottom:8px;
width:200px; padding:10px;
background:#222; border:1px solid #555; border-radius:4px; color:#eee;
display:none; z-index:2147483647;}
.nai-popup:before {content:'';position:absolute;bottom:-6px;right:14px;
border:6px solid transparent;
border-top-color:#555;}
.nai-remain-row {margin-top:6px;display:flex;align-items:center;gap:6px;font-size:13px;white-space:nowrap;float:left}
.nai-remain-row input[type="checkbox"] { margin: 6px; accent-color: #f5f3c2; }
.nai-suggest-box {position:fixed; z-index:2147483647; background:#222; border:1px solid #555;
border-radius:4px; max-height:180px; overflow-y:auto; font-size:13px; color:#eee}
.nai-suggest-item {padding:4px 8px; cursor:pointer}
.nai-suggest-item.active {background-color: #444}
@keyframes Flash {0%{background:#444} 100%{background:#242424}}
`);
// ブラウザストレージ保存用定数
const PREFIX = 'naiPromptPreset:';
// トークンをメタデータに残すかどうかのトグル用定数
const TOKEN_REMAIN_TRG = 'naiRemainTokenTrigger';
// デバッグモードのトグル用定数
const DEBUG_MODE_TRG = 'debugModeTrigger';
// JSZip
if (typeof JSZip !== 'undefined') {
unsafeWindow.JSZip = JSZip;
console.log('[PresetMgr] JSZip successfully attached to page context (unsafeWindow).');
} else {
console.error('[PresetMgr] JSZip is not defined in userscript scope despite @require. Check @require path and script manager.');
}
/*
* イベント
* naiRemainUpdate → トークンをメタデータに残すかどうかのトグル
* naiPresetUpdate → プリセット辞書を更新する
* naiDebugUpdate → デバッグモードのトグル
*
*/
class UIManager {
constructor(root) {
this.panel = this.injectUI(root);
this.jsonMgr = jsonManagerSingleton;
}
injectUI(root) {
const panelID = 'nai-preset-panel-injected';
const existed = document.getElementById(panelID);
if (existed) return existed;
if(root) {
const panel = this.createPresetManagerUI();
panel.id = panelID;
root.insertAdjacentElement('afterend', panel);
return panel;
}
return null;
}
destroy(){
if (!this.panel) return;
/* gear のポップアップのクリックハンドラを外す */
if (this._onDocClick){
document.removeEventListener('click', this._onDocClick, false);
this._onDocClick = null;
}
/* パネルDOMを除去 */
if (this.panel.isConnected) this.panel.remove();
this.panel = null;
}
createPresetManagerUI() {
/* container */
const panel = document.createElement('div');
panel.className = 'nai-preset-panel';
panel.style.position = 'relative';
/* title */
panel.innerHTML = `
<div class="nai-preset-title">Prompt Preset / Wildcards Manager</div>
<!-- 歯車 -->
<div class="nai-gear-wrap">
<button class="nai-gear-btn" title="Settings">
<svg viewBox="0 0 24 24" width="16" height="16" fill="currentColor">
<path d="M19.14 12.94a7.93 7.93 0 0 0 .05-.94 7.93 7.93 0 0 0-.05-.94l2.11-1.65a.5.5 0 0 0 .12-.65l-2-3.46a.5.5 0 0 0-.6-.22l-2.49 1a7.2 7.2 0 0 0-1.63-.94l-.38-2.65A.5.5 0 0 0 13.7 3h-3.4a.5.5 0 0 0-.49.41l-.38 2.65a7.2 7.2 0 0 0-1.63.94l-2.49-1a.5.5 0 0 0-.6.22l-2 3.46a.5.5 0 0 0 .12.65l2.11 1.65c-.04.31-.05.63-.05.94s.01.63.05.94l-2.11 1.65a.5.5 0 0 0-.12.65l2 3.46c.13.23.39.32.6.22l2.49-1c.5.4 1.05.72 1.63.94l.38 2.65c.05.24.25.41.49.41h3.4c.24 0 .44-.17.49-.41l.38-2.65c.58-.22 1.13-.54 1.63-.94l2.49 1a.5.5 0 0 0 .6-.22l2-3.46a.5.5 0 0 0-.12-.65l-2.11-1.65ZM12 15.5A3.5 3.5 0 1 1 12 8.5a3.5 3.5 0 0 1 0 7Z"/>
</svg>
</button>
</div>
<!-- ポップアップメニュー -->
<div class="nai-popup">
<h3 style="margin:0 0 10px;font-size:16px">Settings</h3>
<button class="nai-btn nai-set-import" style="width:100%;margin-bottom:8px">Import Preset</button>
<button class="nai-btn nai-set-export" style="width:100%;margin-bottom:8px">Export Preset</button>
<button class="nai-btn nai-set-clear" style="width:100%;color:red">Clear All Preset</button>
<!-- Remain-Token switch -->
<div class="nai-remain-row">
<input type="checkbox" id="nai-remain-check">
<span>Remain Preset Token</span>
</div>
<div class="nai-remain-row">
<input type="checkbox" id="nai-debug-mode-check">
<span>Enable Debug Logging</span>
</div>
<!-- hidden file picker -->
<input type="file" accept=".json,.txt" class="nai-file-input" style="display:none">
</div>
<textarea class="nai-preset-textarea" placeholder="masterpiece, best quality, oil painting (medium)"></textarea>
<div class="nai-preset-controls">
<input class="nai-preset-input" placeholder="Preset name (Do not contain '__' and space)">
<button class="nai-btn nai-btn-add">ADD</button>
<button class="nai-btn nai-btn-clear">CLEAR</button>
<button class="nai-btn nai-btn-toggle">▴</button>
</div>
<div class="nai-preset-list"></div>`;
/* preset data list */
const list = panel.querySelector('.nai-preset-list');
/* load existing presets */
GM_listValues()
.filter(k => k.startsWith(PREFIX))
.forEach(k => {
const presetName = k.slice(PREFIX.length);
list.appendChild(this.makeListItem(presetName));
});
/* ADD button */
panel.querySelector('.nai-btn-add').onclick = () => {
const name = panel.querySelector('.nai-preset-input').value.trim();
if (!name) return;
const validationRe = /^[A-Za-z0-9_.-]+$/;
if (!validationRe.test(name)) return;
const presetText = panel.querySelector('.nai-preset-textarea').value;
if (!presetText) return;
if (name.includes('__')) {
alert(
'[NovelAI Prompt Preset Manager]\n' +
'ERROR: Do not use double-underscore (__) to the preset name.\n\n' +
'This symbol should only be used to enclose actual tokens.'
);
return;
}
const key = PREFIX + name;
const alreadyExists = GM_getValue(key, null) !== null;
/* save preset */
GM_setValue(key, presetText);
if(alreadyExists) {
const item = [...list.children]
.find(el => el.querySelector('span')?.textContent === name);
if(item) {
item.style.animation = 'Flash 0.4s';
setTimeout(() => item.style.animation = '', 400);
}
} else {
/* create new entry */
list.appendChild(this.makeListItem(name));
}
this.jsonMgr.updateDict();; // update dict
/* clear textarea */
panel.querySelector('.nai-preset-input').value = '';
};
/* CLEAR button (clears textarea and preset name) */
panel.querySelector('.nai-btn-clear').onclick = () => {
panel.querySelector('.nai-preset-textarea').value = '';
panel.querySelector('.nai-preset-input').value = '';
};
/* toggle textarea visibility */
panel.querySelector('.nai-btn-toggle').onclick = (e) => {
const ta = panel.querySelector('.nai-preset-textarea');
const hidden = ta.style.display === 'none';
ta.style.display = hidden ? '' : 'none';
e.target.textContent = hidden ? '▴' : '▾';
};
/* checkbox handler – show / hide × button */
list.addEventListener('change', (e) => {
if (e.target.matches('input[type="checkbox"]')) {
const item = e.target.closest('.nai-preset-item');
const btn = item.querySelector('.nai-btn-remove');
btn.style.display = e.target.checked ? 'inline' : 'none';
if(e.target.checked) {
const name = item.querySelector('span').textContent;
/* load preset */
const presetText = GM_getValue(PREFIX + name, '');
const ta = panel.querySelector('.nai-preset-textarea');
ta.value = presetText;
//ta.value = ta.value ? `${ta.value.replace(/\s+$/, '')}, ${presetText}` : presetText;
const nameBox = panel.querySelector('.nai-preset-input');
nameBox.value = name;
/* uncheck all other checkboxes */
const allCheckbocks = document.querySelectorAll('.nai-preset-item input[type="checkbox"]');
const allBtns = document.querySelectorAll('.nai-btn-remove');
allCheckbocks.forEach((el) => {
if (el !== e.target) {
el.checked = false;
}
});
allBtns.forEach((el) => {
if (el !== btn) {
el.style.display = 'none';
}
});
}
}
});
/* remove handler */
list.addEventListener('click', (e) => {
if (!e.target.matches('.nai-btn-remove')) return;
e.stopPropagation();
e.preventDefault();
const item = e.target.closest('.nai-preset-item');
const name = item.querySelector('span').textContent;
/* confirm just in case */
if (!confirm('[NovelAI Prompt Preset Manager]\nDelete preset?: '+ name)) return;
/* delete from browser storage */
GM_deleteValue(PREFIX + name);
/* delete from UI */
item.remove();
this.jsonMgr.updateDict(); // update dict
});
/* gear button handler */
const gearBtn = panel.querySelector('.nai-gear-btn');
const popup = panel.querySelector('.nai-popup');
gearBtn.addEventListener('click', (e) => {
e.stopPropagation();
popup.style.display = popup.style.display === 'block' ? 'none' : 'block';
});
/* ========== Settings buttons ========== */
const importBtn = panel.querySelector('.nai-set-import');
const exportBtn = panel.querySelector('.nai-set-export');
const clearBtn = panel.querySelector('.nai-set-clear');
const fileInput = panel.querySelector('.nai-file-input');
/* --- Import ------------------------------------------ */
importBtn.addEventListener('click', () => fileInput.click());
fileInput.addEventListener('change', () => {
if (!fileInput.files || !fileInput.files[0]) return;
const reader = new FileReader();
let importCount = 0;
reader.onload = () => {
try {
importCount = 0;
/* ファイル形式: { "name":"prompt", ... } の JSON */
const obj = JSON.parse(reader.result);
/* ブラウザストレージに保存 */
Object.entries(obj).forEach(([name, prompt]) => {
if (typeof prompt !== 'string') return;
const key = PREFIX + name;
if(GM_getValue(key, null) === null) {
GM_setValue(key, prompt);
list.appendChild(this.makeListItem(name));
importCount++;
}
});
} catch(err) {
alert('[NovelAI Prompt Preset Manager]\nFailed to import: ' + err.message);
}
if (importCount > 0) {
console.log(`[NovelAI Prompt Preset Manager]\nImported ${importCount} new preset(s)!`);
this.jsonMgr.updateDict(); // update dict
}
/* clear file input */
fileInput.value = '';
};
reader.readAsText(fileInput.files[0]);
});
/* --- Export ------------------------------------------ */
exportBtn.addEventListener('click', () => {
const data = {};
GM_listValues()
.filter(k => k.startsWith(PREFIX))
.forEach(k => {
const presetName = k.slice(PREFIX.length);
data[presetName] = GM_getValue(k, '');
});
const blob = new Blob([JSON.stringify(data, null, 2)], {type:'application/json'});
const url = URL.createObjectURL(blob);
const a = document.createElement('a');
a.href = url;
/* file name */
a.download = 'nai-prompt-presets_' +
new Date()
.toLocaleString('sv-SE', { hour12: false })
.replace(' ', '_')
.replaceAll(':', '-')
+ '.json';
document.body.appendChild(a);
a.click();
a.remove();
URL.revokeObjectURL(url);
});
/* --- Clear All --------------------------------------- */
clearBtn.addEventListener('click', () => {
if (!confirm('[NovelAI Prompt Preset Manager]\nDelete ALL saved presets?')) return;
/* ストレージを削除 */
GM_listValues()
.filter(k => k.startsWith(PREFIX))
.forEach(k => GM_deleteValue(k));
/* UI もクリア */
list.innerHTML = '';
alert('[NovelAI Prompt Preset Manager]\nAll presets cleared.');
this.jsonMgr.updateDict(); // update dict
});
/* hide popup on click outside */
this._onDocClick = (e)=>{
if (!panel.contains(e.target)) popup.style.display = 'none';
};
document.addEventListener('click', this._onDocClick, false);
/* ----- Remain-Token checkbox ----- */
const remainChk = panel.querySelector('#nai-remain-check');
remainChk.checked = GM_getValue(TOKEN_REMAIN_TRG, false); // 復元
remainChk.addEventListener('change', () =>{
GM_setValue(TOKEN_REMAIN_TRG, remainChk.checked); // 保存
window.dispatchEvent(new CustomEvent('naiRemainUpdate',{detail: remainChk.checked}));
});
/* ----- Debug-Mode checkbox ----- */
const debugChk = panel.querySelector('#nai-debug-mode-check');
debugChk.checked = GM_getValue(DEBUG_MODE_TRG, false); //復元
debugChk.addEventListener('change', () => {
GM_setValue(DEBUG_MODE_TRG, debugChk.checked); //保存
window.dispatchEvent(new CustomEvent('naiDebugUpdate', {detail: debugChk.checked}))
});
return panel;
}
/* helper to create a preset list entry */
makeListItem(name) {
const wrapper = document.createElement('label');
wrapper.className = 'nai-preset-item';
wrapper.innerHTML = `
<input type="checkbox">
<span>${name}</span>
<button class="nai-btn-remove">×</button>
`;
return wrapper;
}
}
class SuggestionManager {
constructor(editor, jsonMgr) {
this.jsonMgr = jsonMgr;
this.editor = editor;
this.box = this.createBox();
this.bind();
}
createBox() {
const div = document.createElement('div');
div.className = 'nai-suggest-box';
div.style.display = 'none';
document.body.appendChild(div);
return div;
}
/* サジェストボックスのイベントをバインド */
bind() {
this._updateHandler = () => this.update();
this._navHandler = (e) => this.nav(e);
this._hideHandler = () => this.hide();
this._mousedownHandler = (e) => {
if (e.target.classList.contains('nai-suggest-item')) {
e.preventDefault();
this.choose(e.target, e.shiftKey); //Shiftキーの状態を渡す
}
};
this._mouseoverHandler = (e) => {
if (e.target.classList.contains('nai-suggest-item')) {
e.preventDefault();
this.selIdx = [...this.box.children].indexOf(e.target);
this.highlight();
}
};
this._escapeKeyHandler = (e) => { if (e.key === 'Escape') this.hide(); };
this.editor.addEventListener('input', this._updateHandler);
this.editor.addEventListener('keydown', this._navHandler);
this.editor.addEventListener('click', this._hideHandler);
this.box.addEventListener('mousedown', this._mousedownHandler);
this.box.addEventListener('mouseover', this._mouseoverHandler);
document.addEventListener('keydown', this._escapeKeyHandler);
}
/* caret 左側文字列を取得 */
textBeforeCaret() {
const sel = window.getSelection();
if (!sel || !sel.anchorNode || !this.editor.contains(sel.anchorNode)) return '';
const rng = sel.getRangeAt(0).cloneRange();
rng.collapse(true);
rng.setStart(this.editor, 0);
return rng.toString();
}
/* 候補計算 & 表示 */
update() {
const txt = this.textBeforeCaret();
const dict = this.jsonMgr.getDict();
/* 1) __foo__bar → 値候補 */
const mVal = txt.match(/__([A-Za-z0-9_-]+)__(\w*)$/);
if (mVal && dict[mVal[1]]) {
const [ , key, part ] = mVal;
const list = dict[key]
.split(/\r?\n/)
.filter(Boolean)
.filter(l => l.toLowerCase().includes(part.toLowerCase()))
.slice(0, 100);
if (list.length) return this.render(list.map(t => ({type:'value', text:t, originalKey: key })));
}
/* 2) __foo → トークン候補 */
const mTok = txt.match(/__([A-Za-z0-9_-]*)$/);
if (mTok) {
const prefix = mTok[1].toLowerCase();
let keys = Object.keys(dict);
keys = prefix ? keys.filter(k => k.toLowerCase().startsWith(prefix)) : keys;
keys.sort(); keys = keys.slice(0, 100);
if (keys.length) return this.render(keys.map(k => ({type:'token', text:`__${k}__`})));
}
this.hide();
}
/* 候補を表示 */
render(items) {
this.box.innerHTML = '';
items.forEach((itemData) => {
const div = document.createElement('div');
div.className = 'nai-suggest-item';
div.textContent = itemData.text;
div.dataset.type = itemData.type; // 'token' または 'value'
if (itemData.type === 'value' && itemData.originalKey) {
div.dataset.originalKey = itemData.originalKey;
}
this.box.appendChild(div);
});
this.selIdx = 0;
this.highlight();
const sel = window.getSelection();
if (sel.rangeCount > 0) {
const rect = sel.getRangeAt(0).getBoundingClientRect();
this.box.style.left = `${rect.left + window.scrollX}px`;
this.box.style.top = `${rect.bottom + window.scrollY + 4}px`;
this.box.style.display = 'block';
} else {
this.hide();
}
}
/* ↑↓ Tab Space (Shift+Space対応) */
nav(e) {
if (this.box.style.display === 'none') return;
const items = [...this.box.children];
if (!items.length) return;
if (e.key === 'ArrowDown') { e.preventDefault(); this.selIdx = (this.selIdx + 1) % items.length; }
if (e.key === 'ArrowUp') { e.preventDefault(); this.selIdx = (this.selIdx - 1 + items.length) % items.length; }
if (e.key === 'Tab' || e.key === ' ') { // Spaceキーの場合、Shiftキーの状態も渡す
e.preventDefault();
this.choose(items[this.selIdx], e.shiftKey);
}
this.highlight();
}
/* 候補を選択して置換 */
choose(itemElement, shiftPressed = false) {
const sel = window.getSelection();
if (!sel || !sel.rangeCount) return this.hide();
const fullTextBeforeCaret = this.textBeforeCaret();
const itemType = itemElement.dataset.type;
const suggestionText = itemElement.textContent; // サジェストボックスに表示されているテキスト
let textToInsert = "";
let charactersToDelete = 0;
const newlineReplaceRegex = /\r?\n/g;
const newlineTestRegex = /\r?\n/;
if (itemType === 'value') {
const valueTriggerRegex = /__([A-Za-z0-9_.-]+)__(\w*)$/;
const currentTriggerMatch = fullTextBeforeCaret.match(valueTriggerRegex);
if (currentTriggerMatch) {
charactersToDelete = currentTriggerMatch[0].length;
} else {
charactersToDelete = 0;
}
/** Shiftキーが押された場合 */
if (shiftPressed) {
const presetName = itemElement.dataset.originalKey;
const presetContent = this.jsonMgr.getDict()[presetName];
if (typeof presetContent === 'string') {
if (newlineTestRegex.test(presetContent)) {
// 改行があればランダムプロンプト形式に加工
textToInsert = '||' + presetContent.replace(newlineReplaceRegex, '|') + '||';
} else {
textToInsert = presetContent;
}
} else {
textToInsert = suggestionText;
}
} else {
// 通常のトークン選択 (Tab または Spaceのみ)
textToInsert = suggestionText;
}
} else if (itemType === 'token') {
const tokenTriggerRegex = /__([A-Za-z0-9_.-]*)$/;
const fullTokenTriggerRegex = /__([A-Za-z0-9_.-]+)__$/;
let currentTriggerMatch = fullTextBeforeCaret.match(fullTokenTriggerRegex);
if (currentTriggerMatch) {
// カーソル前が __FOO__
charactersToDelete = currentTriggerMatch[0].length;
} else {
// カーソル前が __FOO
currentTriggerMatch = fullTextBeforeCaret.match(tokenTriggerRegex);
if (currentTriggerMatch) {
charactersToDelete = currentTriggerMatch[0].length;
} else {
charactersToDelete = 0;
}
}
/** Shiftキーが押された場合 */
if (shiftPressed) {
const presetName = suggestionText.slice(2, -2);
const presetContent = this.jsonMgr.getDict()[presetName];
if (typeof presetContent === 'string') {
if (newlineTestRegex.test(presetContent)) {
// 改行があればランダムプロンプト形式に加工
textToInsert = '||' + presetContent.replace(newlineReplaceRegex, '|') + '||';
} else {
textToInsert = presetContent;
}
} else {
textToInsert = suggestionText;
}
} else {
// 通常のトークン選択 (Tab または Spaceのみ)
textToInsert = suggestionText;
}
}
if (charactersToDelete > 0 || textToInsert) {
for (let i = 0; i < charactersToDelete; i++) sel.modify('extend', 'backward', 'character');
document.execCommand('insertText', false, textToInsert);
}
this.hide();
}
/* 選択中の候補をハイライト */
highlight() {
const items = [...this.box.children];
items.forEach((d,i)=> {
d.classList.toggle('active', i===this.selIdx);
});
// 自動スクロール
if (items.length > 0 && this.selIdx >= 0 && this.selIdx < items.length) {
const selectedItem = items[this.selIdx];
const box = this.box;
const itemTop = selectedItem.offsetTop;
const itemBottom = itemTop + selectedItem.offsetHeight;
const boxScrollTop = box.scrollTop;
const boxVisibleHeight = box.clientHeight;
if (itemTop < boxScrollTop) {
box.scrollTop = itemTop;
} else if (itemBottom > boxScrollTop + boxVisibleHeight) {
box.scrollTop = itemBottom - boxVisibleHeight;
}
}
}
/* 選択を解除して非表示 */
hide() { this.box.style.display = 'none'; this.selIdx = -1; }
/** 生成した要素・リスナを片付けてインスタンスを無効化 */
destroy() {
this.editor.removeEventListener('input', this._updateHandler);
this.editor.removeEventListener('keydown', this._navHandler);
this.editor.removeEventListener('click', this._hideHandler);
this.box.removeEventListener('mousedown', this._mousedownHandler);
this.box.removeEventListener('mouseover', this._mouseoverHandler);
document.removeEventListener('keydown', this._escapeKeyHandler);
this.box.remove();
}
}
class ProseMirrorObserver {
constructor(jsonMgr) {
this.jsonMgr = jsonMgr;
this.map = new Map();
this.mo = new MutationObserver(m=>this.handle(m));
this.start();
}
start() {
this.mo.observe(document.documentElement,{childList:true,subtree:true});
}
attach(node){
if (!node.isContentEditable) return;
if (this.map.has(node)) return;
const sm = new SuggestionManager(node, this.jsonMgr);
this.map.set(node, sm);
//console.log('[PresetMgr] SuggestionManager added', node);
}
detach(node){
const sm = this.map.get(node);
if (!sm) return;
sm.destroy();
this.map.delete(node);
//console.log('[PresetMgr] SuggestionManager removed', node);
}
handle(muts){
muts.forEach(m=>{
m.addedNodes.forEach(n=>{
if (n.nodeType!==1) return;
let i = 0;
const elementsSet = new Set(n.querySelectorAll?.('div.ProseMirror[contenteditable]'));
if(elementsSet && elementsSet.size !== 0) {
elementsSet.forEach(el=>{
// this is so annoying
if (!this.map.has(el) && i < Math.round(elementsSet.size/2) ) { this.attach(el); i++;}
});
elementsSet.clear();
} else return;
});
m.removedNodes.forEach(n=>{
if (n.nodeType!==1) return;
if (n.classList?.contains('ProseMirror')) this.detach(n);
});
});
}
}
class PromptBoxObserver {
constructor(){
this.map = new Map();
this.mo = new MutationObserver(m=>this.handle(m));
this.start();
}
start(){
this.mo.observe(document.documentElement,{childList:true,subtree:true});
}
attach(root){
const prev = this.map.get(root);
/* パネルが消えていれば破棄して作り直す */
if (prev && !prev.panel?.isConnected) {
prev.destroy();
this.map.delete(root);
}
if (!this.map.has(root)) {
const ui = new UIManager(root);
this.map.set(root, ui);
console.log('[NovelAI Prompt Preset Manager] UI attached.', root);
}
}
detach(root){
const ui = this.map.get(root);
if (!ui) return;
ui.destroy();
this.map.delete(root);
console.log('[NovelAI Prompt Preset Manager] UI detached.', root);
}
handle(muts){
muts.forEach(m=>{
let i = 0;
m.addedNodes.forEach(n=>{
if (n.nodeType!==1) return;
n.querySelectorAll?.('.prompt-input-box-prompt,.prompt-input-box-プロンプト,.prompt-input-box-ベースプロンプト,.prompt-input-box-base-prompt'
).forEach(el=>{
// this is so annoying
if (!this.map.has(el) && i==0 ) { this.attach(el); i++; }
});
});
m.removedNodes.forEach(n=>{
if (n.nodeType!==1) return;
n.querySelectorAll?.('.prompt-input-box-prompt,.prompt-input-box-プロンプト,.prompt-input-box-ベースプロンプト,.prompt-input-box-base-prompt'
).forEach(el=>{ this.detach(el); });
});
});
}
}
class JsonManager {
constructor() {
this.TARGET_PATH = '/ai/generate-image';
this._dictCache = this.buildDict();
this.installPatch();
}
/* GM_storage → {TOKEN: "value"} へ変換 */
buildDict() {
const dict = {};
GM_listValues()
.filter(k => k.startsWith(PREFIX))
.forEach(k => dict[k.slice(PREFIX.length)] = GM_getValue(k, ''));
console.log('[NovelAI Prompt Preset Manager] Preset dict built.');
return dict;
}
/* ページ側へ JS を注入 */
installPatch() {
if (this._patchInstalled) return;
this._patchInstalled = true;
const TARGET = this.TARGET_PATH;
const naiRemainValue = GM_getValue(TOKEN_REMAIN_TRG, false);
const debugModeValue = GM_getValue(DEBUG_MODE_TRG, false);
const initialDict = JSON.stringify(this._dictCache)
.replace(/\\/g, '\\\\').replace(/`/g, '\\`').replace(/\$\{/g,'\\${');
const patchCode = `
(function(){
window.__naiPresetDict = JSON.parse(\`${initialDict}\`);
window.__naiRemain = ${naiRemainValue};
window.__naiDebugMode = ${debugModeValue};
const debugLog = (...args) => { if (window.__naiDebugMode) console.log(...args); };
const errorLog = (...args) => console.error(...args);
window.addEventListener('naiRemainUpdate', e => { window.__naiRemain = e.detail; });
window.addEventListener('naiPresetUpdate', e => { window.__naiPresetDict = e.detail; });
window.addEventListener('naiDebugUpdate', e => {window.__naiDebugMode = e.detail; });
const tokenRe = /__([A-Za-z0-9_., :/|ぁ-んァ-ヶ一-龯★☆-]+?)__/g;
const replace = s => {
return s.replace(tokenRe, (match, tokenName) => {
if (Object.prototype.hasOwnProperty.call(window.__naiPresetDict, tokenName)) {
let presetValue = window.__naiPresetDict[tokenName];
if (typeof presetValue === 'string') {
const newlineRegex = new RegExp('\\\\r?\\\\n', 'g');
const newlineTestRegex = new RegExp('\\\\r?\\\\n');
if (newlineTestRegex.test(presetValue)){
return '||' + presetValue.replace(newlineRegex, '|') + '||';
}
else return presetValue;
}
return presetValue;
}
return match;
});
};
const deep = o => (typeof o==='string') ? replace(o)
: Array.isArray(o) ? o.map(deep)
: o && typeof o==='object'
? Object.fromEntries(Object.entries(o).map(([k,v])=>[k,deep(v)]))
: o;
const origFetch = window.fetch;
window.fetch = async function(input, init) {
const url = typeof input === 'string' ? input : input.url;
const method = init?.method || (input instanceof Request ? input.method : 'GET').toUpperCase();
if (method === 'POST' && url.includes('${TARGET}')) {
const body = init?.body || (input instanceof Request ? input.body : null);
if (!body) return origFetch.call(this, input, init);
console.log('[NovelAI Prompt Preset Manager] Intercepting POST to ${TARGET}');
const bodyText = typeof body === 'string' ? body : await new Response(body).text();
if (window.__naiRemain) {
try {
const jsonData = JSON.parse(bodyText);
const params = jsonData.parameters || {};
const safeCopy = (obj) => obj ? JSON.parse(JSON.stringify(obj)) : null;
window.__naiLastPromptData = {
inputPrompt : jsonData.input ?? '',
caption : safeCopy(params.v4_prompt?.caption),
negCaption : safeCopy(params.v4_negative_prompt?.caption),
negative : jsonData.negative_prompt ?? ''
};
debugLog('[PresetMgr] Stored raw prompt data for patching.');
} catch(e) { console.error('[PresetMgr] Error parsing prompt data:', e); }
}
const modifiedBody = JSON.stringify(deep(JSON.parse(bodyText)));
if( modifiedBody === bodyText ) {
debugLog('[PresetMgr] No changes in body, skipping patching.');
return origFetch.call(this, input, init);
}
let finalInput, finalInit;
if (typeof input === 'string') {
finalInput = input;
finalInit = {...init, body: modifiedBody};
} else {
finalInput = new Request(input, { body: modifiedBody });
finalInit = undefined;
}
const res = await origFetch.call(this, finalInput, finalInit);
if (window.__naiRemain && res.ok && res.headers.get('Content-Type')?.includes('binary/octet-stream')) {
debugLog('[PresetMgr] binary/octet-stream response. Trying to process as ZIP.');
try {
const zipBuffer = await res.clone().arrayBuffer();
debugLog('[PresetMgr] Got zipBuffer, length:', zipBuffer.byteLength);
if (typeof window.JSZip === 'undefined') {
console.error('[PresetMgr] window.JSZip is undefined. Cannot process ZIP.');
return res;
}
debugLog('[PresetMgr] JSZip found in window. Loading ZIP...');
const zip = await window.JSZip.loadAsync(zipBuffer);
debugLog('[PresetMgr] ZIP loaded. Searching for PNG file...');
const pngFile = zip.file(/image_\\d+\\.png/)[0];
if (pngFile) {
debugLog('[PresetMgr] Found PNG in ZIP:', pngFile.name);
try {
debugLog('[PresetMgr] Attempting pngFile.async("arraybuffer")...');
// Promiseを一度変数に受けてからawaitする
const arrayBufferPromise = pngFile.async('arraybuffer');
const pngFileBuffer = await arrayBufferPromise;
debugLog('[PresetMgr] Successfully got pngFileBuffer, length:', pngFileBuffer.byteLength);
const patchedBuffer = patchPng(pngFileBuffer);
if (patchedBuffer) {
zip.file(pngFile.name, patchedBuffer);
debugLog('[PresetMgr] PNG patched. Generating new ZIP...');
const newZipBuffer = await zip.generateAsync({type: 'arraybuffer', compression: "DEFLATE", compressionOptions: {level: 1}});
debugLog('[PresetMgr] Repacked zip with patched PNG. Returning new response.');
const newHeaders = new Headers(res.headers);
return new Response(newZipBuffer, { status: res.status, statusText: res.statusText, headers: newHeaders });
} else {
errorLog('[PresetMgr] patchPng returned null or undefined. Returning original response (from cloned zipBuffer).');
const originalResponseCloneForFallback = new Response(zipBuffer, { status: res.status, statusText: res.statusText, headers: res.headers });
return originalResponseCloneForFallback;
}
} catch (e_png_processing) {
console.error('[PresetMgr] Error during pngFile.async or subsequent patching/zipping:', e_png_processing);
// エラーが発生した場合、元のZIPデータからレスポンスを再生成して返す
const errorFallbackResponse = new Response(zipBuffer, { status: res.status, statusText: res.statusText, headers: res.headers });
return errorFallbackResponse;
}
} else {
errorLog('[PresetMgr] No PNG file found in the ZIP. Returning original response.');
}
} catch (e_zip_load) {
console.error('[PresetMgr] Failed to load/process ZIP from binary/octet-stream:', e_zip_load);
}
}
return res;
} else {
return origFetch.call(this, input, init);
}
};
/*── PNG書き換えとバイナリヘルパー関数 ──*/
function patchPng(arrayBuf) {
debugLog('[PresetMgr] Patching PNG metadata...');
let data = new Uint8Array(arrayBuf);
const raw = window.__naiLastPromptData;
if (!raw) { return null; }
let p = 8;
let modified = false; // 変更があったかどうかを追跡するフラグ
while (p < data.length) {
const originalLen = readUint32(data, p);
const type = String.fromCharCode(...data.subarray(p + 4, p + 8));
if (type === 'tEXt') {
let q = p + 8;
let key = '';
while (data[q] && q < p + 8 + originalLen) { // バッファオーバーランを防止
key += String.fromCharCode(data[q++]);
}
// 'Description'キーの処理
if (key === 'Description' && raw.inputPrompt) {
debugLog('[PresetMgr] Found "tEXt" chunk with "Description" key.');
q++;
const textOff = q;
const oldLen = originalLen - (key.length + 1);
const newTxt = new TextEncoder().encode(raw.inputPrompt);
const delta = newTxt.length - oldLen;
const out = new Uint8Array(data.length + delta);
const head = p + 8 + key.length + 1;
out.set(data.subarray(0, head), 0);
out.set(newTxt, head);
out.set(data.subarray(head + oldLen), head + newTxt.length);
const newChunkLen = newTxt.length + key.length + 1;
writeUint32(out, p, newChunkLen);
const crc = crc32(out, p + 4, 4 + newChunkLen);
writeUint32(out, p + 8 + newChunkLen, crc);
debugLog('[PresetMgr] PNG description patched.');
data = out; // メインのバッファを更新
modified = true;
}
// 'Comment'キーの処理
else if (key === 'Comment') {
debugLog('[PresetMgr] Found "tEXt" chunk with "Comment" key.');
q++;
const textOff = q;
const oldLen = originalLen - (key.length + 1);
const oldTxt = new TextDecoder().decode(data.subarray(textOff, textOff + oldLen));
const meta = JSON.parse(oldTxt);
// metaデータを更新
if (raw.inputPrompt) { meta.prompt = raw.inputPrompt; }
if (raw.caption && meta.v4_prompt?.caption) {
meta.v4_prompt.caption.base_caption = raw.caption.base_caption ?? meta.v4_prompt.caption.base_caption;
meta.v4_prompt.caption.char_captions = raw.caption.char_captions ?? meta.v4_prompt.caption.char_captions;
}
if (raw.negCaption && meta.v4_negative_prompt?.caption) {
meta.v4_negative_prompt.caption.base_caption = raw.negCaption.base_caption ?? meta.v4_negative_prompt.caption.base_caption;
meta.v4_negative_prompt.caption.char_captions = raw.negCaption.char_captions ?? meta.v4_negative_prompt.caption.char_captions;
}
if (raw.negative) { meta.uc = raw.negative }
const newTxt = new TextEncoder().encode(JSON.stringify(meta));
const delta = newTxt.length - oldLen;
const out = new Uint8Array(data.length + delta);
const head = p + 8 + key.length + 1;
out.set(data.subarray(0, head), 0);
out.set(newTxt, head);
out.set(data.subarray(head + oldLen), head + newTxt.length);
const newChunkLen = newTxt.length + key.length + 1;
writeUint32(out, p, newChunkLen);
const crc = crc32(out, p + 4, 4 + newChunkLen);
writeUint32(out, p + 8 + newChunkLen, crc);
debugLog('[PresetMgr] PNG metadata patched.');
data = out; // メインのバッファを更新
modified = true;
}
}
// 更新された可能性のあるチャンクの長さを使ってポインタを進める
p += 12 + readUint32(data, p);
}
if (modified) {
console.log('[NovelAI Prompt Preset Manager] PNG metadata patching finished.');
window.__naiLastPromptData = null; // すべての処理が終わったのでクリア
return data.buffer;
}
debugLog('[PresetMgr] Target chunk (tEXt with Comment or Description) not found.');
return null;
}
function readUint32(u8, idx) { return (u8[idx] << 24) | (u8[idx+1] << 16) | (u8[idx+2] << 8) | u8[idx+3]; }
function writeUint32(u8, idx, v) { u8[idx] = v >>> 24; u8[idx+1] = v >>> 16; u8[idx+2] = v >>> 8; u8[idx+3] = v & 0xFF; }
function crc32(u8, off, len) {
let c = ~0 >>> 0;
for (let i = 0; i < len; i++) {
c ^= u8[off + i];
for (let k = 0; k < 8; k++) c = (c & 1) ? (c >>> 1) ^ 0xEDB88320 : (c >>> 1);
}
return (~c) >>> 0;
}
})();`;
const scr = document.createElement('script');
scr.textContent = patchCode;
document.documentElement.appendChild(scr);
scr.remove();
console.log('[NovelAI Prompt Preset Manager] Patches installed to handle JSON/PNG.');
}
/* プリセット辞書を更新する */
updateDict() {
this._dictCache = this.buildDict(); //キャッシュを更新
window.dispatchEvent(
new CustomEvent('naiPresetUpdate', { detail: this._dictCache })
);
}
/* キャッシュされた辞書を返す */
getDict() {
return this._dictCache;
}
}
const jsonManagerSingleton = new JsonManager(); //json manager初期化
(function () {
'use strict';
window.__naiPmObserver ??
(window.__naiPmObserver = new ProseMirrorObserver(jsonManagerSingleton));
window.__naiPromptObserver ??=
(window.__naiPromptObserver = new PromptBoxObserver());
})();
