xiaoxue_admin
/
memoryweave
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
memoryweave/plugins/obsidian/main.js

114 lines
36 KiB
JavaScript
Raw Blame History

var G=Object.defineProperty;var oe=Object.getOwnPropertyDescriptor;var se=Object.getOwnPropertyNames;var ce=Object.prototype.hasOwnProperty;var le=(a,n,e)=>n in a?G(a,n,{enumerable:!0,configurable:!0,writable:!0,value:e}):a[n]=e;var de=(a,n)=>()=>(a&&(n=a(a=0)),n);var X=(a,n)=>{for(var e in n)G(a,e,{get:n[e],enumerable:!0})},pe=(a,n,e,t)=>{if(n&&typeof n=="object"||typeof n=="function")for(let r of se(n))!ce.call(a,r)&&r!==e&&G(a,r,{get:()=>n[r],enumerable:!(t=oe(n,r))||t.enumerable});return a};var Z=a=>pe(G({},"__esModule",{value:!0}),a);var p=(a,n,e)=>le(a,typeof n!="symbol"?n+"":n,e);var te={};X(te,{fetchByEntity:()=>W,fetchCategories:()=>K,fetchDistillQuota:()=>B,fetchDistillStatus:()=>V,fetchGraphExport:()=>D,fetchMemories:()=>H,fetchNeighbors:()=>$,fetchPageRank:()=>R,fetchStats:()=>ue,getApiBase:()=>J,getApiKey:()=>ee,getRuntimeNamespace:()=>v,searchMemories:()=>A,setRuntimeConfig:()=>C,writeMemory:()=>j});function J(){return(S.apiUrl||he).replace(/\/$/,"")}function ee(){return S.apiKey||ge}function C(a,n,e){S.apiUrl=a,S.apiKey=n,S.namespace=e}function v(){return S.namespace||"hermes-main"}async function z(a,n={}){let e=n.method??"GET",t={"X-API-Key":ee(),"Content-Type":"application/json"},r={method:e,headers:t};n.body!==void 0&&(r.body=JSON.stringify(n.body));let o=await fetch(`${J()}${a}`,r);if(!o.ok)throw new Error(`API ${a} failed: ${o.status} ${o.statusText}`);let s=await o.text();return s?JSON.parse(s):{}}async function ue(){return z("/api/v1/stats")}async function H(a){let n=new URLSearchParams,e=a.namespace||v();e&&n.set("namespace",e),a.limit&&n.set("limit",String(a.limit)),a.offset!==void 0&&n.set("offset",String(a.offset)),a.category&&n.set("category",a.category);let t=n.toString();return z(`/api/v1/memories?${t}`)}async function A(a,n){return z("/api/v1/search/recall",{method:"POST",body:{query:a,namespace:n||v(),top_k:20}})}async function $(a){return z(`/api/v1/graph/navigate?entity=${encodeURIComponent(a)}&depth=1`)}async function D(a,n=100){return z(`/api/v1/graph/export?namespace=${a||v()}&limit=${n}`)}async function V(){return z("/api/v1/distill/status")}async function B(){return z("/api/v1/distill/quota")}async function R(a=30){return z(`/api/v1/graph/pagerank?limit=${a}`)}async function W(a){return z(`/api/v1/memories/by-entity/${encodeURIComponent(a)}`)}async function j(a,n="normal",e){return z("/api/v1/memories",{method:"POST",body:{content:a,category:n,namespace:e||v(),source:"obsidian-plugin"}})}async function K(){try{let a=await H({limit:500}),n=new Set;for(let e of a.memories)e.category&&n.add(e.category);return Array.from(n).sort()}catch{return[]}}var he,ge,S,T=de(()=>{he="http://localhost:7821",ge="zhiyi-dev-key-2026",S={}});var ye={};X(ye,{default:()=>I});module.exports=Z(ye);var x=require("obsidian");var ie=require("obsidian");T();var M="zhiyi-memory-view",O=30,N=class extends ie.ItemView{constructor(e){super(e);p(this,"memories",[]);p(this,"currentPage",0);p(this,"totalMemories",0);p(this,"status_interval",null);p(this,"categories",[]);p(this,"activeCategory","");p(this,"expandedId",null)}getViewType(){return M}getDisplayText(){return"\u7EC7\u5FC6\u8BB0\u5FC6"}async onOpen(){this.render(),this.startStatusPoll()}async onClose(){this.status_interval!==null&&window.clearInterval(this.status_interval)}startStatusPoll(){let e=window.setInterval(()=>this.renderDistillBanner(),3e4);this.status_interval=e}async renderDistillBanner(){let e=this.container.querySelector(".zhiyi-distill-banner");if(e)try{let[t,r]=await Promise.all([V(),B()]);t.queue_len>0?(e.innerHTML=`<span class="zhiyi-warn">\u26A0\uFE0F \u84B8\u998F\u961F\u5217: ${t.queue_len} \u6761</span>`,e.style.display="block"):e.style.display="none"}catch{}}async loadCategories(){try{this.categories=await K()}catch{this.categories=[]}}async render(){this.container.empty(),this.container.createEl("style",{text:`
.zhiyi-memory-view { height: 100%; display: flex; flex-direction: column; font-size: 13px; }
.zhiyi-header { padding: 8px 12px 4px; border-bottom: 1px solid var(--background-modifier-border); }
.zhiyi-header-row1 { display: flex; align-items: center; gap: 6px; }
.zhiyi-header h1 { font-size: 14px; font-weight: 600; margin: 0; flex: 1; }
.zhiyi-stats { color: var(--text-muted); font-size: 11px; margin: 3px 0 0; }
.zhiyi-toolbar { display: flex; gap: 4px; padding: 4px 12px; border-bottom: 1px solid var(--background-modifier-border); flex-wrap: wrap; align-items: center; }
.zhiyi-toolbar select { background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; color: var(--text-normal); font-size: 11px; padding: 2px 4px; max-width: 120px; }
.zhiyi-filter-chip { padding: 2px 8px; border-radius: 10px; font-size: 10px; border: 1px solid var(--background-modifier-border); background: var(--background-secondary); color: var(--text-muted); cursor: pointer; white-space: nowrap; }
.zhiyi-filter-chip:hover { background: var(--background-modifier-hover); }
.zhiyi-filter-chip.active { background: var(--text-accent); color: var(--bg-on-accent, #fff); border-color: var(--text-accent); }
.zhiyi-btn-icon { padding: 2px 6px; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; cursor: pointer; color: var(--text-muted); font-size: 11px; }
.zhiyi-btn-icon:hover { background: var(--background-modifier-hover); color: var(--text-normal); }
.zhiyi-distill-banner { display: none; background: #c44; color: white; padding: 4px 12px; font-size: 12px; font-weight: 600; cursor: pointer; }
.zhiyi-body { flex: 1; overflow-y: auto; }
.zhiyi-item { padding: 8px 12px; border-bottom: 1px solid var(--background-modifier-border); cursor: pointer; }
.zhiyi-item:hover { background: var(--background-modifier-hover); }
.zhiyi-item-header { display: flex; justify-content: space-between; align-items: center; margin-bottom: 3px; }
.zhiyi-item-cat { font-size: 10px; color: var(--text-muted); background: var(--background-secondary); padding: 1px 5px; border-radius: 2px; }
.zhiyi-item-right { display: flex; align-items: center; gap: 6px; }
.zhiyi-quality-bar { width: 40px; height: 4px; background: var(--background-modifier-border); border-radius: 2px; overflow: hidden; }
.zhiyi-quality-fill { height: 100%; border-radius: 2px; }
.zhiyi-item-content { font-size: 12px; color: var(--text-normal); line-height: 1.4; }
.zhiyi-item-footer { display: flex; justify-content: space-between; align-items: center; margin-top: 3px; }
.zhiyi-item-time { font-size: 10px; color: var(--text-muted); }
.zhiyi-item-meta { font-size: 10px; color: var(--text-faint); font-family: monospace; }
.zhiyi-item-actions { display: flex; gap: 6px; opacity: 0; transition: opacity 0.15s; }
.zhiyi-item:hover .zhiyi-item-actions { opacity: 1; }
.zhiyi-action-btn { font-size: 10px; color: var(--text-accent); cursor: pointer; background: none; border: none; padding: 0; }
.zhiyi-action-btn:hover { text-decoration: underline; }
.zhiyi-pagination { display: flex; gap: 6px; padding: 8px 12px; border-top: 1px solid var(--background-modifier-border); align-items: center; }
.zhiyi-pagination button { padding: 4px 10px; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; cursor: pointer; font-size: 11px; }
.zhiyi-pagination button:disabled { opacity: 0.4; cursor: not-allowed; }
.zhiyi-pagination .page-info { flex: 1; text-align: center; font-size: 11px; color: var(--text-muted); }
.zhiyi-loading { padding: 30px; text-align: center; color: var(--text-muted); }
.zhiyi-empty { padding: 30px; text-align: center; color: var(--text-muted); }
/* Modal */
.zhiyi-detail-modal { position: fixed; inset: 0; background: rgba(0,0,0,0.5); z-index: 1000; display: flex; align-items: center; justify-content: center; }
.zhiyi-detail-box { background: var(--modal-background); border: 1px solid var(--background-modifier-border); border-radius: 8px; max-width: 600px; width: 90%; max-height: 80vh; display: flex; flex-direction: column; box-shadow: 0 8px 32px rgba(0,0,0,0.4); }
.zhiyi-detail-header { display: flex; justify-content: space-between; align-items: center; padding: 12px 16px; border-bottom: 1px solid var(--background-modifier-border); }
.zhiyi-detail-header h3 { margin: 0; font-size: 14px; font-weight: 600; }
.zhiyi-detail-close { background: none; border: none; cursor: pointer; color: var(--text-muted); font-size: 18px; padding: 0; }
.zhiyi-detail-body { flex: 1; overflow-y: auto; padding: 16px; }
.zhiyi-detail-meta { display: flex; gap: 12px; margin-bottom: 12px; flex-wrap: wrap; }
.zhiyi-detail-meta span { font-size: 11px; color: var(--text-muted); background: var(--background-secondary); padding: 2px 6px; border-radius: 3px; }
.zhiyi-detail-content { font-size: 13px; line-height: 1.6; color: var(--text-normal); white-space: pre-wrap; word-break: break-word; }
.zhiyi-detail-footer { padding: 10px 16px; border-top: 1px solid var(--background-modifier-border); display: flex; gap: 8px; justify-content: flex-end; }
.zhiyi-detail-footer button { padding: 4px 12px; border-radius: 4px; cursor: pointer; font-size: 12px; }
/* Write dialog */
.zhiyi-write-section { padding: 8px 12px; border-bottom: 1px solid var(--background-modifier-border); }
.zhiyi-write-input { width: 100%; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; color: var(--text-normal); padding: 6px 8px; font-size: 12px; resize: vertical; min-height: 50px; box-sizing: border-box; }
.zhiyi-write-input:focus { outline: none; border-color: var(--text-accent); }
.zhiyi-write-meta { display: flex; gap: 6px; margin-top: 4px; align-items: center; }
.zhiyi-write-meta select { background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; color: var(--text-normal); font-size: 11px; padding: 2px 4px; }
.zhiyi-write-meta button { padding: 3px 10px; background: var(--text-accent); color: var(--bg-on-accent, #fff); border: none; border-radius: 4px; cursor: pointer; font-size: 11px; }
.zhiyi-write-meta button:disabled { opacity: 0.5; cursor: not-allowed; }
.zhiyi-write-hint { font-size: 10px; color: var(--text-faint); margin-left: auto; }
`}),this.container.createEl("div",{cls:"zhiyi-memory-view"}).append(this.container.createEl("div",{cls:"zhiyi-distill-banner"}),this.container.createEl("div",{cls:"zhiyi-header"}).append(Object.assign(this.container.createEl("div",{cls:"zhiyi-header-row1"}),{innerHTML:"<h1>\u{1F9E0} \u7EC7\u5FC6\u8BB0\u5FC6</h1>"}),this.container.createEl("p",{cls:"zhiyi-stats",text:"\u52A0\u8F7D\u4E2D..."})),this.container.createEl("div",{cls:"zhiyi-toolbar"}),this.container.createEl("div",{cls:"zhiyi-body"}),this.container.createEl("div",{cls:"zhiyi-pagination"})),await this.loadCategories(),await this.loadMemories(),await this.renderDistillBanner()}async loadMemories(){let e=this.container.querySelector(".zhiyi-body"),t=this.container.querySelector(".zhiyi-stats"),r=this.container.querySelector(".zhiyi-toolbar");e.innerHTML='<div class="zhiyi-loading">\u52A0\u8F7D\u4E2D\u2026</div>';try{let o=v(),s=await H({namespace:o,limit:O,offset:this.currentPage*O,category:this.activeCategory||void 0});this.memories=s.memories||[],this.totalMemories=s.total||0,t.setText(`${this.totalMemories} \u6761\u8BB0\u5FC6 \xB7 \u7B2C ${this.currentPage+1} \u9875 \xB7 ${o}`),this.renderToolbar(r),this.renderList()}catch(o){e.innerHTML=`<div class="zhiyi-empty">\u52A0\u8F7D\u5931\u8D25: ${o.message}</div>`}}renderToolbar(e){e.empty();let t=e.createEl("select");t.createEl("option",{value:"hermes-main",text:"hermes-main"}),t.createEl("option",{value:"openclaw",text:"openclaw"}),t.value=v(),t.onchange=()=>{let c=t.value;window.zhiyi_namespace=c,this.currentPage=0,this.loadMemories()};let r=e.createEl("span",{cls:"zhiyi-filter-chip"+(this.activeCategory===""?" active":""),text:"\u5168\u90E8"});r.onclick=()=>{this.activeCategory="",this.currentPage=0,this.loadMemories()};for(let c of this.categories.slice(0,8)){let l=e.createEl("span",{cls:"zhiyi-filter-chip"+(this.activeCategory===c?" active":""),text:c});l.onclick=()=>{this.activeCategory=c,this.currentPage=0,this.loadMemories()}}let o=e.createEl("button",{cls:"zhiyi-btn-icon",text:"\u21BB \u5237\u65B0"});o.onclick=()=>{this.currentPage=0,this.loadMemories()};let s=e.createEl("button",{cls:"zhiyi-btn-icon",text:"+ \u5199\u8BB0\u5FC6"});s.onclick=()=>this.showWriteSection(e)}showWriteSection(e){let t=e.querySelector(".zhiyi-write-section");if(t){t.remove();return}let r=e.createEl("div",{cls:"zhiyi-write-section"}),o=r.createEl("textarea",{cls:"zhiyi-write-input",attr:{placeholder:"\u8F93\u5165\u8BB0\u5FC6\u5185\u5BB9\uFF0C\u6309 Enter \u6216\u70B9\u51FB\u300C\u5199\u5165\u300D\u63D0\u4EA4\u2026"}}),s=r.createEl("div",{cls:"zhiyi-write-meta"}),c=s.createEl("select");c.createEl("option",{value:"normal",text:"\u5206\u7C7B"});for(let d of this.categories)c.createEl("option",{value:d,text:d});let l=s.createEl("span",{cls:"zhiyi-write-hint",text:"Enter \u5199\u5165 \xB7 Esc \u53D6\u6D88"}),h=s.createEl("button",{text:"\u5199\u5165"}),m=async()=>{let d=o.value.trim();if(d){h.setAttr("disabled","true");try{await j(d,c.value||"normal"),o.value="",r.remove(),this.currentPage=0,await this.loadCategories(),await this.loadMemories()}catch(y){l.setText(`\u5931\u8D25: ${y.message}`)}h.removeAttr("disabled")}};h.onclick=m,o.onkeydown=d=>{d.key==="Enter"&&!d.shiftKey&&(d.preventDefault(),m()),d.key==="Escape"&&r.remove()},o.focus()}renderList(){let e=this.container.querySelector(".zhiyi-body"),t=this.container.querySelector(".zhiyi-pagination");e.empty(),this.memories.length===0&&e.createEl("div",{cls:"zhiyi-empty",text:"\u6682\u65E0\u8BB0\u5FC6"});for(let c of this.memories){let l=e.createEl("div",{cls:"zhiyi-item"}),h=l.createEl("div",{cls:"zhiyi-item-header"});h.createEl("span",{cls:"zhiyi-item-cat",text:c.category||"normal"});let m=h.createEl("div",{cls:"zhiyi-item-right"}),d=c.quality_score??0,y=d>.7?"#34c759":d>.4?"#f6c945":"#ff6b6b",w=m.createEl("div",{cls:"zhiyi-quality-bar"}).createEl("div",{cls:"zhiyi-quality-fill"});w.style.width=`${Math.round(d*100)}%`,w.style.background=y,m.createEl("span",{cls:"zhiyi-item-score",text:(d*100).toFixed(0)+"%"}),l.createEl("div",{cls:"zhiyi-item-content",text:this.truncate(c.content,160)});let b=l.createEl("div",{cls:"zhiyi-item-footer"});b.createEl("span",{cls:"zhiyi-item-time",text:this.formatTime(c.created_at)});let P=b.createEl("span",{cls:"zhiyi-item-meta",text:c.id.slice(0,16)}),E=b.createEl("div",{cls:"zhiyi-item-actions"});E.createEl("button",{cls:"zhiyi-action-btn",text:"\u5C55\u5F00"}).onclick=g=>{g.stopPropagation(),this.showDetail(c)},E.createEl("button",{cls:"zhiyi-action-btn",text:"\u590D\u5236"}).onclick=g=>{g.stopPropagation(),navigator.clipboard.writeText(c.content).catch(()=>{})},l.onclick=()=>this.showDetail(c)}let r=Math.ceil(this.totalMemories/O)||1;t.empty();let o=t.createEl("button",{text:"\u25C0"});o.setAttr("disabled",this.currentPage===0?"true":""),o.onclick=()=>{this.currentPage--,this.loadMemories()},t.createEl("span",{cls:"page-info",text:`${this.currentPage+1} / ${r}`});let s=t.createEl("button",{text:"\u25B6"});s.setAttr("disabled",this.currentPage>=r-1?"true":""),s.onclick=()=>{this.currentPage++,this.loadMemories()}}showDetail(e){let t=this.container.createEl("div",{cls:"zhiyi-detail-modal"}),r=t.createEl("div",{cls:"zhiyi-detail-box"}),o=r.createEl("div",{cls:"zhiyi-detail-header"});o.createEl("h3",{text:"\u8BB0\u5FC6\u8BE6\u60C5"});let s=o.createEl("button",{cls:"zhiyi-detail-close",text:"\xD7"});s.onclick=()=>t.remove(),t.onclick=m=>{m.target===t&&t.remove()};let c=r.createEl("div",{cls:"zhiyi-detail-body"}),l=c.createEl("div",{cls:"zhiyi-detail-meta"});l.createEl("span",{text:`\u5206\u7C7B: ${e.category||"normal"}`}),l.createEl("span",{text:`\u8D28\u91CF: ${((e.quality_score??0)*100).toFixed(1)}%`}),l.createEl("span",{text:`Namespace: ${e.namespace}`}),l.createEl("span",{text:`\u65F6\u95F4: ${this.formatTime(e.created_at)}`}),e.recall_count!==void 0&&l.createEl("span",{text:`\u53EC\u56DE: ${e.recall_count}`}),c.createEl("div",{cls:"zhiyi-detail-content",text:e.content});let h=r.createEl("div",{cls:"zhiyi-detail-footer"});h.createEl("button",{text:"\u590D\u5236\u5185\u5BB9",attr:{style:"background:var(--background-secondary);border:1px solid var(--background-modifier-border);color:var(--text-normal)"}}).onclick=()=>{navigator.clipboard.writeText(e.content).catch(()=>{})},h.createEl("button",{text:"\u5173\u95ED",attr:{style:"background:var(--text-accent);border:none;color:var(--bg-on-accent,#fff)"}}).onclick=()=>t.remove()}truncate(e,t){return e?e.length>t?e.slice(0,t)+"\u2026":e:""}formatTime(e){if(!e)return"";try{return new Date(e).toLocaleString("zh-CN",{month:"2-digit",day:"2-digit",hour:"2-digit",minute:"2-digit"})}catch{return e}}};var re=require("obsidian");T();async function me(){return window.d3?window.d3:new Promise((a,n)=>{let e=document.createElement("script");e.src="https://cdn.jsdelivr.net/npm/d3@7/dist/d3.min.js",e.onload=()=>a(window.d3),e.onerror=n,document.head.appendChild(e)})}var L="zhiyi-graph-view",_=class extends re.ItemView{constructor(e){super(e);p(this,"currentEntity","");p(this,"d3",null);p(this,"svg",null);p(this,"simulation",null);p(this,"topEntities",[])}getViewType(){return L}getDisplayText(){return"\u7EC7\u5FC6\u56FE\u8C31"}async onOpen(){this.render()}async onClose(){this.simulation?.stop()}async render(){this.container.empty(),this.container.createEl("style",{text:`
.zhiyi-graph-view { height: 100%; display: flex; flex-direction: column; }
.zhiyi-graph-toolbar { display: flex; gap: 6px; padding: 6px 10px; border-bottom: 1px solid var(--background-modifier-border); align-items: center; }
.zhiyi-graph-toolbar input { flex: 1; padding: 4px 8px; font-size: 12px; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; color: var(--text-normal); }
.zhiyi-graph-toolbar button { padding: 4px 10px; font-size: 12px; cursor: pointer; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; }
.zhiyi-graph-toolbar .info { font-size: 11px; color: var(--text-muted); padding: 0 4px; }
.zhiyi-graph-body { flex: 1; display: flex; overflow: hidden; }
.zhiyi-graph-entity-list { width: 160px; border-right: 1px solid var(--background-modifier-border); overflow-y: auto; padding: 4px 0; flex-shrink: 0; }
.zhiyi-graph-entity-list h4 { margin: 0; padding: 4px 8px; font-size: 11px; color: var(--text-muted); font-weight: 600; border-bottom: 1px solid var(--background-modifier-border); }
.zhiyi-entity-item { padding: 4px 8px; font-size: 11px; cursor: pointer; color: var(--text-normal); display: flex; justify-content: space-between; align-items: center; }
.zhiyi-entity-item:hover { background: var(--background-modifier-hover); }
.zhiyi-entity-item.active { color: var(--text-accent); background: var(--background-modifier-hover); }
.zhiyi-entity-score { font-size: 9px; color: var(--text-faint); }
.zhiyi-graph-canvas { flex: 1; position: relative; overflow: hidden; }
.zhiyi-graph-canvas svg { width: 100%; height: 100%; }
.zhiyi-graph-detail { position: absolute; bottom: 0; left: 0; right: 0; background: var(--background-secondary); border-top: 1px solid var(--background-modifier-border); padding: 8px 12px; font-size: 11px; max-height: 120px; overflow-y: auto; }
.zhiyi-graph-detail-title { font-weight: 600; margin-bottom: 4px; }
.zhiyi-graph-node-label { font-size: 10px; fill: var(--text-muted); pointer-events: none; }
.zhiyi-legend { position: absolute; top: 8px; right: 8px; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; padding: 6px 8px; font-size: 10px; }
.zhiyi-legend-item { display: flex; align-items: center; gap: 4px; margin: 2px 0; }
.zhiyi-legend-dot { width: 8px; height: 8px; border-radius: 50%; display: inline-block; }
.zhiyi-tooltip { position: absolute; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; padding: 6px 8px; font-size: 11px; pointer-events: none; opacity: 0; transition: opacity 0.15s; max-width: 200px; z-index: 10; }
`});let e=this.container.createEl("div",{cls:"zhiyi-graph-view"}),t=e.createEl("div",{cls:"zhiyi-graph-toolbar"}),r=t.createEl("input",{attr:{placeholder:"\u8F93\u5165\u5B9E\u4F53\u540D\u79F0\uFF0C\u56DE\u8F66\u67E5\u770B ego-network\uFF08\u5982: \u7267\u5C18\uFF09"}}),o=t.createEl("button",{text:"\u52A0\u8F7D\u56FE\u8C31"});t.createEl("span",{cls:"info",text:"\u70B9\u51FB\u8282\u70B9\u67E5\u770B\u8BE6\u60C5"});let s=t.createEl("button",{text:"Top \u8282\u70B9"}),c=t.createEl("button",{text:"\u5237\u65B0\u5217\u8868"}),l=e.createEl("div",{cls:"zhiyi-graph-body"}),h=l.createEl("div",{cls:"zhiyi-graph-entity-list"});h.createEl("h4",{text:"\u{1F4CA} Top \u5B9E\u4F53"});let m=h.createEl("div",{cls:"zhiyi-graph-entity-list-body"}),d=l.createEl("div",{cls:"zhiyi-graph-canvas"}),y=d.createEl("div",{cls:"zhiyi-tooltip"}),f=d.createEl("div",{cls:"zhiyi-legend"});f.innerHTML=`
<div class="zhiyi-legend-item"><span class="zhiyi-legend-dot" style="background:#4a9eff"></span> \u9ED8\u8BA4</div>
<div class="zhiyi-legend-item"><span class="zhiyi-legend-dot" style="background:#f6c945"></span> \u4EBA\u7269</div>
<div class="zhiyi-legend-item"><span class="zhiyi-legend-dot" style="background:#34c759"></span> \u9879\u76EE</div>
<div class="zhiyi-legend-item"><span class="zhiyi-legend-dot" style="background:#ff6b6b"></span> \u4E8B\u4EF6</div>
`;let w=e.createEl("div",{cls:"zhiyi-graph-detail",attr:{style:"display:none"}}),b=w.createEl("div",{cls:"zhiyi-graph-detail-title"}),P=w.createEl("div",{cls:"zhiyi-graph-detail-content"});this.d3=await me(),this.svg=this.d3.select(d.createEl("svg")),await this.renderGlobalGraph(),await this.loadEntityList(m);let E=async g=>{r.value=g,await this.renderEgoGraph(g,d,y,w,b,P),m.querySelectorAll(".zhiyi-entity-item").forEach(k=>{k.classList.toggle("active",k.dataset.entity===g)})};o.onclick=async()=>{r.value.trim()&&await E(r.value.trim())},r.onkeydown=async g=>{g.key==="Enter"&&r.value.trim()&&await E(r.value.trim())},s.onclick=async()=>{try{let k=(await R(20)).pagerank?.[0]?.entity||"\u7267\u5C18";await E(k)}catch{}},c.onclick=async()=>{await this.loadEntityList(m)}}async loadEntityList(e){try{let t=await R(30);if(this.topEntities=t.pagerank||[],e.empty(),this.topEntities.length===0){e.createEl("div",{text:"\u6682\u65E0\u6570\u636E",attrs:{style:"padding:8px;font-size:11px;color:var(--text-muted)"}});return}for(let r of this.topEntities){let o=e.createEl("div",{cls:"zhiyi-entity-item",attr:{"data-entity":r.entity}});o.createEl("span",{text:r.entity.length>14?r.entity.slice(0,14)+"\u2026":r.entity}),o.createEl("span",{cls:"zhiyi-entity-score",text:r.score.toFixed(2)}),o.onclick=async()=>{let c=this.container.querySelector(".zhiyi-graph-toolbar")?.querySelector("input");c&&(c.value=r.entity,await this.renderEgoGraph(r.entity,null,null,this.container.querySelector(".zhiyi-graph-detail"),this.container.querySelector(".zhiyi-graph-detail-title"),this.container.querySelector(".zhiyi-graph-detail-content"))),e.querySelectorAll(".zhiyi-entity-item").forEach(l=>l.classList.remove("active")),o.classList.add("active")}}}catch{e.empty(),e.createEl("div",{text:"\u52A0\u8F7D\u5931\u8D25",attrs:{style:"padding:8px;font-size:11px;color:var(--text-muted)"}})}}async renderGlobalGraph(){try{let e=v(),t=await D(e,80);await this.renderD3Graph(t.nodes,t.edges,null)}catch{}}async renderEgoGraph(e,t,r,o,s,c){this.currentEntity=e,this.simulation?.stop();try{let[l,h]=await Promise.all([$(e),W(e)]),d=[{id:e,label:e},...l.nodes.filter(f=>f.id!==e)];await this.renderD3Graph(d,l.edges,e),o.style.display="block",s.setText(`\u{1F4CC} ${e}`);let y=(h.memories||[]).slice(0,5);c.innerHTML=y.length>0?y.map(f=>`<div style="margin:2px 0">\u2022 ${this.truncate(f.content,80)}</div>`).join(""):"<div>\u65E0\u5173\u8054\u8BB0\u5FC6</div>"}catch(l){let h=t.createEl("div",{text:`\u52A0\u8F7D\u5931\u8D25: ${l.message}`,cls:"zhiyi-empty"});setTimeout(()=>h.remove(),3e3)}}async renderD3Graph(e,t,r){if(!this.d3||!this.svg)return;let o=this.d3,s=this.svg;s.selectAll("*").remove();let c=s.node().clientWidth||600,l=s.node().clientHeight||400,h={\u4EBA\u7269:"#f6c945",\u4EBA\u7269_name:"#f6c945",project:"#34c759",\u9879\u76EE:"#34c759",event:"#ff6b6b",\u4E8B\u4EF6:"#ff6b6b",concept:"#a855f7",\u6982\u5FF5:"#a855f7",location:"#06b6d4",distilled:"#4a9eff",default:"#4a9eff"},m=i=>i.id===r?"#ff4d4d":i.category&&h[i.category]?h[i.category]:h[i.weight!==void 0&&i.weight>5?"distilled":"default"],d=e.map(i=>({id:i.id,label:i.label||i.id,category:i.category,weight:i.weight,isCentral:i.id===r})),y=new Map(d.map(i=>[i.id,i])),f=t.filter(i=>y.has(i.source)&&y.has(i.target)).map(i=>({source:i.source,target:i.target,relation:i.relation,weight:i.weight})),w=o.zoom().scaleExtent([.2,5]).on("zoom",i=>{b.attr("transform",i.transform)});s.call(w);let b=s.append("g");s.append("defs").append("marker").attr("id","arrowhead").attr("viewBox","-0 -5 10 10").attr("refX",15).attr("refY",0).attr("orient","auto").attr("markerWidth",6).attr("markerHeight",6).append("path").attr("d","M 0,-5 L 10,0 L 0,5").attr("fill","#666");let P=b.append("g").selectAll("line").data(f).enter().append("line").attr("stroke","#999").attr("stroke-opacity",.6).attr("stroke-width",i=>Math.min(2,i.weight||1)).attr("marker-end","url(#arrowhead)"),E=b.append("g").selectAll("text").data(f).enter().append("text").text(i=>i.relation||"").attr("text-anchor","middle").attr("dy",-3).attr("font-size","8px").attr("fill","#999").attr("pointer-events","none").style("display",i=>i.relation?"block":"none"),g=b.append("g").selectAll("g").data(d).enter().append("g").style("cursor","pointer");g.append("circle").attr("r",i=>i.isCentral?12:6+(i.weight||1)*.5).attr("fill",i=>m(i)).attr("stroke",i=>i.isCentral?"#ff4d4d":"white").attr("stroke-width",i=>i.isCentral?2:1.5),g.append("text").text(i=>i.label.length>12?i.label.slice(0,12)+"\u2026":i.label).attr("dy",-8).attr("text-anchor","middle").attr("class","zhiyi-graph-node-label").style("font-size",i=>i.isCentral?"11px":"9px").style("fill","var(--text-muted)");let k=o.drag().on("start",(i,u)=>{i.active||this.simulation.alphaTarget(.3).restart(),u.fx=u.x,u.fy=u.y}).on("drag",(i,u)=>{u.fx=i.x,u.fy=i.y}).on("end",(i,u)=>{i.active||this.simulation.alphaTarget(0),u.fx=null,u.fy=null});g.call(k),g.on("click",(i,u)=>{if(u.id===this.currentEntity)return;let F=this.container.querySelector(".zhiyi-graph-toolbar")?.querySelector("input");F&&(F.value=u.id,this.renderEgoGraph(u.id,this.container.querySelector(".zhiyi-graph-canvas"),this.container.querySelector(".zhiyi-tooltip"),this.container.querySelector(".zhiyi-graph-detail"),this.container.querySelector(".zhiyi-graph-detail-title"),this.container.querySelector(".zhiyi-graph-detail-content"))),this.container.querySelector(".zhiyi-graph-entity-list-body")?.querySelectorAll(".zhiyi-entity-item").forEach(Q=>{Q.classList.toggle("active",Q.dataset.entity===u.id)})}),g.on("mouseover",(i,u)=>{tooltip.style.opacity="1",tooltip.style.left=i.offsetX+10+"px",tooltip.style.top=i.offsetY-10+"px",tooltip.setText(`${u.label}${u.weight!==void 0?` (\xD7${u.weight})`:""}`)}),g.on("mouseout",()=>{tooltip.style.opacity="0"}),this.simulation=o.forceSimulation(d).force("link",o.forceLink(f).id(i=>i.id).distance(80)).force("charge",o.forceManyBody().strength(-200)).force("center",o.forceCenter(c/2,l/2)).force("collision",o.forceCollide(20)).on("tick",()=>{P.attr("x1",i=>i.source.x).attr("y1",i=>i.source.y).attr("x2",i=>i.target.x).attr("y2",i=>i.target.y),E.attr("x",i=>(i.source.x+i.target.x)/2).attr("y",i=>(i.source.y+i.target.y)/2),g.attr("transform",i=>`translate(${i.x},${i.y})`)})}truncate(e,t){return e?e.length>t?e.slice(0,t)+"\u2026":e:""}};var ae=require("obsidian");T();var q=class extends ae.Modal{constructor(e){super(e);p(this,"results",[]);p(this,"inputEl");p(this,"listEl");p(this,"loading",!1);p(this,"debounceTimer",null);p(this,"onSelect")}onOpen(){this.contentEl.createEl("style",{text:`
.zhiyi-search-modal { width: 600px; max-height: 80vh; display: flex; flex-direction: column; }
.zhiyi-search-header {
display: flex; align-items: center; gap: 8px;
padding: 12px 16px; border-bottom: 1px solid var(--background-modifier-border);
}
.zhiyi-search-header h2 { margin: 0; font-size: 14px; font-weight: 600; flex: 1; }
.zhiyi-search-input {
flex: 1; padding: 8px 12px; font-size: 14px;
background: var(--background-secondary);
border: 1px solid var(--background-modifier-border);
border-radius: 6px; color: var(--text-normal); outline: none;
}
.zhiyi-search-input:focus { border-color: var(--text-accent); }
.zhiyi-search-results { flex: 1; overflow-y: auto; padding: 8px 0; }
.zhiyi-search-result {
padding: 10px 16px; border-bottom: 1px solid var(--background-modifier-border);
cursor: pointer;
}
.zhiyi-search-result:hover { background: var(--background-modifier-hover); }
.zhiyi-search-result-header { display: flex; justify-content: space-between; align-items: center; margin-bottom: 4px; }
.zhiyi-search-score { font-size: 10px; color: var(--text-muted); }
.zhiyi-search-cat { font-size: 10px; background: var(--background-secondary); padding: 1px 4px; border-radius: 2px; color: var(--text-muted); }
.zhiyi-search-content { font-size: 12px; color: var(--text-normal); line-height: 1.5; }
.zhiyi-search-id { font-size: 10px; color: var(--text-faint); font-family: monospace; margin-top: 3px; }
.zhiyi-search-loading { padding: 20px; text-align: center; color: var(--text-muted); }
.zhiyi-search-empty { padding: 20px; text-align: center; color: var(--text-muted); }
.zhiyi-search-hint { padding: 8px 16px; font-size: 11px; color: var(--text-muted); text-align: center; }
`});let e=this.contentEl.createEl("div",{cls:"zhiyi-search-modal"}),t=e.createEl("div",{cls:"zhiyi-search-header"});t.createEl("h2",{text:"\u{1F50D} \u7EC7\u5FC6\u8BED\u4E49\u641C\u7D22"}),this.inputEl=t.createEl("input",{cls:"zhiyi-search-input",attr:{placeholder:"\u8F93\u5165\u5173\u952E\u8BCD\uFF0C\u8BED\u4E49\u641C\u7D22\u8BB0\u5FC6\u2026",autofocus:"true"}}),this.listEl=e.createEl("div",{cls:"zhiyi-search-results"}),e.createEl("div",{cls:"zhiyi-search-hint",text:"\u5B9E\u65F6\u641C\u7D22 \xB7 \u6309 Enter \u9009\u62E9\u7B2C\u4E00\u6761\u7ED3\u679C"}),this.inputEl.oninput=()=>{this.debounceTimer!==null&&window.clearTimeout(this.debounceTimer),this.debounceTimer=window.setTimeout(()=>this.doSearch(),300)},this.inputEl.onkeydown=r=>{r.key==="Enter"&&this.results.length>0&&this.selectResult(this.results[0]),r.key==="Escape"&&this.close()}}async doSearch(){let e=this.inputEl.value.trim();if(!e){this.results=[],this.listEl.empty();return}this.listEl.innerHTML='<div class="zhiyi-search-loading">\u641C\u7D22\u4E2D\u2026</div>',this.loading=!0;try{let t=await A(e,"hermes-main");this.results=t.memories||[],this.renderResults()}catch(t){this.listEl.innerHTML=`<div class="zhiyi-search-empty">\u641C\u7D22\u5931\u8D25: ${t.message}</div>`}}renderResults(){if(this.listEl.empty(),this.results.length===0){this.listEl.createEl("div",{cls:"zhiyi-search-empty",text:"\u672A\u627E\u5230\u7ED3\u679C"});return}for(let e of this.results){let t=this.listEl.createEl("div",{cls:"zhiyi-search-result"}),r=t.createEl("div",{cls:"zhiyi-search-result-header"});r.createEl("span",{cls:"zhiyi-search-cat",text:e.category||"normal"}),r.createEl("span",{cls:"zhiyi-search-score",text:` relevance: ${(e.score??0).toFixed(3)}`}),t.createEl("div",{cls:"zhiyi-search-content",text:this.truncate(e.content,200)}),t.createEl("div",{cls:"zhiyi-search-id",text:e.id}),t.onclick=()=>this.selectResult(e)}}selectResult(e){this.onSelect?.(e.id,e.content),this.close()}truncate(e,t){return e?e.length>t?e.slice(0,t)+"\u2026":e:""}onClose(){this.listEl.empty()}};T();var ne="zhiyi-settings.json",U={apiUrl:"http://localhost:7821",apiKey:"zhiyi-dev-key-2026",namespace:"hermes-main"},I=class extends x.Plugin{constructor(){super(...arguments);p(this,"settings",{...U});p(this,"ribbonIcon",null)}async onload(){await this.loadSettings(),this.registerView(M,e=>new N(e)),this.registerView(L,e=>new _(e)),this.addCommand({id:"zhiyi-open-memory-view",name:"\u6253\u5F00\u7EC7\u5FC6\u8BB0\u5FC6\u9762\u677F",callback:()=>this.openMemoryView()}),this.addCommand({id:"zhiyi-open-graph-view",name:"\u6253\u5F00\u7EC7\u5FC6\u56FE\u8C31\u9762\u677F",callback:()=>this.openGraphView()}),this.addCommand({id:"zhiyi-search-memories",name:"\u641C\u7D22\u7EC7\u5FC6\u8BB0\u5FC6",callback:()=>this.openSearchModal()}),this.addCommand({id:"zhiyi-write-memory",name:"\u5C06\u9009\u4E2D\u5185\u5BB9\u5199\u5165\u7EC7\u5FC6",editorCallback:e=>{let t=e.getSelection();t&&this.openWriteModal(t)}}),this.ribbonIcon=this.addStatusBarItem(),this.ribbonIcon.setIcon("brain"),this.ribbonIcon.setTooltip("\u7EC7\u5FC6\u8BB0\u5FC6"),this.ribbonIcon.onClick(()=>this.openMemoryView()),this.addStatusBarItem().setText("\u{1F9E0} \u7EC7\u5FC6 v1.1"),this.addSettingTab(new Y(this.app,this))}onunload(){this.app.workspace.getLeavesOfType(M).forEach(e=>e.detach()),this.app.workspace.getLeavesOfType(L).forEach(e=>e.detach())}async loadSettings(){try{let e=await this.loadData(ne);e&&(this.settings={...U,...e})}catch{this.settings={...U}}C(this.settings.apiUrl,this.settings.apiKey,this.settings.namespace)}async saveSettings(){await this.saveData(this.settings,ne),C(this.settings.apiUrl,this.settings.apiKey,this.settings.namespace)}getSettings(){return this.settings}async openMemoryView(){let e=this.app.workspace.getLeaf("right");await e.setViewState({type:M,active:!0}),this.app.workspace.revealLeaf(e)}async openGraphView(){let e=this.app.workspace.getLeaf("right");await e.setViewState({type:L,active:!0}),this.app.workspace.revealLeaf(e)}openSearchModal(){let e=new q(this.app);e.onSelect=(t,r)=>{navigator.clipboard.writeText(r).catch(()=>{}),this.app.notify(`\u5DF2\u590D\u5236: ${r.slice(0,80)}\u2026`)},e.open()}openWriteModal(e=""){let{writeMemory:t}=(T(),Z(te)),r=e||window.prompt("\u8F93\u5165\u8BB0\u5FC6\u5185\u5BB9\uFF1A");if(!r?.trim())return;let o=window.prompt("\u5206\u7C7B\uFF08\u53EF\u9009\uFF0C\u5982 normal/user_pref/project\uFF09\uFF1A","normal")||"normal";t(r.trim(),o,this.settings.namespace).then(()=>{this.app.notify("\u2705 \u8BB0\u5FC6\u5DF2\u5199\u5165\u7EC7\u5FC6")}).catch(s=>{this.app.notify(`\u274C \u5199\u5165\u5931\u8D25: ${s.message}`)})}},Y=class extends x.PluginSettingTab{constructor(e,t){super(e,t);p(this,"plugin",t)}display(){let{containerEl:e}=this;e.empty(),e.createEl("h2",{text:"\u{1F9E0} \u7EC7\u5FC6\u8BBE\u7F6E"}),new x.Setting(e).setName("API \u5730\u5740").setDesc("\u7EC7\u5FC6 Go API \u670D\u52A1\u5730\u5740\uFF08\u9ED8\u8BA4 http://localhost:7821\uFF09").addText(t=>{t.setPlaceholder("http://localhost:7821").setValue(this.plugin.getSettings().apiUrl).onChange(async r=>{this.plugin.getSettings().apiUrl=r,await this.plugin.saveSettings()})}),new x.Setting(e).setName("API Key").setDesc("\u7EC7\u5FC6\u8BA4\u8BC1\u5BC6\u94A5").addText(t=>{t.setPlaceholder("zhiyi-dev-key-2026").setValue(this.plugin.getSettings().apiKey).onChange(async r=>{this.plugin.getSettings().apiKey=r,await this.plugin.saveSettings()})}),new x.Setting(e).setName("\u9ED8\u8BA4 Namespace").setDesc("\u641C\u7D22\u548C\u67E5\u8BE2\u4F7F\u7528\u7684\u9ED8\u8BA4\u547D\u540D\u7A7A\u95F4").addText(t=>{t.setValue(this.plugin.getSettings().namespace).onChange(async r=>{this.plugin.getSettings().namespace=r,await this.plugin.saveSettings()})}),new x.Setting(e).setName("\u8BB0\u5FC6\u9762\u677F").addButton(t=>t.setButtonText("\u6253\u5F00 \u{1F9E0}").onClick(()=>{this.plugin.openMemoryView()})),new x.Setting(e).setName("\u56FE\u8C31\u9762\u677F").addButton(t=>t.setButtonText("\u6253\u5F00 \u{1F4D0}").onClick(()=>{this.plugin.openGraphView()})),new x.Setting(e).setName("\u8BF4\u660E").setDesc("Ctrl+P \u2192 \u7EC7\u5FC6 \u53EF\u5FEB\u901F\u641C\u7D22\u8BB0\u5FC6\u3002\u9009\u4E2D\u6587\u5B57\u540E\u53F3\u952E\u6216\u547D\u4EE4\u9762\u677F\u9009\u62E9\u300C\u5C06\u9009\u4E2D\u5185\u5BB9\u5199\u5165\u7EC7\u5FC6\u300D\u53EF\u4FDD\u5B58\u7B14\u8BB0\u3002")}};
Reference in New Issue View Git Blame Copy Permalink