☰

AI提示词助手

AI提示词管理和快速输入工具

  1. // ==UserScript==
  2. // @name         AI提示词助手
  3. // @namespace    http://tampermonkey.net/
  4. // @version      0.2
  5. // @description  AI提示词管理和快速输入工具
  6. // @author       初沐 (https://github.com/chumu)
  7. // @license      MIT
  8. // @match        https://pro.yrai.cc/new
  9. // @grant        GM_addStyle
  10. // @grant        GM_download
  11. // @grant        GM_xmlhttpRequest
  12. // @connect      cdn.jsdelivr.net
  13. // @run-at       document-end
  14. // ==/UserScript==
  15.  
  16. (function() {
  17.     'use strict';
  18.  
  19.     // 首先加载xlsx库
  20.     function loadXLSX() {
  21.         return new Promise((resolve, reject) => {
  22.             GM_xmlhttpRequest({
  23.                 method: 'GET',
  24.                 url: 'https://cdn.jsdelivr.net/npm/xlsx/dist/xlsx.full.min.js',
  25.                 onload: function(response) {
  26.                     try {
  27.                         // 创建script标签并执行代码
  28.                         const script = document.createElement('script');
  29.                         script.textContent = response.responseText;
  30.                         document.head.appendChild(script);
  31.                         resolve();
  32.                     } catch (error) {
  33.                         reject(error);
  34.                     }
  35.                 },
  36.                 onerror: function(error) {
  37.                     reject(error);
  38.                 }
  39.             });
  40.         });
  41.     }
  42.  
  43.     // 等待xlsx库加载完成后初始化UI
  44.     loadXLSX().then(() => {
  45.         // 原有的初始化代码
  46.         createUI();
  47.     }).catch(error => {
  48.         console.error('加载xlsx库失败:', error);
  49.         // 即使加载失败也创建UI，只是Excel功能不可用
  50.         createUI();
  51.     });
  52.  
  53.     // 添加样式
  54.     GM_addStyle(`
  55.         .prompt-helper {
  56.             position: fixed;
  57.             right: 20px;
  58.             top: 20px;
  59.             background: white;
  60.             border: 1px solid #ccc;
  61.             padding: 20px;
  62.             border-radius: 12px;
  63.             box-shadow: 0 4px 20px rgba(0,0,0,0.15);
  64.             z-index: 9999;
  65.             width: 380px;
  66.             font-family: Arial, sans-serif;
  67.             transition: transform 0.3s ease;
  68.         }
  69.         .prompt-helper.collapsed {
  70.             transform: translateX(calc(100% + 20px));
  71.         }
  72.         .toggle-button {
  73.             position: fixed;
  74.             right: 20px;
  75.             top: 20px;
  76.             z-index: 9998;
  77.             padding: 8px 15px;
  78.             background: #4a90e2;
  79.             color: white;
  80.             border: none;
  81.             border-radius: 6px;
  82.             cursor: pointer;
  83.             box-shadow: 0 2px 8px rgba(0,0,0,0.1);
  84.         }
  85.         .toggle-button:hover {
  86.             background: #357abd;
  87.         }
  88.         .manage-categories {
  89.             margin-top: 10px;
  90.             padding: 10px;
  91.             background: #f9f9f9;
  92.             border-radius: 8px;
  93.             display: none;
  94.         }
  95.         .manage-categories.active {
  96.             display: block;
  97.         }
  98.         .category-list {
  99.             margin: 10px 0;
  100.             padding: 5px;
  101.             background: white;
  102.             border: 1px solid #eee;
  103.             border-radius: 6px;
  104.             max-height: 150px;
  105.             overflow-y: auto;
  106.         }
  107.         .category-item {
  108.             display: flex;
  109.             justify-content: space-between;
  110.             align-items: center;
  111.             padding: 8px;
  112.             border-bottom: 1px solid #eee;
  113.         }
  114.         .category-item:last-child {
  115.             border-bottom: none;
  116.         }
  117.         .prompt-helper h3 {
  118.             margin: 0 0 15px 0;
  119.             color: #333;
  120.             font-size: 18px;
  121.             border-bottom: 2px solid #eee;
  122.             padding-bottom: 10px;
  123.             display: flex;
  124.             align-items: center;
  125.             justify-content: space-between;
  126.         }
  127.         .prompt-helper h3 .title {
  128.             flex: 1;
  129.         }
  130.         .prompt-helper h3 .buttons {
  131.             display: flex;
  132.             gap: 8px;
  133.             align-items: center;
  134.         }
  135.         .prompt-helper h3 button {
  136.             padding: 4px 12px;
  137.             border: 1px solid #ddd;
  138.             border-radius: 4px;
  139.             background: #f5f5f5;
  140.             cursor: pointer;
  141.             font-size: 14px;
  142.             line-height: 1.5;
  143.             transition: all 0.2s;
  144.         }
  145.         .prompt-helper h3 button:hover {
  146.             background: #e5e5e5;
  147.         }
  148.         .prompt-helper button {
  149.             margin: 5px;
  150.             padding: 6px 12px;
  151.             border: 1px solid #ddd;
  152.             border-radius: 6px;
  153.             background: #f5f5f5;
  154.             cursor: pointer;
  155.             transition: all 0.2s;
  156.         }
  157.         .prompt-helper button:hover {
  158.             background: #e5e5e5;
  159.             transform: translateY(-1px);
  160.         }
  161.         .prompt-helper input, .prompt-helper select, .prompt-helper textarea {
  162.             padding: 6px 10px;
  163.             border: 1px solid #ddd;
  164.             border-radius: 6px;
  165.             margin-bottom: 8px;
  166.         }
  167.         .prompt-list {
  168.             margin-top: 15px;
  169.             max-height: 400px;
  170.             overflow-y: auto;
  171.         }
  172.         .prompt-item {
  173.             margin: 8px 0;
  174.             padding: 10px;
  175.             border: 1px solid #eee;
  176.             border-radius: 8px;
  177.             background: #fafafa;
  178.         }
  179.         .prompt-item:hover {
  180.             border-color: #ddd;
  181.             background: #fff;
  182.             box-shadow: 0 2px 8px rgba(0,0,0,0.05);
  183.         }
  184.         .prompt-content {
  185.             margin: 5px 0;
  186.             font-size: 14px;
  187.             color: #666;
  188.         }
  189.         .prompt-comment {
  190.             font-size: 12px;
  191.             color: #888;
  192.             margin-top: 5px;
  193.             font-style: italic;
  194.         }
  195.         .category-section {
  196.             margin: 15px 0;
  197.             padding: 10px;
  198.             border: 1px solid #eee;
  199.             border-radius: 8px;
  200.         }
  201.         .category-title {
  202.             font-weight: bold;
  203.             color: #444;
  204.             margin-bottom: 10px;
  205.         }
  206.         .search-box {
  207.             width: 100%;
  208.             margin-bottom: 15px;
  209.         }
  210.         .controls {
  211.             display: flex;
  212.             gap: 8px;
  213.             margin-bottom: 10px;
  214.         }
  215.         .add-prompt-form {
  216.             display: none;
  217.             padding: 10px;
  218.             background: #f9f9f9;
  219.             border-radius: 8px;
  220.             margin: 10px 0;
  221.         }
  222.         .add-prompt-form.active {
  223.             display: block;
  224.         }
  225.         #addCategory {
  226.             font-size: 12px;
  227.             padding: 4px 8px;
  228.         }
  229.         #newCategory {
  230.             font-size: 12px;
  231.         }
  232.         .import-section {
  233.             margin: 10px 0;
  234.             padding: 10px;
  235.             background: #f9f9f9;
  236.             border-radius: 8px;
  237.             border: 2px dashed #ccc;
  238.             text-align: center;
  239.             transition: all 0.3s;
  240.             overflow: hidden;
  241.         }
  242.         .import-section.collapsed {
  243.             padding: 0;
  244.             height: 0;
  245.             border: none;
  246.             margin: 0;
  247.         }
  248.         .import-section.dragover {
  249.             background: #e9e9e9;
  250.             border-color: #4a90e2;
  251.         }
  252.         .import-toggle {
  253.             display: flex;
  254.             align-items: center;
  255.             padding: 8px;
  256.             background: #f5f5f5;
  257.             border-radius: 4px;
  258.             cursor: pointer;
  259.             user-select: none;
  260.             margin-bottom: 10px;
  261.             transition: all 0.3s;
  262.         }
  263.         .import-toggle:hover {
  264.             background: #e8e8e8;
  265.         }
  266.         .import-toggle .arrow {
  267.             width: 0;
  268.             height: 0;
  269.             border-left: 5px solid transparent;
  270.             border-right: 5px solid transparent;
  271.             border-top: 5px solid #666;
  272.             margin-right: 8px;
  273.             transition: transform 0.3s;
  274.         }
  275.         .import-toggle.collapsed .arrow {
  276.             transform: rotate(-90deg);
  277.         }
  278.         .import-toggle-text {
  279.             color: #666;
  280.             font-size: 14px;
  281.         }
  282.         #fileInput {
  283.             display: none;
  284.         }
  285.         .display-mode-switch {
  286.             display: flex;
  287.             align-items: center;
  288.             gap: 8px;
  289.             margin: 10px 0;
  290.             padding: 8px;
  291.             background: #f5f5f5;
  292.             border-radius: 6px;
  293.         }
  294.         .display-mode-switch label {
  295.             font-size: 14px;
  296.             color: #666;
  297.         }
  298.         .prompt-content.simplified {
  299.             max-height: 1.5em;
  300.             overflow: hidden;
  301.             text-overflow: ellipsis;
  302.             white-space: nowrap;
  303.             cursor: pointer;
  304.         }
  305.         .prompt-content.simplified:hover {
  306.             color: #4a90e2;
  307.         }
  308.         .settings-menu {
  309.             display: none;
  310.             position: absolute;
  311.             top: 50px;
  312.             right: 20px;
  313.             background: white;
  314.             border: 1px solid #ddd;
  315.             border-radius: 8px;
  316.             padding: 15px;
  317.             box-shadow: 0 2px 10px rgba(0,0,0,0.1);
  318.             z-index: 10001;
  319.             width: 250px;
  320.         }
  321.         .settings-menu.active {
  322.             display: block;
  323.         }
  324.         .settings-menu h4 {
  325.             margin: 0 0 10px 0;
  326.             padding-bottom: 8px;
  327.             border-bottom: 1px solid #eee;
  328.             color: #333;
  329.         }
  330.         .settings-item {
  331.             margin: 8px 0;
  332.         }
  333.         .settings-item label {
  334.             display: flex;
  335.             align-items: center;
  336.             gap: 8px;
  337.             color: #666;
  338.             font-size: 14px;
  339.         }
  340.         .settings-item input[type="checkbox"] {
  341.             margin: 0;
  342.         }
  343.         .prompt-item {
  344.             position: relative;
  345.         }
  346.         .prompt-name {
  347.             font-weight: bold;
  348.             margin-bottom: 5px;
  349.         }
  350.         .prompt-content {
  351.             margin: 5px 0;
  352.             color: #333;
  353.         }
  354.         .prompt-content.simplified {
  355.             max-height: 1.5em;
  356.             overflow: hidden;
  357.             text-overflow: ellipsis;
  358.             white-space: nowrap;
  359.             cursor: pointer;
  360.         }
  361.         .prompt-comment {
  362.             font-size: 12px;
  363.             color: #666;
  364.             font-style: italic;
  365.             margin-top: 3px;
  366.         }
  367.         .prompt-category {
  368.             font-size: 12px;
  369.             color: #4a90e2;
  370.             margin-top: 3px;
  371.         }
  372.         .settings-menu h4 {
  373.             margin: 0 0 10px 0;
  374.             padding-bottom: 8px;
  375.             border-bottom: 1px solid #eee;
  376.             color: #333;
  377.         }
  378.         .settings-menu .author-info {
  379.             margin-top: 15px;
  380.             padding-top: 15px;
  381.             border-top: 1px solid #eee;
  382.             font-size: 13px;
  383.             color: #666;
  384.         }
  385.         .settings-menu .author-info p {
  386.             margin: 5px 0;
  387.             line-height: 1.4;
  388.         }
  389.         .settings-menu .author-info a {
  390.             color: #4a90e2;
  391.             text-decoration: none;
  392.         }
  393.         .settings-menu .author-info a:hover {
  394.             text-decoration: underline;
  395.         }
  396.     `);
  397.  
  398.     // 获取保存的分类
  399.     function getCategories() {
  400.         return JSON.parse(localStorage.getItem('aiPromptCategories') || '["洗稿", "创作", "翻译"]');
  401.     }
  402.  
  403.     // 保存分类
  404.     function saveCategories(categories) {
  405.         localStorage.setItem('aiPromptCategories', JSON.stringify(categories));
  406.     }
  407.  
  408.     // 更新分类列表
  409.     function updateCategoryList() {
  410.         const categoryList = document.getElementById('categoryList');
  411.         const categories = getCategories();
  412.         
  413.         categoryList.innerHTML = categories.map(category => `
  414.             <div class="category-item" data-category="${category}">
  415.                 <span class="category-name">${category}</span>
  416.                 <button class="btn btn-danger" onclick="deleteCategory('${category}')">删除</button>
  417.             </div>
  418.         `).join('');
  419.     }
  420.  
  421.     // 删除分类
  422.     function deleteCategory(name) {
  423.         if (confirm(`确定要删除分类"${name}"吗？`)) {
  424.             const categories = getCategories();
  425.             const newCategories = categories.filter(c => c !== name);
  426.             saveCategories(newCategories);
  427.             updateCategoryList();
  428.             updateCategorySelectors();
  429.         }
  430.     }
  431.  
  432.     // 清空表单
  433.     function clearForm() {
  434.         document.getElementById('promptName').value = '';
  435.         document.getElementById('promptContent').value = '';
  436.         document.getElementById('promptComment').value = '';
  437.     }
  438.  
  439.     // 保存提示词
  440.     function savePrompt(name, content, comment, category) {
  441.         const prompts = JSON.parse(localStorage.getItem('aiPrompts') || '{}');
  442.         prompts[name] = {
  443.             content: content,
  444.             comment: comment || '',
  445.             category: category
  446.         };
  447.         localStorage.setItem('aiPrompts', JSON.stringify(prompts));
  448.     }
  449.  
  450.     // 删除提示词
  451.     function deletePrompt(name) {
  452.         const prompts = JSON.parse(localStorage.getItem('aiPrompts') || '{}');
  453.         if (prompts[name]) {
  454.             delete prompts[name];
  455.             localStorage.setItem('aiPrompts', JSON.stringify(prompts));
  456.             loadPrompts();
  457.             showToast('提示词已删除');
  458.         }
  459.     }
  460.  
  461.     // 复制提示词内容
  462.     function copyPromptContent(content) {
  463.         // 确保content不为空
  464.         if (!content) {
  465.             console.error('复制内容为空');
  466.             return;
  467.         }
  468.  
  469.         // 使用 navigator.clipboard API
  470.         if (navigator.clipboard && window.isSecureContext) {
  471.             navigator.clipboard.writeText(content).then(() => {
  472.                 showToast('复制成功');
  473.             }).catch((err) => {
  474.                 console.error('Clipboard API failed:', err);
  475.                 fallbackCopy(content);
  476.             });
  477.         } else {
  478.             fallbackCopy(content);
  479.         }
  480.     }
  481.  
  482.     // 回退的复制方法
  483.     function fallbackCopy(content) {
  484.         const textarea = document.createElement('textarea');
  485.         textarea.value = content;
  486.         textarea.style.position = 'fixed';
  487.         textarea.style.left = '-9999px';
  488.         textarea.style.top = '0';
  489.         document.body.appendChild(textarea);
  490.         textarea.focus();
  491.         textarea.select();
  492.  
  493.         try {
  494.             document.execCommand('copy');
  495.             showToast('复制成功');
  496.         } catch (err) {
  497.             console.error('复制失败:', err);
  498.             showToast('复制失败');
  499.         }
  500.  
  501.         document.body.removeChild(textarea);
  502.     }
  503.  
  504.     // 显示提示信息
  505.     function showToast(message) {
  506.         const toast = document.createElement('div');
  507.         toast.style.cssText = `
  508.             position: fixed;
  509.             bottom: 20px;
  510.             left: 50%;
  511.             transform: translateX(-50%);
  512.             background: rgba(0, 0, 0, 0.7);
  513.             color: white;
  514.             padding: 8px 16px;
  515.             border-radius: 4px;
  516.             font-size: 14px;
  517.             z-index: 10000;
  518.         `;
  519.         toast.textContent = message;
  520.         document.body.appendChild(toast);
  521.         setTimeout(() => document.body.removeChild(toast), 2000);
  522.     }
  523.  
  524.     // 获取显示设置
  525.     function getDisplaySettings() {
  526.         return JSON.parse(localStorage.getItem('promptDisplaySettings') || JSON.stringify({
  527.             showName: true,
  528.             showContent: true,
  529.             showComment: true,
  530.             showCategory: true,
  531.             isSimplified: false
  532.         }));
  533.     }
  534.  
  535.     // 保存显示设置
  536.     function saveDisplaySettings(settings) {
  537.         localStorage.setItem('promptDisplaySettings', JSON.stringify(settings));
  538.     }
  539.  
  540.     // 加载提示词
  541.     function loadPrompts(searchTerm = '') {
  542.         const promptList = document.getElementById('promptList');
  543.         const prompts = JSON.parse(localStorage.getItem('aiPrompts') || '{}');
  544.         const selectedCategory = document.getElementById('categorySelect').value;
  545.         const settings = getDisplaySettings();
  546.         
  547.         promptList.innerHTML = '';
  548.         
  549.         // 按分类组织提示词
  550.         const categorizedPrompts = {};
  551.         
  552.         Object.entries(prompts).forEach(([name, data]) => {
  553.             if (searchTerm && 
  554.                 !name.toLowerCase().includes(searchTerm.toLowerCase()) && 
  555.                 !data.content.toLowerCase().includes(searchTerm.toLowerCase())) {
  556.                 return;
  557.             }
  558.             
  559.             if (selectedCategory !== 'all' && data.category !== selectedCategory) {
  560.                 return;
  561.             }
  562.  
  563.             if (!categorizedPrompts[data.category]) {
  564.                 categorizedPrompts[data.category] = [];
  565.             }
  566.             categorizedPrompts[data.category].push({name, ...data});
  567.         });
  568.  
  569.         Object.entries(categorizedPrompts).forEach(([category, items]) => {
  570.             const categorySection = document.createElement('div');
  571.             categorySection.className = 'category-section';
  572.             categorySection.innerHTML = `<div class="category-title">${category}</div>`;
  573.  
  574.             items.forEach(item => {
  575.                 const div = document.createElement('div');
  576.                 div.className = 'prompt-item';
  577.                 
  578.                 // 根据设置显示内容
  579.                 if (settings.showName) {
  580.                     const nameDiv = document.createElement('div');
  581.                     nameDiv.className = 'prompt-name';
  582.                     nameDiv.textContent = item.name;
  583.                     div.appendChild(nameDiv);
  584.                 }
  585.                 
  586.                 if (settings.showContent) {
  587.                     const contentDiv = document.createElement('div');
  588.                     contentDiv.className = `prompt-content ${settings.isSimplified ? 'simplified' : ''}`;
  589.                     contentDiv.textContent = item.content;
  590.                     if (settings.isSimplified) {
  591.                         contentDiv.title = item.content;
  592.                         contentDiv.addEventListener('click', function() {
  593.                             this.classList.toggle('simplified');
  594.                         });
  595.                     }
  596.                     div.appendChild(contentDiv);
  597.                 }
  598.                 
  599.                 if (settings.showComment && item.comment) {
  600.                     const commentDiv = document.createElement('div');
  601.                     commentDiv.className = 'prompt-comment';
  602.                     commentDiv.textContent = `注释: ${item.comment}`;
  603.                     div.appendChild(commentDiv);
  604.                 }
  605.  
  606.                 if (settings.showCategory) {
  607.                     const categoryDiv = document.createElement('div');
  608.                     categoryDiv.className = 'prompt-category';
  609.                     categoryDiv.textContent = `分类: ${item.category}`;
  610.                     div.appendChild(categoryDiv);
  611.                 }
  612.                 
  613.                 // 创建按钮容器
  614.                 const buttonDiv = document.createElement('div');
  615.                 buttonDiv.style.textAlign = 'right';
  616.                 
  617.                 // 创建复制按钮
  618.                 const copyButton = document.createElement('button');
  619.                 copyButton.textContent = '复制';
  620.                 copyButton.dataset.content = item.content;
  621.                 copyButton.addEventListener('click', function() {
  622.                     copyPromptContent(this.dataset.content);
  623.                 });
  624.                 
  625.                 // 创建删除按钮
  626.                 const deleteButton = document.createElement('button');
  627.                 deleteButton.textContent = '删除';
  628.                 deleteButton.dataset.name = item.name;
  629.                 deleteButton.addEventListener('click', function() {
  630.                     deletePrompt(this.dataset.name);
  631.                 });
  632.                 
  633.                 buttonDiv.appendChild(copyButton);
  634.                 buttonDiv.appendChild(deleteButton);
  635.                 div.appendChild(buttonDiv);
  636.                 
  637.                 categorySection.appendChild(div);
  638.             });
  639.  
  640.             promptList.appendChild(categorySection);
  641.         });
  642.     }
  643.  
  644.     // 初始化事件
  645.     function initEvents() {
  646.         // 搜索功能
  647.         document.getElementById('searchPrompt').addEventListener('input', function(e) {
  648.             loadPrompts(e.target.value);
  649.         });
  650.  
  651.         // 分类筛选
  652.         document.getElementById('categorySelect').addEventListener('change', function() {
  653.             loadPrompts(document.getElementById('searchPrompt').value);
  654.         });
  655.  
  656.         // 新增提示词表单显示/隐藏
  657.         document.getElementById('addNewPrompt').addEventListener('click', function() {
  658.             document.getElementById('addPromptForm').classList.add('active');
  659.         });
  660.  
  661.         document.getElementById('cancelPrompt').addEventListener('click', function() {
  662.             document.getElementById('addPromptForm').classList.remove('active');
  663.             clearForm();
  664.         });
  665.  
  666.         // 保存提示词
  667.         document.getElementById('savePrompt').addEventListener('click', function() {
  668.             const name = document.getElementById('promptName').value;
  669.             const content = document.getElementById('promptContent').value;
  670.             const comment = document.getElementById('promptComment').value;
  671.             const category = document.getElementById('promptCategory').value;
  672.  
  673.             if (name && content) {
  674.                 savePrompt(name, content, comment, category);
  675.                 clearForm();
  676.                 document.getElementById('addPromptForm').classList.remove('active');
  677.                 loadPrompts();
  678.             }
  679.         });
  680.  
  681.         // 管理分类按钮
  682.         document.getElementById('manageCategories').addEventListener('click', function() {
  683.             document.getElementById('manageCategoriesForm').classList.toggle('active');
  684.         });
  685.  
  686.         // 添加新分类
  687.         document.getElementById('addCategory').addEventListener('click', function() {
  688.             const newCategoryInput = document.getElementById('newCategory');
  689.             const newCategory = newCategoryInput.value.trim();
  690.             
  691.             if (newCategory && !getCategories().includes(newCategory)) {
  692.                 const categories = getCategories();
  693.                 categories.push(newCategory);
  694.                 saveCategories(categories);
  695.                 
  696.                 updateCategorySelectors();
  697.                 updateCategoryList();
  698.                 newCategoryInput.value = '';
  699.             }
  700.         });
  701.  
  702.         // 窗口控制
  703.         document.getElementById('toggleHelper').addEventListener('click', function() {
  704.             const helper = document.querySelector('.prompt-helper');
  705.             const toggleButton = document.querySelector('.toggle-button');
  706.             helper.classList.toggle('collapsed');
  707.             toggleButton.style.display = helper.classList.contains('collapsed') ? 'block' : 'none';
  708.         });
  709.  
  710.         // 为promptList添加事件委托
  711.         document.getElementById('promptList').addEventListener('click', function(e) {
  712.             if (e.target.tagName === 'BUTTON') {
  713.                 if (e.target.textContent === '复制') {
  714.                     copyPromptContent(decodeURIComponent(e.target.getAttribute('data-content')));
  715.                 } else if (e.target.textContent === '删除') {
  716.                     deletePrompt(decodeURIComponent(e.target.getAttribute('data-name')));
  717.                 }
  718.             }
  719.         });
  720.  
  721.         // 导入文件按钮点击事件
  722.         document.getElementById('importBtn').addEventListener('click', function() {
  723.             document.getElementById('fileInput').click();
  724.         });
  725.  
  726.         // 文件选择变化事件
  727.         document.getElementById('fileInput').addEventListener('change', handleFileImport);
  728.  
  729.         // 下载模板按钮点击事件
  730.         document.getElementById('downloadTemplate').addEventListener('click', downloadTemplate);
  731.  
  732.         // 添加拖拽相关事件
  733.         const importSection = document.querySelector('.import-section');
  734.         
  735.         ['dragenter', 'dragover', 'dragleave', 'drop'].forEach(eventName => {
  736.             importSection.addEventListener(eventName, preventDefaults, false);
  737.             document.body.addEventListener(eventName, preventDefaults, false);
  738.         });
  739.  
  740.         function preventDefaults(e) {
  741.             e.preventDefault();
  742.             e.stopPropagation();
  743.         }
  744.  
  745.         ['dragenter', 'dragover'].forEach(eventName => {
  746.             importSection.addEventListener(eventName, highlight, false);
  747.         });
  748.  
  749.         ['dragleave', 'drop'].forEach(eventName => {
  750.             importSection.addEventListener(eventName, unhighlight, false);
  751.         });
  752.  
  753.         function highlight(e) {
  754.             importSection.classList.add('dragover');
  755.         }
  756.  
  757.         function unhighlight(e) {
  758.             importSection.classList.remove('dragover');
  759.         }
  760.  
  761.         importSection.addEventListener('drop', handleDrop, false);
  762.  
  763.         function handleDrop(e) {
  764.             const dt = e.dataTransfer;
  765.             const files = dt.files;
  766.             if (files.length) {
  767.                 handleFileImport({ target: { files: [files[0]] } });
  768.             }
  769.         }
  770.  
  771.         // 设置按钮点击事件
  772.         document.getElementById('settingsBtn').addEventListener('click', function(e) {
  773.             e.stopPropagation();
  774.             document.getElementById('settingsMenu').classList.toggle('active');
  775.         });
  776.  
  777.         // 点击其他地方关闭设置菜单
  778.         document.addEventListener('click', function(e) {
  779.             const settingsMenu = document.getElementById('settingsMenu');
  780.             if (!settingsMenu.contains(e.target) && e.target.id !== 'settingsBtn') {
  781.                 settingsMenu.classList.remove('active');
  782.             }
  783.         });
  784.  
  785.         // 设置选项变化事件
  786.         ['showName', 'showContent', 'showComment', 'showCategory', 'isSimplified'].forEach(id => {
  787.             document.getElementById(id).addEventListener('change', function() {
  788.                 const settings = getDisplaySettings();
  789.                 settings[id] = this.checked;
  790.                 saveDisplaySettings(settings);
  791.                 loadPrompts(document.getElementById('searchPrompt').value);
  792.             });
  793.         });
  794.  
  795.         // 导入区域显示/隐藏
  796.         document.getElementById('importToggle').addEventListener('click', function() {
  797.             const importSection = document.getElementById('importSection');
  798.             const importToggle = document.getElementById('importToggle');
  799.             importSection.classList.toggle('collapsed');
  800.             importToggle.classList.toggle('collapsed');
  801.         });
  802.     }
  803.  
  804.     // 更新分类选择器
  805.     function updateCategorySelectors() {
  806.         const categories = getCategories();
  807.         const selectors = document.querySelectorAll('#categorySelect, #promptCategory');
  808.         
  809.         selectors.forEach(selector => {
  810.             const currentValue = selector.value;
  811.             selector.innerHTML = selector.id === 'categorySelect' ? 
  812.                 '<option value="all">所有分类</option>' : '';
  813.             
  814.             categories.forEach(category => {
  815.                 selector.innerHTML += `<option value="${category}">${category}</option>`;
  816.             });
  817.             
  818.             if (currentValue && categories.includes(currentValue)) {
  819.                 selector.value = currentValue;
  820.             }
  821.         });
  822.     }
  823.  
  824.     // 下载Excel模板
  825.     function downloadTemplate() {
  826.         const template = [
  827.             ['名称', '内容', '注释', '分类'],
  828.             ['示例1', '这是一个示例提示词', '这是注释', '创作'],
  829.             ['示例2', '这是另一个示例', '可选的注释', '洗稿']
  830.         ];
  831.  
  832.         // 创建工作簿
  833.         const wb = XLSX.utils.book_new();
  834.         const ws = XLSX.utils.aoa_to_sheet(template);
  835.         XLSX.utils.book_append_sheet(wb, ws, "提示词");
  836.  
  837.         // 生成Excel文件并下载
  838.         XLSX.writeFile(wb, '提示词模板.xlsx');
  839.     }
  840.  
  841.     // 处理导入的文件
  842.     function handleFileImport(event) {
  843.         const file = event.target.files[0];
  844.         if (!file) return;
  845.  
  846.         const fileExtension = file.name.split('.').pop().toLowerCase();
  847.  
  848.         if (fileExtension === 'xlsx' || fileExtension === 'xls') {
  849.             // 处理Excel文件
  850.             const reader = new FileReader();
  851.             reader.onload = function(e) {
  852.                 try {
  853.                     const data = new Uint8Array(e.target.result);
  854.                     const workbook = XLSX.read(data, {type: 'array'});
  855.                     const firstSheetName = workbook.SheetNames[0];
  856.                     const worksheet = workbook.Sheets[firstSheetName];
  857.                     
  858.                     // 转换为数组格式
  859.                     const rows = XLSX.utils.sheet_to_json(worksheet, {header: 1});
  860.                     
  861.                     if (rows.length < 2) {
  862.                         showToast('Excel文件内容为空');
  863.                         return;
  864.                     }
  865.  
  866.                     const prompts = JSON.parse(localStorage.getItem('aiPrompts') || '{}');
  867.                     let importCount = 0;
  868.  
  869.                     // 跳过标题行
  870.                     for (let i = 1; i < rows.length; i++) {
  871.                         const row = rows[i];
  872.                         if (!row || row.length === 0) continue;
  873.  
  874.                         const [name, content, comment, category] = row;
  875.                         if (name && content) {
  876.                             prompts[name] = {
  877.                                 content: content,
  878.                                 comment: comment || '',
  879.                                 category: category || '其他'
  880.                             };
  881.                             importCount++;
  882.                         }
  883.                     }
  884.  
  885.                     localStorage.setItem('aiPrompts', JSON.stringify(prompts));
  886.                     showToast(`成功导入 ${importCount} 个提示词`);
  887.                     loadPrompts();
  888.                 } catch (err) {
  889.                     console.error('Excel解析失败:', err);
  890.                     showToast('Excel文件格式错误，请检查格式是否正确');
  891.                 }
  892.             };
  893.             reader.readAsArrayBuffer(file);
  894.         } else if (fileExtension === 'csv') {
  895.             // 处理CSV文件的现有代码
  896.             handleCSVImport(file);
  897.         } else {
  898.             showToast('不支持的文件格式，请使用.xlsx、.xls或.csv格式');
  899.         }
  900.     }
  901.  
  902.     // 处理CSV文件导入
  903.     function handleCSVImport(file) {
  904.         // 尝试不同的编码
  905.         function tryReadFile(encoding) {
  906.             return new Promise((resolve, reject) => {
  907.                 const reader = new FileReader();
  908.                 reader.onload = function(e) {
  909.                     try {
  910.                         const text = e.target.result;
  911.                         // 检查是否包含乱码字符
  912.                         if (text.includes('�')) {
  913.                             reject('encoding_error');
  914.                             return;
  915.                         }
  916.                         resolve(text);
  917.                     } catch (err) {
  918.                         reject(err);
  919.                     }
  920.                 };
  921.                 reader.onerror = reject;
  922.                 reader.readAsText(file, encoding);
  923.             });
  924.         }
  925.  
  926.         // 按顺序尝试不同的编码
  927.         const encodings = ['UTF-8', 'GBK', 'GB2312', 'BIG5'];
  928.         let currentEncodingIndex = 0;
  929.  
  930.         function tryNextEncoding() {
  931.             if (currentEncodingIndex >= encodings.length) {
  932.                 showToast('无法正确读取文件，请确保文件编码正确');
  933.                 return;
  934.             }
  935.  
  936.             const currentEncoding = encodings[currentEncodingIndex];
  937.             tryReadFile(currentEncoding)
  938.                 .then(text => {
  939.                     try {
  940.                         // 处理 BOM 标记
  941.                         const content = text.replace(/^\uFEFF/, '');
  942.                         const rows = content.split(/\r?\n/);
  943.                         const prompts = JSON.parse(localStorage.getItem('aiPrompts') || '{}');
  944.                         let importCount = 0;
  945.  
  946.                         // 跳过标题行
  947.                         for (let i = 1; i < rows.length; i++) {
  948.                             const row = rows[i].trim();
  949.                             if (!row) continue;
  950.  
  951.                             // 使用更可靠的CSV解析方式
  952.                             let columns = [];
  953.                             let currentColumn = '';
  954.                             let inQuotes = false;
  955.  
  956.                             // 手动解析CSV，处理引号内的逗号
  957.                             for (let j = 0; j < row.length; j++) {
  958.                                 const char = row[j];
  959.                                 if (char === '"') {
  960.                                     inQuotes = !inQuotes;
  961.                                 } else if (char === ',' && !inQuotes) {
  962.                                     columns.push(currentColumn.trim());
  963.                                     currentColumn = '';
  964.                                 } else {
  965.                                     currentColumn += char;
  966.                                 }
  967.                             }
  968.                             columns.push(currentColumn.trim());
  969.  
  970.                             // 清理引号
  971.                             columns = columns.map(col => col.replace(/^["']|["']$/g, ''));
  972.  
  973.                             if (columns.length >= 2) {
  974.                                 const [name, content, comment, category] = columns;
  975.                                 if (name && content) {
  976.                                     prompts[name] = {
  977.                                         content: content,
  978.                                         comment: comment || '',
  979.                                         category: category || '其他'
  980.                                     };
  981.                                     importCount++;
  982.                                 }
  983.                             }
  984.                         }
  985.  
  986.                         localStorage.setItem('aiPrompts', JSON.stringify(prompts));
  987.                         showToast(`成功导入 ${importCount} 个提示词`);
  988.                         loadPrompts();
  989.                     } catch (err) {
  990.                         console.error('解析失败:', err);
  991.                         showToast('文件格式错误，请检查CSV格式是否正确');
  992.                     }
  993.                 })
  994.                 .catch(err => {
  995.                     if (err === 'encoding_error') {
  996.                         currentEncodingIndex++;
  997.                         tryNextEncoding();
  998.                     } else {
  999.                         console.error('导入失败:', err);
  1000.                         showToast('导入失败，请检查文件格式');
  1001.                     }
  1002.                 });
  1003.         }
  1004.  
  1005.         tryNextEncoding();
  1006.     }
  1007.  
  1008.     // 创建UI
  1009.     function createUI() {
  1010.         // 添加切换按钮
  1011.         const toggleButton = document.createElement('button');
  1012.         toggleButton.className = 'toggle-button';
  1013.         toggleButton.textContent = '提示词助手';
  1014.         toggleButton.onclick = function() {
  1015.             const helper = document.querySelector('.prompt-helper');
  1016.             helper.classList.toggle('collapsed');
  1017.             this.style.display = helper.classList.contains('collapsed') ? 'block' : 'none';
  1018.         };
  1019.         document.body.appendChild(toggleButton);
  1020.  
  1021.         const container = document.createElement('div');
  1022.         container.className = 'prompt-helper';
  1023.         container.innerHTML = `
  1024.             <h3>
  1025.                 <span class="title">AI提示词助手</span>
  1026.                 <div class="buttons">
  1027.                     <button id="settingsBtn">设置</button>
  1028.                     <button id="toggleHelper">收起</button>
  1029.                 </div>
  1030.             </h3>
  1031.             <div class="settings-menu" id="settingsMenu">
  1032.                 <h4>显示设置</h4>
  1033.                 <div class="settings-item">
  1034.                     <label>
  1035.                         <input type="checkbox" id="showName" ${getDisplaySettings().showName ? 'checked' : ''}>
  1036.                         显示名称
  1037.                     </label>
  1038.                 </div>
  1039.                 <div class="settings-item">
  1040.                     <label>
  1041.                         <input type="checkbox" id="showContent" ${getDisplaySettings().showContent ? 'checked' : ''}>
  1042.                         显示内容
  1043.                     </label>
  1044.                 </div>
  1045.                 <div class="settings-item">
  1046.                     <label>
  1047.                         <input type="checkbox" id="showComment" ${getDisplaySettings().showComment ? 'checked' : ''}>
  1048.                         显示注释
  1049.                     </label>
  1050.                 </div>
  1051.                 <div class="settings-item">
  1052.                     <label>
  1053.                         <input type="checkbox" id="showCategory" ${getDisplaySettings().showCategory ? 'checked' : ''}>
  1054.                         显示分类
  1055.                     </label>
  1056.                 </div>
  1057.                 <div class="settings-item">
  1058.                     <label>
  1059.                         <input type="checkbox" id="isSimplified" ${getDisplaySettings().isSimplified ? 'checked' : ''}>
  1060.                         简化显示（点击展开）
  1061.                     </label>
  1062.                 </div>
  1063.                 <div class="author-info">
  1064.                     <p>作者：初沐</p>
  1065.                     <p>联系方式：<a href="mailto:1501942742@qq.com">1501942742@qq.com</a></p>
  1066.                     <p>如有问题或建议，欢迎反馈</p>
  1067.                 </div>
  1068.             </div>
  1069.             <div class="import-toggle" id="importToggle">
  1070.                 <div class="arrow"></div>
  1071.                 <span class="import-toggle-text">导入提示词</span>
  1072.             </div>
  1073.             <div class="import-section collapsed" id="importSection">
  1074.                 <p>拖拽文件到此处或点击下方按钮导入</p>
  1075.                 <input type="file" id="fileInput" accept=".csv,.xlsx,.xls">
  1076.                 <button id="importBtn">导入提示词</button>
  1077.                 <button id="downloadTemplate">下载模板</button>
  1078.             </div>
  1079.             <input type="text" id="searchPrompt" class="search-box" placeholder="搜索提示词...">
  1080.             <div class="controls">
  1081.                 <select id="categorySelect" style="flex: 1">
  1082.                     <option value="all">所有分类</option>
  1083.                     ${getCategories().map(category => `<option value="${category}">${category}</option>`).join('')}
  1084.                 </select>
  1085.                 <button id="addNewPrompt">新增提示词</button>
  1086.                 <button id="manageCategories">分类管理</button>
  1087.             </div>
  1088.             <div id="manageCategoriesForm" class="manage-categories">
  1089.                 <div style="display: flex; gap: 8px; margin-bottom: 10px;">
  1090.                     <input type="text" id="newCategory" placeholder="输入新分类名称" style="flex: 1">
  1091.                     <button id="addCategory">添加分类</button>
  1092.                 </div>
  1093.                 <div class="category-list" id="categoryList"></div>
  1094.             </div>
  1095.             <div id="addPromptForm" class="add-prompt-form">
  1096.                 <input type="text" id="promptName" placeholder="提示词名称" style="width: 100%">
  1097.                 <textarea id="promptContent" placeholder="提示词内容" style="width: 100%; height: 80px; margin: 8px 0"></textarea>
  1098.                 <textarea id="promptComment" placeholder="注释（可选）" style="width: 100%; height: 40px"></textarea>
  1099.                 <select id="promptCategory" style="width: 100%; margin: 8px 0">
  1100.                     <option value="洗稿">洗稿</option>
  1101.                     <option value="创作">创作</option>
  1102.                     <option value="翻译">翻译</option>
  1103.                     <option value="其他">其他</option>
  1104.                 </select>
  1105.                 <div style="text-align: right">
  1106.                     <button id="cancelPrompt">取消</button>
  1107.                     <button id="savePrompt">保存</button>
  1108.                 </div>
  1109.             </div>
  1110.             <div class="prompt-list" id="promptList"></div>
  1111.         `;
  1112.         document.body.appendChild(container);
  1113.  
  1114.         // 初始化事件
  1115.         initEvents();
  1116.         loadPrompts();
  1117.         updateCategoryList();
  1118.     }
  1119.  
  1120.     // 初始化
  1121.     createUI();
  1122. })();

欢迎访问Greasy Fork镜像

土豆服务器，请按需使用

镜像地址随时可能被墙，建议加群获取最新地址

加入Q群

QingJ © 2025

镜像随时可能失效，请加Q群300939539或关注我们的公众号极客氢云获取最新地址