self.send(belongs_to :me)

class Console < Sinatra::Base
  require 'json'
  require 'yaml'

  BASE_PATH = File.expand_path(File.dirname(__FILE__) + "/web")
  # BASE_PATH is The folder the the console will only access
  # Also, enable Sinatra Cookies and Sessions

  get '/' do
    session[:current_path] = BASE_PATH
    haml :index
  end

  before do
    set_current_dir
    @p = env["rack.request.query_hash"] #accessing `q` and `XXX` like ?q=XXX
    # I dont know if theres another way
  end

  def set_current_dir #set @current for the current folder we are
    @current = session[:current_path]
  end  

  def f_level(path1_string, path2_string)
    # Check http://stackoverflow.com/questions/4353410/any-function-in-ruby-determine-the-difference-between-folders-level
    path1, path2 = [path1_string, path2_string].map { |s| s.split(File::SEPARATOR) }
    minsize = [path1, path2].map(&:size).min
    path1.first(minsize) == path2.first(minsize) ? path1.size - path2.size : nil
  end

  get '/api/v0.1/:command' do
    content_type :json
    msg = {} # a hash for returning errors, messages etc.

    f = "#{@current}/#{@p['q']}"
    f = "#{@current}" if @p['q'].nil?

    unless File.exist?(f)
      msg['error'] = "No such file or directory - #{@p['q']} in ~/#{@current[@current.index('web'), @current.length]}"
      return msg.to_json
    end

    level = f_level(File.expand_path(f), BASE_PATH)

    if level.nil? || level < 0 || level.nil? || @p['q'] && @p['q'][0] == '/'
      # this prevents user to go up the folder `/web` Or that you specified in BASE_PATH
      msg['error'] = "Unauthorized to access #{@p['q']}"
      return msg.to_json
    end

    f = File.expand_path(f)

    case params[:command]
    when "pwd"
      ("/LOTS_OF_FOLDERS/#{@current[@current.index('web'), @current.length]}").to_json

    when "cd"
      if File.directory?(f)
        session[:current_path] = f
        session[:current_path] = BASE_PATH if @p['q'] == ""
        f = session[:current_path]
        f.to_json
      else
        msg['error'] = "cd: #{@p['q']}: Not a directory"
        msg.to_json
      end

    when "ls"
      if File.directory?(f)
        Dir.glob("#{f}/*").map{ |x| File.basename(x)}.to_json
      else
        msg['error'] = "ls: #{@p['q']}: Not a directory"
        msg.to_json
      end

    when "cat" #show file content
      if File.file?(f)
        file = File.open(f)
        result = ""
        result << file.gets.strip while !file.eof?
        result.strip.to_json
      else
        msg['error'] = "cat: #{@p['q']}: Is a directory"
        msg.to_json
        # "cat: No such file or directory - #{@p['q']} in #{@current}".to_json
      end

    else
      "invalid all"
    end
  end
end

require 'rubygems'
require 'sinatra'
require 'haml'
require 'sass'
require './app'
run Console