styd/apexcharts.rb

# frozen_string_literal: true

module ApexCharts::Charts::Features
  module Annotations
    def annotation(axis, value:, text:, color: nil, **options)
      @annotations ||= {}

      raise "unrecognized axis: #{axis}" unless %i[xaxis yaxis points].include? axis

      @annotations[axis] ||= []
      @annotations[axis] << annotation_value(axis, value).merge(
        if axis == :points
          {marker: {size: 8, fillColor: 'white', strokeColor: color, radius: 2}, **options}
        else
          {borderColor: color, fillColor: color, opacity: 0.2}
        end
      ).merge(
        annotation_label(text, color, **options)
      )
    end

    def x_annotation(**args)
      annotation :xaxis, **args
    end

    def y_annotation(**args)
      annotation :yaxis, **args
    end

    def point_annotation(**args)
      annotation :points, **args
    end

  private

    def annotation_value(axis, value)
      axis = axis.to_s.delete_suffix('axis').to_sym
      case value
      when Range
        value = ApexCharts::Utils::DateTime.convert_range(value)

        case axis
        when :x
          {x: value.first, x2: value.last}
        when :y
          {y: value.first, y2: value.last}
        else
          raise "axis :points doesn't accept value of type Range"
        end
      else
        if axis == :points
          value.map! {|x| ApexCharts::Utils::DateTime.convert(x) }
          {x: value.first, y: value.last}
        else
          value = ApexCharts::Utils::DateTime.convert(value)
          {axis => value}
        end
      end
    end

    def annotation_label(text, color, **options)
      {
        label: {
          borderColor: color,
          style: {
            color: 'white',
            background: color
          },
          text: text
        }, **options
      }
    end

    include ApexCharts::Utils::Hash
  end
end