feat: paste inside current page root when nothing is selected

meetalva · May 24, 2018 · 5248cca · 5248cca
1 parent e704609
commit 5248cca
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/src/store/view-store.ts b/src/store/view-store.ts
@@ -621,12 +621,22 @@ export class ViewStore {
 
 	public pasteAfterSelectedElement(): Model.Element | undefined {
 		const selectedElement = this.getSelectedElement();
+		const page = this.getCurrentPage();
+		const rootElement = page ? page.getRoot() : undefined;
 
-		if (!selectedElement) {
+		if (!selectedElement && !rootElement) {
 			return;
 		}
 
-		return this.pasteAfterElement(selectedElement);
+		if (selectedElement) {
+			return this.pasteAfterElement(selectedElement);
+		}
+
+		if (rootElement) {
+			return this.pasteInsideElement(rootElement);
+		}
+
+		return;
 	}
 
 	public pasteInsideElement(element: Model.Element): Model.Element | undefined {
@@ -794,6 +804,7 @@ export class ViewStore {
 	}
 
 	public setActivePageByIndex(index: number): void {
+		this.selectedElement = undefined;
 		this.activePage = index;
 	}