# frozen_string_literal: true

module RubyJard
  class Screens
    ##
    # Display source code of current stopping line and surrounding lines
    class SourceScreen < RubyJard::Screen
      def initialize(*args)
        super
        @frame_file = @session.current_frame&.frame_file
        @frame_line = @session.current_frame&.frame_line

        if !@frame_file.nil? && !@frame_line.nil?
          @path_decorator = RubyJard::Decorators::PathDecorator.new
          @loc_decorator = RubyJard::Decorators::LocDecorator.new
          @source_decorator = RubyJard::Decorators::SourceDecorator.new(@frame_file, @frame_line, @layout.height)
        end

        @selected = 0
      end

      def title
        return 'Source' if @frame_file.nil? || @frame_line.nil?

        _, path_label = @path_decorator.decorate(@frame_file, @frame_line)
        ['Source', path_label]
      end

      def build
        return 'Source' if @frame_file.nil? || @frame_line.nil?

        # TODO: screen now supports window.
        codes = @source_decorator.codes
        @rows = codes.map.with_index do |loc, index|
          lineno = @source_decorator.window_start + index
          RubyJard::Row.new(
            line_limit: 3,
            columns: [
              RubyJard::Column.new(
                spans: [
                  span_mark(lineno),
                  span_lineno(lineno)
                ]
              ),
              RubyJard::Column.new(
                word_wrap: RubyJard::Column::WORD_WRAP_BREAK_WORD,
                spans: loc_spans(loc)
              )
            ]
          )
        end
      end

      private

      def handle_anonymous_evaluation
        @rows = [
          RubyJard::Row.new(
            line_limit: 3,
            columns: [
              RubyJard::Column.new(
                spans: [
                  RubyJard::Span.new(
                    content: 'This section is anonymous!',
                    styles: :normal_token
                  )
                ]
              )
            ]
          ),
          RubyJard::Row.new(
            line_limit: 3,
            columns: [
              RubyJard::Column.new(
                spans: [
                  RubyJard::Span.new(
                    content: 'Maybe it is dynamically evaluated, or called via ruby-e, without file information.',
                    styles: :source_lineno
                  )
                ]
              )
            ]
          )
        ]
      end

      def span_mark(lineno)
        RubyJard::Span.new(
          margin_right: 1,
          content: @frame_line == lineno ? '⮕' : ' ',
          styles: :text_selected
        )
      end

      def span_lineno(lineno)
        padded_lineno = lineno.to_s.rjust(@source_decorator.window_end.to_s.length)
        RubyJard::Span.new(
          content: padded_lineno,
          styles: @frame_line == lineno ? :text_selected : :text_dim
        )
      end

      def loc_spans(loc)
        return [] if @frame_file.nil?

        spans, _tokens = @loc_decorator.decorate(loc, @frame_file)
        spans
      end
    end
  end
end

RubyJard::Screens.add_screen('source', RubyJard::Screens::SourceScreen)