trackplot 0.1.0

data/app/helpers/trackplot/chart_helper.rb

@@ -0,0 +1,9 @@
+module Trackplot
+  module ChartHelper
+    def trackplot_chart(data, **options, &block)
+      builder = ChartBuilder.new(data, **options)
+      capture(builder, &block) if block_given?
+      builder.render(self)
+    end
+  end
+end

data/config/importmap.rb

@@ -0,0 +1,2 @@
+pin "d3", to: "https://cdn.jsdelivr.net/npm/d3@7/+esm", preload: false
+pin "trackplot", to: "trackplot/index.js", preload: true

data/lib/generators/trackplot/install_generator.rb

@@ -0,0 +1,23 @@
+module Trackplot
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      desc "Install Trackplot into your Rails application"
+
+      def add_import
+        js_file = Rails.root.join("app/javascript/application.js")
+
+        if File.exist?(js_file)
+          content = File.read(js_file)
+          if content.include?('import "trackplot"')
+            say "trackplot import already present in application.js", :yellow
+          else
+            append_to_file js_file, "\nimport \"trackplot\"\n"
+            say "Added trackplot import to application.js", :green
+          end
+        else
+          say "Could not find app/javascript/application.js — please add `import \"trackplot\"` manually", :red
+        end
+      end
+    end
+  end
+end

data/lib/trackplot/chart_builder.rb

@@ -0,0 +1,119 @@
+require "securerandom"
+require "json"
+
+module Trackplot
+  class ChartBuilder
+    attr_reader :data, :options, :components
+
+    def initialize(data, **options)
+      @data = DataAdapter.normalize(data)
+      @options = options
+      @components = []
+    end
+
+    def line(data_key, **opts)
+      @components << Components::Line.new(data_key, **opts)
+      nil
+    end
+
+    def bar(data_key, **opts)
+      @components << Components::Bar.new(data_key, **opts)
+      nil
+    end
+
+    def area(data_key, **opts)
+      @components << Components::Area.new(data_key, **opts)
+      nil
+    end
+
+    def pie(data_key, **opts)
+      @components << Components::Pie.new(data_key, **opts)
+      nil
+    end
+
+    def axis(direction, **opts)
+      @components << Components::Axis.new(direction, **opts)
+      nil
+    end
+
+    def tooltip(**opts)
+      @components << Components::Tooltip.new(**opts)
+      nil
+    end
+
+    def legend(**opts)
+      @components << Components::Legend.new(**opts)
+      nil
+    end
+
+    def grid(**opts)
+      @components << Components::Grid.new(**opts)
+      nil
+    end
+
+    def scatter(data_key, **opts)
+      @components << Components::Scatter.new(data_key, **opts)
+      nil
+    end
+
+    def radar(data_key, **opts)
+      @components << Components::Radar.new(data_key, **opts)
+      nil
+    end
+
+    def horizontal_bar(data_key, **opts)
+      @components << Components::HorizontalBar.new(data_key, **opts)
+      nil
+    end
+
+    def candlestick(**opts)
+      @components << Components::Candlestick.new(**opts)
+      nil
+    end
+
+    def funnel(data_key, **opts)
+      @components << Components::Funnel.new(data_key, **opts)
+      nil
+    end
+
+    def reference_line(**opts)
+      @components << Components::ReferenceLine.new(**opts)
+      nil
+    end
+
+    def render(view_context)
+      chart_id = options[:id] || "trackplot-#{SecureRandom.hex(8)}"
+      config = build_config
+
+      view_context.content_tag(
+        "trackplot-chart",
+        nil,
+        id: chart_id,
+        config: config.to_json,
+        style: chart_style,
+        class: css_classes
+      )
+    end
+
+    private
+
+    def build_config
+      {
+        data: data,
+        components: components.map(&:to_config),
+        animate: options.fetch(:animate, true),
+        theme: Theme.resolve(options[:theme])
+      }
+    end
+
+    def chart_style
+      width = options[:width] || "100%"
+      height = options[:height] || "400px"
+      "display:block;width:#{width};height:#{height};"
+    end
+
+    def css_classes
+      ["trackplot-chart", options[:class]].compact.join(" ")
+    end
+  end
+end

data/lib/trackplot/components/area.rb

