From 99430e0edc60588b921938ef3c9bd44488348efb Mon Sep 17 00:00:00 2001
From: MapleLeaf <19603573+itsMapleLeaf@users.noreply.github.com>
Date: Sat, 25 Dec 2021 01:24:52 -0600
Subject: [PATCH] refactor: rendering button

---
 playground/counter.tsx        |  4 +--
 playground/main.tsx           |  4 ++-
 src.new/components/button.tsx | 28 ++++++++++++++++++
 src.new/components/text.tsx   | 14 ---------
 src.new/jsx.d.ts              | 14 +++++++++
 src.new/node.ts               |  3 ++
 src.new/reconciler.ts         |  9 ++++--
 src.new/root-node.ts          | 53 ++++++++++++++++++++++++++++-------
 src.new/text-node.ts          |  9 ++++--
 9 files changed, 107 insertions(+), 31 deletions(-)
 create mode 100644 src.new/components/button.tsx
 delete mode 100644 src.new/components/text.tsx
 create mode 100644 src.new/jsx.d.ts
 create mode 100644 src.new/node.ts

diff --git a/playground/counter.tsx b/playground/counter.tsx
index fd30bba..1f642c0 100644
--- a/playground/counter.tsx
+++ b/playground/counter.tsx
@@ -1,12 +1,12 @@
 import * as React from "react"
-import { Button } from "../src/main.js"
+import { Button } from "../src.new/components/button.js"
 
 export function Counter() {
   const [count, setCount] = React.useState(0)
   return (
     <>
       this button was clicked {count} times
-      <Button onClick={() => setCount(count + 1)}>clicc</Button>
+      <Button label="clicc" onClick={() => setCount(count + 1)} />
     </>
   )
 }
diff --git a/playground/main.tsx b/playground/main.tsx
index d2f3e0b..a9f78b0 100644
--- a/playground/main.tsx
+++ b/playground/main.tsx
@@ -1,7 +1,9 @@
 import { Client } from "discord.js"
 import "dotenv/config"
+import React from "react"
 import { InstanceManager } from "../src.new/main.js"
 import { createCommandHandler } from "./command-handler.js"
