1 files changed, 34 insertions, 0 deletions
diff --git a/source/features/embed-gist-via-iframe.tsx b/source/features/embed-gist-via-iframe.tsx
new file mode 100644
index 00000000..f0368fda
--- /dev/null
+++ b/source/features/embed-gist-via-iframe.tsx
@@ -0,0 +1,34 @@
+import select from 'select-dom';
+import features from '../libs/features';
+
+function init(): void {
+	const embedViaScript = select('.file-navigation-option [value^="<script"]')!;
+	const embedViaIframe = embedViaScript.cloneNode(true) as HTMLButtonElement;
+
+	// Remove analytics attributes
+	embedViaIframe.removeAttribute('data-hydro-click');
+	embedViaIframe.removeAttribute('data-hydro-click-hmac');
+
+	// Set required content
+	embedViaIframe.setAttribute('aria-checked', 'false');
+	embedViaIframe.value = `<iframe src="${location.origin}${location.pathname}.pibb"></iframe>`;
+	select('.select-menu-item-heading', embedViaIframe)!.textContent = 'Embed via <iframe>';
+	select('.description', embedViaIframe)!.textContent = 'Embed this gist in your website via <iframe>.';
+
+	// Modify description of the original embed type to distinguish the two items
+	select('.select-menu-item-heading', embedViaScript)!.textContent = 'Embed via <script>';
+	select('.description', embedViaScript)!.textContent = 'Embed this gist in your website via <script>.';
+
+	embedViaScript.after(embedViaIframe);
+}
+
+features.add({
+	id: __featureName__,
+	description: 'Adds a menu item to embed a gist via <iframe>.',
+	screenshot: 'https://user-images.githubusercontent.com/44045911/63633382-6a1b6200-c67a-11e9-9038-aedd62e4f6a8.png',
+	include: [
+		features.isGist
+	],
+	load: features.onDomReady,
+	init
+});