prosemirror-suggestcat-plugin 0.1.8 → 0.1.11

package/README.md

@@ -1,6 +1,6 @@
 # prosemirror-suggestcat-plugin
 
-![made by Emergence Engineering](https://emergence-engineering.com/ee-logo.svg)
+[![made by Emergence Engineering](https://emergence-engineering.com/ee-logo.svg)](https://emergence-engineering.com)
 
 [**Made by Emergence-Engineering**](https://emergence-engineering.com/)
 
@@ -172,17 +172,17 @@ view.dispatch(
 
 ```ts
 export enum TaskType {
-    Complete = "Complete",
-    Improve = "Improve",
-    MakeLonger = "MakeLonger",
-    MakeShorter = "MakeShorter",
-    Simplify = "Simplify",
-    Explain = "Explain",
-    ActionItems = "ActionItems",
+  Complete = "Complete",
+  Improve = "Improve",
+  MakeLonger = "MakeLonger",
+  MakeShorter = "MakeShorter",
+  Simplify = "Simplify",
+  Explain = "Explain",
+  ActionItems = "ActionItems",
 }
 export enum OpenAiPromptsWithParam {
-    ChangeTone = "ChangeTone",
-    Translate = "Translate",
+  ChangeTone = "ChangeTone",
+  Translate = "Translate",
 }
 
 export enum Status {

package/dist/completePlugin.js

@@ -0,0 +1,125 @@
+/* eslint-disable no-console */
+import { Fragment, Slice } from "prosemirror-model";
+import { Plugin } from "prosemirror-state";
+import { defaultCompleteOptions } from "./defaults";
+import { completeRequest, makeShorterLonger } from "./makeTaksRequest";
+import { OpenAiPromptsWithoutParam, OpenAiPromptsWithParam, Status, } from "./types";
+import { completePluginKey, isCompleteMeta } from "./utils";
+/*
+ * idle - initial state, waiting for task
+ * new - make request in view
+ * streaming - update state in apply method and wait for finished
+ * finished - update state in apply method and wait for accepted or rejected
+ * accepted - insert/replace text in view and set status to done, also used for clearing an error
+ * rejected - set status to done
+ * done - used for clearing pluginState, do not use, it is just for internal state management
+ * error - whenever an error occurs, this field gets populated so you can display it for the user, use `accepted`* meta to clear it
+ */
+export const completePlugin = (apiKey, options = defaultCompleteOptions) => {
+    let streaming = false;
+    return new Plugin({
+        key: completePluginKey,
+        state: {
+            init() {
+                return { status: Status.idle };
+            },
+            apply(tr, pluginState, prevState, state) {
+                const meta = tr.getMeta(completePluginKey);
+                if (pluginState.status === Status.done ||
+                    pluginState.status === Status.rejected) {
+                    streaming = false;
+                    return { status: Status.idle };
+                }
+                if ((meta === null || meta === void 0 ? void 0 : meta.status) === Status.cancelled) {
+                    return Object.assign(Object.assign(Object.assign({}, pluginState), meta), { isCancelled: true });
+                }
+                if (meta && isCompleteMeta(meta)) {
+                    if (pluginState.type && meta.type !== pluginState.type) {
+                        return pluginState;
+                    }
+                    return Object.assign(Object.assign({}, pluginState), meta);
+                }
+                return pluginState;
+            },
+        },
+        view() {
+            return {
+                update(view, prevState) {
+                    const pluginState = completePluginKey.getState(view.state);
+                    /* eslint-disable prefer-destructuring */
+                    let tr = view.state.tr;
+                    if ((pluginState === null || pluginState === void 0 ? void 0 : pluginState.status) === Status.new && !streaming) {
+                        switch (pluginState.type) {
+                            case OpenAiPromptsWithoutParam.Complete:
+                                streaming = true;
+                                completeRequest(pluginState, view, apiKey);
+                                console.log("complete");
+                                break;
+                            case OpenAiPromptsWithoutParam.MakeLonger:
+                            case OpenAiPromptsWithoutParam.MakeShorter:
+                            case OpenAiPromptsWithoutParam.Improve:
+                            case OpenAiPromptsWithoutParam.Simplify:
+                            case OpenAiPromptsWithoutParam.Explain:
+                            case OpenAiPromptsWithoutParam.ActionItems:
+                            case OpenAiPromptsWithParam.Translate:
+                            case OpenAiPromptsWithParam.ChangeTone:
+                                streaming = true;
+                                makeShorterLonger(pluginState.type, pluginState, view, apiKey, options.maxSelection, pluginState.params);
+                                console.log("improve selected text");
+                                break;
+                            default:
+                                break;
+                        }
+                    }
+                    if ((pluginState === null || pluginState === void 0 ? void 0 : pluginState.status) === Status.accepted) {
+                        if (pluginState.error) {
+                            tr.setMeta(completePluginKey, {
+                                type: pluginState.type,
+                                status: Status.done,
+                            });
+                            view.dispatch(tr);
+                            return;
+                        }
+                        switch (pluginState.type) {
+                            case OpenAiPromptsWithoutParam.Complete:
+                                tr = tr.insertText(pluginState.result || "", view.state.doc.nodeSize - 2);
+                                tr.setMeta(completePluginKey, {
+                                    type: OpenAiPromptsWithoutParam.Complete,
+                                    status: Status.done,
+                                });
+                                view.dispatch(tr);
+                                view.focus();
+                                console.log("complete accepted V1");
+                                break;
+                            case OpenAiPromptsWithoutParam.MakeLonger:
+                            case OpenAiPromptsWithoutParam.MakeShorter:
+                            case OpenAiPromptsWithoutParam.Improve:
+                            case OpenAiPromptsWithoutParam.Simplify:
+                            case OpenAiPromptsWithoutParam.Explain:
+                            case OpenAiPromptsWithoutParam.ActionItems:
+                            case OpenAiPromptsWithParam.Translate:
+                            case OpenAiPromptsWithParam.ChangeTone:
+                                if (pluginState.selection && pluginState.result) {
+                                    const content = pluginState.result;
+                                    const paragraphs = content.split("\n\n");
+                                    const paragraphNodes = paragraphs.map((paragraph) => view.state.schema.node("paragraph", null, view.state.schema.text(paragraph)));
+                                    const fragment = Fragment.fromArray(paragraphNodes);
+                                    tr.selection.replace(tr, new Slice(fragment, 0, 0));
+                                }
+                                tr.setMeta(completePluginKey, {
+                                    type: pluginState.type,
+                                    status: Status.done,
+                                });
+                                view.dispatch(tr);
+                                view.focus();
+                                console.log("makeShorterLonger acceptedV1");
+                                break;
+                            default:
+                                break;
+                        }
+                    }
+                },
+            };
+        },
+    });
+};

package/dist/createUpdatePopup.js

@@ -0,0 +1,34 @@
+import { GrammarSuggestElementClass } from "./types";
+export const createUpdatePopup = (view, decoration, pos, applySuggestion, discardSuggestion) => {
+    const root = document.createElement("div");
+    root.className = "grammar-suggest-tooltip";
+    const { spec } = decoration;
+    const boundingRect = view.dom.getBoundingClientRect();
+    root.id = GrammarSuggestElementClass.grammarSuggestPopup;
+    const coords = view.coordsAtPos(pos);
+    const editorTopScroll = view.dom.scrollTop || 0;
+    const editorLeftScroll = view.dom.scrollLeft || 0;
+    root.style.left = `${coords.left - boundingRect.left + editorLeftScroll}px`;
+    root.style.top = `${coords.bottom - boundingRect.top + editorTopScroll + 5}px`;
+    const applyButton = document.createElement("div");
+    applyButton.className = "grammar-suggest-tooltip-apply";
+    applyButton.innerText = spec.text || spec.originalText;
+    console.log({ spec });
+    if (!spec.text) {
+        applyButton.style.textDecoration = "line-through";
+        applyButton.style.color = "red";
+    }
+    applyButton.onclick = () => {
+        applySuggestion(view, decoration);
+    };
+    root.appendChild(applyButton);
+    const discardButton = document.createElement("div");
+    discardButton.innerHTML =
+        "<svg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 24 24' width='24' height='24'><path d='M12 22C6.47715 22 2 17.5228 2 12C2 6.47715 6.47715 2 12 2C17.5228 2 22 6.47715 22 12C22 17.5228 17.5228 22 12 22ZM12 20C16.4183 20 20 16.4183 20 12C20 7.58172 16.4183 4 12 4C7.58172 4 4 7.58172 4 12C4 16.4183 7.58172 20 12 20ZM12 10.5858L14.8284 7.75736L16.2426 9.17157L13.4142 12L16.2426 14.8284L14.8284 16.2426L12 13.4142L9.17157 16.2426L7.75736 14.8284L10.5858 12L7.75736 9.17157L9.17157 7.75736L12 10.5858Z'></path></svg>";
+    discardButton.className = "grammar-suggest-tooltip-discard";
+    discardButton.onclick = () => {
+        discardSuggestion(view, decoration);
+    };
+    root.appendChild(discardButton);
+    return root;
+};

package/dist/eventHandlers.js

@@ -0,0 +1,125 @@
+import { Decoration, DecorationSet } from "prosemirror-view";
+import { TextSelection } from "prosemirror-state";
+import { Mapping, StepMap } from "prosemirror-transform";
+import { Fragment, Slice } from "prosemirror-model";
+import { GrammarSuggestMetaType, } from "./types";
+import { getChangedRegions, getTextWithNewlines, grammarSuggestPluginKey, } from "./utils";
+import { getDiff, isIdentity } from "./mergeDiffs";
+import { docToTextWithMapping, textPosToDocPos } from "./mapping";
+export const handleUpdate = (pluginState, meta, tr) => {
+    // Add decorations
+    const { changedRegion, fix, mapping, text } = meta;
+    const newDecorations = getDiff(changedRegion.newText, fix.result)
+        .filter((i) => !isIdentity(i))
+        .filter((i) => i.original !== `${i.replacement}\n`)
+        .map(({ from, to, original, replacement }) => {
+        const decorationFrom = textPosToDocPos(changedRegion.start + from, mapping);
+        const decorationTo = textPosToDocPos(changedRegion.start + (replacement.endsWith("\n") ? to - 1 : to), mapping);
+        const decorationText = replacement.endsWith("\n")
+            ? replacement.slice(0, -1)
+            : replacement;
+        const spec = {
+            text: decorationText,
+            originalText: original,
+            id: {},
+        };
+        return Decoration.inline(decorationFrom, decorationTo, {
+            class: `grammarSuggestion ${replacement === "" ? "removalSuggestion" : ""}`,
+        }, spec);
+    });
+    return Object.assign(Object.assign({}, pluginState), { decorations: pluginState.decorations.add(tr.doc, newDecorations), lastText: text });
+};
+export const handleAccept = (pluginState, meta, tr) => {
+    const newDecorations = pluginState.decorations.remove(pluginState.decorations.find(0, tr.doc.nodeSize, (spec) => spec.id === meta.id));
+    return Object.assign(Object.assign({}, pluginState), { lastText: getTextWithNewlines(tr.doc), decorations: newDecorations.map(tr.mapping, tr.doc), popupDecoration: DecorationSet.empty });
+};
+export const handleDocChange = (pluginState, tr, oldState, withYjs) => {
+    // Do it more efficiently. There's no need to calculate the whole mapping etc.
+    const oldText = docToTextWithMapping(oldState.doc).text;
+    const { text, mapping } = docToTextWithMapping(tr.doc);
+    if (text === oldText)
+        return pluginState;
+    let pmMapping = tr.mapping;
+    if (withYjs) {
+        const diffStart = tr.doc.content.findDiffStart(oldState.doc.content);
+        const diffEnd = oldState.doc.content.findDiffEnd(tr.doc.content);
+        const map = diffEnd && diffStart
+            ? new StepMap([diffStart, diffEnd.a - diffStart, diffEnd.b - diffStart])
+            : new StepMap([0, 0, 0]);
+        pmMapping = new Mapping([map]);
+    }
+    const changedRegion = getChangedRegions(oldText, text);
+    const mappedDecorations = pluginState.decorations.map(pmMapping, tr.doc);
+    const decorationsStart = textPosToDocPos(changedRegion.start, mapping);
+    const decorationsEnd = textPosToDocPos(changedRegion.end, mapping);
+    const mappedPopupDecoration = pluginState.popupDecoration.map(pmMapping, tr.doc);
+    return Object.assign(Object.assign({}, pluginState), { decorations: mappedDecorations.remove(mappedDecorations.find(decorationsStart, decorationsEnd)), popupDecoration: mappedPopupDecoration.remove(mappedPopupDecoration.find(decorationsStart, decorationsEnd)) });
+};
+const applySuggestion = (view, decoration) => {
+    var _a;
+    const currentDecoration = (_a = grammarSuggestPluginKey
+        .getState(view.state)) === null || _a === void 0 ? void 0 : _a.decorations.find(0, view.state.doc.nodeSize, (spec) => spec.id === decoration.spec.id)[0];
+    if (!currentDecoration)
+        return;
+    const { text } = currentDecoration.spec;
+    const { from, to } = currentDecoration;
+    const meta = {
+        type: GrammarSuggestMetaType.acceptSuggestion,
+        id: decoration.spec.id,
+    };
+    const paragraphs = text.split("\n");
+    const paragraphNodes = paragraphs.map((paragraph) => view.state.schema.node("paragraph", null, view.state.schema.text(paragraph)));
+    const fragment = Fragment.fromArray(paragraphNodes);
+    let { tr } = view.state;
+    tr = tr.setSelection(TextSelection.create(view.state.doc, from, to));
+    tr.selection.replace(tr, new Slice(fragment, 1, 1));
+    tr.setMeta(grammarSuggestPluginKey, meta);
+    view.dispatch(tr);
+};
+const discardSuggestion = (view, decoration) => {
+    const { spec } = decoration;
+    const meta = {
+        type: GrammarSuggestMetaType.discardSuggestion,
+        id: spec.id,
+    };
+    const tr = view.state.tr.setMeta(grammarSuggestPluginKey, meta);
+    view.dispatch(tr);
+};
+export const handleOpenSuggestion = (pluginState, meta, tr, options) => {
+    const { decoration } = meta;
+    const spec = { id: decoration.spec.id };
+    return Object.assign(Object.assign({}, pluginState), { popupDecoration: DecorationSet.create(tr.doc, [
+            Decoration.widget(decoration.from, (view, getPos) => {
+                const pos = getPos();
+                if (!pos)
+                    return document.createElement("div");
+                return options.createUpdatePopup(view, decoration, pos, applySuggestion, discardSuggestion);
+            }, Object.assign(Object.assign({}, spec), { stopEvent: () => true })),
+        ]) });
+};
+export const handleClick = (view, pos) => {
+    const pluginState = grammarSuggestPluginKey.getState(view.state);
+    if (!pluginState)
+        return false;
+    const { decorations } = pluginState;
+    const decoration = decorations.find(pos, pos)[0];
+    if (!decoration) {
+        const meta = {
+            type: GrammarSuggestMetaType.closeSuggestion,
+        };
+        view.dispatch(view.state.tr.setMeta(grammarSuggestPluginKey, meta));
+        return false;
+    }
+    const popupDecoration = pluginState.popupDecoration.find()[0];
+    if ((popupDecoration === null || popupDecoration === void 0 ? void 0 : popupDecoration.spec.id) === decoration.spec.id)
+        return false;
+    const meta = {
+        type: GrammarSuggestMetaType.openSuggestion,
+        decoration,
+    };
+    view.dispatch(view.state.tr.setMeta(grammarSuggestPluginKey, meta));
+    return false;
+};
+export const handleDiscardSuggestion = (pluginState, meta, tr) => {
+    return Object.assign(Object.assign({}, pluginState), { decorations: pluginState.decorations.remove(pluginState.decorations.find(0, tr.doc.nodeSize, (spec) => spec.id === meta.id)), popupDecoration: DecorationSet.empty });
+};

package/dist/index.es.js

@@ -1 +1 @@
-import{PluginKey as e,Plugin as t}from"prosemirror-state";import{Decoration as n,DecorationSet as o}from"prosemirror-view";import r from"fast-diff";import{__awaiter as s}from"tslib";import i from"lodash.debounce";import{StepMap as a,Mapping as c}from"prosemirror-transform";import{Fragment as l,Slice as d}from"prosemirror-model";var p,g,u,m,h,f,x;!function(e){e.suggestionUpdate="suggestionUpdate",e.acceptSuggestion="acceptSuggestion",e.openSuggestion="openSuggestion",e.closeSuggestion="closeSuggestion",e.discardSuggestion="discardSuggestion"}(p||(p={})),function(e){e.grammarSuggestPopup="grammar-suggest-popup"}(g||(g={})),function(e){e.Casual="Casual",e.Confident="Confident",e.Straightforward="Straightforward",e.Friendly="Friendly"}(u||(u={})),function(e){e.English="English",e.Spanish="Spanish",e.French="French",e.German="German",e.Italian="Italian",e.Portuguese="Portuguese",e.Dutch="Dutch",e.Russian="Russian",e.Chinese="Chinese",e.Korean="Korean",e.Japanese="Japanese"}(m||(m={})),function(e){e.Complete="Complete",e.Improve="Improve",e.MakeLonger="MakeLonger",e.MakeShorter="MakeShorter",e.Simplify="Simplify",e.Explain="Explain",e.ActionItems="ActionItems"}(h||(h={})),function(e){e.ChangeTone="ChangeTone",e.Translate="Translate"}(f||(f={})),function(e){e.idle="idle",e.new="new",e.streaming="streaming",e.finished="finished",e.accepted="accepted",e.cancelled="cancelled",e.rejected="rejected",e.done="done",e.error="error"}(x||(x={}));const v=new e("completePlugin"),y=new e("grammarSuggestPlugin"),S=e=>{let t="";return e.descendants(((e,n)=>{e.isText&&(t+=`${e.text}\n`)})),t},w=(e,t)=>{if(e===t)return{start:e.length,end:e.length,oldStart:e.length,oldEnd:e.length,oldText:"",newText:""};const n=r(e,t),o=n[0],s=o[0]===r.EQUAL?o[1].length-1:0,i=((e,t)=>{if(t[t.length-1][0]!==r.EQUAL)return e.length;const n=t.slice(0,t.length-1);let o=0;for(const e of n){const[t,n]=e;t===r.EQUAL&&(o+=n.length),t===r.DELETE&&(o+=n.length)}return o})(e,n),a=e.lastIndexOf("\n",s),c=e.indexOf("\n",i),l=-1===a?0:a+1,d=-1===c?e.length:c,p=e.slice(l,d),g=t.slice(l,t.length-(e.length-d));return{start:l,end:l+g.length,oldStart:l,oldEnd:d,oldText:p,newText:g}},b=[{docPos:1,textPos:0}],j=e=>{let t="";const n=[];return e.descendants(((e,o)=>{e.isText&&(n.push({docPos:o,textPos:t.length}),t+=`${e.text}\n`)})),{text:t,mapping:n.length?n:b}},C=(e,t)=>{for(let n=0;n<t.length;n++)if(e>=t[n].textPos&&(void 0===t[n+1]||e<t[n+1].textPos))return t[n].docPos+(e-t[n].textPos);throw new Error("textPositionToDocumentPosition: textPos not found in mapping")},T=(e,t)=>s(void 0,void 0,void 0,(function*(){const n=[...t.split("\n")];return fetch("https://prosemirror-ai-plugin.web.app/api/suggestion",{method:"POST",cache:"no-cache",headers:{"Content-Type":"application/json",Authorization:`Bearer ${e}`},body:JSON.stringify({model:"gpt-3.5-turbo",modelParams:{input:n}})}).then((e=>e.ok?e.json():Promise.reject(e))).then((e=>!e||e.some((e=>(e=>{try{JSON.parse(e)}catch(e){return!0}return!1})(e)))?{fixed:!1,result:t}:{result:e.map((e=>JSON.parse(e).result)).join("\n"),fixed:!0})).catch((e=>(e.text().then((t=>{console.error({status:e.status,text:t})})),{fixed:!1,result:t})))})),O=e=>e.replacement===e.original,E=e=>e.reduce(((e,t)=>{const n=e[e.length-1],o=e.slice(0,e.length-1);if(!n)return[t];return[...o,...((e,t)=>{if(e.to!==t.from)throw new Error(`Replace pairs must be adjacent\n\n, ${JSON.stringify({leftReplace:e,rightReplace:t})}`);if(e.replacement.endsWith("\n"))return[e,t];if(O(e)&&O(t))return[{from:e.from,to:t.to,original:e.original+t.original,replacement:e.replacement+t.replacement}];if(O(e)){if(t.replacement.startsWith(" ")&&""===t.original||t.original.startsWith(" ")&&""===t.replacement)return[e,t];const n=e.original.split(" "),o=n.pop()||"",r=n.join(" ")+(n.length?" ":"");return[...r.length?[{from:e.from,to:e.from+r.length,original:r,replacement:r}]:[],{from:e.from+r.length,to:t.to,original:o+t.original,replacement:o+t.replacement}]}if(O(t)){if(e.replacement.endsWith(" ")&&""===e.original||e.original.endsWith(" ")&&""===e.replacement)return[e,t];const n=t.original.split(" "),o=n.shift()||"",r=(n.length?" ":"")+n.join(" ");return" "===r?[{from:e.from,to:t.to,original:e.original+t.original,replacement:e.replacement+t.replacement}]:[{from:e.from,to:e.to+o.length,original:e.original+o,replacement:e.replacement+o},...r.length?[{from:e.to+o.length,to:t.to,original:r,replacement:r}]:[]]}return[{from:e.from,to:t.to,original:e.original+t.original,replacement:e.replacement+t.replacement}]})(n,t)]}),[]),M=(e,t)=>{const n=(e=>{let t=0;return e.map((([e,n])=>{switch(e){case r.EQUAL:const e=n.lastIndexOf("\n"),o={from:t,to:t+n.length,original:n,replacement:n};if(-1!==e&&"\n"!==n){const o=[{from:t,to:t+e+1,original:n.slice(0,e+1),replacement:n.slice(0,e+1)},{from:t+e+1,to:t+n.length,original:n.slice(e+1),replacement:n.slice(e+1)}];return t+=n.length,o}return t+=n.length,[o];case r.DELETE:const s={from:t,to:t+n.length,original:n,replacement:""};return t+=n.length,[s];case r.INSERT:return[{from:t,to:t,original:"",replacement:n}]}})).flat()})(r(e,t));return E(E(n))},P=(e,t)=>{var n;const o=null===(n=y.getState(e.state))||void 0===n?void 0:n.decorations.find(0,e.state.doc.nodeSize,(e=>e.id===t.spec.id))[0];if(!o)return;const{text:r}=o.spec,{from:s,to:i}=o,a={type:p.acceptSuggestion,id:t.spec.id},c=e.state.tr.insertText(r,s,i).setMeta(y,a);e.dispatch(c)},L=(e,t)=>{const{spec:n}=t,o={type:p.discardSuggestion,id:n.id},r=e.state.tr.setMeta(y,o);e.dispatch(r)},k=(e,t)=>{const n=y.getState(e.state);if(!n)return!1;const{decorations:o}=n,r=o.find(t,t)[0];if(!r){const t={type:p.closeSuggestion};return e.dispatch(e.state.tr.setMeta(y,t)),!1}const s=n.popupDecoration.find()[0];if((null==s?void 0:s.spec.id)===r.spec.id)return!1;const i={type:p.openSuggestion,decoration:r};return e.dispatch(e.state.tr.setMeta(y,i)),!1},D={debounceMs:2e3,createUpdatePopup:(e,t,n,o,r)=>{const s=document.createElement("div");s.className="grammar-suggest-tooltip";const{spec:i}=t,a=e.dom.getBoundingClientRect();s.id=g.grammarSuggestPopup;const c=e.coordsAtPos(n),l=e.dom.scrollTop||0,d=e.dom.scrollLeft||0;s.style.left=`${c.left-a.left+d}px`,s.style.top=`${c.bottom-a.top+l+5}px`;const p=document.createElement("div");p.className="grammar-suggest-tooltip-apply",p.innerText=i.text||i.originalText,i.text||(p.style.textDecoration="line-through",p.style.color="red"),p.onclick=()=>{o(e,t)},s.appendChild(p);const u=document.createElement("div");return u.innerHTML="<svg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 24 24' width='24' height='24'><path d='M12 22C6.47715 22 2 17.5228 2 12C2 6.47715 6.47715 2 12 2C17.5228 2 22 6.47715 22 12C22 17.5228 17.5228 22 12 22ZM12 20C16.4183 20 20 16.4183 20 12C20 7.58172 16.4183 4 12 4C7.58172 4 4 7.58172 4 12C4 16.4183 7.58172 20 12 20ZM12 10.5858L14.8284 7.75736L16.2426 9.17157L13.4142 12L16.2426 14.8284L14.8284 16.2426L12 13.4142L9.17157 16.2426L7.75736 14.8284L10.5858 12L7.75736 9.17157L9.17157 7.75736L12 10.5858Z'></path></svg>",u.className="grammar-suggest-tooltip-discard",u.onclick=()=>{r(e,t)},s.appendChild(u),s},withYjs:!1},A={maxSelection:1e3},I=(e,r=D)=>{let s=!1;return new t({key:y,state:{init:()=>({lastText:"",decorations:o.empty,popupDecoration:o.empty}),apply(e,t,s,i){const l=e.getMeta(y);return(null==l?void 0:l.type)===p.suggestionUpdate?((e,t,o)=>{const{changedRegion:r,fix:s,mapping:i,text:a}=t,c=M(r.newText,s.result).filter((e=>!O(e))).filter((e=>e.original!==`${e.replacement}\n`)).map((({from:e,to:t,original:o,replacement:s})=>{const a=C(r.start+e,i),c=C(r.start+(s.endsWith("\n")?t-1:t),i),l={text:s.endsWith("\n")?s.slice(0,-1):s,originalText:o,id:{}};return n.inline(a,c,{class:"grammarSuggestion "+(""===s?"removalSuggestion":"")},l)}));return Object.assign(Object.assign({},e),{decorations:e.decorations.add(o.doc,c),lastText:a})})(t,l,e):(null==l?void 0:l.type)===p.acceptSuggestion?((e,t,n)=>{const r=e.decorations.remove(e.decorations.find(0,n.doc.nodeSize,(e=>e.id===t.id)));return Object.assign(Object.assign({},e),{lastText:S(n.doc),decorations:r.map(n.mapping,n.doc),popupDecoration:o.empty})})(t,l,e):(null==l?void 0:l.type)===p.openSuggestion?((e,t,r,s)=>{const{decoration:i}=t,a={id:i.spec.id};return Object.assign(Object.assign({},e),{popupDecoration:o.create(r.doc,[n.widget(i.from,((e,t)=>{const n=t();return n?s.createUpdatePopup(e,i,n,P,L):document.createElement("div")}),Object.assign(Object.assign({},a),{stopEvent:()=>!0}))])})})(t,l,e,r):(null==l?void 0:l.type)===p.closeSuggestion?Object.assign(Object.assign({},t),{popupDecoration:o.empty}):(null==l?void 0:l.type)===p.discardSuggestion?((e,t,n)=>Object.assign(Object.assign({},e),{decorations:e.decorations.remove(e.decorations.find(0,n.doc.nodeSize,(e=>e.id===t.id))),popupDecoration:o.empty}))(t,l,e):e.docChanged?((e,t,n,o)=>{const r=j(n.doc).text,{text:s,mapping:i}=j(t.doc);if(s===r)return e;let l=t.mapping;if(o){const e=t.doc.content.findDiffStart(n.doc.content),o=n.doc.content.findDiffEnd(t.doc.content),r=new a(o&&e?[e,o.a-e,o.b-e]:[0,0,0]);l=new c([r])}const d=w(r,s),p=e.decorations.map(l,t.doc),g=C(d.start,i),u=C(d.end,i),m=e.popupDecoration.map(l,t.doc);return Object.assign(Object.assign({},e),{decorations:p.remove(p.find(g,u)),popupDecoration:m.remove(m.find(g,u))})})(t,e,s,r.withYjs):t}},props:{handleClick:k,decorations:e=>{const t=y.getState(e);if(!t)return null;const n=v.getState(e);return(null==n?void 0:n.status)!==x.idle?t.decorations:t.decorations.add(e.doc,t.popupDecoration.find())}},view(){const t=((e,t)=>i((e=>{var n;const o=j(e.state.doc),r=(null===(n=y.getState(e.state))||void 0===n?void 0:n.lastText)||"",s=w(r,o.text);T(t,s.newText).then((t=>{if(S(e.state.doc)!==o.text)return;const n={type:p.suggestionUpdate,fix:t,changedRegion:s,mapping:o.mapping,text:o.text};e.dispatch(e.state.tr.setMeta(y,n))})).catch((e=>{console.error("Grammar suggest API error",e)}))}),e.debounceMs))(r,e);return{update(e,n){const o=y.getState(e.state);e.state.doc.textBetween(0,e.state.doc.nodeSize-2,"/n")===n.doc.textBetween(0,n.doc.nodeSize-2,"/n")&&s||!o||o.lastText===S(e.state.doc)||(s=!0,t(e))}}}})},R=(e,t,n,o,r,i,a)=>s(void 0,void 0,void 0,(function*(){var e;let n="";const c=new AbortController;try{const l=null===(e=(yield fetch("https://suggestion-gw5lxik4dq-uc.a.run.app",{method:"POST",cache:"no-cache",headers:{"Content-Type":"application/json",Authorization:"Bearer -qKivjCv6MfQSmgF438PjEY7RnLfqoVe"},signal:c.signal,body:JSON.stringify({model:"gpt-3.5-turbo",modelParams:{input:[t],task:r,params:a}})})).body)||void 0===e?void 0:e.getReader(),d=({done:e,value:t})=>s(void 0,void 0,void 0,(function*(){var s;if(e)return;if(null===(s=v.getState(o.state))||void 0===s?void 0:s.isCancelled)return void(yield c.abort());const a=(new TextDecoder).decode(t);try{n+=a,o.dispatch(o.state.tr.setMeta(v,Object.assign({type:r,status:x.streaming,result:n},i&&{selection:i})))}catch(e){console.error("Could not parse stream message",a,e)}return null==l?void 0:l.read().then(d)}));yield null==l?void 0:l.read().then(d),o.dispatch(o.state.tr.setMeta(v,Object.assign({type:r,status:x.finished,result:n},i&&{selection:i})))}catch(e){console.error("Error:",e)}})),U=(e,n=A)=>{let o=!1;return new t({key:v,state:{init:()=>({status:x.idle}),apply(e,t,n,r){const s=e.getMeta(v);return t.status===x.done||t.status===x.rejected?(o=!1,{status:x.idle}):(null==s?void 0:s.status)===x.cancelled?Object.assign(Object.assign(Object.assign({},t),s),{isCancelled:!0}):s&&(e=>Object.values(f).includes(e.type)||Object.values(h).includes(e.type))(s)?t.type&&s.type!==t.type?t:Object.assign(Object.assign({},t),s):t}},view:()=>({update(e,t){const r=v.getState(e.state);let i=e.state.tr;if((null==r?void 0:r.status)===x.new&&!o)switch(r.type){case h.Complete:o=!0,((e,t,n)=>{s(void 0,void 0,void 0,(function*(){const{doc:e}=t.state,n=[];e.descendants((e=>{"paragraph"===e.type.name&&n.push(e.textContent)}));let o="";o=n.length>=2?n.slice(-2).join(" "):n.join(" "),R(0,o,0,t,h.Complete)}))})(0,e);break;case h.MakeLonger:case h.MakeShorter:case h.Improve:case h.Simplify:case h.Explain:case h.ActionItems:case f.Translate:case f.ChangeTone:o=!0,((e,t,n,o,r,s)=>{const i=n.state.selection;if(!i)return void n.dispatch(n.state.tr.setMeta(v,{task:e,status:x.done}));if(n.state.doc.textBetween(i.from,i.to).length>r)return void n.dispatch(n.state.tr.setMeta(v,{type:e,status:x.error,error:"Selection is too big"}));const a=n.state.doc.textBetween(i.from,i.to,"\n");R(0,a,0,n,e,i,s)})(r.type,0,e,0,n.maxSelection,r.params)}if((null==r?void 0:r.status)===x.accepted){if(r.error)return i.setMeta(v,{type:r.type,status:x.done}),void e.dispatch(i);switch(r.type){case h.Complete:i=i.insertText(r.result||"",e.state.doc.nodeSize-2),i.setMeta(v,{type:h.Complete,status:x.done}),e.dispatch(i),e.focus();break;case h.MakeLonger:case h.MakeShorter:case h.Improve:case h.Simplify:case h.Explain:case h.ActionItems:case f.Translate:case f.ChangeTone:if(r.selection&&r.result){const t=r.result.split("\n\n").map((t=>e.state.schema.node("paragraph",null,e.state.schema.text(t)))),n=l.fromArray(t);i.selection.replace(i,new d(n,0,0))}i.setMeta(v,{type:r.type,status:x.done}),e.dispatch(i),e.focus()}}}})})};export{u as MoodParamType,f as OpenAiPromptsWithParam,h as OpenAiPromptsWithoutParam,x as Status,m as TranslationTargetLanguage,U as completePlugin,v as completePluginKey,A as defaultCompleteOptions,D as defaultOptions,I as grammarSuggestPlugin};
+import{PluginKey as e,TextSelection as t,Plugin as n}from"prosemirror-state";import{Decoration as o,DecorationSet as r}from"prosemirror-view";import s from"fast-diff";import{__awaiter as a}from"tslib";import i from"lodash.debounce";import{StepMap as c,Mapping as l}from"prosemirror-transform";import{Fragment as d,Slice as p}from"prosemirror-model";var g,u,m,h,f,x,v;!function(e){e.suggestionUpdate="suggestionUpdate",e.acceptSuggestion="acceptSuggestion",e.openSuggestion="openSuggestion",e.closeSuggestion="closeSuggestion",e.discardSuggestion="discardSuggestion"}(g||(g={})),function(e){e.grammarSuggestPopup="grammar-suggest-popup"}(u||(u={})),function(e){e.Casual="Casual",e.Confident="Confident",e.Straightforward="Straightforward",e.Friendly="Friendly"}(m||(m={})),function(e){e.English="English",e.Spanish="Spanish",e.French="French",e.German="German",e.Italian="Italian",e.Portuguese="Portuguese",e.Dutch="Dutch",e.Russian="Russian",e.Chinese="Chinese",e.Korean="Korean",e.Japanese="Japanese"}(h||(h={})),function(e){e.Complete="Complete",e.Improve="Improve",e.MakeLonger="MakeLonger",e.MakeShorter="MakeShorter",e.Simplify="Simplify",e.Explain="Explain",e.ActionItems="ActionItems"}(f||(f={})),function(e){e.ChangeTone="ChangeTone",e.Translate="Translate"}(x||(x={})),function(e){e.idle="idle",e.new="new",e.streaming="streaming",e.finished="finished",e.accepted="accepted",e.cancelled="cancelled",e.rejected="rejected",e.done="done",e.error="error"}(v||(v={}));const y=new e("completePlugin"),S=new e("grammarSuggestPlugin"),w=e=>{let t="";return e.descendants(((e,n)=>{e.isText&&(t+=`${e.text}\n`)})),t},b=(e,t)=>{if(e===t)return{start:e.length,end:e.length,oldStart:e.length,oldEnd:e.length,oldText:"",newText:""};const n=s(e,t),o=n[0],r=o[0]===s.EQUAL?o[1].length-1:0,a=((e,t)=>{if(t[t.length-1][0]!==s.EQUAL)return e.length;const n=t.slice(0,t.length-1);let o=0;for(const e of n){const[t,n]=e;t===s.EQUAL&&(o+=n.length),t===s.DELETE&&(o+=n.length)}return o})(e,n),i=e.lastIndexOf("\n",r),c=e.indexOf("\n",a),l=-1===i?0:i+1,d=-1===c?e.length:c,p=e.slice(l,d),g=t.slice(l,t.length-(e.length-d));return{start:l,end:l+g.length,oldStart:l,oldEnd:d,oldText:p,newText:g}},j=[{docPos:1,textPos:0}],C=e=>{let t="";const n=[];return e.descendants(((e,o)=>{e.isText&&(n.push({docPos:o,textPos:t.length}),t+=`${e.text}\n`)})),{text:t,mapping:n.length?n:j}},T=(e,t)=>{for(let n=0;n<t.length;n++)if(e>=t[n].textPos&&(void 0===t[n+1]||e<t[n+1].textPos))return t[n].docPos+(e-t[n].textPos);throw new Error("textPositionToDocumentPosition: textPos not found in mapping")},O=(e,t)=>a(void 0,void 0,void 0,(function*(){const n=[...t.split("\n")];return fetch("https://prosemirror-ai-plugin.web.app/api/suggestion",{method:"POST",cache:"no-cache",headers:{"Content-Type":"application/json",Authorization:`Bearer ${e}`},body:JSON.stringify({model:"gpt-3.5-turbo",modelParams:{input:n}})}).then((e=>e.ok?e.json():Promise.reject(e))).then((e=>!e||e.some((e=>(e=>{try{JSON.parse(e)}catch(e){return!0}return!1})(e)))?{fixed:!1,result:t}:{result:e.map((e=>JSON.parse(e).result)).join("\n"),fixed:!0})).catch((e=>(e.text().then((t=>{console.error({status:e.status,text:t})})),{fixed:!1,result:t})))})),E=e=>e.replacement===e.original,M=e=>e.reduce(((e,t)=>{const n=e[e.length-1],o=e.slice(0,e.length-1);if(!n)return[t];return[...o,...((e,t)=>{if(e.to!==t.from)throw new Error(`Replace pairs must be adjacent\n\n, ${JSON.stringify({leftReplace:e,rightReplace:t})}`);if(e.replacement.endsWith("\n"))return[e,t];if(E(e)&&E(t))return[{from:e.from,to:t.to,original:e.original+t.original,replacement:e.replacement+t.replacement}];if(E(e)){if(t.replacement.startsWith(" ")&&""===t.original||t.original.startsWith(" ")&&""===t.replacement)return[e,t];const n=e.original.split(" "),o=n.pop()||"",r=n.join(" ")+(n.length?" ":"");return[...r.length?[{from:e.from,to:e.from+r.length,original:r,replacement:r}]:[],{from:e.from+r.length,to:t.to,original:o+t.original,replacement:o+t.replacement}]}if(E(t)){if(e.replacement.endsWith(" ")&&""===e.original||e.original.endsWith(" ")&&""===e.replacement)return[e,t];const n=t.original.split(" "),o=n.shift()||"",r=(n.length?" ":"")+n.join(" ");return" "===r?[{from:e.from,to:t.to,original:e.original+t.original,replacement:e.replacement+t.replacement}]:[{from:e.from,to:e.to+o.length,original:e.original+o,replacement:e.replacement+o},...r.length?[{from:e.to+o.length,to:t.to,original:r,replacement:r}]:[]]}return[{from:e.from,to:t.to,original:e.original+t.original,replacement:e.replacement+t.replacement}]})(n,t)]}),[]),P=(e,t)=>{const n=(e=>{let t=0;return e.map((([e,n])=>{switch(e){case s.EQUAL:const e=n.lastIndexOf("\n"),o={from:t,to:t+n.length,original:n,replacement:n};if(-1!==e&&"\n"!==n){const o=[{from:t,to:t+e+1,original:n.slice(0,e+1),replacement:n.slice(0,e+1)},{from:t+e+1,to:t+n.length,original:n.slice(e+1),replacement:n.slice(e+1)}];return t+=n.length,o}return t+=n.length,[o];case s.DELETE:const r={from:t,to:t+n.length,original:n,replacement:""};return t+=n.length,[r];case s.INSERT:return[{from:t,to:t,original:"",replacement:n}]}})).flat()})(s(e,t));return M(M(n))},L=(e,n)=>{var o;const r=null===(o=S.getState(e.state))||void 0===o?void 0:o.decorations.find(0,e.state.doc.nodeSize,(e=>e.id===n.spec.id))[0];if(!r)return;const{text:s}=r.spec,{from:a,to:i}=r,c={type:g.acceptSuggestion,id:n.spec.id},l=s.split("\n").map((t=>e.state.schema.node("paragraph",null,e.state.schema.text(t)))),u=d.fromArray(l);let{tr:m}=e.state;m=m.setSelection(t.create(e.state.doc,a,i)),m.selection.replace(m,new p(u,1,1)),m.setMeta(S,c),e.dispatch(m)},k=(e,t)=>{const{spec:n}=t,o={type:g.discardSuggestion,id:n.id},r=e.state.tr.setMeta(S,o);e.dispatch(r)},D=(e,t)=>{const n=S.getState(e.state);if(!n)return!1;const{decorations:o}=n,r=o.find(t,t)[0];if(!r){const t={type:g.closeSuggestion};return e.dispatch(e.state.tr.setMeta(S,t)),!1}const s=n.popupDecoration.find()[0];if((null==s?void 0:s.spec.id)===r.spec.id)return!1;const a={type:g.openSuggestion,decoration:r};return e.dispatch(e.state.tr.setMeta(S,a)),!1},A={debounceMs:2e3,createUpdatePopup:(e,t,n,o,r)=>{const s=document.createElement("div");s.className="grammar-suggest-tooltip";const{spec:a}=t,i=e.dom.getBoundingClientRect();s.id=u.grammarSuggestPopup;const c=e.coordsAtPos(n),l=e.dom.scrollTop||0,d=e.dom.scrollLeft||0;s.style.left=`${c.left-i.left+d}px`,s.style.top=`${c.bottom-i.top+l+5}px`;const p=document.createElement("div");p.className="grammar-suggest-tooltip-apply",p.innerText=a.text||a.originalText,a.text||(p.style.textDecoration="line-through",p.style.color="red"),p.onclick=()=>{o(e,t)},s.appendChild(p);const g=document.createElement("div");return g.innerHTML="<svg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 24 24' width='24' height='24'><path d='M12 22C6.47715 22 2 17.5228 2 12C2 6.47715 6.47715 2 12 2C17.5228 2 22 6.47715 22 12C22 17.5228 17.5228 22 12 22ZM12 20C16.4183 20 20 16.4183 20 12C20 7.58172 16.4183 4 12 4C7.58172 4 4 7.58172 4 12C4 16.4183 7.58172 20 12 20ZM12 10.5858L14.8284 7.75736L16.2426 9.17157L13.4142 12L16.2426 14.8284L14.8284 16.2426L12 13.4142L9.17157 16.2426L7.75736 14.8284L10.5858 12L7.75736 9.17157L9.17157 7.75736L12 10.5858Z'></path></svg>",g.className="grammar-suggest-tooltip-discard",g.onclick=()=>{r(e,t)},s.appendChild(g),s},withYjs:!1},I={maxSelection:1e3},R=(e,t=A)=>{let s=!1;return new n({key:S,state:{init:()=>({lastText:"",decorations:r.empty,popupDecoration:r.empty}),apply(e,n,s,a){const i=e.getMeta(S);return(null==i?void 0:i.type)===g.suggestionUpdate?((e,t,n)=>{const{changedRegion:r,fix:s,mapping:a,text:i}=t,c=P(r.newText,s.result).filter((e=>!E(e))).filter((e=>e.original!==`${e.replacement}\n`)).map((({from:e,to:t,original:n,replacement:s})=>{const i=T(r.start+e,a),c=T(r.start+(s.endsWith("\n")?t-1:t),a),l={text:s.endsWith("\n")?s.slice(0,-1):s,originalText:n,id:{}};return o.inline(i,c,{class:"grammarSuggestion "+(""===s?"removalSuggestion":"")},l)}));return Object.assign(Object.assign({},e),{decorations:e.decorations.add(n.doc,c),lastText:i})})(n,i,e):(null==i?void 0:i.type)===g.acceptSuggestion?((e,t,n)=>{const o=e.decorations.remove(e.decorations.find(0,n.doc.nodeSize,(e=>e.id===t.id)));return Object.assign(Object.assign({},e),{lastText:w(n.doc),decorations:o.map(n.mapping,n.doc),popupDecoration:r.empty})})(n,i,e):(null==i?void 0:i.type)===g.openSuggestion?((e,t,n,s)=>{const{decoration:a}=t,i={id:a.spec.id};return Object.assign(Object.assign({},e),{popupDecoration:r.create(n.doc,[o.widget(a.from,((e,t)=>{const n=t();return n?s.createUpdatePopup(e,a,n,L,k):document.createElement("div")}),Object.assign(Object.assign({},i),{stopEvent:()=>!0}))])})})(n,i,e,t):(null==i?void 0:i.type)===g.closeSuggestion?Object.assign(Object.assign({},n),{popupDecoration:r.empty}):(null==i?void 0:i.type)===g.discardSuggestion?((e,t,n)=>Object.assign(Object.assign({},e),{decorations:e.decorations.remove(e.decorations.find(0,n.doc.nodeSize,(e=>e.id===t.id))),popupDecoration:r.empty}))(n,i,e):e.docChanged?((e,t,n,o)=>{const r=C(n.doc).text,{text:s,mapping:a}=C(t.doc);if(s===r)return e;let i=t.mapping;if(o){const e=t.doc.content.findDiffStart(n.doc.content),o=n.doc.content.findDiffEnd(t.doc.content),r=new c(o&&e?[e,o.a-e,o.b-e]:[0,0,0]);i=new l([r])}const d=b(r,s),p=e.decorations.map(i,t.doc),g=T(d.start,a),u=T(d.end,a),m=e.popupDecoration.map(i,t.doc);return Object.assign(Object.assign({},e),{decorations:p.remove(p.find(g,u)),popupDecoration:m.remove(m.find(g,u))})})(n,e,s,t.withYjs):n}},props:{handleClick:D,decorations:e=>{const t=S.getState(e);if(!t)return null;const n=y.getState(e);return(null==n?void 0:n.status)!==v.idle?t.decorations:t.decorations.add(e.doc,t.popupDecoration.find())}},view(){const n=((e,t)=>i((e=>{var n;const o=C(e.state.doc),r=(null===(n=S.getState(e.state))||void 0===n?void 0:n.lastText)||"",s=b(r,o.text);O(t,s.newText).then((t=>{if(w(e.state.doc)!==o.text)return;const n={type:g.suggestionUpdate,fix:t,changedRegion:s,mapping:o.mapping,text:o.text};e.dispatch(e.state.tr.setMeta(S,n))})).catch((e=>{console.error("Grammar suggest API error",e)}))}),e.debounceMs))(t,e);return{update(e,t){const o=S.getState(e.state);e.state.doc.textBetween(0,e.state.doc.nodeSize-2,"/n")===t.doc.textBetween(0,t.doc.nodeSize-2,"/n")&&s||!o||o.lastText===w(e.state.doc)||(s=!0,n(e))}}}})},U=(e,t,n,o,r,s,i)=>a(void 0,void 0,void 0,(function*(){var e;let n="";const c=new AbortController;try{const l=null===(e=(yield fetch("https://suggestion-gw5lxik4dq-uc.a.run.app",{method:"POST",cache:"no-cache",headers:{"Content-Type":"application/json",Authorization:"Bearer -qKivjCv6MfQSmgF438PjEY7RnLfqoVe"},signal:c.signal,body:JSON.stringify({model:"gpt-3.5-turbo",modelParams:{input:[t],task:r,params:i}})})).body)||void 0===e?void 0:e.getReader(),d=({done:e,value:t})=>a(void 0,void 0,void 0,(function*(){var a;if(e)return;if(null===(a=y.getState(o.state))||void 0===a?void 0:a.isCancelled)return void(yield c.abort());const i=(new TextDecoder).decode(t);try{n+=i,o.dispatch(o.state.tr.setMeta(y,Object.assign({type:r,status:v.streaming,result:n},s&&{selection:s})))}catch(e){console.error("Could not parse stream message",i,e)}return null==l?void 0:l.read().then(d)}));yield null==l?void 0:l.read().then(d),o.dispatch(o.state.tr.setMeta(y,Object.assign({type:r,status:v.finished,result:n},s&&{selection:s})))}catch(e){console.error("Error:",e)}})),N=(e,t=I)=>{let o=!1;return new n({key:y,state:{init:()=>({status:v.idle}),apply(e,t,n,r){const s=e.getMeta(y);return t.status===v.done||t.status===v.rejected?(o=!1,{status:v.idle}):(null==s?void 0:s.status)===v.cancelled?Object.assign(Object.assign(Object.assign({},t),s),{isCancelled:!0}):s&&(e=>Object.values(x).includes(e.type)||Object.values(f).includes(e.type))(s)?t.type&&s.type!==t.type?t:Object.assign(Object.assign({},t),s):t}},view:()=>({update(e,n){const r=y.getState(e.state);let s=e.state.tr;if((null==r?void 0:r.status)===v.new&&!o)switch(r.type){case f.Complete:o=!0,((e,t,n)=>{a(void 0,void 0,void 0,(function*(){const{doc:e}=t.state,n=[];e.descendants((e=>{"paragraph"===e.type.name&&n.push(e.textContent)}));let o="";o=n.length>=2?n.slice(-2).join(" "):n.join(" "),U(0,o,0,t,f.Complete)}))})(0,e);break;case f.MakeLonger:case f.MakeShorter:case f.Improve:case f.Simplify:case f.Explain:case f.ActionItems:case x.Translate:case x.ChangeTone:o=!0,((e,t,n,o,r,s)=>{const a=n.state.selection;if(!a)return void n.dispatch(n.state.tr.setMeta(y,{task:e,status:v.done}));if(n.state.doc.textBetween(a.from,a.to).length>r)return void n.dispatch(n.state.tr.setMeta(y,{type:e,status:v.error,error:"Selection is too big"}));const i=n.state.doc.textBetween(a.from,a.to,"\n");U(0,i,0,n,e,a,s)})(r.type,0,e,0,t.maxSelection,r.params)}if((null==r?void 0:r.status)===v.accepted){if(r.error)return s.setMeta(y,{type:r.type,status:v.done}),void e.dispatch(s);switch(r.type){case f.Complete:s=s.insertText(r.result||"",e.state.doc.nodeSize-2),s.setMeta(y,{type:f.Complete,status:v.done}),e.dispatch(s),e.focus();break;case f.MakeLonger:case f.MakeShorter:case f.Improve:case f.Simplify:case f.Explain:case f.ActionItems:case x.Translate:case x.ChangeTone:if(r.selection&&r.result){const t=r.result.split("\n\n").map((t=>e.state.schema.node("paragraph",null,e.state.schema.text(t)))),n=d.fromArray(t);s.selection.replace(s,new p(n,0,0))}s.setMeta(y,{type:r.type,status:v.done}),e.dispatch(s),e.focus()}}}})})};export{m as MoodParamType,x as OpenAiPromptsWithParam,f as OpenAiPromptsWithoutParam,v as Status,h as TranslationTargetLanguage,N as completePlugin,y as completePluginKey,I as defaultCompleteOptions,A as defaultOptions,R as grammarSuggestPlugin};

package/dist/index.js

@@ -1 +1 @@
-"use strict";var t,e,o,n,r,s,a,i=require("prosemirror-state"),p=require("prosemirror-view"),c=require("fast-diff"),l=require("tslib"),d=require("lodash.debounce"),u=require("prosemirror-transform"),g=require("prosemirror-model");!function(t){t.suggestionUpdate="suggestionUpdate",t.acceptSuggestion="acceptSuggestion",t.openSuggestion="openSuggestion",t.closeSuggestion="closeSuggestion",t.discardSuggestion="discardSuggestion"}(t||(t={})),function(t){t.grammarSuggestPopup="grammar-suggest-popup"}(e||(e={})),exports.MoodParamType=void 0,(o=exports.MoodParamType||(exports.MoodParamType={})).Casual="Casual",o.Confident="Confident",o.Straightforward="Straightforward",o.Friendly="Friendly",exports.TranslationTargetLanguage=void 0,(n=exports.TranslationTargetLanguage||(exports.TranslationTargetLanguage={})).English="English",n.Spanish="Spanish",n.French="French",n.German="German",n.Italian="Italian",n.Portuguese="Portuguese",n.Dutch="Dutch",n.Russian="Russian",n.Chinese="Chinese",n.Korean="Korean",n.Japanese="Japanese",exports.OpenAiPromptsWithoutParam=void 0,(r=exports.OpenAiPromptsWithoutParam||(exports.OpenAiPromptsWithoutParam={})).Complete="Complete",r.Improve="Improve",r.MakeLonger="MakeLonger",r.MakeShorter="MakeShorter",r.Simplify="Simplify",r.Explain="Explain",r.ActionItems="ActionItems",exports.OpenAiPromptsWithParam=void 0,(s=exports.OpenAiPromptsWithParam||(exports.OpenAiPromptsWithParam={})).ChangeTone="ChangeTone",s.Translate="Translate",exports.Status=void 0,(a=exports.Status||(exports.Status={})).idle="idle",a.new="new",a.streaming="streaming",a.finished="finished",a.accepted="accepted",a.cancelled="cancelled",a.rejected="rejected",a.done="done",a.error="error";const m=new i.PluginKey("completePlugin"),h=new i.PluginKey("grammarSuggestPlugin"),x=t=>{let e="";return t.descendants(((t,o)=>{t.isText&&(e+=`${t.text}\n`)})),e},f=(t,e)=>{if(t===e)return{start:t.length,end:t.length,oldStart:t.length,oldEnd:t.length,oldText:"",newText:""};const o=c(t,e),n=o[0],r=n[0]===c.EQUAL?n[1].length-1:0,s=((t,e)=>{if(e[e.length-1][0]!==c.EQUAL)return t.length;const o=e.slice(0,e.length-1);let n=0;for(const t of o){const[e,o]=t;e===c.EQUAL&&(n+=o.length),e===c.DELETE&&(n+=o.length)}return n})(t,o),a=t.lastIndexOf("\n",r),i=t.indexOf("\n",s),p=-1===a?0:a+1,l=-1===i?t.length:i,d=t.slice(p,l),u=e.slice(p,e.length-(t.length-l));return{start:p,end:p+u.length,oldStart:p,oldEnd:l,oldText:d,newText:u}},P=[{docPos:1,textPos:0}],S=t=>{let e="";const o=[];return t.descendants(((t,n)=>{t.isText&&(o.push({docPos:n,textPos:e.length}),e+=`${t.text}\n`)})),{text:e,mapping:o.length?o:P}},v=(t,e)=>{for(let o=0;o<e.length;o++)if(t>=e[o].textPos&&(void 0===e[o+1]||t<e[o+1].textPos))return e[o].docPos+(t-e[o].textPos);throw new Error("textPositionToDocumentPosition: textPos not found in mapping")},y=(t,e)=>l.__awaiter(void 0,void 0,void 0,(function*(){const o=[...e.split("\n")];return fetch("https://prosemirror-ai-plugin.web.app/api/suggestion",{method:"POST",cache:"no-cache",headers:{"Content-Type":"application/json",Authorization:`Bearer ${t}`},body:JSON.stringify({model:"gpt-3.5-turbo",modelParams:{input:o}})}).then((t=>t.ok?t.json():Promise.reject(t))).then((t=>!t||t.some((t=>(t=>{try{JSON.parse(t)}catch(t){return!0}return!1})(t)))?{fixed:!1,result:e}:{result:t.map((t=>JSON.parse(t).result)).join("\n"),fixed:!0})).catch((t=>(t.text().then((e=>{console.error({status:t.status,text:e})})),{fixed:!1,result:e})))})),O=t=>t.replacement===t.original,w=t=>t.reduce(((t,e)=>{const o=t[t.length-1],n=t.slice(0,t.length-1);if(!o)return[e];return[...n,...((t,e)=>{if(t.to!==e.from)throw new Error(`Replace pairs must be adjacent\n\n, ${JSON.stringify({leftReplace:t,rightReplace:e})}`);if(t.replacement.endsWith("\n"))return[t,e];if(O(t)&&O(e))return[{from:t.from,to:e.to,original:t.original+e.original,replacement:t.replacement+e.replacement}];if(O(t)){if(e.replacement.startsWith(" ")&&""===e.original||e.original.startsWith(" ")&&""===e.replacement)return[t,e];const o=t.original.split(" "),n=o.pop()||"",r=o.join(" ")+(o.length?" ":"");return[...r.length?[{from:t.from,to:t.from+r.length,original:r,replacement:r}]:[],{from:t.from+r.length,to:e.to,original:n+e.original,replacement:n+e.replacement}]}if(O(e)){if(t.replacement.endsWith(" ")&&""===t.original||t.original.endsWith(" ")&&""===t.replacement)return[t,e];const o=e.original.split(" "),n=o.shift()||"",r=(o.length?" ":"")+o.join(" ");return" "===r?[{from:t.from,to:e.to,original:t.original+e.original,replacement:t.replacement+e.replacement}]:[{from:t.from,to:t.to+n.length,original:t.original+n,replacement:t.replacement+n},...r.length?[{from:t.to+n.length,to:e.to,original:r,replacement:r}]:[]]}return[{from:t.from,to:e.to,original:t.original+e.original,replacement:t.replacement+e.replacement}]})(o,e)]}),[]),T=(t,e)=>{const o=(t=>{let e=0;return t.map((([t,o])=>{switch(t){case c.EQUAL:const t=o.lastIndexOf("\n"),n={from:e,to:e+o.length,original:o,replacement:o};if(-1!==t&&"\n"!==o){const n=[{from:e,to:e+t+1,original:o.slice(0,t+1),replacement:o.slice(0,t+1)},{from:e+t+1,to:e+o.length,original:o.slice(t+1),replacement:o.slice(t+1)}];return e+=o.length,n}return e+=o.length,[n];case c.DELETE:const r={from:e,to:e+o.length,original:o,replacement:""};return e+=o.length,[r];case c.INSERT:return[{from:e,to:e,original:"",replacement:o}]}})).flat()})(c(t,e));return w(w(o))},A=(e,o)=>{var n;const r=null===(n=h.getState(e.state))||void 0===n?void 0:n.decorations.find(0,e.state.doc.nodeSize,(t=>t.id===o.spec.id))[0];if(!r)return;const{text:s}=r.spec,{from:a,to:i}=r,p={type:t.acceptSuggestion,id:o.spec.id},c=e.state.tr.insertText(s,a,i).setMeta(h,p);e.dispatch(c)},b=(e,o)=>{const{spec:n}=o,r={type:t.discardSuggestion,id:n.id},s=e.state.tr.setMeta(h,r);e.dispatch(s)},j=(e,o)=>{const n=h.getState(e.state);if(!n)return!1;const{decorations:r}=n,s=r.find(o,o)[0];if(!s){const o={type:t.closeSuggestion};return e.dispatch(e.state.tr.setMeta(h,o)),!1}const a=n.popupDecoration.find()[0];if((null==a?void 0:a.spec.id)===s.spec.id)return!1;const i={type:t.openSuggestion,decoration:s};return e.dispatch(e.state.tr.setMeta(h,i)),!1},C={debounceMs:2e3,createUpdatePopup:(t,o,n,r,s)=>{const a=document.createElement("div");a.className="grammar-suggest-tooltip";const{spec:i}=o,p=t.dom.getBoundingClientRect();a.id=e.grammarSuggestPopup;const c=t.coordsAtPos(n),l=t.dom.scrollTop||0,d=t.dom.scrollLeft||0;a.style.left=`${c.left-p.left+d}px`,a.style.top=`${c.bottom-p.top+l+5}px`;const u=document.createElement("div");u.className="grammar-suggest-tooltip-apply",u.innerText=i.text||i.originalText,i.text||(u.style.textDecoration="line-through",u.style.color="red"),u.onclick=()=>{r(t,o)},a.appendChild(u);const g=document.createElement("div");return g.innerHTML="<svg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 24 24' width='24' height='24'><path d='M12 22C6.47715 22 2 17.5228 2 12C2 6.47715 6.47715 2 12 2C17.5228 2 22 6.47715 22 12C22 17.5228 17.5228 22 12 22ZM12 20C16.4183 20 20 16.4183 20 12C20 7.58172 16.4183 4 12 4C7.58172 4 4 7.58172 4 12C4 16.4183 7.58172 20 12 20ZM12 10.5858L14.8284 7.75736L16.2426 9.17157L13.4142 12L16.2426 14.8284L14.8284 16.2426L12 13.4142L9.17157 16.2426L7.75736 14.8284L10.5858 12L7.75736 9.17157L9.17157 7.75736L12 10.5858Z'></path></svg>",g.className="grammar-suggest-tooltip-discard",g.onclick=()=>{s(t,o)},a.appendChild(g),a},withYjs:!1},M={maxSelection:1e3},W=(t,e,o,n,r,s,a)=>l.__awaiter(void 0,void 0,void 0,(function*(){var t;let o="";const i=new AbortController;try{const p=null===(t=(yield fetch("https://suggestion-gw5lxik4dq-uc.a.run.app",{method:"POST",cache:"no-cache",headers:{"Content-Type":"application/json",Authorization:"Bearer -qKivjCv6MfQSmgF438PjEY7RnLfqoVe"},signal:i.signal,body:JSON.stringify({model:"gpt-3.5-turbo",modelParams:{input:[e],task:r,params:a}})})).body)||void 0===t?void 0:t.getReader(),c=({done:t,value:e})=>l.__awaiter(void 0,void 0,void 0,(function*(){var a;if(t)return;if(null===(a=m.getState(n.state))||void 0===a?void 0:a.isCancelled)return void(yield i.abort());const l=(new TextDecoder).decode(e);try{o+=l,n.dispatch(n.state.tr.setMeta(m,Object.assign({type:r,status:exports.Status.streaming,result:o},s&&{selection:s})))}catch(t){console.error("Could not parse stream message",l,t)}return null==p?void 0:p.read().then(c)}));yield null==p?void 0:p.read().then(c),n.dispatch(n.state.tr.setMeta(m,Object.assign({type:r,status:exports.Status.finished,result:o},s&&{selection:s})))}catch(t){console.error("Error:",t)}}));exports.completePlugin=(t,e=M)=>{let o=!1;return new i.Plugin({key:m,state:{init:()=>({status:exports.Status.idle}),apply(t,e,n,r){const s=t.getMeta(m);return e.status===exports.Status.done||e.status===exports.Status.rejected?(o=!1,{status:exports.Status.idle}):(null==s?void 0:s.status)===exports.Status.cancelled?Object.assign(Object.assign(Object.assign({},e),s),{isCancelled:!0}):s&&(t=>Object.values(exports.OpenAiPromptsWithParam).includes(t.type)||Object.values(exports.OpenAiPromptsWithoutParam).includes(t.type))(s)?e.type&&s.type!==e.type?e:Object.assign(Object.assign({},e),s):e}},view:()=>({update(t,n){const r=m.getState(t.state);let s=t.state.tr;if((null==r?void 0:r.status)===exports.Status.new&&!o)switch(r.type){case exports.OpenAiPromptsWithoutParam.Complete:o=!0,((t,e,o)=>{l.__awaiter(void 0,void 0,void 0,(function*(){const{doc:t}=e.state,o=[];t.descendants((t=>{"paragraph"===t.type.name&&o.push(t.textContent)}));let n="";n=o.length>=2?o.slice(-2).join(" "):o.join(" "),W(0,n,0,e,exports.OpenAiPromptsWithoutParam.Complete)}))})(0,t);break;case exports.OpenAiPromptsWithoutParam.MakeLonger:case exports.OpenAiPromptsWithoutParam.MakeShorter:case exports.OpenAiPromptsWithoutParam.Improve:case exports.OpenAiPromptsWithoutParam.Simplify:case exports.OpenAiPromptsWithoutParam.Explain:case exports.OpenAiPromptsWithoutParam.ActionItems:case exports.OpenAiPromptsWithParam.Translate:case exports.OpenAiPromptsWithParam.ChangeTone:o=!0,((t,e,o,n,r,s)=>{const a=o.state.selection;if(!a)return void o.dispatch(o.state.tr.setMeta(m,{task:t,status:exports.Status.done}));if(o.state.doc.textBetween(a.from,a.to).length>r)return void o.dispatch(o.state.tr.setMeta(m,{type:t,status:exports.Status.error,error:"Selection is too big"}));const i=o.state.doc.textBetween(a.from,a.to,"\n");W(0,i,0,o,t,a,s)})(r.type,0,t,0,e.maxSelection,r.params)}if((null==r?void 0:r.status)===exports.Status.accepted){if(r.error)return s.setMeta(m,{type:r.type,status:exports.Status.done}),void t.dispatch(s);switch(r.type){case exports.OpenAiPromptsWithoutParam.Complete:s=s.insertText(r.result||"",t.state.doc.nodeSize-2),s.setMeta(m,{type:exports.OpenAiPromptsWithoutParam.Complete,status:exports.Status.done}),t.dispatch(s),t.focus();break;case exports.OpenAiPromptsWithoutParam.MakeLonger:case exports.OpenAiPromptsWithoutParam.MakeShorter:case exports.OpenAiPromptsWithoutParam.Improve:case exports.OpenAiPromptsWithoutParam.Simplify:case exports.OpenAiPromptsWithoutParam.Explain:case exports.OpenAiPromptsWithoutParam.ActionItems:case exports.OpenAiPromptsWithParam.Translate:case exports.OpenAiPromptsWithParam.ChangeTone:if(r.selection&&r.result){const e=r.result.split("\n\n").map((e=>t.state.schema.node("paragraph",null,t.state.schema.text(e)))),o=g.Fragment.fromArray(e);s.selection.replace(s,new g.Slice(o,0,0))}s.setMeta(m,{type:r.type,status:exports.Status.done}),t.dispatch(s),t.focus()}}}})})},exports.completePluginKey=m,exports.defaultCompleteOptions=M,exports.defaultOptions=C,exports.grammarSuggestPlugin=(e,o=C)=>{let n=!1;return new i.Plugin({key:h,state:{init:()=>({lastText:"",decorations:p.DecorationSet.empty,popupDecoration:p.DecorationSet.empty}),apply(e,n,r,s){const a=e.getMeta(h);return(null==a?void 0:a.type)===t.suggestionUpdate?((t,e,o)=>{const{changedRegion:n,fix:r,mapping:s,text:a}=e,i=T(n.newText,r.result).filter((t=>!O(t))).filter((t=>t.original!==`${t.replacement}\n`)).map((({from:t,to:e,original:o,replacement:r})=>{const a=v(n.start+t,s),i=v(n.start+(r.endsWith("\n")?e-1:e),s),c={text:r.endsWith("\n")?r.slice(0,-1):r,originalText:o,id:{}};return p.Decoration.inline(a,i,{class:"grammarSuggestion "+(""===r?"removalSuggestion":"")},c)}));return Object.assign(Object.assign({},t),{decorations:t.decorations.add(o.doc,i),lastText:a})})(n,a,e):(null==a?void 0:a.type)===t.acceptSuggestion?((t,e,o)=>{const n=t.decorations.remove(t.decorations.find(0,o.doc.nodeSize,(t=>t.id===e.id)));return Object.assign(Object.assign({},t),{lastText:x(o.doc),decorations:n.map(o.mapping,o.doc),popupDecoration:p.DecorationSet.empty})})(n,a,e):(null==a?void 0:a.type)===t.openSuggestion?((t,e,o,n)=>{const{decoration:r}=e,s={id:r.spec.id};return Object.assign(Object.assign({},t),{popupDecoration:p.DecorationSet.create(o.doc,[p.Decoration.widget(r.from,((t,e)=>{const o=e();return o?n.createUpdatePopup(t,r,o,A,b):document.createElement("div")}),Object.assign(Object.assign({},s),{stopEvent:()=>!0}))])})})(n,a,e,o):(null==a?void 0:a.type)===t.closeSuggestion?Object.assign(Object.assign({},n),{popupDecoration:p.DecorationSet.empty}):(null==a?void 0:a.type)===t.discardSuggestion?((t,e,o)=>Object.assign(Object.assign({},t),{decorations:t.decorations.remove(t.decorations.find(0,o.doc.nodeSize,(t=>t.id===e.id))),popupDecoration:p.DecorationSet.empty}))(n,a,e):e.docChanged?((t,e,o,n)=>{const r=S(o.doc).text,{text:s,mapping:a}=S(e.doc);if(s===r)return t;let i=e.mapping;if(n){const t=e.doc.content.findDiffStart(o.doc.content),n=o.doc.content.findDiffEnd(e.doc.content),r=n&&t?new u.StepMap([t,n.a-t,n.b-t]):new u.StepMap([0,0,0]);i=new u.Mapping([r])}const p=f(r,s),c=t.decorations.map(i,e.doc),l=v(p.start,a),d=v(p.end,a),g=t.popupDecoration.map(i,e.doc);return Object.assign(Object.assign({},t),{decorations:c.remove(c.find(l,d)),popupDecoration:g.remove(g.find(l,d))})})(n,e,r,o.withYjs):n}},props:{handleClick:j,decorations:t=>{const e=h.getState(t);if(!e)return null;const o=m.getState(t);return(null==o?void 0:o.status)!==exports.Status.idle?e.decorations:e.decorations.add(t.doc,e.popupDecoration.find())}},view(){const r=((e,o)=>d((e=>{var n;const r=S(e.state.doc),s=(null===(n=h.getState(e.state))||void 0===n?void 0:n.lastText)||"",a=f(s,r.text);y(o,a.newText).then((o=>{if(x(e.state.doc)!==r.text)return;const n={type:t.suggestionUpdate,fix:o,changedRegion:a,mapping:r.mapping,text:r.text};e.dispatch(e.state.tr.setMeta(h,n))})).catch((t=>{console.error("Grammar suggest API error",t)}))}),e.debounceMs))(o,e);return{update(t,e){const o=h.getState(t.state);t.state.doc.textBetween(0,t.state.doc.nodeSize-2,"/n")===e.doc.textBetween(0,e.doc.nodeSize-2,"/n")&&n||!o||o.lastText===x(t.state.doc)||(n=!0,r(t))}}}})};
+"use strict";var t,e,o,n,r,s,a,i=require("prosemirror-state"),p=require("prosemirror-view"),c=require("fast-diff"),l=require("tslib"),d=require("lodash.debounce"),u=require("prosemirror-transform"),g=require("prosemirror-model");!function(t){t.suggestionUpdate="suggestionUpdate",t.acceptSuggestion="acceptSuggestion",t.openSuggestion="openSuggestion",t.closeSuggestion="closeSuggestion",t.discardSuggestion="discardSuggestion"}(t||(t={})),function(t){t.grammarSuggestPopup="grammar-suggest-popup"}(e||(e={})),exports.MoodParamType=void 0,(o=exports.MoodParamType||(exports.MoodParamType={})).Casual="Casual",o.Confident="Confident",o.Straightforward="Straightforward",o.Friendly="Friendly",exports.TranslationTargetLanguage=void 0,(n=exports.TranslationTargetLanguage||(exports.TranslationTargetLanguage={})).English="English",n.Spanish="Spanish",n.French="French",n.German="German",n.Italian="Italian",n.Portuguese="Portuguese",n.Dutch="Dutch",n.Russian="Russian",n.Chinese="Chinese",n.Korean="Korean",n.Japanese="Japanese",exports.OpenAiPromptsWithoutParam=void 0,(r=exports.OpenAiPromptsWithoutParam||(exports.OpenAiPromptsWithoutParam={})).Complete="Complete",r.Improve="Improve",r.MakeLonger="MakeLonger",r.MakeShorter="MakeShorter",r.Simplify="Simplify",r.Explain="Explain",r.ActionItems="ActionItems",exports.OpenAiPromptsWithParam=void 0,(s=exports.OpenAiPromptsWithParam||(exports.OpenAiPromptsWithParam={})).ChangeTone="ChangeTone",s.Translate="Translate",exports.Status=void 0,(a=exports.Status||(exports.Status={})).idle="idle",a.new="new",a.streaming="streaming",a.finished="finished",a.accepted="accepted",a.cancelled="cancelled",a.rejected="rejected",a.done="done",a.error="error";const m=new i.PluginKey("completePlugin"),h=new i.PluginKey("grammarSuggestPlugin"),x=t=>{let e="";return t.descendants(((t,o)=>{t.isText&&(e+=`${t.text}\n`)})),e},f=(t,e)=>{if(t===e)return{start:t.length,end:t.length,oldStart:t.length,oldEnd:t.length,oldText:"",newText:""};const o=c(t,e),n=o[0],r=n[0]===c.EQUAL?n[1].length-1:0,s=((t,e)=>{if(e[e.length-1][0]!==c.EQUAL)return t.length;const o=e.slice(0,e.length-1);let n=0;for(const t of o){const[e,o]=t;e===c.EQUAL&&(n+=o.length),e===c.DELETE&&(n+=o.length)}return n})(t,o),a=t.lastIndexOf("\n",r),i=t.indexOf("\n",s),p=-1===a?0:a+1,l=-1===i?t.length:i,d=t.slice(p,l),u=e.slice(p,e.length-(t.length-l));return{start:p,end:p+u.length,oldStart:p,oldEnd:l,oldText:d,newText:u}},P=[{docPos:1,textPos:0}],S=t=>{let e="";const o=[];return t.descendants(((t,n)=>{t.isText&&(o.push({docPos:n,textPos:e.length}),e+=`${t.text}\n`)})),{text:e,mapping:o.length?o:P}},v=(t,e)=>{for(let o=0;o<e.length;o++)if(t>=e[o].textPos&&(void 0===e[o+1]||t<e[o+1].textPos))return e[o].docPos+(t-e[o].textPos);throw new Error("textPositionToDocumentPosition: textPos not found in mapping")},y=(t,e)=>l.__awaiter(void 0,void 0,void 0,(function*(){const o=[...e.split("\n")];return fetch("https://prosemirror-ai-plugin.web.app/api/suggestion",{method:"POST",cache:"no-cache",headers:{"Content-Type":"application/json",Authorization:`Bearer ${t}`},body:JSON.stringify({model:"gpt-3.5-turbo",modelParams:{input:o}})}).then((t=>t.ok?t.json():Promise.reject(t))).then((t=>!t||t.some((t=>(t=>{try{JSON.parse(t)}catch(t){return!0}return!1})(t)))?{fixed:!1,result:e}:{result:t.map((t=>JSON.parse(t).result)).join("\n"),fixed:!0})).catch((t=>(t.text().then((e=>{console.error({status:t.status,text:e})})),{fixed:!1,result:e})))})),O=t=>t.replacement===t.original,w=t=>t.reduce(((t,e)=>{const o=t[t.length-1],n=t.slice(0,t.length-1);if(!o)return[e];return[...n,...((t,e)=>{if(t.to!==e.from)throw new Error(`Replace pairs must be adjacent\n\n, ${JSON.stringify({leftReplace:t,rightReplace:e})}`);if(t.replacement.endsWith("\n"))return[t,e];if(O(t)&&O(e))return[{from:t.from,to:e.to,original:t.original+e.original,replacement:t.replacement+e.replacement}];if(O(t)){if(e.replacement.startsWith(" ")&&""===e.original||e.original.startsWith(" ")&&""===e.replacement)return[t,e];const o=t.original.split(" "),n=o.pop()||"",r=o.join(" ")+(o.length?" ":"");return[...r.length?[{from:t.from,to:t.from+r.length,original:r,replacement:r}]:[],{from:t.from+r.length,to:e.to,original:n+e.original,replacement:n+e.replacement}]}if(O(e)){if(t.replacement.endsWith(" ")&&""===t.original||t.original.endsWith(" ")&&""===t.replacement)return[t,e];const o=e.original.split(" "),n=o.shift()||"",r=(o.length?" ":"")+o.join(" ");return" "===r?[{from:t.from,to:e.to,original:t.original+e.original,replacement:t.replacement+e.replacement}]:[{from:t.from,to:t.to+n.length,original:t.original+n,replacement:t.replacement+n},...r.length?[{from:t.to+n.length,to:e.to,original:r,replacement:r}]:[]]}return[{from:t.from,to:e.to,original:t.original+e.original,replacement:t.replacement+e.replacement}]})(o,e)]}),[]),T=(t,e)=>{const o=(t=>{let e=0;return t.map((([t,o])=>{switch(t){case c.EQUAL:const t=o.lastIndexOf("\n"),n={from:e,to:e+o.length,original:o,replacement:o};if(-1!==t&&"\n"!==o){const n=[{from:e,to:e+t+1,original:o.slice(0,t+1),replacement:o.slice(0,t+1)},{from:e+t+1,to:e+o.length,original:o.slice(t+1),replacement:o.slice(t+1)}];return e+=o.length,n}return e+=o.length,[n];case c.DELETE:const r={from:e,to:e+o.length,original:o,replacement:""};return e+=o.length,[r];case c.INSERT:return[{from:e,to:e,original:"",replacement:o}]}})).flat()})(c(t,e));return w(w(o))},A=(e,o)=>{var n;const r=null===(n=h.getState(e.state))||void 0===n?void 0:n.decorations.find(0,e.state.doc.nodeSize,(t=>t.id===o.spec.id))[0];if(!r)return;const{text:s}=r.spec,{from:a,to:p}=r,c={type:t.acceptSuggestion,id:o.spec.id},l=s.split("\n").map((t=>e.state.schema.node("paragraph",null,e.state.schema.text(t)))),d=g.Fragment.fromArray(l);let{tr:u}=e.state;u=u.setSelection(i.TextSelection.create(e.state.doc,a,p)),u.selection.replace(u,new g.Slice(d,1,1)),u.setMeta(h,c),e.dispatch(u)},b=(e,o)=>{const{spec:n}=o,r={type:t.discardSuggestion,id:n.id},s=e.state.tr.setMeta(h,r);e.dispatch(s)},j=(e,o)=>{const n=h.getState(e.state);if(!n)return!1;const{decorations:r}=n,s=r.find(o,o)[0];if(!s){const o={type:t.closeSuggestion};return e.dispatch(e.state.tr.setMeta(h,o)),!1}const a=n.popupDecoration.find()[0];if((null==a?void 0:a.spec.id)===s.spec.id)return!1;const i={type:t.openSuggestion,decoration:s};return e.dispatch(e.state.tr.setMeta(h,i)),!1},C={debounceMs:2e3,createUpdatePopup:(t,o,n,r,s)=>{const a=document.createElement("div");a.className="grammar-suggest-tooltip";const{spec:i}=o,p=t.dom.getBoundingClientRect();a.id=e.grammarSuggestPopup;const c=t.coordsAtPos(n),l=t.dom.scrollTop||0,d=t.dom.scrollLeft||0;a.style.left=`${c.left-p.left+d}px`,a.style.top=`${c.bottom-p.top+l+5}px`;const u=document.createElement("div");u.className="grammar-suggest-tooltip-apply",u.innerText=i.text||i.originalText,i.text||(u.style.textDecoration="line-through",u.style.color="red"),u.onclick=()=>{r(t,o)},a.appendChild(u);const g=document.createElement("div");return g.innerHTML="<svg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 24 24' width='24' height='24'><path d='M12 22C6.47715 22 2 17.5228 2 12C2 6.47715 6.47715 2 12 2C17.5228 2 22 6.47715 22 12C22 17.5228 17.5228 22 12 22ZM12 20C16.4183 20 20 16.4183 20 12C20 7.58172 16.4183 4 12 4C7.58172 4 4 7.58172 4 12C4 16.4183 7.58172 20 12 20ZM12 10.5858L14.8284 7.75736L16.2426 9.17157L13.4142 12L16.2426 14.8284L14.8284 16.2426L12 13.4142L9.17157 16.2426L7.75736 14.8284L10.5858 12L7.75736 9.17157L9.17157 7.75736L12 10.5858Z'></path></svg>",g.className="grammar-suggest-tooltip-discard",g.onclick=()=>{s(t,o)},a.appendChild(g),a},withYjs:!1},M={maxSelection:1e3},W=(t,e,o,n,r,s,a)=>l.__awaiter(void 0,void 0,void 0,(function*(){var t;let o="";const i=new AbortController;try{const p=null===(t=(yield fetch("https://suggestion-gw5lxik4dq-uc.a.run.app",{method:"POST",cache:"no-cache",headers:{"Content-Type":"application/json",Authorization:"Bearer -qKivjCv6MfQSmgF438PjEY7RnLfqoVe"},signal:i.signal,body:JSON.stringify({model:"gpt-3.5-turbo",modelParams:{input:[e],task:r,params:a}})})).body)||void 0===t?void 0:t.getReader(),c=({done:t,value:e})=>l.__awaiter(void 0,void 0,void 0,(function*(){var a;if(t)return;if(null===(a=m.getState(n.state))||void 0===a?void 0:a.isCancelled)return void(yield i.abort());const l=(new TextDecoder).decode(e);try{o+=l,n.dispatch(n.state.tr.setMeta(m,Object.assign({type:r,status:exports.Status.streaming,result:o},s&&{selection:s})))}catch(t){console.error("Could not parse stream message",l,t)}return null==p?void 0:p.read().then(c)}));yield null==p?void 0:p.read().then(c),n.dispatch(n.state.tr.setMeta(m,Object.assign({type:r,status:exports.Status.finished,result:o},s&&{selection:s})))}catch(t){console.error("Error:",t)}}));exports.completePlugin=(t,e=M)=>{let o=!1;return new i.Plugin({key:m,state:{init:()=>({status:exports.Status.idle}),apply(t,e,n,r){const s=t.getMeta(m);return e.status===exports.Status.done||e.status===exports.Status.rejected?(o=!1,{status:exports.Status.idle}):(null==s?void 0:s.status)===exports.Status.cancelled?Object.assign(Object.assign(Object.assign({},e),s),{isCancelled:!0}):s&&(t=>Object.values(exports.OpenAiPromptsWithParam).includes(t.type)||Object.values(exports.OpenAiPromptsWithoutParam).includes(t.type))(s)?e.type&&s.type!==e.type?e:Object.assign(Object.assign({},e),s):e}},view:()=>({update(t,n){const r=m.getState(t.state);let s=t.state.tr;if((null==r?void 0:r.status)===exports.Status.new&&!o)switch(r.type){case exports.OpenAiPromptsWithoutParam.Complete:o=!0,((t,e,o)=>{l.__awaiter(void 0,void 0,void 0,(function*(){const{doc:t}=e.state,o=[];t.descendants((t=>{"paragraph"===t.type.name&&o.push(t.textContent)}));let n="";n=o.length>=2?o.slice(-2).join(" "):o.join(" "),W(0,n,0,e,exports.OpenAiPromptsWithoutParam.Complete)}))})(0,t);break;case exports.OpenAiPromptsWithoutParam.MakeLonger:case exports.OpenAiPromptsWithoutParam.MakeShorter:case exports.OpenAiPromptsWithoutParam.Improve:case exports.OpenAiPromptsWithoutParam.Simplify:case exports.OpenAiPromptsWithoutParam.Explain:case exports.OpenAiPromptsWithoutParam.ActionItems:case exports.OpenAiPromptsWithParam.Translate:case exports.OpenAiPromptsWithParam.ChangeTone:o=!0,((t,e,o,n,r,s)=>{const a=o.state.selection;if(!a)return void o.dispatch(o.state.tr.setMeta(m,{task:t,status:exports.Status.done}));if(o.state.doc.textBetween(a.from,a.to).length>r)return void o.dispatch(o.state.tr.setMeta(m,{type:t,status:exports.Status.error,error:"Selection is too big"}));const i=o.state.doc.textBetween(a.from,a.to,"\n");W(0,i,0,o,t,a,s)})(r.type,0,t,0,e.maxSelection,r.params)}if((null==r?void 0:r.status)===exports.Status.accepted){if(r.error)return s.setMeta(m,{type:r.type,status:exports.Status.done}),void t.dispatch(s);switch(r.type){case exports.OpenAiPromptsWithoutParam.Complete:s=s.insertText(r.result||"",t.state.doc.nodeSize-2),s.setMeta(m,{type:exports.OpenAiPromptsWithoutParam.Complete,status:exports.Status.done}),t.dispatch(s),t.focus();break;case exports.OpenAiPromptsWithoutParam.MakeLonger:case exports.OpenAiPromptsWithoutParam.MakeShorter:case exports.OpenAiPromptsWithoutParam.Improve:case exports.OpenAiPromptsWithoutParam.Simplify:case exports.OpenAiPromptsWithoutParam.Explain:case exports.OpenAiPromptsWithoutParam.ActionItems:case exports.OpenAiPromptsWithParam.Translate:case exports.OpenAiPromptsWithParam.ChangeTone:if(r.selection&&r.result){const e=r.result.split("\n\n").map((e=>t.state.schema.node("paragraph",null,t.state.schema.text(e)))),o=g.Fragment.fromArray(e);s.selection.replace(s,new g.Slice(o,0,0))}s.setMeta(m,{type:r.type,status:exports.Status.done}),t.dispatch(s),t.focus()}}}})})},exports.completePluginKey=m,exports.defaultCompleteOptions=M,exports.defaultOptions=C,exports.grammarSuggestPlugin=(e,o=C)=>{let n=!1;return new i.Plugin({key:h,state:{init:()=>({lastText:"",decorations:p.DecorationSet.empty,popupDecoration:p.DecorationSet.empty}),apply(e,n,r,s){const a=e.getMeta(h);return(null==a?void 0:a.type)===t.suggestionUpdate?((t,e,o)=>{const{changedRegion:n,fix:r,mapping:s,text:a}=e,i=T(n.newText,r.result).filter((t=>!O(t))).filter((t=>t.original!==`${t.replacement}\n`)).map((({from:t,to:e,original:o,replacement:r})=>{const a=v(n.start+t,s),i=v(n.start+(r.endsWith("\n")?e-1:e),s),c={text:r.endsWith("\n")?r.slice(0,-1):r,originalText:o,id:{}};return p.Decoration.inline(a,i,{class:"grammarSuggestion "+(""===r?"removalSuggestion":"")},c)}));return Object.assign(Object.assign({},t),{decorations:t.decorations.add(o.doc,i),lastText:a})})(n,a,e):(null==a?void 0:a.type)===t.acceptSuggestion?((t,e,o)=>{const n=t.decorations.remove(t.decorations.find(0,o.doc.nodeSize,(t=>t.id===e.id)));return Object.assign(Object.assign({},t),{lastText:x(o.doc),decorations:n.map(o.mapping,o.doc),popupDecoration:p.DecorationSet.empty})})(n,a,e):(null==a?void 0:a.type)===t.openSuggestion?((t,e,o,n)=>{const{decoration:r}=e,s={id:r.spec.id};return Object.assign(Object.assign({},t),{popupDecoration:p.DecorationSet.create(o.doc,[p.Decoration.widget(r.from,((t,e)=>{const o=e();return o?n.createUpdatePopup(t,r,o,A,b):document.createElement("div")}),Object.assign(Object.assign({},s),{stopEvent:()=>!0}))])})})(n,a,e,o):(null==a?void 0:a.type)===t.closeSuggestion?Object.assign(Object.assign({},n),{popupDecoration:p.DecorationSet.empty}):(null==a?void 0:a.type)===t.discardSuggestion?((t,e,o)=>Object.assign(Object.assign({},t),{decorations:t.decorations.remove(t.decorations.find(0,o.doc.nodeSize,(t=>t.id===e.id))),popupDecoration:p.DecorationSet.empty}))(n,a,e):e.docChanged?((t,e,o,n)=>{const r=S(o.doc).text,{text:s,mapping:a}=S(e.doc);if(s===r)return t;let i=e.mapping;if(n){const t=e.doc.content.findDiffStart(o.doc.content),n=o.doc.content.findDiffEnd(e.doc.content),r=n&&t?new u.StepMap([t,n.a-t,n.b-t]):new u.StepMap([0,0,0]);i=new u.Mapping([r])}const p=f(r,s),c=t.decorations.map(i,e.doc),l=v(p.start,a),d=v(p.end,a),g=t.popupDecoration.map(i,e.doc);return Object.assign(Object.assign({},t),{decorations:c.remove(c.find(l,d)),popupDecoration:g.remove(g.find(l,d))})})(n,e,r,o.withYjs):n}},props:{handleClick:j,decorations:t=>{const e=h.getState(t);if(!e)return null;const o=m.getState(t);return(null==o?void 0:o.status)!==exports.Status.idle?e.decorations:e.decorations.add(t.doc,e.popupDecoration.find())}},view(){const r=((e,o)=>d((e=>{var n;const r=S(e.state.doc),s=(null===(n=h.getState(e.state))||void 0===n?void 0:n.lastText)||"",a=f(s,r.text);y(o,a.newText).then((o=>{if(x(e.state.doc)!==r.text)return;const n={type:t.suggestionUpdate,fix:o,changedRegion:a,mapping:r.mapping,text:r.text};e.dispatch(e.state.tr.setMeta(h,n))})).catch((t=>{console.error("Grammar suggest API error",t)}))}),e.debounceMs))(o,e);return{update(t,e){const o=h.getState(t.state);t.state.doc.textBetween(0,t.state.doc.nodeSize-2,"/n")===e.doc.textBetween(0,e.doc.nodeSize-2,"/n")&&n||!o||o.lastText===x(t.state.doc)||(n=!0,r(t))}}}})};

package/dist/makeRequest.js

@@ -0,0 +1,96 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import debounce from "lodash.debounce";
+import { getChangedRegions, getTextWithNewlines, grammarSuggestPluginKey, } from "./utils";
+import { GrammarSuggestMetaType, } from "./types";
+import { docToTextWithMapping } from "./mapping";
+const isJsonString = (str) => {
+    try {
+        JSON.parse(str);
+    }
+    catch (e) {
+        return true;
+    }
+    return false;
+};
+const myApiRequest = (apiKey, text) => __awaiter(void 0, void 0, void 0, function* () {
+    const input = [...text.split("\n")];
+    // return fetch("https://prosemirror-ai-plugin.web.app/api/suggestion", {
+    return fetch("http://127.0.0.1:5001/prosemirror-ai-plugin/us-central1/suggestion", {
+        method: "POST",
+        cache: "no-cache",
+        headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${apiKey}`,
+        },
+        body: JSON.stringify({
+            model: "gpt-3.5-turbo",
+            modelParams: {
+                input,
+            },
+        }),
+    })
+        .then((response) => {
+        if (response.ok) {
+            return response.json();
+        }
+        return Promise.reject(response);
+    })
+        .then((jsonData) => {
+        if (!jsonData || jsonData.some((i) => isJsonString(i))) {
+            return {
+                fixed: false,
+                result: text,
+            };
+        }
+        return {
+            result: jsonData
+                .map((data) => JSON.parse(data).result)
+                .join("\n"),
+            fixed: true,
+        };
+    })
+        .catch((e) => {
+        e.text().then((text) => {
+            console.error({ status: e.status, text });
+        });
+        return {
+            fixed: false,
+            result: text,
+        };
+    });
+});
+export const createMakeRequest = (options, apiKey) => debounce((view) => {
+    var _a;
+    // The document changed, start API request
+    const versionAtRequestStart = docToTextWithMapping(view.state.doc);
+    const oldText = ((_a = grammarSuggestPluginKey.getState(view.state)) === null || _a === void 0 ? void 0 : _a.lastText) || "";
+    const changedRegion = getChangedRegions(oldText, versionAtRequestStart.text);
+    myApiRequest(apiKey, changedRegion.newText)
+        .then((fix) => {
+        // Check if the document version has changed while we were waiting
+        if (getTextWithNewlines(view.state.doc) !== versionAtRequestStart.text) {
+            // The state changed, abort
+            return;
+        }
+        // State did not change, update the plugin state
+        const meta = {
+            type: GrammarSuggestMetaType.suggestionUpdate,
+            fix,
+            changedRegion,
+            mapping: versionAtRequestStart.mapping,
+            text: versionAtRequestStart.text,
+        };
+        view.dispatch(view.state.tr.setMeta(grammarSuggestPluginKey, meta));
+    })
+        .catch((error) => {
+        console.error("Grammar suggest API error", error);
+    });
+}, options.debounceMs);

