feat: add window.screenshot for svg-convertible chrome screenshots (#…

…617)
meetalva · Sep 19, 2018 · fd259fe · fd259fe
1 parent f8e4fa0
commit fd259fe
Show file tree

Hide file tree

Showing 3 changed files with 61 additions and 2 deletions.
diff --git a/src/electron/create-app-message-handler.ts b/src/electron/create-app-message-handler.ts
@@ -1,4 +1,7 @@
 import * as Electron from 'electron';
+import * as Fs from 'fs';
+import * as Os from 'os';
+import * as Path from 'path';
 import * as Message from '../message';
 import * as Model from '../model';
 import * as uuid from 'uuid';
@@ -144,6 +147,40 @@ export async function createAppMessageHandler(
 					default:
 						return;
 				}
+				break;
+			}
+			case Message.MessageType.ChromeScreenShot: {
+				if (!ctx.win) {
+					return;
+				}
+
+				ctx.win.webContents.printToPDF(
+					{
+						marginsType: 1,
+						printBackground: true,
+						pageSize: {
+							// px => micron
+							width: message.payload.width * 265,
+							height: message.payload.height * 265
+							// tslint:disable-next-line:no-any
+						} as any
+					},
+					(err, data) => {
+						if (err) {
+							return console.error(err.message);
+						}
+
+						const targetPath = Path.join(Os.tmpdir(), `${uuid.v4()}.pdf`);
+
+						Fs.writeFile(targetPath, data, error => {
+							if (error) {
+								return console.error(error);
+							}
+
+							Electron.shell.openExternal(`file://${targetPath}`);
+						});
+					}
+				);
 			}
 		}
 	};

diff --git a/src/message/message.ts b/src/message/message.ts
@@ -75,7 +75,8 @@ export enum MessageType {
 	ChangeUserStore = 'user-store-change',
 	SelectElement = 'select-element',
 	HighlightElement = 'highlight-element',
-	WindowClose = 'window-close'
+	WindowClose = 'window-close',
+	ChromeScreenShot = 'chrome-screenshot',
 }
 
 export type Message =
@@ -151,7 +152,8 @@ export type Message =
 	| ChangeUserStore
 	| SelectElement
 	| HighlightElement
-	| WindowClose;
+	| WindowClose
+	| ChromeScreenShot;
 
 export type ActivatePage = Envelope<MessageType.ActivatePage, { id: string }>;
 export type AppLoaded = EmptyEnvelope<MessageType.AppLoaded>;
@@ -422,3 +424,10 @@ export type MobxAddMessage = Envelope<MessageType.MobxAdd, MobxAddPayload>;
 export type MobxDeleteMessage = Envelope<MessageType.MobxDelete, MobxDeletePayload>;
 export type MobxSpliceMessage = Envelope<MessageType.MobxSplice, MobxSplicePayload>;
 export type WindowClose = EmptyEnvelope<MessageType.WindowClose>;
+export type ChromeScreenShot = Envelope<
+	MessageType.ChromeScreenShot,
+	{
+		width: number;
+		height: number;
+	}
+>;
diff --git a/src/renderer/renderer.tsx b/src/renderer/renderer.tsx
@@ -34,6 +34,19 @@ export function startRenderer(): void {
 
 	// tslint:disable-next-line:no-any
 	(window as any).store = store;
+
+	// tslint:disable-next-line:no-any
+	(window as any).screenshot = () => {
+		sender.send({
+			id: uuid.v4(),
+			type: MessageType.ChromeScreenShot,
+			payload: {
+				width: window.innerWidth,
+				height: window.innerHeight
+			}
+		});
+	};
+
 	console.log('Access ViewStore at .store');
 
 	createListeners({ store });