From 166a9ca150df365da01c312e09e33f9de4f03a75 Mon Sep 17 00:00:00 2001
From: Chris Griffing <cmgriffing@gmail.com>
Date: Mon, 25 Nov 2024 18:18:35 -0800
Subject: [PATCH] feat: begin adding line number support but stopped because of
 strange line wrapping issues

---
 playgrounds/app/src/components/Editor.tsx | 61 +++++++++++++++++++++--
 playgrounds/app/src/routes/index.tsx      |  2 +
 playgrounds/app/src/types.ts              |  3 ++
 3 files changed, 62 insertions(+), 4 deletions(-)

diff --git a/playgrounds/app/src/components/Editor.tsx b/playgrounds/app/src/components/Editor.tsx
index d65ddb5..ab9ed12 100644
--- a/playgrounds/app/src/components/Editor.tsx
+++ b/playgrounds/app/src/components/Editor.tsx
@@ -236,6 +236,7 @@ export default function Editor(props: EditorProps) {
               backgroundGradientColorStart: props.snippetSettings.bgGradientColorStart,
               backgroundGradientColorEnd: props.snippetSettings.bgGradientColorEnd,
               backgroundGradientDirection: props.snippetSettings.bgGradientDirection,
+              lineNumberColor: props.snippetSettings.lineNumberColor,
             },
           },
         )
@@ -352,6 +353,50 @@ export default function Editor(props: EditorProps) {
                 collapsible
                 defaultValue={['background', 'layout', 'shadow', 'font']}
               >
+                <AccordionItem value="background">
+                  <AccordionTrigger>Line Numbers</AccordionTrigger>
+                  <AccordionContent>
+                    <div class="flex flex-col gap-4">
+                      <div class="flex flex-row items-center justify-between">
+                        <Label
+                          for="line-numbers-checkbox"
+                          onClick={() =>
+                            props.setSnippetSettings(
+                              'lineNumbersEnabled',
+                              !props.snippetSettings.lineNumbersEnabled,
+                            )
+                          }
+                        >
+                          Show Line Numbers
+                        </Label>
+                        <Checkbox
+                          id="line-numbers-checkbox"
+                          checked={props.snippetSettings.lineNumbersEnabled}
+                          onChange={() => {
+                            props.setSnippetSettings(
+                              'lineNumbersEnabled',
+                              !props.snippetSettings.lineNumbersEnabled,
+                            )
+                          }}
+                        />
+                      </div>
+                      <div class="flex flex-row items-center justify-between">
+                        <Label for="line-number-color" class="font-normal">
+                          Color
+                        </Label>
+                        <input
+                          id="line-number-color"
+                          class="h-6 w-6 rounded"
+                          type="color"
+                          value={props.snippetSettings.lineNumberColor}
+                          onInput={e => {
+                            props.setSnippetSettings('lineNumberColor', e.target.value)
+                          }}
+                        />
+                      </div>
+                    </div>
+                  </AccordionContent>
+                </AccordionItem>
                 <AccordionItem value="background">
                   <AccordionTrigger>Background</AccordionTrigger>
                   <AccordionContent>
@@ -563,7 +608,7 @@ export default function Editor(props: EditorProps) {
                           }}
                         />
                       </div>
-
+                      //
                       <div class="flex flex-row items-center justify-between">
                         <Label for="shadow-color-input" class="font-normal">
                           Color
@@ -839,7 +884,7 @@ export default function Editor(props: EditorProps) {
                                 options={{
                                   duration: 800,
                                   stagger: 0,
-                                  lineNumbers: false,
+                                  lineNumbers: props.snippetSettings.lineNumbersEnabled,
                                 }}
                               />
                               {/* The hidden shiki that we use to generate the magic move elements */}
@@ -859,7 +904,7 @@ export default function Editor(props: EditorProps) {
                                   options={{
                                     duration: 800,
                                     stagger: 0,
-                                    lineNumbers: false,
+                                    lineNumbers: props.snippetSettings.lineNumbersEnabled,
                                     onAnimationStart: async (elements, maxContainerDimensions) => {
                                       if (elements.length === 0) {
                                         return
@@ -1178,6 +1223,7 @@ async function createAnimationFrame(
   height: number = 100,
   config: AnimationFrameConfig,
 ) {
+  console.log({ elements })
   const { yPadding, xPadding } = config.layout
   const { shadowEnabled, shadowOffsetY, shadowBlur, shadowColor, shadowOpacity } = config.shadow
   const {
@@ -1189,6 +1235,7 @@ async function createAnimationFrame(
     backgroundGradientColorStart,
     backgroundGradientColorEnd,
     backgroundGradientDirection,
+    lineNumberColor,
   } = config.styling
 
   const canvas = document.createElement('canvas')
@@ -1260,10 +1307,16 @@ async function createAnimationFrame(
     const opacity = interpolate(frame, [0, animationFrames], [el.opacity.start, el.opacity.end], {
       easing: Easing.inOut(Easing.quad),
     })
+
+    let fallbackColor = 'rgba(0,0,0,0)'
+    if (el.el.classList.contains('shiki-magic-move-line-number')) {
+      fallbackColor = lineNumberColor || '#aaaaaa'
+    }
+
     const color = interpolateColors(
       frame,
       [0, animationFrames],
-      [el.color.start || 'rgba(0,0,0,0)', el.color.end || 'rgba(0,0,0,0)'],
+      [el.color.start || fallbackColor, el.color.end || fallbackColor],
     )
 
     ctx.font = `${fontSize} ${fontFamily}`
diff --git a/playgrounds/app/src/routes/index.tsx b/playgrounds/app/src/routes/index.tsx
index 9419fce..afa4c88 100644
--- a/playgrounds/app/src/routes/index.tsx
+++ b/playgrounds/app/src/routes/index.tsx
@@ -50,6 +50,8 @@ export default function Home() {
       theme: 'nord',
       fontSize: 16,
       fontFamily: 'Fira Code',
+      lineNumbersEnabled: true,
+      lineNumberColor: '#aaaaaa',
     }),
     { name: 'snippetSettings' },
   )
diff --git a/playgrounds/app/src/types.ts b/playgrounds/app/src/types.ts
index 3afe77a..3c6fdf0 100644
--- a/playgrounds/app/src/types.ts
+++ b/playgrounds/app/src/types.ts
@@ -27,6 +27,8 @@ export interface SnippetSettings {
   theme: string
   fontSize: number
   fontFamily: string
+  lineNumbersEnabled: boolean
+  lineNumberColor: string
 }
 
 export interface Snippet extends SnippetSettings {
@@ -58,6 +60,7 @@ export interface AnimationFrameStyling {
   backgroundGradientColorStart: string
   backgroundGradientColorEnd: string
   backgroundGradientDirection: number
+  lineNumberColor: string
 }
 
 export interface AnimationFrameConfig {