amp 0.5.0

data/lib/amp/server/fancy_views/stylesheet.sass

@@ -0,0 +1,175 @@
+!lightblue = #ddddff
+!medblue   = #bbbbff
+!darkblue  = #aaaaff
+=pushleft
+  :position relative
+  :float left
+  :clear left
+body
+  :font-family "Lucida Grande"
+  :line-height 1.1em
+  :font-size 80%
+  a
+    :text-decoration none
+    :color #3333dd
+    &:hover
+      :text-decoration underline
+#main_container
+  :position relative
+
+h2.header
+  +pushleft
+#navbar
+  :position absolute
+  :top 10px
+  :left 10px
+  :background-color #ddd
+  :width 140px
+  ul
+    :padding 0
+    :text-align left
+    :font-size 12px
+    :font-family "Lucida Grande"
+    li
+      :padding 6px
+      :background-color #ddd
+      :border none
+      a
+        :text-decoration none
+        :color black
+    li:hover, li.selected
+      :background-color #f8f8f8
+      :border-left solid 2px #444
+      :padding-left 4px
+#navbar ul, #navbar li
+  :list-style-type none
+  :margin 0
+#main
+  :padding-left 160px
+.shortlog
+  +pushleft
+  tr
+    :background-color #DDFFE3
+    :border-bottom solid 1px #555
+    :padding 3px 0
+    :width 100%
+    :float left
+    &:hover, &.selected
+      :background-color #bbddc3
+  .index
+    :width 35px
+  .user
+    :width 150px
+  .date
+    :width 150px
+  .description a
+    :text-decoration none
+    &:hover
+      :text-decoration underline
+.pagebutton
+  +pushleft
+  :background-color = !lightblue
+  :padding 3px
+  :margin 2px
+  :text-decoration none
+  :color black
+  &:hover
+    :background-color = !medblue
+    :padding 2px
+    :border solid 1px = !darkblue
+.browser-list
+  +pushleft
+  tr
+    :background-color #DDFFE3
+    :border-bottom solid 1px #555
+    :padding 3px 0
+    :width 100%
+    :float left
+    &:hover, &.selected
+      :background-color #bbddc3
+  .filename
+    :width 200px
+    a
+      :color = #3333cc
+      :text-decoration none
+  .date
+    :width 25%
+  .message
+    :width 50%
+.browser-file
+  +pushleft
+  :width 100%
+  .source-holder
+    :position relative
+    :float left
+    :clear left
+    :background-color #eee
+    :border solid 1px #ddd
+    :width 90%
+    :padding 15px
+    :overflow-x auto
+    pre
+      +pushleft
+      :font-size 14px
+      :line-height 1.1em
+      :width 100%
+      :overflow-x auto
+ul.versions
+  +pushleft
+  :width 800px
+  :list-style-type none
+  li
+    :position relative
+    :float left
+    :clear none
+    :width 120px
+    :text-align center
+    :padding-top 2px
+    :margin 1px
+    :height 20px
+    :font-size 12px
+    :list-style-type none
+    :background-color = !medblue
+    a
+      :color black
+      :text-decoration none
+    &:hover
+      :background-color = !lightblue
+      :border = solid 1px !darkblue
+      :width 118px
+      :height 19px
+      :padding-top 1px
+.changeset-desc
+  +pushleft
+  blockquote
+    +pushleft
+    :border-left solid 2px #444
+    :margin 1em 0 1em 0
+    :padding 0px 0px 0px 8px
+  .two-column
+    +pushleft
+    td.label
+      :text-align right
+      :color #999
+    td.data
+      :text-align left
+      :color #222
+.diff-list
+  +pushleft
+  :width 100%
+  &>li
+    :list-style-type none
+.diff
+  :width 90%
+  :overflow-x auto
+.diff, .diff li, pre
+  :list-style-type none
+  :margin 0
+  :padding 1px 0px
+  :size 12px
+.diff-add
+  :background-color #aaffaa
+.diff-del
+  :background-color #ffaaaa
+.diff-ctx
+  :background-color #aaaaaa

data/lib/amp/server/http_server.rb

