# 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 "em_rugged"
require "json"

module Dolt
  module Sinatra
    module Actions
      # Built-in redirect seems to not work with Sinatra::Async, it throws
      # an error.
      def redirect(url)
        response.status = 302
        response["Location"] = url
        body ""
      end

      def error(error, repo, ref)
        response["Content-Type"] = "text/html"
        body(renderer.render(:"500", {
                               :error => error,
                               :repository => repo,
                               :ref => ref
                             }))
      end

      def raw(repo, ref, path)
        blob(repo, ref, path, {
               :template => :raw,
               :content_type => "text/plain",
               :template_options => { :layout => nil }
             })
      end

      def blob(repo, ref, path, options = { :template => :blob, :content_type => "text/html" })
        actions.blob(repo, ref, path) do |err, data|
          return error(err, repo, ref) if !err.nil?
          blob = data[:blob]
          return redirect(tree_url(repo, ref, path)) if !blob.is_a?(Rugged::Blob)
          response["Content-Type"] = options[:content_type]
          tpl_options = options[:template_options] || {}
          body(renderer.render(options[:template], data, tpl_options))
        end
      end

      def tree(repo, ref, path)
        actions.tree(repo, ref, path) do |err, data|
          return error(err, repo, ref) if !err.nil?
          tree = data[:tree]
          return redirect(blob_url(repo, ref, path)) if tree.class.to_s !~ /\bTree/
          response["Content-Type"] = "text/html"
          body(renderer.render(:tree, data))
        end
      end

      def blame(repo, ref, path)
        actions.blame(repo, ref, path) do |err, data|
          return error(err, repo, ref) if !err.nil?
          response["Content-Type"] = "text/html"
          body(renderer.render(:blame, data))
        end
      end

      def history(repo, ref, path, count)
        actions.history(repo, ref, path, count) do |err, data|
          return error(err, repo, ref) if !err.nil?
          response["Content-Type"] = "text/html"
          body(renderer.render(:commits, data))
        end
      end

      def refs(repo)
        actions.refs(repo) do |err, data|
          return error(err, repo, ref) if !err.nil?
          response["Content-Type"] = "application/json"
          body(renderer.render(:refs, data, :layout => nil))
        end
      end

      def tree_history(repo, ref, path, count = 1)
        actions.tree_history(repo, ref, path, count) do |err, data|
          if !err.nil?
            error(err, repo, ref)
          else
            response["Content-Type"] = "application/json"
            body(renderer.render(:tree_history, data, :layout => nil))
          end
        end
      end
    end
  end
end