package/dist/utils.js

@@ -0,0 +1,82 @@
+import { PluginKey } from "prosemirror-state";
+import diff from "fast-diff";
+import { OpenAiPromptsWithoutParam, OpenAiPromptsWithParam, } from "./types";
+export const completePluginKey = new PluginKey("completePlugin");
+export const isCompleteMeta = (meta) => {
+    return (Object.values(OpenAiPromptsWithParam).includes(meta.type) ||
+        Object.values(OpenAiPromptsWithoutParam).includes(meta.type));
+};
+export const grammarSuggestPluginKey = new PluginKey("grammarSuggestPlugin");
+export const getTextWithNewlines = (node) => {
+    let text = "";
+    node.descendants((n, pos) => {
+        if (n.isText) {
+            text += `${n.text}\n`;
+        }
+    });
+    console.log({ text });
+    return text;
+};
+const getLastChangePos = (oldText, changes) => {
+    const lastChange = changes[changes.length - 1];
+    if (lastChange[0] !== diff.EQUAL)
+        return oldText.length;
+    const changesWithoutLast = changes.slice(0, changes.length - 1);
+    let lastChangePos = 0;
+    for (const change of changesWithoutLast) {
+        const [type, text] = change;
+        if (type === diff.EQUAL) {
+            lastChangePos += text.length;
+        }
+        if (type === diff.DELETE) {
+            lastChangePos += text.length;
+        }
+    }
+    return lastChangePos;
+};
+// TODO: when enters are pressed the new paragraphs are not part of the region
+// - Make a diff between the last stored text and the current one.
+// - We get the end of the beginning identity and the start of the end identity.
+// - Get the first & last newlines before the start and after the end.
+export const getChangedRegions = (oldText, newText) => {
+    if (oldText === newText)
+        return {
+            start: oldText.length,
+            end: oldText.length,
+            oldStart: oldText.length,
+            oldEnd: oldText.length,
+            oldText: "",
+            newText: "",
+        };
+    const changes = diff(oldText, newText);
+    const fistChange = changes[0];
+    const firstChangePos = fistChange[0] === diff.EQUAL ? fistChange[1].length - 1 : 0;
+    const lastChangePos = getLastChangePos(oldText, changes);
+    const firstNewlinePosFromFirstChange = oldText.lastIndexOf("\n", firstChangePos);
+    const lastNewlinePosFromLastChange = oldText.indexOf("\n", lastChangePos);
+    const oldStart = firstNewlinePosFromFirstChange === -1
+        ? 0
+        : firstNewlinePosFromFirstChange + 1;
+    const oldEnd = lastNewlinePosFromLastChange === -1
+        ? oldText.length
+        : lastNewlinePosFromLastChange;
+    // console.log({
+    //   oldText,
+    //   lastChangePos,
+    //   lastNewlinePosFromLastChange,
+    //   oldEnd,
+    //   firstChangePos,
+    //   firstNewlinePosFromFirstChange,
+    //   oldStart,
+    // });
+    const oldChange = oldText.slice(oldStart, oldEnd);
+    const newChange = newText.slice(oldStart, newText.length - (oldText.length - oldEnd));
+    return {
+        start: oldStart,
+        end: oldStart + newChange.length,
+        oldStart,
+        oldEnd,
+        oldText: oldChange,
+        newText: newChange,
+    };
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prosemirror-suggestcat-plugin",
-  "version": "0.1.8",
+  "version": "0.1.11",
   "description": "",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",