muggle-string 0.1.0 → 0.2.1

package/README.md

@@ -18,53 +18,55 @@ The example is base-on [magic-string readme](https://github.com/rich-harris/magi
 
 ```ts
 import {
-  toString,
-  overwrite, // not yet support
+	toString,
+	replace,
 } from 'muggle-string';
 
 /** @type {import('@volar/vue-language-core').VueLanguagePlugin} */
 const plugin = () => {
 	return {
+		name: 'example-vue-language-plugin',
+		version: 1,
 		resolveEmbeddedFile(fileName, sfc, embeddedFile) {
 			if (embeddedFile.fileName.replace(fileName, '').match(/^\.(js|ts|jsx|tsx)$/)) {
 
-        const s = embeddedFile.content;
-        toString(s); // 'problems = 99'
+				const s = embeddedFile.content;
+				toString(s); // 'problems = 99'
 
-        overwrite(s, 0, 8, 'answer'); // not yet support
-        toString(s); // 'answer = 99'
+				replace(s, 'problems', 'answer');
+				toString(s); // 'answer = 99'
 
-        overwrite(s, 11, 13, '42'); // not yet support
-        toString(s); // 'answer = 42'
+				replace(s, '99', '42');
+				toString(s); // 'answer = 42'
 
-        // add string by Array method directly
-        s.unshift('var ');
-        s.push(';');
-        toString(s); // 'var answer = 42;'
+				// add string by Array method directly
+				s.unshift('var ');
+				s.push(';');
+				toString(s); // 'var answer = 42;'
 
-        for (const block of sfc.customBlocks) {
-          if (block.type === 'more-script') {
-            s.push([
-              block.content, // text to add
-              block.name, // source
-              0, // content offset in source
-              {
-                 // langauge capabilities to enable in this segment
-                hover: true,
-                references: true,
-                definition: true,
-                diagnostic: true,
-                rename: true,
-                completion: true,
-                semanticTokens: true,
-              },
-            ]);
-            toString(s); // 'var answer = 42;console.log( answer )'
-          }
-        }
-      }
-    }
-  };
+				for (const block of sfc.customBlocks) {
+					if (block.type === 'more-script') {
+						s.push([
+							block.content, // text to add
+							block.name, // source
+							0, // content offset in source
+							{
+								// language capabilities to enable in this segment
+								hover: true,
+								references: true,
+								definition: true,
+								diagnostic: true,
+								rename: true,
+								completion: true,
+								semanticTokens: true,
+							},
+						]);
+						toString(s); // 'var answer = 42;console.log( answer )'
+					}
+				}
+			}
+		}
+	};
 };
 module.exports = plugin;
 ```

package/out/binarySearch.d.ts

@@ -0,0 +1 @@
+export declare function binarySearch(offsets: number[], start: number): number;

package/out/binarySearch.js

@@ -0,0 +1,25 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.binarySearch = void 0;
+function binarySearch(offsets, start) {
+    let low = 0;
+    let high = offsets.length - 1;
+    while (low <= high) {
+        const mid = Math.floor((low + high) / 2);
+        const midValue = offsets[mid];
+        if (midValue < start) {
+            low = mid + 1;
+        }
+        else if (midValue > start) {
+            high = mid - 1;
+        }
+        else {
+            low = mid;
+            high = mid;
+            break;
+        }
+    }
+    return Math.max(Math.min(low, high, offsets.length - 1), 0);
+}
+exports.binarySearch = binarySearch;
+//# sourceMappingURL=binarySearch.js.map

package/out/index.d.ts

@@ -1,18 +1,28 @@
-export declare type Segment<T = '--'> = T extends '--' ? SegmentWithoutData : SegmentWithData<T>;
-export declare type SegmentWithoutData = [
+declare const NO_DATA_SYMBOL: unique symbol;
+export type Segment<T = typeof NO_DATA_SYMBOL> = T extends typeof NO_DATA_SYMBOL ? SegmentWithoutData : SegmentWithData<T>;
+export type SegmentWithoutData = WithString<_SegmentWithoutData>;
+export type SegmentWithData<T = any> = WithString<_SegmentWithData<T>>;
+type WithString<T> = T | string;
+type _SegmentWithoutData = [
     string,
     // text
     string | undefined,
     // source
     number | [number, number]
-] | string;
-export declare type SegmentWithData<T> = [
+];
+type _SegmentWithData<T = any> = [
     string,
     // text
     string | undefined,
     // source
     number | [number, number],
     T
-] | string;
+];
 export declare function getLength(segments: Segment<any>[]): number;
-export declare function toString(segments: Segment<any>[]): string;
+export declare function toString<T extends Segment<any>>(segments: T[]): string;
+export declare function create(source: string): Segment[];
+export declare function replace<T extends Segment<any>>(segments: T[], pattern: string | RegExp, ...replacers: (string | ((match: string) => T))[]): void;
+export declare function replaceAll<T extends Segment<any>>(segments: T[], pattern: RegExp, ...replacers: (string | ((match: string) => T))[]): void;
+export declare function replaceSourceRange<T extends Segment<any>>(segments: T[], source: string | undefined, startOffset: number, endOffset: number, ...newSegments: T[]): void;
+export declare function replaceRange<T extends Segment<any>>(segments: T[], startOffset: number, endOffset: number, ...newSegments: T[]): void;
+export {};

package/out/index.js

@@ -1,6 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.toString = exports.getLength = void 0;
+exports.replaceRange = exports.replaceSourceRange = exports.replaceAll = exports.replace = exports.create = exports.toString = exports.getLength = void 0;
+const binarySearch_1 = require("./binarySearch");
+const NO_DATA_SYMBOL = Symbol();
 function getLength(segments) {
     let length = 0;
     for (const segment of segments) {
@@ -13,3 +15,136 @@ function toString(segments) {
     return segments.map(s => typeof s === 'string' ? s : s[0]).join('');
 }
 exports.toString = toString;
+function create(source) {
+    return [[source, undefined, 0]];
+}
+exports.create = create;
+function replace(segments, pattern, ...replacers) {
+    const str = toString(segments);
+    const match = str.match(pattern);
+    if (match && match.index !== undefined) {
+        const startOffset = match.index;
+        const endOffset = startOffset + match[0].length;
+        replaceRange(segments, startOffset, endOffset, ...replacers.map(replacer => typeof replacer === 'string' ? replacer : replacer(match[0])));
+    }
+}
+exports.replace = replace;
+function replaceAll(segments, pattern, ...replacers) {
+    const str = toString(segments);
+    const allMatch = str.matchAll(pattern);
+    let length = str.length;
+    let lengthDiff = 0;
+    for (const match of allMatch) {
+        if (match.index !== undefined) {
+            const startOffset = match.index + lengthDiff;
+            const endOffset = startOffset + match[0].length;
+            replaceRange(segments, startOffset, endOffset, ...replacers.map(replacer => typeof replacer === 'string' ? replacer : replacer(match[0])));
+            const newLength = getLength(segments);
+            lengthDiff += newLength - length;
+            length = newLength;
+        }
+    }
+}
+exports.replaceAll = replaceAll;
+function replaceSourceRange(segments, source, startOffset, endOffset, ...newSegments) {
+    for (const segment of segments) {
+        if (typeof segment === 'string') {
+            continue;
+        }
+        if (segment[1] === source) {
+            const segmentStart = typeof segment[2] === 'number' ? segment[2] : segment[2][0];
+            const segmentEnd = typeof segment[2] === 'number' ? segment[2] + segment[0].length : segment[2][1];
+            if (segmentStart <= startOffset && segmentEnd >= endOffset) {
+                const inserts = [];
+                if (startOffset > segmentStart) {
+                    inserts.push(trimSegmentEnd(segment, startOffset - segmentStart));
+                }
+                for (const newSegment of newSegments) {
+                    inserts.push(newSegment);
+                }
+                if (endOffset < segmentEnd) {
+                    inserts.push(trimSegmentStart(segment, endOffset - segmentEnd));
+                }
+                combineStrings(inserts);
+                segments.splice(segments.indexOf(segment), 1, ...inserts);
+                return;
+            }
+        }
+    }
+    throw new Error(`Source range not found: ${source} ${startOffset} ${endOffset}`);
+}
+exports.replaceSourceRange = replaceSourceRange;
+function replaceRange(segments, startOffset, endOffset, ...newSegments) {
+    const offsets = toOffsets(segments);
+    const startIndex = (0, binarySearch_1.binarySearch)(offsets, startOffset);
+    const endIndex = (0, binarySearch_1.binarySearch)(offsets, endOffset);
+    const startSegment = segments[startIndex];
+    const endSegment = segments[endIndex];
+    const startSegmentStart = offsets[startIndex];
+    const endSegmentStart = offsets[endIndex];
+    const endSegmentEnd = offsets[endIndex] + (typeof endSegment === 'string' ? endSegment.length : endSegment[0].length);
+    const inserts = [];
+    if (startOffset > startSegmentStart) {
+        inserts.push(trimSegmentEnd(startSegment, startOffset - startSegmentStart));
+    }
+    for (const newSegment of newSegments) {
+        inserts.push(newSegment);
+    }
+    if (endOffset < endSegmentEnd) {
+        inserts.push(trimSegmentStart(endSegment, endOffset - endSegmentStart));
+    }
+    combineStrings(inserts);
+    segments.splice(startIndex, endIndex - startIndex + 1, ...inserts);
+}
+exports.replaceRange = replaceRange;
+function combineStrings(segments) {
+    for (let i = segments.length - 1; i >= 1; i--) {
+        if (typeof segments[i] === 'string' && typeof segments[i - 1] === 'string') {
+            segments[i - 1] = segments[i - 1] + segments[i];
+            segments.splice(i, 1);
+        }
+    }
+}
+function trimSegmentEnd(segment, trimEnd) {
+    if (typeof segment === 'string') {
+        return segment.slice(0, trimEnd);
+    }
+    const originalString = segment[0];
+    const originalRange = segment[2];
+    const newString = originalString.slice(0, trimEnd);
+    const newRange = typeof originalRange === 'number' ? originalRange : [originalRange[0], originalRange[1] - (originalString.length - newString.length)];
+    return [
+        newString,
+        segment[1],
+        newRange,
+        ...segment.slice(3),
+    ];
+}
+function trimSegmentStart(segment, trimStart) {
+    if (typeof segment === 'string') {
+        return segment.slice(trimStart);
+    }
+    const originalString = segment[0];
+    const originalRange = segment[2];
+    const newString = originalString.slice(trimStart);
+    if (trimStart < 0) {
+        trimStart += originalString.length;
+    }
+    const newRange = typeof originalRange === 'number' ? originalRange + trimStart : [originalRange[0] + trimStart, originalRange[1]];
+    return [
+        newString,
+        segment[1],
+        newRange,
+        ...segment.slice(3),
+    ];
+}
+function toOffsets(segments) {
+    const offsets = [];
+    let offset = 0;
+    for (const segment of segments) {
+        offsets.push(offset);
+        offset += typeof segment == 'string' ? segment.length : segment[0].length;
+    }
+    return offsets;
+}
+//# sourceMappingURL=index.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "muggle-string",
-	"version": "0.1.0",
+	"version": "0.2.1",
 	"main": "out/index.js",
 	"license": "MIT",
 	"files": [
@@ -11,7 +11,12 @@
 		"type": "git",
 		"url": "https://github.com/johnsoncodehk/muggle-string.git"
 	},
+	"scripts": {
+		"test": "vitest run"
+	},
 	"devDependencies": {
-		"typescript": "latest"
+		"typescript": "latest",
+		"vitest": "latest",
+		"vite": "latest"
 	}
 }