fix syntax highlighter example to fully support pygments-stylesheet attribute

1 files changed, 10 insertions, 7 deletions

diff --git a/docs/modules/syntax-highlighting/examples/pygments-syntax-highlighter-with-custom-stylesheet.rb b/docs/modules/syntax-highlighting/examples/pygments-syntax-highlighter-with-custom-stylesheet.rb
index 9c6fdae1..ef32705e 100644
--- a/docs/modules/syntax-highlighting/examples/pygments-syntax-highlighter-with-custom-stylesheet.rb
+++ b/docs/modules/syntax-highlighting/examples/pygments-syntax-highlighter-with-custom-stylesheet.rb
@@ -1,12 +1,15 @@
-class MyPygmentsAdapter < (Asciidoctor::SyntaxHighlighter.for 'pygments')
+class ExtendedPygmentsSyntaxHighlighter < (Asciidoctor::SyntaxHighlighter.for 'pygments')
   register_for :pygments
 
-  def write_stylesheet? doc
-    false
-  end
-
   def docinfo location, doc, opts
-    slash = opts[:self_closing_tag_slash]
-    %(<link rel="stylesheet" href="/styles/syntax-theme.css"#{slash}>)
+    stylesheet = doc.attr 'pygments-stylesheet', './pygments.css'
+    if opts[:linkcss]
+      %(<link rel="stylesheet" href="#{stylesheet}"#{opts[:self_closing_tag_slash]}>)
+    else
+      stylesheet = doc.normalize_system_path stylesheet
+      %(<style>
+#{doc.read_asset stylesheet, label: 'stylesheet', normalize: true}
+</style>)
+    end
   end
 end