ligarb 0.4.0 → 0.6.0

data/lib/ligarb/template.rb

@@ -12,7 +12,7 @@ module Ligarb
       @css_path      = File.join(ASSETS_DIR, "style.css")
     end
 
-    def render(config:, chapters:, structure:, assets:, index_entries: [])
+    def render(config:, chapters:, structure:, assets:, index_entries: [], bibliography: [])
       css = File.read(@css_path)
       template = File.read(@template_path)
 
@@ -34,12 +34,18 @@ module Ligarb
       b.local_variable_set(:ai_generated, config.ai_generated)
       b.local_variable_set(:footer, config.effective_footer)
       b.local_variable_set(:index_tree, build_index_tree(index_entries, chapters))
+      b.local_variable_set(:bibliography, bibliography)
 
       ERB.new(template, trim_mode: "-").result(b)
     end
 
     private
 
+    # HTML-escape helper for ERB templates (available via binding)
+    def h(s)
+      ERB::Util.html_escape(s.to_s)
+    end
+
     # Build a sorted tree structure for the index.
     # Returns: { "A" => [ { term: "Algorithm", refs: [...] },
     #                      { term: "Array", refs: [...], children: [ { term: "sort", refs: [...] } ] } ],

data/lib/ligarb/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Ligarb
-  VERSION = "0.4.0"
+  VERSION = "0.6.0"
 end

data/lib/ligarb/writer.rb

@@ -1,10 +1,13 @@
 # frozen_string_literal: true
 
 require "yaml"
+require "json"
 require "fileutils"
 
 module Ligarb
   class Writer
+    class WriterError < RuntimeError; end
+
     BRIEF_FIELDS_FOR_BOOK_YML = %w[author output_dir chapter_numbers style repository].freeze
 
     def initialize(brief_path, no_build: false)
@@ -19,8 +22,7 @@ module Ligarb
       book_yml_path = File.join(output_dir, "book.yml")
 
       if File.exist?(book_yml_path)
-        $stderr.puts "Error: #{book_yml_path} already exists. Remove it first to regenerate."
-        exit 1
+        raise WriterError, "#{book_yml_path} already exists. Remove it first to regenerate."
       end
 
       FileUtils.mkdir_p(output_dir)
@@ -28,8 +30,7 @@ module Ligarb
       run_claude(prompt)
 
       unless File.exist?(book_yml_path)
-        $stderr.puts "Error: Claude did not generate book.yml in #{output_dir}"
-        exit 1
+        raise WriterError, "Claude did not generate book.yml in #{output_dir}"
       end
 
       puts "Book files generated in #{output_dir}"
@@ -39,6 +40,10 @@ module Ligarb
         require_relative "builder"
         Builder.new(book_yml_path).build
       end
+
+      git_commit_initial(output_dir, brief["title"]) if git_available?(output_dir)
+
+      book_yml_path
     end
 
     def self.init_brief(directory = nil)
@@ -47,8 +52,7 @@ module Ligarb
       path = File.join(target, "brief.yml")
 
       if File.exist?(path)
-        $stderr.puts "Error: #{path} already exists."
-        exit 1
+        raise WriterError, "#{path} already exists."
       end
 
       FileUtils.mkdir_p(target)
@@ -88,28 +92,48 @@ module Ligarb
       puts "Created #{path}"
       puts "Created #{claude_md}" if created_claude_md
       brief_arg = directory ? " #{path}" : ""
+      unless system("git", "rev-parse", "--git-dir", out: File::NULL, err: File::NULL, chdir: target)
+        gitignore = File.join(target, ".gitignore")
+        unless File.exist?(gitignore)
+          File.write(gitignore, "# Generated by ligarb - remove lines as needed\nbuild/\n.ligarb/\n")
+        end
+        system("git", "init", chdir: target)
+        puts "Initialized git repository"
+      end
+
       puts "Edit brief.yml, then run 'ligarb write#{brief_arg}' to generate the book."
     end
 
     private
 
+    def git_available?(dir)
+      system("git", "rev-parse", "--git-dir", out: File::NULL, err: File::NULL, chdir: dir)
+    end
+
+    def git_commit_initial(dir, title)
+      system("git", "add", "-A", chdir: dir, out: File::NULL, err: File::NULL)
+      system("git", "commit", "-m", "[ligarb] Generate book: #{title}",
+             chdir: dir, out: File::NULL, err: File::NULL)
+    end
+
     def check_claude_installed!
+      claude_path = ENV["PATH"].to_s.split(File::PATH_SEPARATOR).find { |dir| File.executable?(File.join(dir, "claude")) }
+      unless claude_path
+        raise WriterError, "'claude' command not found. Install Claude Code first."
+      end
       unless system("claude", "--version", out: File::NULL, err: File::NULL)
