# encoding: utf-8
#--
#   Copyright (C) 2012 Gitorious AS
#
#   This program is free software: you can redistribute it and/or modify
#   it under the terms of the GNU Affero General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU Affero General Public License for more details.
#
#   You should have received a copy of the GNU Affero General Public License
#   along with this program.  If not, see <http://www.gnu.org/licenses/>.
#++
require "test_helper"
require "dolt/sinatra/actions"

class DummySinatraApp
  include Dolt::Sinatra::Actions
  attr_reader :actions, :renderer

  def initialize(actions, renderer)
    @actions = actions
    @renderer = renderer
  end

  def body(str = nil)
    @body = str if !str.nil?
    @body
  end

  def response
    if !@response
      @response = {}
      def @response.status; @status; end
      def @response.status=(status); @status = status; end
    end
    @response
  end

  def tree_url(repo, ref, path)
    "/#{repo}/tree/#{ref}:#{path}"
  end

  def blob_url(repo, ref, path)
    "/#{repo}/blob/#{ref}:#{path}"
  end

  def tree_entry_url(repo, ref, path)
    "/#{repo}/source/#{ref}:#{path}"
  end

  def blame_url(repo, ref, path)
    "/#{repo}/blame/#{ref}:#{path}"
  end

  def history_url(repo, ref, path)
    "/#{repo}/history/#{ref}:#{path}"
  end

  def tree_history_url(repo, ref, path)
    "/#{repo}/tree_history/#{ref}:#{path}"
  end

  def raw_url(repo, ref, path)
    "/#{repo}/raw/#{ref}:#{path}"
  end
end

class RedirectingDummySinatraApp < DummySinatraApp
  def redirect_refs?; true; end
end

class Renderer
  def initialize(body = ""); @body = body; end

  def render(action, data, options = {})
    @action = action
    @data = data
    "#{action}:#@body"
  end
end

class BlobStub
  def is_a?(type)
    type == Rugged::Blob
  end
end

class TreeStub
  def is_a?(type)
    type == Rugged::Tree
  end
end

class Actions
  attr_reader :repo, :ref, :path

  def initialize(response)
    @response = response
  end

  def blob(repo, ref, path, &block)
    respond(:blob, repo, ref, path, &block)
  end

  def tree(repo, ref, path, &block)
    respond(:tree, repo, ref, path, &block)
  end

  def tree_entry(repo, ref, path, &block)
    respond(:tree_entry, repo, ref, path, &block)
  end

  def raw(repo, ref, path, &block)
    respond(:raw, repo, ref, path, &block)
  end

  def blame(repo, ref, path, &block)
    respond(:blame, repo, ref, path, &block)
  end

  def history(repo, ref, path, limit, &block)
    respond(:history, repo, ref, path, &block)
  end

  def refs(repo, &block)
    respond(:refs, repo, nil, nil, &block)
  end

  def tree_history(repo, ref, path, count, &block)
    respond(:tree_history, repo, ref, path, &block)
  end

  def respond(type, repo, ref, path, &block)
    @repo = repo
    @ref = ref
    @path = path
    data = { :ref => ref, :repository => repo }
    data[type != :tree_entry ? type : (@response.class.to_s =~ /Tree/ ? :tree : :blob)] = @response
    block.call(nil, data)
  end

  def rev_parse_oid(repo, ref)
    "a" * 40
  end
end