+import { Counter } from "./counter.js"
 
 const client = new Client({
   intents: ["GUILDS"],
@@ -14,7 +16,7 @@ createCommandHandler(client, [
     name: "counter",
     description: "shows a counter button",
     run: (interaction) => {
-      manager.create(interaction).render("hi world")
+      manager.create(interaction).render(<Counter />)
     },
   },
 ])
diff --git a/src.new/components/button.tsx b/src.new/components/button.tsx
new file mode 100644
index 0000000..b79a2e4
--- /dev/null
+++ b/src.new/components/button.tsx
@@ -0,0 +1,28 @@
+import type {
+  EmojiResolvable,
+  MessageButtonStyle,
+  MessageComponentInteraction,
+} from "discord.js"
+import React from "react"
+import { Node } from "../node.js"
+
+export type ButtonProps = {
+  label?: string
+  style?: Exclude<Lowercase<MessageButtonStyle>, "link">
+  disabled?: boolean
+  emoji?: EmojiResolvable
+  onClick?: (interaction: MessageComponentInteraction) => void
+}
+
+export const ButtonTag = "reacord-button"
+
+export function Button(props: ButtonProps) {
+  return React.createElement(ButtonTag, props)
+}
+
+export class ButtonNode extends Node {
+  readonly name = "button"
+  constructor(public props: ButtonProps) {
+    super()
+  }
+}
diff --git a/src.new/components/text.tsx b/src.new/components/text.tsx
deleted file mode 100644
index 6ce9b59..0000000
--- a/src.new/components/text.tsx
+++ /dev/null
@@ -1,14 +0,0 @@
-import type { ReactNode } from "react"
-import React from "react"
-
-export type TextProps = {
-  children?: ReactNode
-}
-
-export const TextTag = "reacord-text"
-
-export function Text(props: TextProps) {
-  return React.createElement(TextTag, props)
-}
-
-export class TextElementNode {}
diff --git a/src.new/jsx.d.ts b/src.new/jsx.d.ts
new file mode 100644
index 0000000..fc1ca15
--- /dev/null
+++ b/src.new/jsx.d.ts
@@ -0,0 +1,14 @@
+import type { ReactNode } from "react"
+import type { Node } from "./node"
+
+declare global {
+  namespace JSX {
+    // eslint-disable-next-line @typescript-eslint/consistent-type-definitions
+    interface IntrinsicElements {
+      "reacord-element": {
+        createNode: () => Node
+        children?: ReactNode
+      }
+    }
+  }
+}
diff --git a/src.new/node.ts b/src.new/node.ts
new file mode 100644
index 0000000..b38b461
--- /dev/null
+++ b/src.new/node.ts
@@ -0,0 +1,3 @@
+export abstract class Node {
+  abstract get name(): string
+}
diff --git a/src.new/reconciler.ts b/src.new/reconciler.ts
index 089020d..7e40f3a 100644
--- a/src.new/reconciler.ts
+++ b/src.new/reconciler.ts
@@ -1,6 +1,8 @@
 import type { HostConfig } from "react-reconciler"
 import ReactReconciler from "react-reconciler"
 import { raise } from "../src/helpers/raise.js"
+import { ButtonNode } from "./components/button.js"
+import type { Node } from "./node.js"
 import type { RootNode } from "./root-node.js"
 import { TextNode } from "./text-node.js"
 
@@ -8,7 +10,7 @@ const config: HostConfig<
   string, // Type,
   Record<string, unknown>, // Props,
   RootNode, // Container,
-  never, // Instance,
+  Node, // Instance,
   TextNode, // TextInstance,
   never, // SuspenseInstance,
   never, // HydratableInstance,
@@ -32,7 +34,10 @@ const config: HostConfig<
   getRootHostContext: () => ({}),
   getChildHostContext: () => ({}),
 
-  createInstance: () => raise("not implemented"),
+  createInstance: (type, props) => {
+    if (type === "reacord-button") return new ButtonNode(props)
+    raise(`Unknown type: ${type}`)
+  },
   createTextInstance: (text) => new TextNode(text),
   shouldSetTextContent: () => false,
 
diff --git a/src.new/root-node.ts b/src.new/root-node.ts
index 210d91f..84e9560 100644
--- a/src.new/root-node.ts
+++ b/src.new/root-node.ts
@@ -1,12 +1,21 @@
 import type { CommandInteraction, MessageOptions } from "discord.js"
-import type { TextNode } from "./text-node.js"
+import { MessageActionRow } from "discord.js"
+import { nanoid } from "nanoid"
+import { last } from "../src/helpers/last.js"
+import { toUpper } from "../src/helpers/to-upper.js"
+import { ButtonNode } from "./components/button.js"
+import { Node } from "./node.js"
+import { TextNode } from "./text-node.js"
 
-export class RootNode {
-  private children = new Set<TextNode>()
+export class RootNode extends Node {
+  readonly name = "root"
+  private children = new Set<Node>()
 
-  constructor(private interaction: CommandInteraction) {}
+  constructor(private interaction: CommandInteraction) {
+    super()
+  }
 
-  add(child: TextNode) {
+  add(child: Node) {
     this.children.add(child)
   }
 
@@ -14,7 +23,7 @@ export class RootNode {
     this.children.clear()
   }
 
-  remove(child: TextNode) {
+  remove(child: Node) {
     this.children.delete(child)
   }
 
@@ -22,13 +31,37 @@ export class RootNode {
     this.interaction.reply(this.getMessageOptions()).catch(console.error)
   }
 
-  getMessageOptions() {
-    const options: MessageOptions = {}
+  private getMessageOptions(): MessageOptions {
+    let content = ""
+    let components: MessageActionRow[] = []
 
     for (const child of this.children) {
-      options.content = (options.content ?? "") + child.text
+      if (child instanceof TextNode) {
+        content += child.text
+      }
+
+      if (child instanceof ButtonNode) {
+        let actionRow = last(components)
+        if (
+          !actionRow ||
+          actionRow.components.length >= 5 ||
+          actionRow.components[0]?.type === "SELECT_MENU"
+        ) {
+          actionRow = new MessageActionRow()
+          components.push(actionRow)
+        }
+
+        actionRow.addComponents({
+          type: "BUTTON",
+          customId: nanoid(),
+          style: toUpper(child.props.style ?? "secondary"),
+          disabled: child.props.disabled,
+          emoji: child.props.emoji,
+          label: child.props.label,
+        })
+      }
     }
 
-    return options
+    return { content, components }
   }
 }
diff --git a/src.new/text-node.ts b/src.new/text-node.ts
index 7d66981..06bb000 100644
--- a/src.new/text-node.ts
+++ b/src.new/text-node.ts
@@ -1,3 +1,8 @@
-export class TextNode {
-  constructor(public text: string) {}
+import { Node } from "./node.js"
+
+export class TextNode extends Node {
+  readonly name = "text"
+  constructor(public text: string) {
+    super()
+  }
 }