require_relative '../helper'
require "fixtures/show_source_doc_examples"

describe "show-source" do
  before do
    @o = Object.new
    def @o.sample_method
      :sample
    end

    Object.remove_const :Test if Object.const_defined? :Test
    Object.const_set(:Test, Module.new)
  end

  after do
    Pad.clear
  end

  it "should output a method's source" do
    expect(pry_eval(binding, 'show-source @o.sample_method')).to match(/def @o.sample/)
  end

  it "should output help" do
    expect(pry_eval('show-source -h')).to match(/Usage:\s+show-source/)
  end

  it "should output a method's source with line numbers" do
    expect(pry_eval(binding, 'show-source -l @o.sample_method')).to match(/\d+: def @o.sample/)
  end

  it "should output a method's source with line numbers starting at 1" do
    expect(pry_eval(binding, 'show-source -b @o.sample_method')).to match(/1: def @o.sample/)
  end

  it "should output a method's source if inside method and no name given" do
    def @o.sample
      pry_eval(binding, 'show-source').should =~ /def @o.sample/
    end
    @o.sample
  end

  it "should output a method's source inside method using the -l switch" do
    def @o.sample
      pry_eval(binding, 'show-source -l').should =~ /def @o.sample/
    end
    @o.sample
  end

  it "should find methods even if there are spaces in the arguments" do
    def @o.foo(*_bars)
      @foo = "Mr flibble"
      self
    end

    out = pry_eval(binding, "show-source @o.foo('bar', 'baz bam').foo")
    expect(out).to match(/Mr flibble/)
  end

  it "should find methods even if the object overrides method method" do
    _c = Class.new{
      def method;
        98
      end
    }

    expect(pry_eval(binding, "show-source _c.new.method")).to match(/98/)
  end

  it "should not show the source when a non-extant method is requested" do
    _c = Class.new{ def method; 98; end }
    expect(mock_pry(binding, "show-source _c#wrongmethod")).to match(/Couldn't locate/)
  end

  it "should not show the source and deliver an error message without exclamation point" do
    _c = Class.new
    error_message = "Error: Couldn't locate a definition for _c#wrongmethod\n"
    expect(mock_pry(binding, "show-source _c#wrongmethod")).to eq(error_message)
  end


  it "should find instance_methods if the class overrides instance_method" do
    _c = Class.new{
      def method;
        98
      end

      def self.instance_method; 789; end
    }

    expect(pry_eval(binding, "show-source _c#method")).to match(/98/)
  end

  it "should find instance methods with self#moo" do
    _c = Class.new{ def moo; "ve over!"; end }

    expect(pry_eval(binding, "cd _c", "show-source self#moo")).to match(/ve over/)
  end

  it "should not find instance methods with self.moo" do
    _c = Class.new{ def moo; "ve over!"; end }

    expect { pry_eval(binding, 'cd _c', 'show-source self.moo') }.to raise_error(Pry::CommandError, /Couldn't locate/)
  end

  it "should find normal methods with self.moo" do
    _c = Class.new{ def self.moo; "ve over!"; end }

    expect(pry_eval(binding, 'cd _c', 'show-source self.moo')).to match(/ve over/)
  end

  it "should not find normal methods with self#moo" do
    _c = Class.new{ def self.moo; "ve over!"; end }

    expect { pry_eval(binding, 'cd _c', 'show-source self#moo') }.to raise_error(Pry::CommandError, /Couldn't locate/)
  end

  it "should find normal methods (i.e non-instance methods) by default" do
    _c = Class.new{ def self.moo; "ve over!"; end }

    expect(pry_eval(binding, "cd _c", "show-source moo")).to match(/ve over/)
  end

  it "should find instance methods if no normal methods available" do
    _c = Class.new{ def moo; "ve over!"; end }

    expect(pry_eval(binding, "cd _c", "show-source moo")).to match(/ve over/)
  end

  describe "with -e option" do
    before do
      class FooBar
        def bar
          :bar
        end
      end
    end

    after do
      Object.remove_const(:FooBar)
    end

    it "evaluates the argument as ruby and shows the source code for the returned value" do
      ReplTester.start target: binding do
        input 'show-source -e FooBar.new'
        output(/class FooBar/)
      end
    end
  end

  it "should raise a CommandError when super method doesn't exist" do
    def @o.foo(*bars); end

    expect { pry_eval(binding, "show-source --super @o.foo") }.to raise_error(Pry::CommandError, /No superclass found/)
  end

  it "should output the source of a method defined inside Pry" do
    out = pry_eval("def dyn_method\n:test\nend", 'show-source dyn_method')
    expect(out).to match(/def dyn_method/)
    Object.remove_method :dyn_method
  end

  it 'should output source for an instance method defined inside pry' do
    pry_tester.tap do |t|
      t.eval "class Test::A\n  def yo\n  end\nend"
      expect(t.eval('show-source Test::A#yo')).to match(/def yo/)
    end
  end

  it 'should output source for a repl method defined using define_method' do
    pry_tester.tap do |t|
      t.eval "class Test::A\n  define_method(:yup) {}\nend"
      expect(t.eval('show-source Test::A#yup')).to match(/define_method\(:yup\)/)
    end
  end

  it "should output the source of a command defined inside Pry" do
    command_definition = %{
      Pry.config.commands.command "hubba-hubba" do
        puts "that's what she said!"
      end
    }
    out = pry_eval(command_definition, 'show-source hubba-hubba')
    expect(out).to match(/what she said/)
    Pry.config.commands.delete "hubba-hubba"
  end

  context "when there's no source code but the comment exists" do
    before do
      class Foo
        # Bingo.
        def bar; end
      end

      allow_any_instance_of(Pry::Method).to receive(:source).and_return(nil)
    end

    after do
      Object.remove_const(:Foo)
    end

    it "outputs zero line numbers" do
      out = pry_eval('show-source Foo#bar')
      expect(out).to match(/
        Owner:\sFoo
        .+
        Number\sof\slines:\s0
        .+
        \*\*\sWarning:\sCannot\sfind\scode\sfor\s'bar'\s\(source_location\sis\snil\)
      /mx)
    end
  end

  describe "finding super methods with help of `--super` switch" do
    before do
      class Foo
        def foo(*_bars)
          :super_wibble
        end
      end
    end

    after do
      Object.remove_const(:Foo)
    end

    it "finds super methods with explicit method argument" do

      o = Foo.new
      def o.foo(*_bars)
        :wibble
      end

      expect(pry_eval(binding, "show-source --super o.foo")).to match(/:super_wibble/)
    end

    it "finds super methods without explicit method argument" do
      o = Foo.new
      def o.foo(*bars)
        @foo = :wibble
        pry_eval(binding, 'show-source --super')
      end

      expect(o.foo).to match(/:super_wibble/)
    end

    it "finds super methods with multiple --super " do
      o = Foo.new

      o.extend(
        Module.new do
          def foo
            :nibble
          end
        end
      )

      def o.foo(*bars)
        @foo = :wibble
        pry_eval(binding, 'show-source --super --super')
      end

      expect(o.foo).to match(/:super_wibble/)
    end
  end

  describe "on sourcable objects" do
    it "should output source defined inside pry" do
      pry_tester.tap do |t|
        t.eval "hello = proc { puts 'hello world!' }"
        expect(t.eval("show-source hello")).to match(/proc \{ puts/)
      end
    end

    it "should output source for procs/lambdas stored in variables" do
      _hello = proc { puts 'hello world!' }
      expect(pry_eval(binding, 'show-source _hello')).to match(/proc \{ puts/)
    end

    it "should output source for procs/lambdas stored in constants" do
      HELLO = proc { puts 'hello world!' }
      expect(pry_eval(binding, "show-source HELLO")).to match(/proc \{ puts/)
      Object.remove_const(:HELLO)
    end

    it "should output source for method objects" do
      def @o.hi; puts 'hi world'; end
      _meth = @o.method(:hi)
      expect(pry_eval(binding, "show-source _meth")).to match(/puts 'hi world'/)
    end

    describe "on variables that shadow methods" do
      before do
        @t = pry_tester.eval unindent(<<-EOS)
          class ::TestHost
            def hello
              hello = proc { ' smile ' }
              _foo = hello
              pry_tester(binding)
            end
          end
          ::TestHost.new.hello
          EOS
        end

        after do
          Object.remove_const(:TestHost)
        end

        it "source of variable should take precedence over method that is being shadowed" do
          source = @t.eval('show-source hello')
          expect(source).not_to match(/def hello/)
          expect(source).to match(/proc \{ ' smile ' \}/)
        end

        it "source of method being shadowed should take precedence over variable
          if given self.meth_name syntax" do
          expect(@t.eval('show-source self.hello')).to match(/def hello/)
        end
      end
    end

    describe "on variable or constant" do
      before do
        class TestHost
          def hello
            "hi there"
          end
        end
      end

      after do
        Object.remove_const(:TestHost)
      end

      it "should output source of its class if variable doesn't respond to source_location" do
        _test_host = TestHost.new
        expect(pry_eval(binding, 'show-source _test_host')).
          to match(/class TestHost\n.*def hello/)
      end

      it "should output source of its class if constant doesn't respond to source_location" do
        TEST_HOST = TestHost.new
        expect(pry_eval(binding, 'show-source TEST_HOST')).
          to match(/class TestHost\n.*def hello/)
        Object.remove_const(:TEST_HOST)
      end
    end

    describe "on modules" do
      before do
        class ShowSourceTestSuperClass
          def alpha
          end
        end

        class ShowSourceTestClass<ShowSourceTestSuperClass
          def alpha
          end
        end

        module ShowSourceTestSuperModule
          def alpha
          end
        end

        module ShowSourceTestModule
          include ShowSourceTestSuperModule
          def alpha
          end
        end

        ShowSourceTestClassWeirdSyntax = Class.new do
          def beta
          end
        end

        ShowSourceTestModuleWeirdSyntax = Module.new do
          def beta
          end
        end
      end

      after do
        Object.remove_const :ShowSourceTestSuperClass
        Object.remove_const :ShowSourceTestClass
        Object.remove_const :ShowSourceTestClassWeirdSyntax
        Object.remove_const :ShowSourceTestSuperModule
        Object.remove_const :ShowSourceTestModule
        Object.remove_const :ShowSourceTestModuleWeirdSyntax
      end

      describe "basic functionality, should find top-level module definitions" do
        it 'should show source for a class' do
          expect(pry_eval('show-source ShowSourceTestClass')).
            to match(/class ShowSourceTestClass.*?def alpha/m)
        end

        it 'should show source for a super class' do
          expect(pry_eval('show-source -s ShowSourceTestClass')).
            to match(/class ShowSourceTestSuperClass.*?def alpha/m)
        end

        it 'should show source for a module' do
          expect(pry_eval('show-source ShowSourceTestModule')).
            to match(/module ShowSourceTestModule/)
        end

        it 'should show source for an ancestor module' do
          expect(pry_eval('show-source -s ShowSourceTestModule')).
            to match(/module ShowSourceTestSuperModule/)
        end

        it 'should show source for a class when Const = Class.new syntax is used' do
          expect(pry_eval('show-source ShowSourceTestClassWeirdSyntax')).
            to match(/ShowSourceTestClassWeirdSyntax = Class.new/)
        end

        it 'should show source for a super class when Const = Class.new syntax is used' do
          expect(pry_eval('show-source -s ShowSourceTestClassWeirdSyntax')).
            to match(/class Object/)
        end

        it 'should show source for a module when Const = Module.new syntax is used' do
          expect(pry_eval('show-source ShowSourceTestModuleWeirdSyntax')).
            to match(/ShowSourceTestModuleWeirdSyntax = Module.new/)
        end
      end

      before do
        pry_eval unindent(<<-EOS)
        class Dog
          def woof
          end
        end

        class TobinaMyDog < Dog
          def woof
          end
        end
        EOS
      end

      after do
        Object.remove_const :Dog
        Object.remove_const :TobinaMyDog
      end

      describe "in REPL" do
        it 'should find class defined in repl' do
          expect(pry_eval('show-source TobinaMyDog')).to match(/class TobinaMyDog/)
        end

        it 'should find superclass defined in repl' do
          expect(pry_eval('show-source -s TobinaMyDog')).to match(/class Dog/)
        end
      end

      it 'should lookup module name with respect to current context' do

        temporary_constants(:AlphaClass, :BetaClass) do
          class BetaClass
            def alpha
            end
          end

          class AlphaClass
            class BetaClass
              def beta
              end
            end
          end

          expect(pry_eval(AlphaClass, 'show-source BetaClass')).to match(/def beta/)
        end
      end

      it 'should lookup nested modules' do
        temporary_constants(:AlphaClass) do
          class AlphaClass
            class BetaClass
              def beta
              end
            end
          end

          expect(pry_eval('show-source AlphaClass::BetaClass')).to match(/class Beta/)
        end
      end

      # note that pry assumes a class is only monkey-patched at most
      # ONCE per file, so will not find multiple monkeypatches in the
      # SAME file.
      describe "show-source -a" do
        it 'should show the source for all monkeypatches defined in different files' do
          class TestClassForShowSource
            def beta
            end
          end

          result = pry_eval('show-source TestClassForShowSource -a')
          expect(result).to match(/def alpha/)
          expect(result).to match(/def beta/)
        end

        it 'should show the source for a class_eval-based monkeypatch' do
          TestClassForShowSourceClassEval.class_eval do
            def class_eval_method
            end
          end

          result = pry_eval('show-source TestClassForShowSourceClassEval -a')
          expect(result).to match(/def class_eval_method/)
        end

        it 'should ignore -a when object is not a module' do
          TestClassForShowSourceClassEval.class_eval do
            def class_eval_method
              :bing
            end
          end

          result = pry_eval('show-source TestClassForShowSourceClassEval#class_eval_method -a')
          expect(result).to match(/bing/)
        end

        it 'should show the source for an instance_eval-based monkeypatch' do
          TestClassForShowSourceInstanceEval.instance_eval do
            def instance_eval_method
            end
          end

          result = pry_eval('show-source TestClassForShowSourceInstanceEval -a')
          expect(result).to match(/def instance_eval_method/)
        end

        describe "messages relating to -a" do
          it 'indicates all available monkeypatches can be shown with -a when (when -a not used and more than one candidate exists for class)' do
            class TestClassForShowSource
              def gamma
              end
            end

            result = pry_eval('show-source TestClassForShowSource')
            expect(result).to match(/available monkeypatches/)
          end

          it 'shouldnt say anything about monkeypatches when only one candidate exists for selected class' do
            class Aarrrrrghh
              def o;end
            end

            result = pry_eval('show-source Aarrrrrghh')
            expect(result).not_to match(/available monkeypatches/)
            Object.remove_const(:Aarrrrrghh)
          end
        end
      end

      describe "when show-source is invoked without a method or class argument" do
        before do
          module TestHost
            class M
              def alpha; end

              def beta; end
            end

            module C
            end

            module D
              def self.invoked_in_method
                pry_eval(binding, 'show-source')
              end
            end
          end
        end

        after do
          Object.remove_const(:TestHost)
        end

        describe "inside a module" do
          it 'should display module source by default' do
            out = pry_eval(TestHost::M, 'show-source')
            expect(out).to match(/class M/)
            expect(out).to match(/def alpha/)
            expect(out).to match(/def beta/)
          end

          it 'should be unable to find module source if no methods defined' do
            expect { pry_eval(TestHost::C, 'show-source') }.to raise_error(Pry::CommandError, /Couldn't locate/)
          end

          it 'should display method code (rather than class) if Pry started inside method binding' do
            out = TestHost::D.invoked_in_method
            expect(out).to match(/invoked_in_method/)
            expect(out).not_to match(/module D/)
          end

          it 'should display class source when inside instance' do
            out = pry_eval(TestHost::M.new, 'show-source')
            expect(out).to match(/class M/)
            expect(out).to match(/def alpha/)
            expect(out).to match(/def beta/)
          end

          it 'should allow options to be passed' do
            out = pry_eval(TestHost::M, 'show-source -b')
            expect(out).to match(/\d:\s*class M/)
            expect(out).to match(/\d:\s*def alpha/)
            expect(out).to match(/\d:\s*def beta/)
          end

          describe "should skip over broken modules" do
            before do
              module BabyDuck

                module Muesli
                  binding.eval("def a; end", "dummy.rb", 1)
                  binding.eval("def b; end", "dummy.rb", 2)
                  binding.eval("def c; end", "dummy.rb", 3)
                end

                module Muesli
                  def d; end

                  def e; end
                end
              end
            end

            after do
              Object.remove_const(:BabyDuck)
            end

            it 'should return source for first valid module' do
              out = pry_eval('show-source BabyDuck::Muesli')
              expect(out).to match(/def d; end/)
              expect(out).not_to match(/def a; end/)
            end
          end

          describe "monkey-patched C modules" do
            # Monkey-patch Array and add 15 methods, so its internal rank is
            # high enough to make this definition primary.
            class Array
              15.times do |i|
                define_method(:"doge#{i}") do
                  :"doge#{i}"
                end
              end
            end

            describe "when current context is a C object" do
              it "should display a warning, and not monkey-patched definition" do
                out = pry_eval([1, 2, 3], 'show-source')
                expect(out).not_to match(/doge/)
                expect(out).to match(/warning/i)
              end

              it "recommends to use the --all switch when other candidates are found" do
                out = pry_eval([], 'show-source')
                expect(out).to match(/'--all' switch/i)
              end
            end

            describe "when current context is something other than a C object" do
              it "should display a candidate, not a warning" do
                out = pry_eval('show-source Array')
                expect(out).to match(/doge/)
                expect(out).not_to match(/warning/i)
              end
            end
          end
        end
      end
    end

    describe "on commands" do
      before do
        @oldset = Pry.config.commands
        @set = Pry.config.commands = Pry::CommandSet.new do
          import Pry::Commands
        end
      end

      after do
        Pry.config.commands = @oldset
      end

      describe "block commands" do
        it 'should show source for an ordinary command' do
          @set.command "foo", :body_of_foo do; end

          expect(pry_eval('show-source foo')).to match(/:body_of_foo/)
        end

        it "should output source of commands using special characters" do
          @set.command "!%$", "I gots the yellow fever" do; end

          expect(pry_eval('show-source !%$')).to match(/yellow fever/)
        end

        it 'should show source for a command with spaces in its name' do
          @set.command "foo bar", :body_of_foo_bar do; end

          expect(pry_eval('show-source foo bar')).to match(/:body_of_foo_bar/)
        end

        it 'should show source for a command by listing name' do
          @set.command(/foo(.*)/, :body_of_foo_bar_regex, listing: "bar") do; end

          expect(pry_eval('show-source bar')).to match(/:body_of_foo_bar_regex/)
        end
      end

      describe "create_command commands" do
        it 'should show source for a command' do
          @set.create_command "foo", "babble" do
            def process() :body_of_foo end
          end
          expect(pry_eval('show-source foo')).to match(/:body_of_foo/)
        end

        it 'should show source for a command defined inside pry' do
          pry_eval %{
          _pry_.commands.create_command "foo", "babble" do
            def process() :body_of_foo end
          end
        }
        expect(pry_eval('show-source foo')).to match(/:body_of_foo/)
      end
    end

    describe "real class-based commands" do
      before do
        class ::TemporaryCommand < Pry::ClassCommand
          match 'temp-command'
          def process() :body_of_temp end
        end

        Pry.config.commands.add_command(::TemporaryCommand)
      end

      after do
        Object.remove_const(:TemporaryCommand)
      end

      it 'should show source for a command' do
        expect(pry_eval('show-source temp-command')).to match(/:body_of_temp/)
      end

      it 'should show source for a command defined inside pry' do
        pry_eval %{
          class ::TemporaryCommandInPry < Pry::ClassCommand
            match 'temp-command-in-pry'
            def process() :body_of_temp end
          end
        }
        Pry.config.commands.add_command(::TemporaryCommandInPry)
        expect(pry_eval('show-source temp-command-in-pry')).to match(/:body_of_temp/)
        Object.remove_const(:TemporaryCommandInPry)
      end
    end
  end

  describe "should set _file_ and _dir_" do
    it 'should set _file_ and _dir_ to file containing method source' do
      t = pry_tester
      t.process_command "show-source TestClassForShowSource#alpha"
      expect(t.pry.last_file).to match(/show_source_doc_examples/)
      expect(t.pry.last_dir).to match(/fixtures/)
    end
  end

  describe "can't find class/module code" do
    describe "for classes" do
      before do
        module Jesus
          module Pig
            def lillybing; :lillybing; end
          end

          class Brian; end
          class Jingle
            def a; :doink; end
          end

          class Jangle < Jingle; include Pig; end
          class Bangle < Jangle; end
        end
      end

      after do
        Object.remove_const(:Jesus)
      end

      it 'shows superclass code' do
        t = pry_tester
        t.process_command "show-source Jesus::Jangle"
        expect(t.last_output).to match(/doink/)
      end

      it 'ignores included modules' do
        t = pry_tester
        t.process_command "show-source Jesus::Jangle"
        expect(t.last_output).not_to match(/lillybing/)
      end

      it 'errors when class has no superclass to show' do
        t = pry_tester
        expect { t.process_command "show-source Jesus::Brian" }.to raise_error(Pry::CommandError, /Couldn't locate/)
      end

      it 'shows warning when reverting to superclass code' do
        t = pry_tester
        t.process_command "show-source Jesus::Jangle"
        expect(t.last_output).to match(/Warning.*?Cannot find.*?Jesus::Jangle.*Showing.*Jesus::Jingle instead/)
      end

      it 'shows nth level superclass code (when no intermediary superclasses have code either)' do
        t = pry_tester
        t.process_command "show-source Jesus::Bangle"
        expect(t.last_output).to match(/doink/)
      end

      it 'shows correct warning when reverting to nth level superclass' do
        t = pry_tester
        t.process_command "show-source Jesus::Bangle"
        expect(t.last_output).to match(/Warning.*?Cannot find.*?Jesus::Bangle.*Showing.*Jesus::Jingle instead/)
      end
    end

    describe "for modules" do
      before do
        module Jesus
          module Alpha
            def alpha; :alpha; end
          end

          module Zeta; end

          module Beta
            include Alpha
          end

          module Gamma
            include Beta
          end
        end
      end

      after do
        Object.remove_const(:Jesus)
      end

      it 'shows included module code' do
        t = pry_tester
        t.process_command "show-source Jesus::Beta"
        expect(t.last_output).to match(/alpha/)
      end

      it 'shows warning when reverting to included module code' do
        t = pry_tester
        t.process_command "show-source Jesus::Beta"
        expect(t.last_output).to match(/Warning.*?Cannot find.*?Jesus::Beta.*Showing.*Jesus::Alpha instead/)
      end

      it 'errors when module has no included module to show' do
        t = pry_tester
        expect { t.process_command "show-source Jesus::Zeta" }.to raise_error(Pry::CommandError, /Couldn't locate/)
      end

      it 'shows nth level included module code (when no intermediary modules have code either)' do
        t = pry_tester
        t.process_command "show-source Jesus::Gamma"
        expect(t.last_output).to match(/alpha/)
      end

      it 'shows correct warning when reverting to nth level included module' do
        t = pry_tester
        t.process_command "show-source Jesus::Gamma"
        expect(t.last_output).to match(/Warning.*?Cannot find.*?Jesus::Gamma.*Showing.*Jesus::Alpha instead/)
      end
    end
  end
end