describe Dolt::Sinatra::Actions do
  describe "#blob" do
    it "delegates to actions" do
      actions = Actions.new(BlobStub.new)
      app = DummySinatraApp.new(actions, Renderer.new)
      app.blob("gitorious", "master", "app/models/repository.rb")

      assert_equal "gitorious", actions.repo
      assert_equal "master", actions.ref
      assert_equal "app/models/repository.rb", actions.path
    end

    it "renders the blob template as html" do
      app = DummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Blob"))
      app.blob("gitorious", "master", "app/models/repository.rb")

      assert_equal "text/html; charset=utf-8", app.response["Content-Type"]
      assert_equal "blob:Blob", app.body
    end

    it "redirects tree views to tree action" do
      app = DummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.blob("gitorious", "master", "app/models")

      assert_equal 302, app.response.status
      assert_equal "/gitorious/tree/master:app/models", app.response["Location"]
      assert_equal "", app.body
    end

    it "unescapes ref" do
      actions = Actions.new(BlobStub.new)
      app = DummySinatraApp.new(actions, Renderer.new("Blob"))
      app.blob("gitorious", "issue-%23221", "app/my documents")

      assert_equal "issue-#221", actions.ref
    end

    it "does not redirect ref to oid by default" do
      app = DummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Blob"))
      app.blob("gitorious", "master", "lib/gitorious.rb")

      location = app.response["Location"]
      refute_equal 302, app.response.status
    end

    it "redirects ref to oid if configured so" do
      app = RedirectingDummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Blob"))
      app.blob("gitorious", "master", "lib/gitorious.rb")

      location = app.response["Location"]
      assert_equal 302, app.response.status
      assert_equal "/gitorious/blob/#{'a' * 40}:lib/gitorious.rb", location
      assert_equal "", app.body
    end
  end

  describe "#tree" do
    it "delegates to actions" do
      actions = Actions.new(TreeStub.new)
      app = DummySinatraApp.new(actions, Renderer.new)
      app.tree("gitorious", "master", "app/models")

      assert_equal "gitorious", actions.repo
      assert_equal "master", actions.ref
      assert_equal "app/models", actions.path
    end

    it "renders the tree template as html" do
      app = DummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.tree("gitorious", "master", "app/models")

      assert_equal "text/html; charset=utf-8", app.response["Content-Type"]
      assert_equal "tree:Tree", app.body
    end

    it "redirects blob views to blob action" do
      app = DummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Tree"))
      app.tree("gitorious", "master", "app/models/repository.rb")

      location = app.response["Location"]
      assert_equal 302, app.response.status
      assert_equal "/gitorious/blob/master:app/models/repository.rb", location
      assert_equal "", app.body
    end

    it "sets X-UA-Compatible header" do
      app = DummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.tree("gitorious", "master", "app/models")

      assert_equal "IE=edge", app.response["X-UA-Compatible"]
    end

    it "does not set cache-control header for head ref" do
      app = DummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.tree("gitorious", "master", "app/models")

      assert !app.response.key?("Cache-Control")
    end

    it "sets cache headers for full oid ref" do
      app = DummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.tree("gitorious", "a" * 40, "app/models")

      assert_equal "max-age=315360000, public", app.response["Cache-Control"]
      refute_nil app.response["Expires"]
    end

    it "unescapes ref" do
      actions = Actions.new(TreeStub.new)
      app = DummySinatraApp.new(actions, Renderer.new("Tree"))
      app.tree("gitorious", "issue-%23221", "app")

      assert_equal "issue-#221", actions.ref
    end

    it "redirects ref to oid if configured so" do
      app = RedirectingDummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.tree("gitorious", "master", "lib")

      assert_equal "/gitorious/tree/#{'a' * 40}:lib", app.response["Location"]
    end
  end

  describe "#tree_entry" do
    it "renders trees with the tree template as html" do
      app = DummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.tree_entry("gitorious", "master", "app/models")

      assert_equal "text/html; charset=utf-8", app.response["Content-Type"]
      assert_equal "tree:Tree", app.body
    end

    it "renders trees with the tree template as html" do
      app = DummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Blob"))
      app.tree_entry("gitorious", "master", "app/models")

      assert_equal "text/html; charset=utf-8", app.response["Content-Type"]
      assert_equal "blob:Blob", app.body
    end

    it "unescapes ref" do
      actions = Actions.new(TreeStub.new)
      app = DummySinatraApp.new(actions, Renderer.new("Tree"))
      app.tree_entry("gitorious", "issue-%23221", "app")

      assert_equal "issue-#221", actions.ref
    end

    it "redirects ref to oid if configured so" do
      app = RedirectingDummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.tree_entry("gitorious", "master", "lib")

      assert_equal "/gitorious/source/#{'a' * 40}:lib", app.response["Location"]
    end
  end

  describe "#raw" do
    it "delegates to actions" do
      actions = Actions.new(BlobStub.new)
      app = DummySinatraApp.new(actions, Renderer.new)
      app.raw("gitorious", "master", "app/models/repository.rb")

      assert_equal "gitorious", actions.repo
      assert_equal "master", actions.ref
      assert_equal "app/models/repository.rb", actions.path
    end

    it "renders the raw template as text" do
      app = DummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Text"))
      app.raw("gitorious", "master", "app/models/repository.rb")

      assert_equal "text/plain", app.response["Content-Type"]
      assert_equal "raw:Text", app.body
    end

    it "redirects tree views to tree action" do
      app = DummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.raw("gitorious", "master", "app/models")

      location = app.response["Location"]
      assert_equal 302, app.response.status
      assert_equal "/gitorious/tree/master:app/models", location
      assert_equal "", app.body
    end

    it "unescapes ref" do
      actions = Actions.new(BlobStub.new)
      app = DummySinatraApp.new(actions, Renderer.new("Blob"))
      app.raw("gitorious", "issue-%23221", "app/models/repository.rb")

      assert_equal "issue-#221", actions.ref
    end

    it "redirects ref to oid if configured so" do
      app = RedirectingDummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Blob"))
      app.raw("gitorious", "master", "lib/gitorious.rb")

      assert_equal "/gitorious/raw/#{'a' * 40}:lib/gitorious.rb", app.response["Location"]
    end
  end

  describe "#blame" do
    it "delegates to actions" do
      actions = Actions.new(BlobStub.new)
      app = DummySinatraApp.new(actions, Renderer.new)
      app.blame("gitorious", "master", "app/models/repository.rb")

      assert_equal "gitorious", actions.repo
      assert_equal "master", actions.ref
      assert_equal "app/models/repository.rb", actions.path
    end

    it "renders the blame template as html" do
      app = DummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Text"))
      app.blame("gitorious", "master", "app/models/repository.rb")

      assert_equal "text/html; charset=utf-8", app.response["Content-Type"]
      assert_equal "blame:Text", app.body
    end

    it "unescapes ref" do
      actions = Actions.new(BlobStub.new)
      app = DummySinatraApp.new(actions, Renderer.new("Blob"))
      app.blame("gitorious", "issue-%23221", "app/models/repository.rb")

      assert_equal "issue-#221", actions.ref
    end

    it "redirects ref to oid if configured so" do
      app = RedirectingDummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Blob"))
      app.blame("gitorious", "master", "lib/gitorious.rb")

      assert_equal "/gitorious/blame/#{'a' * 40}:lib/gitorious.rb", app.response["Location"]
    end
  end

  describe "#history" do
    it "delegates to actions" do
      actions = Actions.new(BlobStub.new)
      app = DummySinatraApp.new(actions, Renderer.new)
      app.history("gitorious", "master", "app/models/repository.rb", 10)

      assert_equal "gitorious", actions.repo
      assert_equal "master", actions.ref
      assert_equal "app/models/repository.rb", actions.path
    end

    it "renders the commits template as html" do
      app = DummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Text"))
      app.history("gitorious", "master", "app/models/repository.rb", 10)

      assert_equal "text/html; charset=utf-8", app.response["Content-Type"]
      assert_equal "commits:Text", app.body
    end

    it "unescapes ref" do
      actions = Actions.new(BlobStub.new)
      app = DummySinatraApp.new(actions, Renderer.new("Blob"))
      app.history("gitorious", "issue-%23221", "lib/gitorious.rb", 10)

      assert_equal "issue-#221", actions.ref
    end

    it "redirects ref to oid if configured so" do
      app = RedirectingDummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("Blob"))
      app.history("gitorious", "master", "lib/gitorious.rb", 10)

      assert_equal "/gitorious/history/#{'a' * 40}:lib/gitorious.rb", app.response["Location"]
    end
  end

  describe "#refs" do
    it "renders the refs template as json" do
      app = DummySinatraApp.new(Actions.new(BlobStub.new), Renderer.new("JSON"))
      app.refs("gitorious")

      assert_equal "application/json", app.response["Content-Type"]
      assert_equal "refs:JSON", app.body
    end
  end

  describe "#tree_history" do
    it "renders the tree_history template as json" do
      app = DummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("JSON"))
      app.tree_history("gitorious", "master", "", 1)

      assert_equal "application/json", app.response["Content-Type"]
      assert_equal "tree_history:JSON", app.body
    end

    it "unescapes ref" do
      actions = Actions.new(TreeStub.new)
      app = DummySinatraApp.new(actions, Renderer.new("Tree"))
      app.tree_history("gitorious", "issue-%23221", "app/models")

      assert_equal "issue-#221", actions.ref
    end

    it "redirects ref to oid if configured so" do
      app = RedirectingDummySinatraApp.new(Actions.new(TreeStub.new), Renderer.new("Tree"))
      app.tree_history("gitorious", "master", "lib", 10)

      assert_equal "/gitorious/tree_history/#{'a' * 40}:lib", app.response["Location"]
    end
  end
end