require 'yard'

class ClassLineHandler < YARD::Handlers::Ruby::Base
  handles :class

  def process
    obj = ClassObject.new(namespace, statement[0].source)
    obj.line_end = statement[0].line + statement.source.split("\n").size - 1
  end
end

class ModuleLineHandler < YARD::Handlers::Ruby::Base
  handles :module

  def process
    obj = register ModuleObject.new(namespace, statement[0].source)
    obj.line_end = statement[0].line + statement.source.split("\n").size - 1
  end
end

class LegacyClassLineHandler < YARD::Handlers::Ruby::Legacy::Base
  handles TkCLASS

  def process
    classname = statement.tokens.to_s[/^class\s+(#{NAMESPACEMATCH})/, 1]
    obj = ClassObject.new(namespace, classname)
    obj.line_end = statement.line + statement.to_s(true).split("\n").size - 1
  end
end

class LegacyModuleLineHandler < YARD::Handlers::Ruby::Legacy::Base
  handles TkMODULE

  def process
    modname = statement.tokens.to_s[/^module\s+(#{NAMESPACEMATCH})/, 1]
    obj = register ModuleObject.new(namespace, modname)
    obj.line_end = statement.line + statement.to_s(true).split("\n").size - 1
  end
end

YARD::Parser::SourceParser.parser_type = :ruby18
YARD::Parser::SourceParser.parse_string <<-eof
  module M
    class A
      def a; end
    end
  end
eof

obj = P("M::A")
p obj.line..obj.line_end

obj = P("M")
p obj.line..obj.line_end