Fixed support of non elixir code blocks

CHANGELOG.md

@@ -9,6 +9,10 @@
 > - You can also [buy me a coffee](https://ko-fi.com/C0C316P9FN), as it would encourage me spending
 >   more of my free time to fixing bugs and developing new features 🤗
 
+## v0.8.2 (not released)
+
+- **bugfix**: [fixed support of non elixir code blocks](https://github.com/phenixdigital/phoenix_storybook/issues/570)
+
 ## v0.8.1 (2025-01-20)
 
 - **improvement**: [better formatting of playground's HTML](https://github.com/phenixdigital/phoenix_storybook/pull/562) - 🙏 [@xxdavid](https://github.com/xxdavid)

lib/phoenix_storybook/rendering/code_renderer.ex

@@ -271,10 +271,10 @@ defmodule PhoenixStorybook.Rendering.CodeRenderer do
     code =
       if format? do
         case lang do
-          :elixir -> ElixirLexer.lex(code)
-          :heex -> HEExLexer.lex(code)
+          :elixir -> code |> ElixirLexer.lex() |> HTMLFormatter.format_inner_as_binary([])
+          :heex -> code |> HEExLexer.lex() |> HTMLFormatter.format_inner_as_binary([])
+          _ -> code
         end
-        |> HTMLFormatter.format_inner_as_binary([])
       else
         code
       end

lib/phoenix_storybook/stories/doc.ex

@@ -113,6 +113,7 @@ defmodule PhoenixStorybook.Stories.Doc do
         "elixir" -> :elixir
         "heex" -> :heex
         "" -> code |> String.trim_leading() |> guess_lang()
+        _ -> :unknown
       end
 
     CodeRenderer.render_code_block(code, lang, trim: false)

test/fixtures/components/event_component.ex

@@ -1,6 +1,19 @@
 defmodule EventComponent do
   use Phoenix.Component
 
+  @doc """
+  Component doc
+
+  ```
+  Some code
+  ```
+
+  ```css
+  .my-class {
+    margin: 0;
+  }
+  ```
+  """
   def component(assigns) do
     assigns =
       assigns

test/phoenix_storybook/stories/doc_test.exs

@@ -81,6 +81,16 @@ defmodule PhoenixStorybook.Stories.DocTest do
       assert log =~
                "could not fetch module doc from PhoenixStorybook.Stories.DocTest.NoDocLiveComponent"
     end
+
+    test "it does not crash with css and untyped code blocks" do
+      %Doc{header: header, body: body} =
+        "event/event_component.story.exs"
+        |> compile_story()
+        |> Doc.fetch_doc_as_html()
+
+      refute is_nil(header)
+      refute is_nil(body)
+    end
   end
 
   defp compile_story(path) do