@@ -0,0 +1,140 @@
+require 'rubygems'
+require 'sinatra/base'
+require 'rack/deflater'
+need { 'extension/amp_extension' }
+need { 'extension/authorization' }
+
+module Amp
+  module Servers
+    ##
+    # = HTTPServer
+    # General HTTP Server that serves up a Mercurial repository under the
+    # Mercurial HTTP protocol. Any mercurial client (including Amp) can use
+    # this server for push/pull operations, clones, and so on.
+    # Note: this server only provides methods for dealing with Mercurial clients. 
+    # It is not a full-blown web server with a web interface to the repository.
+    #
+    # It is worth noting that one may serve many repositories using one server
+    # using this class. Spiffy, eh?
+    class HTTPServer < Sinatra::Base
+      register Sinatra::AmpExtension
+      helpers  Sinatra::AmpRepoMethods
+    end
+    
+    ##
+    # = HTTPAuthorizedServer
+    # General HTTP Server that has some form of authentication. Implements
+    # user-logic methods, as every subclass will have the same interface
+    # for adding/removing users. Note: this server only provides methods
+    # for dealing with Mercurial clients. It is not a full-blown web server
+    # with a web interface to the repository. The users you add to an
+    # HTTPAuthorizedServer are the credentials a user must provide when pushing
+    # to a repository (for example), or checking out a private repository.
+    #
+    # It is worth noting that one may serve many repositories using one server
+    # using this class. Spiffy, eh?
+    class HTTPAuthorizedServer < HTTPServer
+      
+      def self.set_storage(manner, opts={})
+        manner = manner.to_sym
+        
+        case manner
+        when :sequel
+          set :user_db_name, (opts[:user_db_name] || "users")
+          extend RepoUserManagement::SequelSQLite3
+        when :memory
+          extend RepoUserManagement::Memory
+        else
+          raise "Unknown storage manner #{manner.inspect}"
+        end
+        
+        # since you can only set the storage once...
+        def self.set_storage(*args); end # this will redefine this method to do nothing
+      end
+      
+      def self.set_permission(style, *args)
+        case style
+        when :writer
+          set_writer *args
+        when :reader
+          set_reader *args
+        else
+          raise ArgumentError.new("Unknown permission level: #{style.to_s.inspect}")
+        end
+      end
+      
+      ##
+      # Sets whether to use digest or basic authentication. May only be called once,
+      # for now.
+      #
+      # @param [Symbol, String] manner the type of authentication
+      def self.set_authentication(manner, opts={})
+        case manner.to_sym
+        when :basic
+          helpers Sinatra::BasicAuthorization
+        when :digest
+          helpers Sinatra::DigestAuthorization
+        end
+        def self.set_authentication; end
+      end
+      
+      set :authorization_realm, "Amp Repository"
+      
+      def repos; self.class.repos; end
+      def users; self.class.users; end
+      
+      ##
+      # This block is run on every single request, before the server code for the request is processed.
+      before do
+        cmd  = params["cmd"]
+        repo = self.class.amp_repositories[request.path_info]
+        
+        if cmd.nil? or cmd.empty? or !repos[repo] or !repos[repo][:private] && command_reads?(cmd)
+          true
+        else
+          login_required
+        end
+      end
+      
+      ##
+      # General helper methods for the server
+      helpers do
+        ##
+        # Is the repository URL private? The URL is expected to be good.
+        #
+        # @param [String] repo the URL to the repository. Expected to be correct
+        def repo_is_private?(repo)
+          repos[repo][:private]
+        end
+        
+        ##
+        # Is the repository URL public?
+        # 
+        # @param [String] repo the URL to the repository. Expected to be correct
+        def repo_is_public?(repo) 
+          !repo_is_private?(repo)
+        end
+        
+        ##
+        # Pay attention kids -- this returns a HASH!!!!!!!!
+        # This is a hash of :user, :read, and :write
+        # 
+        # @param [String] repo path to the repo on the web
+        # @param [String] username the dude we're looking for
+        # @return [{:user => Amp::Servers::User, :read => Boolean, :write => Boolean}]
+        def get_user_and_permissions(repo, username)
+          if u = repos[repo][:users][username]
+            return u
+          end
+          
+          {:user  => User.public_user,
+           :read  => (repos[repo][:private] ? false : true),
+           :write => false
+          }
+        end
+      end
+    end
+    
+  end
+end
+

data/lib/amp/server/repo_user_management.rb

