stabalization

Author: voltsparx

lib/asrfacet_rb/ui/cli.rb

@@ -124,7 +124,7 @@ def delete(target)
       desc "export TARGET", "Export a workspace as json or csv"
       option :format, type: :string, default: "json", enum: %w[json csv], desc: "Export format"
       def export(target)
-        path = workspace_manager.export(target, format: options[:format])
+        path = workspace_manager.export(target, format: export_format)
         puts path
       rescue ASRFacet::Error => e
         ASRFacet::Core::ThreadSafe.print_error(e.message)
@@ -134,6 +134,17 @@ def export(target)
         def workspace_manager
           ASRFacet::Intelligence::SessionManager.new
         end
+
+        def export_format
+          subcommand_format = options[:format].to_s
+          inherited = parent_options.to_h
+          inherited_format = (inherited[:format] || inherited["format"]).to_s
+          return inherited_format if subcommand_format == "json" && %w[json csv].include?(inherited_format)
+
+          subcommand_format
+        rescue StandardError
+          options[:format].to_s
+        end
       end
     end
 

spec/output/output_router_engine_spec.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+# SPDX-License-Identifier: Proprietary
+#
+# ASRFacet-Rb: Attack Surface Reconnaissance Framework
+# Copyright (c) 2026 voltsparx
+#
+# Author: voltsparx
+# Repository: https://github.com/voltsparx/ASRFacet-Rb
+# Contact: voltsparx@gmail.com
+# License: See LICENSE file in the project root
+#
+# This file is part of ASRFacet-Rb and is subject to the terms
+# and conditions defined in the LICENSE file.
+
+require "spec_helper"
+require "tmpdir"
+
+RSpec.describe ASRFacet::Output::OutputRouter do
+  subject(:router) { described_class.new(build_output_store, output_fixture_data[:target], asset_graph: build_output_graph) }
+
+  it "routes PDF and DOCX rendering through the JavaScript bridges when Node.js is available" do
+    pdf_renderer = instance_double(ASRFacet::Output::Js::JsPdfBridge, render: true)
+    docx_renderer = instance_double(ASRFacet::Output::Js::JsDocxBridge, render: true)
+
+    allow(ASRFacet::Output::RuntimeDetector).to receive(:node_available?).and_return(true)
+    allow(ASRFacet::Output::Js::JsPdfBridge).to receive(:new).and_return(pdf_renderer)
+    allow(ASRFacet::Output::Js::JsDocxBridge).to receive(:new).and_return(docx_renderer)
+
+    router.render("pdf", File.join(Dir.tmpdir, "fixture.pdf"))
+    router.render("docx", File.join(Dir.tmpdir, "fixture.docx"))
+
+    expect(ASRFacet::Output::Js::JsPdfBridge).to have_received(:new)
+    expect(ASRFacet::Output::Js::JsDocxBridge).to have_received(:new)
+    expect(pdf_renderer).to have_received(:render)
+    expect(docx_renderer).to have_received(:render)
+  end
+
+  it "falls back to the Ruby PDF and DOCX renderers when Node.js is unavailable" do
+    pdf_renderer = instance_double(ASRFacet::Output::Ruby::PdfRenderer, render: true)
+    docx_renderer = instance_double(ASRFacet::Output::Ruby::DocxRenderer, render: true)
+
+    allow(ASRFacet::Output::RuntimeDetector).to receive(:node_available?).and_return(false)
+    allow(ASRFacet::Output::Ruby::PdfRenderer).to receive(:new).and_return(pdf_renderer)
+    allow(ASRFacet::Output::Ruby::DocxRenderer).to receive(:new).and_return(docx_renderer)
+
+    router.render("pdf", File.join(Dir.tmpdir, "fixture.pdf"))
+    router.render("docx", File.join(Dir.tmpdir, "fixture.docx"))
+
+    expect(ASRFacet::Output::Ruby::PdfRenderer).to have_received(:new)
+    expect(ASRFacet::Output::Ruby::DocxRenderer).to have_received(:new)
+    expect(pdf_renderer).to have_received(:render)
+    expect(docx_renderer).to have_received(:render)
+  end
+
+  it "continues rendering later formats during render_all when one renderer raises an ASRFacet error" do
+    formats = []
+
+    allow(router).to receive(:render).and_wrap_original do |_method, format, path|
+      formats << [format, path]
+      raise ASRFacet::Error, "PDF unavailable" if format == "pdf"
+
+      FileUtils.mkdir_p(File.dirname(path))
+      File.write(path, format)
+    end
+
+    Dir.mktmpdir do |dir|
+      expect { router.render_all(dir) }.not_to raise_error
+    end
+
+    expect(formats.map(&:first)).to eq(%w[txt html json csv pdf docx])
+  end
+end

