Brunobkr/llama.cpp_AlgMor24_github

1/** @typedef {import('./types').Location} Location */2 3/**4 * @param {import('./types').Range} range5 * @param {number} index6 */7function rangeContains(range, index) {8	return range.start <= index && index < range.end;9}10 11/**12 * @param {string} source13 * @param {import('./types').Options} [options]14 */15export function getLocator(source, options = {}) {16	const { offsetLine = 0, offsetColumn = 0 } = options;17 18	let start = 0;19	const ranges = source.split('\n').map((line, i) => {20		const end = start + line.length + 1;21 22		/** @type {import('./types').Range} */23		const range = { start, end, line: i };24 25		start = end;26		return range;27	});28 29	let i = 0;30 31	/**32	 * @param {string | number} search33	 * @param {number} [index]34	 * @returns {Location | undefined}35	 */36	function locator(search, index) {37		if (typeof search === 'string') {38			search = source.indexOf(search, index ?? 0);39		}40 41		if (search === -1) return undefined;42 43		let range = ranges[i];44 45		const d = search >= range.end ? 1 : -1;46 47		while (range) {48			if (rangeContains(range, search)) {49				return {50					line: offsetLine + range.line,51					column: offsetColumn + search - range.start,52					character: search53				};54			}55 56			i += d;57			range = ranges[i];58		}59	}60 61	return locator;62}63 64/**65 * @param {string} source66 * @param {string | number} search67 * @param {import('./types').Options} [options]68 * @returns {Location | undefined}69 */70export function locate(source, search, options) {71	return getLocator(source, options)(search, options && options.startIndex);72}73