@@ -0,0 +1,24 @@
+module Trackplot
+  module Components
+    class Area < Base
+      attr_reader :data_key
+
+      def initialize(data_key, **options)
+        @data_key = data_key
+        super(**options)
+      end
+
+      def to_config
+        {
+          type: "area",
+          data_key: data_key,
+          color: options[:color],
+          curve: options.fetch(:curve, false),
+          opacity: options[:opacity] || 0.3,
+          stroke_width: options[:stroke_width] || 2,
+          stack: options[:stack]
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/axis.rb

@@ -0,0 +1,40 @@
+module Trackplot
+  module Components
+    class Axis < Base
+      FORMAT_SYMBOLS = {
+        currency: "currency",
+        percent: "percent",
+        compact: "compact",
+        decimal: "decimal",
+        integer: "integer"
+      }.freeze
+
+      attr_reader :direction
+
+      def initialize(direction, **options)
+        @direction = direction.to_s
+        super(**options)
+      end
+
+      def to_config
+        {
+          type: "axis",
+          direction: direction,
+          data_key: options[:data_key],
+          label: options[:label],
+          format: resolve_format(options[:format]),
+          tick_count: options[:tick_count],
+          tick_rotation: options[:tick_rotation]
+        }.compact
+      end
+
+      private
+
+      def resolve_format(fmt)
+        return nil if fmt.nil?
+
+        FORMAT_SYMBOLS.fetch(fmt, fmt).to_s
+      end
+    end
+  end
+end

data/lib/trackplot/components/bar.rb

@@ -0,0 +1,23 @@
+module Trackplot
+  module Components
+    class Bar < Base
+      attr_reader :data_key
+
+      def initialize(data_key, **options)
+        @data_key = data_key
+        super(**options)
+      end
+
+      def to_config
+        {
+          type: "bar",
+          data_key: data_key,
+          color: options[:color],
+          opacity: options[:opacity],
+          radius: options[:radius] || 4,
+          stack: options[:stack]
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/base.rb

@@ -0,0 +1,19 @@
+module Trackplot
+  module Components
+    class Base
+      attr_reader :options
+
+      def initialize(**options)
+        @options = options
+      end
+
+      def type
+        self.class.name.demodulize.underscore
+      end
+
+      def to_config
+        {type: type}.merge(options).compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/candlestick.rb

@@ -0,0 +1,17 @@
+module Trackplot
+  module Components
+    class Candlestick < Base
+      def to_config
+        {
+          type: "candlestick",
+          open: options[:open],
+          high: options[:high],
+          low: options[:low],
+          close: options[:close],
+          up_color: options[:up_color] || "#10b981",
+          down_color: options[:down_color] || "#ef4444"
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/funnel.rb

@@ -0,0 +1,20 @@
+module Trackplot
+  module Components
+    class Funnel < Base
+      attr_reader :data_key
+
+      def initialize(data_key, **options)
+        @data_key = data_key
+        super(**options)
+      end
+
+      def to_config
+        {
+          type: "funnel",
+          data_key: data_key,
+          label_key: options[:label_key]
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/grid.rb

@@ -0,0 +1,13 @@
+module Trackplot
+  module Components
+    class Grid < Base
+      def to_config
+        {
+          type: "grid",
+          horizontal: options.fetch(:horizontal, true),
+          vertical: options.fetch(:vertical, false)
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/horizontal_bar.rb

@@ -0,0 +1,22 @@
+module Trackplot
+  module Components
+    class HorizontalBar < Base
+      attr_reader :data_key
+
+      def initialize(data_key, **options)
+        @data_key = data_key
+        super(**options)
+      end
+
+      def to_config
+        {
+          type: "horizontal_bar",
+          data_key: data_key,
+          color: options[:color],
+          opacity: options[:opacity],
+          radius: options[:radius] || 4
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/legend.rb

@@ -0,0 +1,13 @@
+module Trackplot
+  module Components
+    class Legend < Base
+      def to_config
+        {
+          type: "legend",
+          position: options[:position] || "bottom",
+          clickable: options.fetch(:clickable, true)
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/line.rb

@@ -0,0 +1,25 @@
+module Trackplot
+  module Components
+    class Line < Base
+      attr_reader :data_key
+
+      def initialize(data_key, **options)
+        @data_key = data_key
+        super(**options)
+      end
+
+      def to_config
+        {
+          type: "line",
+          data_key: data_key,
+          color: options[:color],
+          curve: options.fetch(:curve, false),
+          dashed: options.fetch(:dashed, false),
+          stroke_width: options[:stroke_width] || 2,
+          dot: options.fetch(:dot, true),
+          dot_size: options[:dot_size] || 4
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/pie.rb

@@ -0,0 +1,23 @@
+module Trackplot
+  module Components
+    class Pie < Base
+      attr_reader :data_key
+
+      def initialize(data_key, **options)
+        @data_key = data_key
+        super(**options)
+      end
+
+      def to_config
+        {
+          type: "pie",
+          data_key: data_key,
+          label_key: options[:label_key],
+          donut: options.fetch(:donut, false),
+          inner_radius: options[:inner_radius],
+          pad_angle: options[:pad_angle] || 0.02
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/radar.rb

@@ -0,0 +1,24 @@
+module Trackplot
+  module Components
+    class Radar < Base
+      attr_reader :data_key
+
+      def initialize(data_key, **options)
+        @data_key = data_key
+        super(**options)
+      end
+
+      def to_config
+        {
+          type: "radar",
+          data_key: data_key,
+          color: options[:color],
+          opacity: options[:opacity] || 0.15,
+          stroke_width: options[:stroke_width] || 2,
+          dot: options.fetch(:dot, true),
+          dot_size: options[:dot_size] || 4
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/reference_line.rb

@@ -0,0 +1,24 @@
+module Trackplot
+  module Components
+    class ReferenceLine < Base
+      def initialize(**options)
+        super
+      end
+
+      def to_config
+        direction = options[:y] ? "y" : "x"
+        value = options[:y] || options[:x]
+
+        {
+          type: "reference_line",
+          direction: direction,
+          value: value,
+          label: options[:label],
+          color: options[:color] || "#ef4444",
+          dashed: options.fetch(:dashed, true),
+          stroke_width: options[:stroke_width] || 1.5
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/scatter.rb

@@ -0,0 +1,24 @@
+module Trackplot
+  module Components
+    class Scatter < Base
+      attr_reader :data_key
+
+      def initialize(data_key, **options)
+        @data_key = data_key
+        super(**options)
+      end
+
+      def to_config
+        {
+          type: "scatter",
+          data_key: data_key,
+          x_key: options[:x_key],
+          color: options[:color],
+          dot_size: options[:dot_size] || 5,
+          opacity: options[:opacity] || 0.7,
+          shape: options[:shape] || "circle"
+        }.compact
+      end
+    end
+  end
+end

data/lib/trackplot/components/tooltip.rb

@@ -0,0 +1,29 @@
+module Trackplot
+  module Components
+    class Tooltip < Base
+      FORMAT_SYMBOLS = {
+        currency: "currency",
+        percent: "percent",
+        compact: "compact",
+        decimal: "decimal",
+        integer: "integer"
+      }.freeze
+
+      def to_config
+        {
+          type: "tooltip",
+          format: resolve_format(options[:format]),
+          label_format: options[:label_format]
+        }.compact
+      end
+
+      private
+
+      def resolve_format(fmt)
+        return nil if fmt.nil?
+
+        FORMAT_SYMBOLS.fetch(fmt, fmt).to_s
+      end
+    end
+  end
+end

data/lib/trackplot/data_adapter.rb

@@ -0,0 +1,33 @@
+module Trackplot
+  class DataAdapter
+    def self.normalize(data)
+      records = case data
+      when ->(d) { defined?(ActiveRecord::Relation) && d.is_a?(ActiveRecord::Relation) }
+        data.map { |record| to_string_hash(record.attributes) }
+      when Array
+        data.map { |item| coerce_record(item) }
+      when Hash
+        [data.transform_keys(&:to_s)]
+      else
+        Array(data).map { |item| coerce_record(item) }
+      end
+
+      records.map { |r| r.transform_keys(&:to_s) }
+    end
+
+    def self.coerce_record(item)
+      case item
+      when Hash
+        item
+      when ->(i) { i.respond_to?(:attributes) }
+        item.attributes
+      when ->(i) { i.respond_to?(:to_h) }
+        item.to_h
+      else
+        raise ArgumentError, "Trackplot: cannot convert #{item.class} to chart data. Expected Hash, ActiveRecord, or an object responding to #to_h."
+      end
+    end
+
+    private_class_method :coerce_record
+  end
+end

data/lib/trackplot/engine.rb

@@ -0,0 +1,22 @@
+module Trackplot
+  class Engine < ::Rails::Engine
+    isolate_namespace Trackplot
+
+    initializer "trackplot.helpers" do
+      ActiveSupport.on_load(:action_view) do
+        include Trackplot::ChartHelper
+      end
+    end
+
+    initializer "trackplot.assets" do |app|
+      app.config.assets.paths << root.join("app/assets/javascripts") if app.config.respond_to?(:assets)
+    end
+
+    initializer "trackplot.importmap", before: "importmap" do |app|
+      if app.config.respond_to?(:importmap)
+        app.config.importmap.paths << root.join("config/importmap.rb")
+        app.config.importmap.cache_sweepers << root.join("app/assets/javascripts")
+      end
+    end
+  end
+end

data/lib/trackplot/theme.rb

@@ -0,0 +1,59 @@
+module Trackplot
+  class Theme
+    PRESETS = {
+      default: {
+        colors: ["#6366f1", "#06b6d4", "#10b981", "#f59e0b", "#ef4444", "#8b5cf6", "#ec4899", "#14b8a6"],
+        background: "transparent",
+        text_color: "#374151",
+        axis_color: "#d1d5db",
+        grid_color: "#e5e7eb",
+        tooltip_bg: "rgba(255, 255, 255, 0.96)",
+        tooltip_text: "#111827",
+        tooltip_border: "#e5e7eb",
+        font: '-apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif'
+      },
+      dark: {
+        colors: ["#818cf8", "#22d3ee", "#34d399", "#fbbf24", "#f87171", "#a78bfa", "#f472b6", "#2dd4bf"],
+        background: "#1e1e2e",
+        text_color: "#e2e8f0",
+        axis_color: "#4a5568",
+        grid_color: "#2d3748",
+        tooltip_bg: "rgba(30, 30, 46, 0.96)",
+        tooltip_text: "#f1f5f9",
+        tooltip_border: "#4a5568",
+        font: '-apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif'
+      },
+      vibrant: {
+        colors: ["#ff6b6b", "#ffd93d", "#6bcb77", "#4d96ff", "#ff922b", "#cc5de8", "#20c997", "#f06595"],
+        background: "transparent",
+        text_color: "#2d3436",
+        axis_color: "#b2bec3",
+        grid_color: "#dfe6e9",
+        tooltip_bg: "rgba(255, 255, 255, 0.96)",
+        tooltip_text: "#2d3436",
+        tooltip_border: "#dfe6e9",
+        font: '-apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif'
+      },
+      minimal: {
+        colors: ["#475569", "#94a3b8", "#334155", "#64748b", "#1e293b", "#78716c", "#57534e", "#a8a29e"],
+        background: "transparent",
+        text_color: "#64748b",
+        axis_color: "#e2e8f0",
+        grid_color: "#f1f5f9",
+        tooltip_bg: "rgba(255, 255, 255, 0.96)",
+        tooltip_text: "#334155",
+        tooltip_border: "#e2e8f0",
+        font: '-apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif'
+      }
+    }.freeze
+
+    def self.resolve(option)
+      case option
+      when nil then PRESETS[:default]
+      when Symbol then PRESETS.fetch(option) { raise ArgumentError, "Unknown theme: #{option}. Available: #{PRESETS.keys.join(", ")}" }
+      when Hash then PRESETS[:default].merge(option.transform_keys(&:to_sym))
+      else raise ArgumentError, "Theme must be a Symbol, Hash, or nil"
+      end
+    end
+  end
+end

data/lib/trackplot/version.rb

@@ -0,0 +1,3 @@
+module Trackplot
+  VERSION = "0.1.0"
+end

data/lib/trackplot.rb

@@ -0,0 +1,26 @@
+require_relative "trackplot/version"
+require_relative "trackplot/engine" if defined?(Rails::Engine)
+
+module Trackplot
+  autoload :ChartBuilder, "trackplot/chart_builder"
+  autoload :DataAdapter, "trackplot/data_adapter"
+  autoload :Theme, "trackplot/theme"
+
+  module Components
+    autoload :Base, "trackplot/components/base"
+    autoload :Line, "trackplot/components/line"
+    autoload :Bar, "trackplot/components/bar"
+    autoload :Area, "trackplot/components/area"
+    autoload :Pie, "trackplot/components/pie"
+    autoload :Axis, "trackplot/components/axis"
+    autoload :Tooltip, "trackplot/components/tooltip"
+    autoload :Legend, "trackplot/components/legend"
+    autoload :Grid, "trackplot/components/grid"
+    autoload :Scatter, "trackplot/components/scatter"
+    autoload :Radar, "trackplot/components/radar"
+    autoload :HorizontalBar, "trackplot/components/horizontal_bar"
+    autoload :Candlestick, "trackplot/components/candlestick"
+    autoload :Funnel, "trackplot/components/funnel"
+    autoload :ReferenceLine, "trackplot/components/reference_line"
+  end
+end