@@ -0,0 +1,287 @@
+##
+# == RepoUserManagement
+# This module manages the repository-user relationships that occur.
+# Information is stored in memory.
+module Amp
+  module Servers
+    module RepoUserManagement
+      module Memory
+        
+        ##
+        # All the repositories stored in a hash.
+        # @example repos['monkey'] = {:users => {String => {:user => Amp::Servers::User,
+        #                                                   :read => Boolean, :write => Boolean}},
+        #                             :private => Boolean}
+        def repos; @repos ||= {}; @repos.default = {}; @repos; end
+            
+        def users; @users ||= {}; end
+        
+        ##
+        # Adds a user to the system. This is not repository specific.
+        #
+        # @param [Hash] user_info The user data, in hash form. Passed to {User.from_hash}
+        # @option user_info [String] :username The username for the user
+        # @option user_info [String] :password The cleartext (unencrypted) password.
+        def add_user(user_info={})
+          new_user = User.from_hash user_info
+          users[new_user.username] = new_user # universal list of users
+        end
+        alias_method :<<, :add_user
+        
+        
+        ##
+        # Adds a given username/password combination to the system, for a given
+        # repository, with write privileges.
+        #
+        # @param [Repository] repository The repository to which are associating the user
+        # @param [Amp::Servers::User] user The user to give write priveleges to
+        def set_writer(repository, user)
+          repos[repository] ||= {:users => {}, :private => false}
+          repos[repository][:users][user.username] ||= {:user => user, :read => true, :write => true}
+          repos[repository][:users][user.username][:read]  = true # these are unnecessary if we are adding a user
+          repos[repository][:users][user.username][:write] = true
+        end
+        
+        ##
+        # Adds a given username/password combination to the system, for a given
+        # repository, with read-only privileges. This will override any other settings
+        # previously set. For example, a call to #set_writer and then #set_reader would
+        # be as though #set_writer never happened.
+        #
+        # @param [Repository] repository The repository to which are associating the user
+        # @param [Amp::Servers::User] user The user to give write priveleges to
+        def set_reader(repository, user)
+          repos[repository] ||= {:users => {}, :private => false}
+          repos[repository][:users][user.username] ||= {:user => user, :read => true, :write => false}
+          repos[repository][:users][user.username][:read]  = true
+          repos[repository][:users][user.username][:write] = false
+        end
+        
+        ##
+        # Sets the given repository's privacy levels. Repositories default to being
+        # public, and must be set to be private using this method. This method can also
+        # be used to later make a private repository public without stopping the server.
+        #
+        # @param [Repository] repository The repository for which to set privacy settings
+        # @param [Boolean] _private (true) whether the repository should be private or not
+        def set_private(repository, _private=true)
+          repos[repository][:private] = !!_private
+        end
+        
+      end
+      
+      module Sequel
+        
+        def self.extended(klass)
+          require 'sequel'
+          
+          # Set up the database
+          # Oh gods please have mercy on my soul
+          const_set 'DB', Sequel.sqlite("amp_serve_#{klass.user_db_name}.db")
+          
+          DB.create_table :users do
+            primary_key :id
+            String :username, :unique => true, :null => false
+            String :password, :null => false
+          end
+          
+          DB.create_table :repos do
+            primary_key :id
+            String  :url,     :unique => true, :null => false
+            String  :path,    :null => false
+            boolean :private, :default => false
+          end
+          
+          DB.create_table :permissions do
+            foreign_key :repo_id, :repos
+            foreign_key :user_id, :users
+            boolean     :read
+            boolean     :write
+          end
+          
+          # find room in your heart to forgive me
+          # constants can't be assigned to in a method
+          const_set 'USERS', DB[:users]
+          const_set 'REPOS', DB[:repos]
+          const_set 'PERMS', DB[:permissions]
+        end
+        
+        ##
+        # All the repositories stored in a hash.
+        # 
+        # @todo slow as fuck
+        # @example repos['monkey'] = {:users => {String => {:user => Amp::Servers::User,
+        #                                                   :read => Boolean, :write => Boolean}},
+        #                             :private => Boolean}
+        def repos
+          REPOS.inject({}) do |h, v|
+            uss = PERMS[:repo_id => v[:id]]
+            q = uss.map do |p|
+              {:user => USERS[:user_id => p[:user_id]],
+               :read => p[:read], :write => p[:write] }
+            end
+            z = q.inject({}) {|h, v| h.merge v[:user][:username] => v }
+            
+            h.merge v[:url] => {:users => z, :private => v[:private]}
+          end
+        end
+            
+        def users
+          USERS.inject({}) do |s, v|
+            s.merge v[:username] => v
+          end
+        end
+                
+        ##
+        # Adds a user to the system. This is not repository specific.
+        #
+        # @param [Hash] user_info The user data, in hash form. Passed to {User.from_hash}
+        # @option user_info [String] :username The username for the user
+        # @option user_info [String] :password The cleartext (unencrypted) password.
+        def add_user(user_info={})
+          USERS << user_info
+        end
+        alias_method :<<, :add_user
+        
+        ##
+        # Adds a given username/password combination to the system, for a given
+        # repository, with write privileges.
+        #
+        # @todo don't know if this actually works
+        # @param [Repository] repository The repository to which are associating the user
+        # @param [Amp::Servers::User] user The user to give write priveleges to
+        def set_writer(repository, user)
+          perm = PERMS.filter 'repo_id = ? and user_id = ?', REPOS.first(:url => repository).id, user[:id]
+          perm.update :read  => true
+          perm.update :write => true
+        end
+        
+        ##
+        # Adds a given username/password combination to the system, for a given
+        # repository, with read-only privileges. This will override any other settings
+        # previously set. For example, a call to #set_writer and then #set_reader would
+        # be as though #set_writer never happened.
+        # 
+        # @todo don't know if this actually works
+        # @param [Repository] repository The repository to which are associating the user
+        # @param [Amp::Servers::User] user The user to give write priveleges to
+        def set_reader(repository, user)
+          perm = PERMS.filter 'repo_id = ? and user_id = ?', REPOS.first(:url => repository).id, user[:id]
+          perm.update :read  => true
+          perm.update :write => false
+        end
+        
+        ##
+        # Sets the given repository's privacy levels. Repositories default to being
+        # public, and must be set to be private using this method. This method can also
+        # be used to later make a private repository public without stopping the server.
+        #
+        # @todo don't know if this actually works
+        # @param [Repository] repository The repository for which to set privacy settings
+        # @param [Boolean] _private (true) whether the repository should be private or not
+        def set_private(repository, _private=true)
+          REPOS[:url => repository] = {:private => true}
+        end
+        
+      end
+      
+      module DataMapper
+        
+        def self.extended(klass)
+          require 'dm-core'
+          
+          users = Class.new do
+            include DataMapper::Resource
+            property :username, String
+            property :password, String
+            
+            has n, :perms
+          end
+          const_set 'User', users
+          
+          perms = Class.new do
+            include DataMapper::Resource
+            property :read
+            property :write
+            
+            belongs_to :user
+            belongs_to :repo
+          end
+          const_set 'Perm', perms
+          
+          repo = Class.new do
+            include DataMapper::Resource
+            property :url,  String
+            property :path, String
+            
+            has n, :perms
+          end
+          const_set 'Repo', repo
+        end
+        
+        ##
+        # All the repositories stored in a hash.
+        # 
+        # @todo slow as fuck
+        # @example repos['monkey'] = {:users => {String => {:user => Amp::Servers::User,
+        #                                                   :read => Boolean, :write => Boolean}},
+        #                             :private => Boolean}
+        def repos
+          
+        end
+            
+        def users
+          
+        end
+                
+        ##
+        # Adds a user to the system. This is not repository specific.
+        #
+        # @param [Hash] user_info The user data, in hash form. Passed to {User.from_hash}
+        # @option user_info [String] :username The username for the user
+        # @option user_info [String] :password The cleartext (unencrypted) password.
+        def add_user(user_info={})
+          
+        end
+        alias_method :<<, :add_user
+        
+        ##
+        # Adds a given username/password combination to the system, for a given
+        # repository, with write privileges.
+        #
+        # @todo don't know if this actually works
+        # @param [Repository] repository The repository to which are associating the user
+        # @param [Amp::Servers::User] user The user to give write priveleges to
+        def set_writer(repository, user)
+          
+        end
+        
+        ##
+        # Adds a given username/password combination to the system, for a given
+        # repository, with read-only privileges. This will override any other settings
+        # previously set. For example, a call to #set_writer and then #set_reader would
+        # be as though #set_writer never happened.
+        # 
+        # @todo don't know if this actually works
+        # @param [Repository] repository The repository to which are associating the user
+        # @param [Amp::Servers::User] user The user to give write priveleges to
+        def set_reader(repository, user)
+          
+        end
+        
+        ##
+        # Sets the given repository's privacy levels. Repositories default to being
+        # public, and must be set to be private using this method. This method can also
+        # be used to later make a private repository public without stopping the server.
+        #
+        # @todo don't know if this actually works
+        # @param [Repository] repository The repository for which to set privacy settings
+        # @param [Boolean] _private (true) whether the repository should be private or not
+        def set_private(repository, _private=true)
+          
+        end
+        
+      end
+    end
+  end
+end