mongo_fe 0.1.0

data/lib/mongo_fe/controllers/collections_controller.rb

@@ -0,0 +1,278 @@
+# encoding: utf-8
+require "json"
+require File.dirname(__FILE__) + '/../application_controller'
+require 'will_paginate'
+require 'will_paginate/view_helpers/sinatra'
+require "will_paginate/bootstrap"
+
+module MongoFe
+  # This controller works as a collection manager for the current db, a database specified by
+  # name as a REST parameter.
+
+  class CollectionsController < ApplicationController
+    namespace '/databases/:db_name/collections' do
+
+      before do
+        session[:db]=params[:db_name]
+      end
+
+      get "/?" do
+        haml :'/collections/index'
+      end
+
+      get "/:collection_name/?" do
+
+        collection_name = params[:collection_name]
+
+        unless collection_name.nil?
+          session[:collection] = collection_name
+          @collection = current_db.collection collection_name
+          @indexes = @collection.index_information
+          @page = params[:page].to_i || 1
+          @query= session[:query]
+
+          begin
+            @total, @documents = MongoFe::MongoDB::SearchDocuments.new(current_db, @collection).list(@query, @page, 10)
+            flash_search_results @query, @total
+          rescue => e
+            session[:query] = nil
+            @total, @documents = MongoFe::MongoDB::SearchDocuments.new(current_db, @collection).list(nil, 1, 10)
+
+            flash[:error] = "Resetting to find all. MongoDB error: #{e.message}"
+          end
+
+        end
+
+        haml :'/collections/index'
+      end
+
+      post "/?" do
+        if (collection_name=params[:collection_name])
+          current_db.create_collection(collection_name)
+          flash[:notice] = "#{collection_name}, successfully created!"
+          redirect "/databases/#{current_db.name}/collections/#{collection_name}"
+        else
+          flash[:error] = "Error: you must provide a collection name."
+          redirect "/databases/#{current_db.name}"
+        end
+      end
+
+      post "/:collection_name/search/?" do |db_name, collection_name|
+
+        if params[:reset_query].nil?
+          json_query = params[:json_query].empty? ? '{}' : params[:json_query].gsub(/'/, "\"")
+          json_query_show = params[:json_query_show]
+          json_query_sort = params[:json_query_sort]
+
+          begin
+            @json_query = JSON.parse json_query
+            puts "{query}: #{@json_query.inspect}"
+            session[:query] = [@json_query, json_query_show, json_query_sort]
+          rescue => e
+            flash[:error] = "Invalid query; #{e.message}"
+          end
+        else
+          session[:query] = nil
+        end
+
+        redirect "/databases/#{current_db.name}/collections/#{collection_name}"
+      end
+
+      delete "/:collection_name/documents*" do |db_name, collection_name, doc_id|
+        if (doc_id = params[:doc_id])
+          begin
+            collection = current_db.collection collection_name
+            collection.remove(:_id => BSON::ObjectId.from_string(doc_id))
+            flash[:notice] = "Document id: #{doc_id}, deleted from: #{db_name}.#{collection.name}"
+          rescue => e
+            flash[:error] = "Cannot delete document id: #{doc_id}, from: #{db_name}.#{collection_name}; #{e.message}"
+          end
+        else
+          flash[:error] = "The document id must be specified"
+        end
+
+        redirect "/databases/#{current_db.name}/collections/#{collection_name}"
+      end
+
+      put "/:collection_name/documents*" do |db_name, collection_name, z|
+        if (doc_id = params[:doc_id])
+          begin
+            @collection = current_db.collection collection_name
+            id = BSON::ObjectId.from_string(doc_id)
+            r = @collection.find(:_id => id).to_a
+            raise "this document is missing!" if r.nil?
+
+            begin
+              if (doc=params[:json_doc_attributes])
+                begin
+                  @doc_id = collection.update({"_id" => id}, JSON.parse(doc)) # rewrite the doc; in the future will use an atomic operator
+                  flash[:notice] = "Document id: #{doc_id}, updated"
+                rescue => e
+                  raise "Invalid JSON structure: #{e.message}"
+                end
+              else
+                raise "Invalid document."
+              end
+            rescue => e
+              flash[:error] = "Cannot update document; #{e.message}"
+            end
+
+          rescue => e
+            flash[:error] = "Cannot update the document with id: #{doc_id}; #{e.message}"
+          end
+        else
+          flash[:error] = "The document id must be specified"
+        end
+
+        redirect "/databases/#{current_db.name}/collections/#{collection_name}"
+      end
+
+      # create a new index
+      post "/:collection_name/indexes*" do |db_name, collection_name, z|
+        @collection = current_db.collection collection_name
+        session[:collection] = @collection.name
+
+        unique = !params[:unique].nil? && params[:unique]
+        sparse = !params[:sparse].nil? && params[:sparse]
+
+        if (fields = params[:index_fields])
+          begin
+            index_name = @collection.create_index(MongoFe::Utils.split_index_specs(fields), {:unique => unique, :sparse => sparse})
+
+            flash[:notice]="The index: '#{index_name}' was successfully created"
+          rescue => e
+            flash[:error]="Cannot create the index; #{e.message}"
+          end
+        else
+          flash[:error]="Specify either a single field name or an array of field name, direction pairs"
+        end
+
+        redirect "/databases/#{current_db.name}/collections/#{collection_name}"
+      end
+
+
+      # delete an existing index
+      #
+      delete "/:collection_name/indexes*" do |db_name, collection_name, z|
+        @collection = current_db.collection collection_name
+
+        if (index_name = params[:index_name])
+          begin
+            @collection.drop_index index_name
+            flash[:notice] = "The index: '#{index_name}', was deleted successfully."
+          rescue => e
+            flash[:error] = "The index: '#{index_name}', cannot be deleted; #{e.message}"
+          end
+
+        else
+          flash[:error] = "Please specify the index name."
+        end
+
+        session[:collection] = @collection.name
+        redirect "/databases/#{current_db.name}/collections/#{collection_name}"
+      end
+
+
+      # create a new document
+      #
+      post "/:collection_name/documents*" do
+        @collection = current_db.collection(params[:collection_name])
+
+        if @collection.nil?
+          flash[:error] = "Error: you must provide a valid collection name."
+          redirect "/databases/#{current_db.name}"
+        else
+          session[:collection] = @collection.name # move this and the one above to a before or such
+
+          begin
+            if (doc=params[:json_doc_attributes])
+              begin
+                @doc_id = @collection.insert(JSON.parse(doc))
+                flash[:notice] = "Document id: #{@doc_id.to_s}, added to: #{current_db.name}.#{@collection.name}"
+              rescue => e
+                raise "Invalid JSON structure; #{e.message}"
+              end
+            else
+              raise "Invalid document."
+            end
+          rescue => e
+            flash[:error] = "Cannot insert document; error: #{e.message}"
+          end
+
+        end
+
+        unless @collection.nil?
+          @page = params[:page].to_i || 1
+          @query= session[:query]
+
+          begin
+            @total, @documents = MongoFe::MongoDB::SearchDocuments.new(current_db, @collection).list(@query, @page, 10)
+            flash_search_results @query, @total
+          rescue => e
+            session[:query] = nil
+            @total, @documents = MongoFe::MongoDB::SearchDocuments.new(current_db, @collection).list(nil, 1, 10)
+
+            flash[:error] = "Resetting to find all. Cause: #{e.message}"
+          end
+
+        end
+
+        haml :'/collections/index'
+      end
+
+      put "/:collection_name/?" do
+        if (collection_name=params[:collection_name])
+          collection = current_db.collection(collection_name)
+
+          if collection.nil?
+            flash[:error] = "The db: #{current_db.name} doesn't know this collection: #{collection_name}"
+          else
+            if (collection_new_name = params[:collection_new_name])
+
+              begin
+                current_db.rename_collection(collection_name, collection_new_name)
+                flash[:notice] = "Collection: #{collection_name}, successfully renamed to: #{collection_new_name}"
+                redirect "/databases/#{current_db.name}/collections/#{collection_new_name}"
+              rescue => e
+                flash[:error] = "Cannot rename: #{collection_name}, to: #{collection_new_name}; error: #{e.message}"
+              end
+
+            else
+              flash[:error] = "You must provide a collection name."
+            end
+            redirect "/databases/#{current_db.name}/collections/#{collection_name}"
+          end
+
+        else
+          flash[:error] = "Error: you must provide a collection name."
+          redirect "/databases/#{current_db.name}"
+        end
+      end
+
+      delete "/:collection_name/?" do
+        db=current_db
+        if (collection_name=params[:collection_name])
+          if (collection = db.collection(collection_name))
+            begin
+              collection.drop
+              flash[:notice] = "#{collection_name}, deleted."
+            rescue => e
+              flash[:error] = "Error: #{e.message}, while trying to delete the collection: #{collection_name}"
+            end
+          end
+        else
+          flash[:error] = "Error: you must provide a collection name."
+        end
+
+        redirect "/databases/#{db.name}"
+      end
+
+    end
+
+    private
+    def flash_search_results(q, total)
+      flash[:notice]="query: #{query.string}<br/>#{total} document(s) found." unless (q.nil? || query.string.start_with?("ex."))
+    end
+  end
+
+end

data/lib/mongo_fe/controllers/databases_controller.rb

@@ -0,0 +1,94 @@
+# encoding: utf-8
+require "mongo"
+require File.dirname(__FILE__) + '/../application_controller'
+
+module MongoFe
+  class DatabasesController < ApplicationController
+    
+    get '/databases/?' do
+      "Move along, nothing to see here!"
+    end
+
+    # info about a specific database
+    get '/databases/:name/?' do      
+      db_name = params[:name]
+      begin          
+        @db = MongoFe::MongoDB.use(db_name)
+        session[:db] = db_name
+        session[:collection] = nil
+        # flash[:notice] = "DB Selected: #{db_name}. <a href='/databases/#{db_name}/info'>More info?</a>"
+        haml :'/databases/info'
+      rescue => e
+        session[:db]=nil
+        flash[:error] = "You must provide a database name; #{e.message}"
+        redirect "/"
+      end
+    end
+
+    # create a new database
+    post '/databases/?' do
+      if (d=params[:name])
+        db = MongoFe::MongoDB.use(d)
+        session[:db]=d
+        flash[:notice] = "A new database: `#{d}`, was created."
+        redirect "/databases/#{d}"
+      else
+        flash[:error] = "Error: you must provide a database name."
+        haml :index
+      end
+    end
+
+    # Delete database
+    delete '/databases/:name/?' do |d|
+      if d
+        begin
+          MongoFe::MongoDB.connection.drop_database(d)
+          flash[:notice] = "Database: #{d} was deleted."
+          session[:db]=nil
+        rescue => e
+          flash[:error] = "Error: #{e.message }; The database: `#{d}` was not deleted."
+        end
+      else
+        flash[:notice] = "Error: you must provide a database name."
+      end
+      haml :index
+    end
+
+    # create a new database user
+    post '/databases/:name/users/?' do |d|
+      if d && username=params[:username]
+        db = MongoFe::MongoDB.use(d)
+        if db.find_user(username)
+          flash[:error] = "the user: '#{username}' is already defined."
+        else
+          is_readonly = params[:readonly].nil? ? false : true
+          db.add_user(username, params[:password], is_readonly)
+          flash[:notice] = "User: '#{username}', was added successfully."
+        end
+      else
+        flash[:error] = "you must provide a user name."
+      end
+
+      redirect "/databases/#{d}"
+    end
+
+    # delete an exiting user
+    delete '/databases/:name/users/?' do |d|
+      if d && username=params[:username]
+        db = MongoFe::MongoDB.use(d)
+        if db.find_user(username)
+          db.remove_user(username)
+          flash[:notice] = "User deleted: '#{username}'"
+        else
+          flash[:error] = "Not a valid username: '#{username}'"
+        end
+      else
+        flash[:error] = "Please provide a user name."
+      end
+
+      redirect "/databases/#{d}"      
+    end
+
+  end
+end
+

data/lib/mongo_fe/helpers/helpers.rb

@@ -0,0 +1,128 @@
+require "json"
+require "bson"
+require 'will_paginate'
+require 'will_paginate/view_helpers/base'
+require 'will_paginate/view_helpers/link_renderer'
+
+module Helpers
+  include Rack::Utils
+  include WillPaginate::ViewHelpers::Base
+
+  alias_method :h, :escape_html
+
+  def current_page
+    url_path request.path_info.sub('/', '')
+  end
+
+  def url_path(*path_parts)
+    [path_prefix, path_parts].join("/").squeeze('/')
+  end
+
+  alias_method :u, :url_path
+
+  def current_db?
+    true if session[:db]
+  end
+
+  def current_db
+    MongoFe::MongoDB.connection.db(session[:db]) if current_db?
+  end
+
+  def current_db_name
+    session[:db] if current_db?
+  end
+
+  def collection
+    current_db.collection(session[:collection]) if current_db? && collection?
+  end
+
+  def collection?
+    true if session[:collection]
+  end
+
+  def link_to(name, location, alternative = false)
+    if alternative && alternative[:condition]
+      "<a href=#{alternative[:location]}>#{alternative[:name]}</a>"
+    else
+      "<a href=#{location}>#{name}</a>"
+    end
+  end
+
+  # prettifying a JSON structure
+  def pretty_json(json)
+    JSON.pretty_generate(json)
+  end
+
+  # truncate a string to max characters and append '...' at the end
+  def truncate(str, max=0)
+    return '' if str.blank?
+
+    if str.length <= 1
+      t_string = str
+    else
+      t_string = str[0..[str.length, max].min]
+      t_string = str.length > t_string.length ? "#{t_string}..." : t_string
+    end
+
+    t_string
+  end
+
+  # poor man flash solution
+  def flash
+    session[:flash] = Hashie::Mash.new if session[:flash].nil?
+    session[:flash]
+  end
+
+  PLACEHOLDER_SORT = "ex. name:ASC, age:DESC"
+  PLACEHOLDER_QUERY_STRING = 'ex. {"name":"foo"}'
+  PLACEHOLDER_SHOW = "ex. name, age"
+
+  # check if there is a query in the session and return its elements as an openstruct object
+  def query
+    if (q = session[:query])
+      OpenStruct.new(:string => q.first.empty? ? '' : q.first.to_json,
+                     :show => q[1].empty? ? '' : q[1],
+                     :sort => q.last.empty? ? '' : q.last)
+    else
+      OpenStruct.new(:string => PLACEHOLDER_QUERY_STRING, :show => PLACEHOLDER_SHOW, :sort => PLACEHOLDER_SORT)
+    end
+  end
+
+  # stolen from http://github.com/cschneid/irclogger/blob/master/lib/partials.rb
+  # and made a lot more robust by me: https://gist.github.com/119874.
+  # Another me, not me :)
+  def partial(template, *args)
+    template_array = template.to_s.split('/')
+    template = template_array[0..-2].join('/') + "/_#{template_array[-1]}"
+    options = args.last.is_a?(Hash) ? args.pop : {}
+    options.merge!(:layout => false)
+    locals = options[:locals] || {}
+
+    if (collection = options.delete(:collection))
+      collection.inject([]) do |buffer, member|
+        buffer << haml(:"#{template}",
+                       options.merge(:layout => false,
+                                     :locals => {template_array[-1].to_sym => member}.merge(locals)))
+      end.join("\n")
+    else
+      haml(:"#{template}", options)
+    end
+  end
+
+  # generate a simple random string
+  def random_string(len)
+    chars = ("a".."z").to_a + ("A".."Z").to_a + ("0".."9").to_a
+    str = ""
+    1.upto(len) { |i| str << chars[rand(chars.size-1)] }
+    str
+  end
+
+
+  MEGABYTE = 1024.0 * 1024.0
+
+  # convert Bytes to MegaBytes
+  def bytes_to_mb bytes=0
+    bytes / MEGABYTE
+  end
+
+end

data/lib/mongo_fe/public/app/application.js

@@ -0,0 +1,3 @@
+var MongoFE = {
+
+};