@univerjs/docs 1.0.0-alpha.1 → 1.0.0-alpha.2

package/lib/es/index.js

@@ -1,4 +1,4 @@
-import { BuildTextUtils, CommandType, DOCS_FORMULA_BAR_EDITOR_UNIT_ID_KEY, DOCS_NORMAL_EDITOR_UNIT_ID_KEY, DeleteDirection, Disposable, DisposableCollection, ICommandService, IConfigService, IUndoRedoService, IUniverInstanceService, Inject, Injector, JSONX, LocaleService, Optional, Plugin, RedoCommandId, RxDisposable, TextX, TextXActionType, UndoCommandId, UniverInstanceType, composeInterceptors, createIdentifier, createInterceptorKey, getRichTextEditPath, isInternalEditorID, merge, remove, toDisposable } from "@univerjs/core";
+import { BuildTextUtils, CommandType, DOCS_FORMULA_BAR_EDITOR_UNIT_ID_KEY, DOCS_NORMAL_EDITOR_UNIT_ID_KEY, DeleteDirection, Disposable, DisposableCollection, DocumentFlavor, ICommandService, IConfigService, IUndoRedoService, IUniverInstanceService, Inject, Injector, JSONX, LocaleService, Optional, Plugin, RedoCommandId, RxDisposable, TextX, TextXActionType, UndoCommandId, UniverInstanceType, composeInterceptors, createIdentifier, createInterceptorKey, createParagraphId, generateRandomId, getRichTextEditPath, isInternalEditorID, merge, remove, toDisposable } from "@univerjs/core";
 import { DocumentSkeleton, DocumentViewModel, IRenderManagerService, NORMAL_TEXT_SELECTION_PLUGIN_STYLE } from "@univerjs/engine-render";
 import { BehaviorSubject, Subject, takeUntil } from "rxjs";
 
