support $$slots in custom elements

src/compiler/compile/render_dom/index.ts

@@ -467,14 +467,20 @@ export default function dom(
 	}
 
 	if (options.customElement) {
+
+		let init_props = x`@attribute_to_object(this.attributes)`;
+		if (uses_slots) {
+			init_props = x`{ ...${init_props}, $$slots: @get_custom_elements_slots(this) }`;
+		}
+
 		const declaration = b`
 			class ${name} extends @SvelteElement {
 				constructor(options) {
 					super();
 
 					${css.code && b`this.shadowRoot.innerHTML = \`<style>${css.code.replace(/\\/g, '\\\\')}${options.dev ? `\n/*# sourceMappingURL=${css.map.toUrl()} */` : ''}</style>\`;`}
 
-					@init(this, { target: this.shadowRoot, props: @attribute_to_object(this.attributes) }, ${definition}, ${has_create_fragment ? 'create_fragment': 'null'}, ${not_equal}, ${prop_indexes}, ${dirty});
+					@init(this, { target: this.shadowRoot, props: ${init_props} }, ${definition}, ${has_create_fragment ? 'create_fragment': 'null'}, ${not_equal}, ${prop_indexes}, ${dirty});
 
 					${dev_props_check}
 

src/runtime/internal/dom.ts

@@ -361,10 +361,18 @@ export class HtmlTag {
 	}
 }
 
-export function attribute_to_object(attributes) {
+export function attribute_to_object(attributes: NamedNodeMap) {
 	const result = {};
 	for (const attribute of attributes) {
 		result[attribute.name] = attribute.value;
 	}
 	return result;
 }
+
+export function get_custom_elements_slots(element: HTMLElement) {
+	const result = {};
+	element.childNodes.forEach((node: Element) => {
+		result[node.slot || 'default'] = true;
+	});
+	return result;
+}

test/custom-elements/samples/$$slot/main.svelte

@@ -0,0 +1,31 @@
+<script>
+	let data = '';
+
+	if ($$slots.b) {
+		data = 'foo';
+	}
+
+	export function getData() {
+		return data;
+	}
+
+	function toString(data) {
+		const result = {};
+		const sortedKeys = Object.keys(data).sort();
+		sortedKeys.forEach(key => result[key] = data[key]);
+		return JSON.stringify(result);
+	}
+</script>
+
+<svelte:options tag="custom-element"/>
+
+<slot></slot>
+<slot name="a"></slot>
+<p>$$slots: {toString($$slots)}</p>
+{#if $$slots.b}
+	<div>
+		<slot name="b"></slot>
+	</div>
+{:else}
+	<p>Slot b is not available</p>
+{/if}

test/custom-elements/samples/$$slot/test.js

@@ -0,0 +1,28 @@
+import * as assert from 'assert';
+import './main.svelte';
+
+export default function (target) {
+	target.innerHTML = `
+		<custom-element><span slot="a">hello world</span><span>bye</span><span>world</span></custom-element>
+		<custom-element><span slot="a">hello world</span><span slot="b">hello world</span><span>bye world</span></custom-element>
+	`;
+
+	const [a, b] = target.querySelectorAll('custom-element');
+
+	assert.htmlEqual(a.shadowRoot.innerHTML, `
+		<slot></slot>
+		<slot name="a"></slot>
+		<p>$$slots: {"a":true,"default":true}</p>
+		<p>Slot b is not available</p>
+	`);
+
+	assert.htmlEqual(b.shadowRoot.innerHTML, `
+		<slot></slot>
+		<slot name="a"></slot>
+		<p>$$slots: {"a":true,"b":true,"default":true}</p>
+		<div><slot name="b"></slot></div>
+	`);
+
+	assert.equal(a.getData(), '');
+	assert.equal(b.getData(), 'foo');
+}