-        $stderr.puts "Error: 'claude' command not found. Install Claude Code first."
-        exit 1
+        raise WriterError, "'claude' command was found but failed to run. Check your Claude Code installation."
       end
     end
 
     def load_brief
       unless File.exist?(@brief_path)
-        $stderr.puts "Error: #{@brief_path} not found."
-        exit 1
+        raise WriterError, "#{@brief_path} not found."
       end
 
       brief = YAML.safe_load_file(@brief_path)
       unless brief.is_a?(Hash) && brief["title"] && !brief["title"].empty?
-        $stderr.puts "Error: 'title' is required in #{@brief_path}."
-        exit 1
+        raise WriterError, "'title' is required in #{@brief_path}."
       end
 
       brief
@@ -148,6 +172,13 @@ module Ligarb
         lines << "In book.yml, set: #{settings}"
       end
 
+      sources = parse_brief_sources(brief)
+      if sources.any?
+        lines << ""
+        lines << "Reference sources (read these files for context):"
+        sources.each { |src| lines << "- #{src[:label]}: #{src[:path]}" }
+      end
+
       lines << ""
       lines << "Create all files in: #{abs_output_dir}"
       lines << "In book.yml, always set: ai_generated: true"
@@ -156,18 +187,65 @@ module Ligarb
       lines << "Each chapter: substantive content with multiple ## sections."
       lines << "Use code blocks, admonitions, mermaid diagrams where appropriate."
       lines << "Chapter filenames: 01-topic.md, 02-topic.md, etc."
+      lines << ""
+      lines << "Create references.bib with real bibliography entries in BibTeX format."
+      lines << "In book.yml, set: bibliography: references.bib"
+      lines << "Cite references in chapter text using [text](#cite:key) syntax."
+      lines << "Search the web to find accurate bibliographic information (DOI, pages, volume, etc.)."
+      lines << "Do NOT fabricate references. Only include real, verifiable publications."
+      lines << "Use entry types: @article, @inproceedings, @book, @misc as appropriate."
+      lines << "Use UTF-8 directly for special characters (no LaTeX commands like \\c{c})."
+      lines << "BibTeX keys: authorsurname+year (e.g. knuth1984, cousot1977)."
 
       lines.join("\n")
     end
 
+    def parse_brief_sources(brief)
+      base_dir = File.dirname(@brief_path)
+      (brief["sources"] || []).map do |entry|
+        case entry
+        when String
+          { path: File.expand_path(entry, base_dir), label: File.basename(entry) }
+        when Hash
+          path = entry["path"] or raise WriterError, "source entry missing 'path'"
+          { path: File.expand_path(path, base_dir), label: entry.fetch("label", File.basename(path)) }
+        else
+          raise WriterError, "invalid source entry: #{entry.inspect}"
+        end
+      end
+    end
+
     def run_claude(prompt)
-      tools = "Write,Bash,WebFetch,WebSearch"
-      allowed = "Write,Bash(mkdir:*),Bash(ls:*),Bash(ligarb:*),WebFetch,WebSearch"
-      cmd = ["claude", "-p", "--verbose", prompt, "--tools", tools, "--allowedTools", allowed]
-      unless system(*cmd)
-        $stderr.puts "Error: Claude process failed."
-        exit 1
+      tools = "Read,Write,Bash,WebFetch,WebSearch"
+      allowed = "Read,Write,Bash(mkdir:*),Bash(ls:*),Bash(ligarb:*),WebFetch,WebSearch"
+      cmd = ["claude", "-p", prompt, "--tools", tools, "--allowedTools", allowed,
+             "--output-format", "stream-json", "--verbose"]
+      puts "Writing with Claude... (this may take a few minutes)"
+      unparsed_lines = []
+      IO.popen(cmd, err: [:child, :out]) do |io|
+        io.each_line do |line|
+          unless parse_stream_event(line)
+            unparsed_lines << line.rstrip
+          end
+        end
+      end
+      unless $?.success?
+        msg = unparsed_lines.reject(&:empty?).last(10).join("\n")
+        raise WriterError, "Claude process failed.#{"\n#{msg}" unless msg.empty?}"
+      end
+    end
+
+    def parse_stream_event(line)
+      json = JSON.parse(line) rescue (return false)
+      case json["type"]
+      when "content_block_start"
+        tool = json.dig("content_block", "tool_use")
+        if tool
+          name = tool["name"]
+          puts "  [#{name}]..." if name
+        end
       end
+      true
     end
   end
 end