@@ -386,14 +386,14 @@ const InsertTextCommand = {
 	id: "doc.command.insert-text",
 	type: CommandType.COMMAND,
 	handler: (accessor, params) => {
-		var _activeRange$segmentI;
+		var _docDataModel$getSelf, _activeRange$segmentI;
 		const commandService = accessor.get(ICommandService);
 		const { range, segmentId, body, unitId, cursorOffset } = params;
 		const docSelectionManagerService = accessor.get(DocSelectionManagerService);
 		const docDataModel = accessor.get(IUniverInstanceService).getUnit(unitId, UniverInstanceType.UNIVER_DOC);
 		if (docDataModel == null) return false;
 		const activeRange = docSelectionManagerService.getActiveTextRange();
-		const originBody = docDataModel.getSelfOrHeaderFooterModel((_activeRange$segmentI = activeRange === null || activeRange === void 0 ? void 0 : activeRange.segmentId) !== null && _activeRange$segmentI !== void 0 ? _activeRange$segmentI : "").getBody();
+		const originBody = (_docDataModel$getSelf = docDataModel.getSelfOrHeaderFooterModel((_activeRange$segmentI = activeRange === null || activeRange === void 0 ? void 0 : activeRange.segmentId) !== null && _activeRange$segmentI !== void 0 ? _activeRange$segmentI : "")) === null || _docDataModel$getSelf === void 0 ? void 0 : _docDataModel$getSelf.getBody();
 		if (originBody == null) return false;
 		const { startOffset, collapsed } = range;
 		const cursorMove = cursorOffset !== null && cursorOffset !== void 0 ? cursorOffset : body.dataStream.length;
@@ -446,12 +446,12 @@ const DeleteTextCommand = {
 	id: "doc.command.delete-text",
 	type: CommandType.COMMAND,
 	handler: (accessor, params) => {
-		var _body$customRanges;
+		var _docDataModel$getSelf2, _body$customRanges;
 		const commandService = accessor.get(ICommandService);
 		const univerInstanceService = accessor.get(IUniverInstanceService);
 		const { range, segmentId, unitId, direction, len = 1 } = params;
 		const docDataModel = univerInstanceService.getUnit(unitId, UniverInstanceType.UNIVER_DOC);
-		const body = docDataModel === null || docDataModel === void 0 ? void 0 : docDataModel.getSelfOrHeaderFooterModel(segmentId).getBody();
+		const body = docDataModel === null || docDataModel === void 0 || (_docDataModel$getSelf2 = docDataModel.getSelfOrHeaderFooterModel(segmentId)) === null || _docDataModel$getSelf2 === void 0 ? void 0 : _docDataModel$getSelf2.getBody();
 		if (docDataModel == null || body == null) return false;
 		const { startOffset } = range;
 		let start = direction === DeleteDirection.LEFT ? startOffset - len : startOffset;
@@ -499,7 +499,7 @@ const UpdateTextCommand = {
 	handler: (accessor, params) => {
 		const { range, segmentId, updateBody, coverType, unitId, textRanges } = params;
 		const commandService = accessor.get(ICommandService);
-		const docDataModel = accessor.get(IUniverInstanceService).getCurrentUniverDocInstance();
+		const docDataModel = accessor.get(IUniverInstanceService).getCurrentUnitOfType(UniverInstanceType.UNIVER_DOC);
 		if (docDataModel == null) return false;
 		const doMutation = {
 			id: RichTextEditingMutation.id,
@@ -529,10 +529,141 @@ const UpdateTextCommand = {
 	}
 };
 
+//#endregion
+//#region src/commands/commands/create-header-footer.command.ts
+let HeaderFooterType = /* @__PURE__ */ function(HeaderFooterType) {
+	HeaderFooterType[HeaderFooterType["FIRST_PAGE_HEADER"] = 0] = "FIRST_PAGE_HEADER";
+	HeaderFooterType[HeaderFooterType["FIRST_PAGE_FOOTER"] = 1] = "FIRST_PAGE_FOOTER";
+	HeaderFooterType[HeaderFooterType["DEFAULT_HEADER"] = 2] = "DEFAULT_HEADER";
+	HeaderFooterType[HeaderFooterType["DEFAULT_FOOTER"] = 3] = "DEFAULT_FOOTER";
+	HeaderFooterType[HeaderFooterType["EVEN_PAGE_HEADER"] = 4] = "EVEN_PAGE_HEADER";
+	HeaderFooterType[HeaderFooterType["EVEN_PAGE_FOOTER"] = 5] = "EVEN_PAGE_FOOTER";
+	return HeaderFooterType;
+}({});
+function getEmptyHeaderFooterBody() {
+	return {
+		dataStream: "\r\n",
+		textRuns: [{
+			st: 0,
+			ed: 0,
+			ts: { fs: 9 }
+		}],
+		customBlocks: [],
+		paragraphs: [{
+			startIndex: 0,
+			paragraphId: createParagraphId(/* @__PURE__ */ new Set()),
+			paragraphStyle: {
+				spaceAbove: { v: 0 },
+				lineSpacing: 1.5,
+				spaceBelow: { v: 0 }
+			}
+		}],
+		sectionBreaks: [{ startIndex: 1 }]
+	};
+}
+function createHeaderFooterAction(segmentId, createType, documentStyle, actions, createMode = "single") {
+	const jsonX = JSONX.getInstance();
+	const ID_LEN = 6;
+	const firstSegmentId = segmentId !== null && segmentId !== void 0 ? segmentId : generateRandomId(ID_LEN);
+	const isHeader = createType === 2 || createType === 0 || createType === 4;
+	const insertAction = jsonX.insertOp([isHeader ? "headers" : "footers", firstSegmentId], {
+		[isHeader ? "headerId" : "footerId"]: firstSegmentId,
+		body: getEmptyHeaderFooterBody()
+	});
+	actions.push(insertAction);
+	let key = "defaultHeaderId";
+	let pairKey = "defaultFooterId";
+	switch (createType) {
+		case 2:
+			key = "defaultHeaderId";
+			pairKey = "defaultFooterId";
+			break;
+		case 3:
+			key = "defaultFooterId";
+			pairKey = "defaultHeaderId";
+			break;
+		case 0:
+			key = "firstPageHeaderId";
+			pairKey = "firstPageFooterId";
+			break;
+		case 1:
+			key = "firstPageFooterId";
+			pairKey = "firstPageHeaderId";
+			break;
+		case 4:
+			key = "evenPageHeaderId";
+			pairKey = "evenPageFooterId";
+			break;
+		case 5:
+			key = "evenPageFooterId";
+			pairKey = "evenPageHeaderId";
+			break;
+		default: throw new Error(`Unknown header footer type: ${createType}`);
+	}
+	const linkedSegmentIds = [[key, firstSegmentId]];
+	if (createMode === "pair" && pairKey != null) {
+		const secondSegmentId = generateRandomId(ID_LEN);
+		const insertPairAction = jsonX.insertOp([isHeader ? "footers" : "headers", secondSegmentId], {
+			[isHeader ? "footerId" : "headerId"]: secondSegmentId,
+			body: getEmptyHeaderFooterBody()
+		});
+		actions.push(insertPairAction);
+		linkedSegmentIds.push([pairKey, secondSegmentId]);
+	}
+	for (const [k, id] of linkedSegmentIds) if (documentStyle[k] != null) {
+		const replaceAction = jsonX.replaceOp(["documentStyle", k], documentStyle[k], id);
+		actions.push(replaceAction);
+	} else {
+		const insertAction = jsonX.insertOp(["documentStyle", k], id);
+		actions.push(insertAction);
+	}
+	return actions;
+}
+const CreateHeaderFooterCommand = {
+	id: "doc.command.create-header-footer",
+	type: CommandType.COMMAND,
+	handler: (accessor, params) => {
+		const commandService = accessor.get(ICommandService);
+		const univerInstanceService = accessor.get(IUniverInstanceService);
+		const { unitId, segmentId, createType, headerFooterProps, createMode = "single" } = params;
+		const docDataModel = univerInstanceService.getUnit(unitId, UniverInstanceType.UNIVER_DOC);
+		if (docDataModel == null) return false;
+		const { documentStyle } = docDataModel.getSnapshot();
+		if (documentStyle.documentFlavor === DocumentFlavor.MODERN) return false;
+		const rawActions = [];
+		const jsonX = JSONX.getInstance();
+		if (createType != null) createHeaderFooterAction(segmentId, createType, documentStyle, rawActions, createMode);
+		if (headerFooterProps != null) Object.keys(headerFooterProps).forEach((key) => {
+			const value = headerFooterProps[key];
+			const oldValue = documentStyle[key];
+			if (value === oldValue) return;
+			const action = oldValue === void 0 ? jsonX.insertOp(["documentStyle", key], value) : jsonX.replaceOp(["documentStyle", key], oldValue, value);
+			rawActions.push(action);
+		});
+		if (rawActions.length === 0) return false;
+		const doMutation = {
+			id: RichTextEditingMutation.id,
+			params: {
+				unitId,
+				actions: rawActions.reduce((acc, cur) => JSONX.compose(acc, cur), null),
+				textRanges: [{
+					startOffset: 0,
+					endOffset: 0,
+					collapsed: true
+				}],
+				debounce: true
+			}
+		};
+		if ((headerFooterProps === null || headerFooterProps === void 0 ? void 0 : headerFooterProps.marginFooter) != null || (headerFooterProps === null || headerFooterProps === void 0 ? void 0 : headerFooterProps.marginHeader) != null) doMutation.params.noNeedSetTextRange = true;
+		const result = commandService.syncExecuteCommand(doMutation.id, doMutation.params);
+		return Boolean(result);
+	}
+};
+
 //#endregion
 //#region package.json
 var name = "@univerjs/docs";
-var version = "1.0.0-alpha.1";
+var version = "1.0.0-alpha.2";
 
 //#endregion
 //#region src/commands/mutations/docs-rename.mutation.ts
@@ -724,7 +855,7 @@ let DocStateChangeManagerService = class DocStateChangeManagerService extends Rx
 	_initialize() {
 		this.disposeWithMe(this._commandService.beforeCommandExecuted((command) => {
 			if (command.id === UndoCommandId || command.id === RedoCommandId) {
-				const univerDoc = this._univerInstanceService.getCurrentUniverDocInstance();
+				const univerDoc = this._univerInstanceService.getCurrentUnitOfType(UniverInstanceType.UNIVER_DOC);
 				if (univerDoc == null) return;
 				const unitId = univerDoc.getUnitId();
 				this._pushHistory(unitId);
@@ -854,6 +985,7 @@ let UniverDocsPlugin = class UniverDocsPlugin extends Plugin {
 			InsertTextCommand,
 			DeleteTextCommand,
 			UpdateTextCommand,
+			CreateHeaderFooterCommand,
 			RichTextEditingMutation,
 			DocsRenameMutation,
 			SetTextSelectionsOperation
@@ -983,7 +1115,7 @@ function addCustomRangeFactory(accessor, param, body) {
 	return doMutation;
 }
 function addCustomRangeBySelectionFactory(accessor, param) {
-	var _selections$;
+	var _selections$, _documentDataModel$ge;
 	const { rangeId, rangeType, wholeEntity, properties, unitId, selections: propSelection } = param;
 	const docSelectionManagerService = accessor.get(DocSelectionManagerService);
 	const univerInstanceService = accessor.get(IUniverInstanceService);
@@ -995,7 +1127,7 @@ function addCustomRangeBySelectionFactory(accessor, param) {
 	if (!(selections === null || selections === void 0 ? void 0 : selections.length)) return false;
 	const documentDataModel = univerInstanceService.getUnit(unitId, UniverInstanceType.UNIVER_DOC);
 	if (!documentDataModel) return false;
-	const body = documentDataModel.getSelfOrHeaderFooterModel(segmentId).getBody();
+	const body = (_documentDataModel$ge = documentDataModel.getSelfOrHeaderFooterModel(segmentId)) === null || _documentDataModel$ge === void 0 ? void 0 : _documentDataModel$ge.getBody();
 	if (!body) return false;
 	const textX = BuildTextUtils.customRange.add({
 		ranges: selections,
@@ -1092,4 +1224,17 @@ function replaceSelectionFactory(accessor, params) {
 }
 
 //#endregion
-export { DOC_INTERCEPTOR_POINT, DeleteTextCommand, DocBlockMoveValidatorService, DocContentInsertService, DocInterceptorService, DocSelectionManagerService, DocSkeletonManagerService, DocStateChangeManagerService, DocStateEmitService, IDocStateChangeInterceptorService, InsertTextCommand, RichTextEditingMutation, SetTextSelectionsOperation, UniverDocsPlugin, UpdateTextCommand, addCustomRangeBySelectionFactory, addCustomRangeFactory, deleteCustomRangeFactory, replaceSelectionFactory };
+//#region src/utils/util.ts
+function consumeContentInsertRange(accessor, unitId) {
+	try {
+		return accessor.get(DocContentInsertService).consumeInsertRange(unitId);
+	} catch {
+		return null;
+	}
+}
+function isHeaderFooterSelection(range) {
+	return Boolean(range === null || range === void 0 ? void 0 : range.segmentId);
+}
+
+//#endregion
+export { CreateHeaderFooterCommand, DOC_INTERCEPTOR_POINT, DeleteTextCommand, DocBlockMoveValidatorService, DocContentInsertService, DocInterceptorService, DocSelectionManagerService, DocSkeletonManagerService, DocStateChangeManagerService, DocStateEmitService, HeaderFooterType, IDocStateChangeInterceptorService, InsertTextCommand, RichTextEditingMutation, SetTextSelectionsOperation, UniverDocsPlugin, UpdateTextCommand, addCustomRangeBySelectionFactory, addCustomRangeFactory, consumeContentInsertRange, deleteCustomRangeFactory, isHeaderFooterSelection, replaceSelectionFactory };