spec/ui/workspace_cli_spec.rb

@@ -0,0 +1,91 @@
+# SPDX-License-Identifier: Proprietary
+#
+# ASRFacet-Rb: Attack Surface Reconnaissance Framework
+# Copyright (c) 2026 voltsparx
+#
+# Author: voltsparx
+# Repository: https://github.com/voltsparx/ASRFacet-Rb
+# Contact: voltsparx@gmail.com
+# License: See LICENSE file in the project root
+#
+# This file is part of ASRFacet-Rb and is subject to the terms
+# and conditions defined in the LICENSE file.
+
+require "spec_helper"
+require "stringio"
+require "tmpdir"
+
+RSpec.describe ASRFacet::UI::CLI do
+  def capture_stdout
+    original = $stdout
+    buffer = StringIO.new
+    $stdout = buffer
+    yield
+    buffer.string
+  ensure
+    $stdout = original
+  end
+
+  def wire_workspace_root(dir)
+    manager = ASRFacet::Intelligence::SessionManager.new(root: dir)
+    allow(ASRFacet::Intelligence::SessionManager).to receive(:new).and_return(manager)
+    allow(ASRFacet::Intelligence::AssetGraph).to receive(:new).and_wrap_original do |method, target, **kwargs|
+      method.call(target, **kwargs.merge(root: dir))
+    end
+    manager
+  end
+
+  it "lists, shows, exports, and deletes persisted workspaces through the CLI" do
+    Dir.mktmpdir do |dir|
+      manager = wire_workspace_root(dir)
+      manager.create("lab.example.com")
+      build_intelligence_graph(root: dir)
+
+      list_output = capture_stdout { described_class.start(["workspace", "list"]) }
+      show_output = capture_stdout { described_class.start(["workspace", "show", "lab.example.com"]) }
+      export_output = capture_stdout { described_class.start(["workspace", "export", "lab.example.com", "--format", "csv"]) }
+
+      expect(list_output).to include("lab.example.com")
+      expect(JSON.parse(show_output)).to include("target" => "lab.example.com")
+
+      export_path = export_output.strip
+      expect(export_path).to end_with(".csv")
+      expect(File).to exist(export_path)
+
+      delete_output = capture_stdout { described_class.start(["workspace", "delete", "lab.example.com"]) }
+      expect(delete_output).to include("Workspace deleted")
+      expect(capture_stdout { described_class.start(["workspace", "show", "lab.example.com"]) }).to include("Workspace not found")
+    end
+  end
+
+  it "tracks diffs, renders graph exports, and lists stored subdomains from a workspace" do
+    Dir.mktmpdir do |dir|
+      manager = wire_workspace_root(dir)
+      manager.create("lab.example.com")
+      build_intelligence_graph("intelligence_graph_previous", root: dir)
+      manager.export("lab.example.com", format: "json")
+      build_intelligence_graph(root: dir)
+
+      track_output = capture_stdout { described_class.start(["track", "lab.example.com", "--since", "2099-01-01"]) }
+      viz_output = capture_stdout { described_class.start(["viz", "lab.example.com", "--format", "mermaid"]) }
+      subs_output = capture_stdout { described_class.start(["subs", "lab.example.com"]) }
+
+      diff = JSON.parse(track_output)
+      expect(diff.dig("summary", "added")).to be > 0
+      expect(viz_output).to include("graph LR")
+      expect(subs_output.lines.map(&:strip)).to eq(%w[api.lab.example.com app.lab.example.com])
+    end
+  end
+
+  it "reports invalid track dates as parse errors without crashing" do
+    Dir.mktmpdir do |dir|
+      manager = wire_workspace_root(dir)
+      manager.create("lab.example.com")
+      build_intelligence_graph(root: dir)
+
+      output = capture_stdout { described_class.start(["track", "lab.example.com", "--since", "not-a-date"]) }
+
+      expect(output).to include("Invalid --since value")
+    end
+  end
+end