runa-chef 0.8.0.1

data/lib/chef/util/file_edit.rb

@@ -0,0 +1,125 @@
+#
+# Author:: Nuo Yan (<nuo@opscode.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'ftools'
+require 'fileutils'
+require 'tempfile'
+
+class Chef
+	class Util
+		class FileEdit
+
+			private
+			
+			attr_accessor :original_pathname, :contents, :file_edited
+
+			public
+			
+			def initialize(filepath)
+				@original_pathname = filepath
+				@file_edited = false
+				
+				raise ArgumentError, "File doesn't exist" unless File.exist? @original_pathname
+				raise ArgumentError, "File is blank" unless (@contents = File.new(@original_pathname).readlines).length > 0
+			end
+			
+			#search the file line by line and match each line with the given regex
+			#if matched, replace the whole line with newline.
+			def search_file_replace_line(regex, newline)
+				search_match(regex, newline, 'r', 1)
+			end
+
+			#search the file line by line and match each line with the given regex
+			#if matched, replace the match (all occurances)  with the replace parameter
+			def search_file_replace(regex, replace)
+				search_match(regex, replace, 'r', 2)
+			end
+			
+			#search the file line by line and match each line with the given regex
+			#if matched, delete the line
+			def search_file_delete_line(regex)
+				search_match(regex, " ", 'd', 1)
+			end
+			
+			#search the file line by line and match each line with the given regex
+			#if matched, delete the match (all occurances) from the line
+			def search_file_delete(regex)
+				search_match(regex, " ", 'd', 2)
+			end
+
+			#search the file line by line and match each line with the given regex
+			#if matched, insert newline after each matching line
+			def insert_line_after_match(regex, newline)
+				search_match(regex, newline, 'i', 0)
+			end
+			 
+			#Make a copy of old_file and write new file out (only if file changed)
+			def write_file
+				
+				# file_edited is false when there was no match in the whole file and thus no contents have changed.
+				if file_edited
+					backup_pathname = original_pathname + ".old"
+					File.copy(original_pathname, backup_pathname)
+					Tempfile.open("w") do |newfile|
+						contents.each do |line|
+							newfile.puts(line)
+						end
+						newfile.flush
+						FileUtils.mv(newfile.path, original_pathname)
+					end
+				end
+				self.file_edited = false
+
+			end
+			
+			private
+			
+			#helper method to do the match, replace, delete, and insert operations
+			#command is the switch of delete, replace, and insert ('d', 'r', 'i')
+			#method is to control operation on whole line or only the match (1 for line, 2 for match)
+			def search_match(regex, replace, command, method)
+				
+				#convert regex to a Regexp object (if not already is one) and store it in exp.
+				exp = Regexp.new(regex)
+
+				#loop through contents and do the appropriate operation depending on 'command' and 'method'
+				new_contents = []
+				
+				contents.each do |line|
+					if line.match(exp) 
+						self.file_edited = true
+						case
+						when command == 'r'
+							new_contents << ((method == 1) ? replace : line.gsub!(exp, replace))
+						when command == 'd'
+							if method == 2
+								new_contents << line.gsub!(exp, "")
+							end
+						when command == 'i'
+							new_contents << line
+							new_contents << replace
+						end
+					else
+						new_contents << line
+					end
+				end
+
+				self.contents = new_contents
+			end
+		end
+	end
+end

data/lib/chef/util/fileedit.rb

@@ -0,0 +1,121 @@
+#
+# Author:: Nuo Yan (<nuo@opscode.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'ftools'
+require 'fileutils'
+require 'tempfile'
+
+class FileEdit
+
+  private
+  
+  attr_accessor :original_pathname, :contents, :file_edited
+
+  public
+  
+  def initialize(filepath)
+    @original_pathname = filepath
+    @file_edited = false
+    
+    raise ArgumentError, "File doesn't exist" unless File.exist? @original_pathname
+    raise ArgumentError, "File is blank" unless (@contents = File.new(@original_pathname).readlines).length > 0
+  end
+  
+  #search the file line by line and match each line with the given regex
+  #if matched, replace the whole line with newline.
+  def search_file_replace_line(regex, newline)
+    search_match(regex, newline, 'r', 1)
+  end
+
+  #search the file line by line and match each line with the given regex
+  #if matched, replace the match (all occurances)  with the replace parameter
+  def search_file_replace(regex, replace)
+    search_match(regex, replace, 'r', 2)
+  end
+  
+  #search the file line by line and match each line with the given regex
+  #if matched, delete the line
+  def search_file_delete_line(regex)
+    search_match(regex, " ", 'd', 1)
+  end
+  
+  #search the file line by line and match each line with the given regex
+  #if matched, delete the match (all occurances) from the line
+  def search_file_delete(regex)
+    search_match(regex, " ", 'd', 2)
+  end
+
+  #search the file line by line and match each line with the given regex
+  #if matched, insert newline after each matching line
+  def insert_line_after_match(regex, newline)
+    search_match(regex, newline, 'i', 0)
+  end
+   
+  #Make a copy of old_file and write new file out (only if file changed)
+  def write_file
+    
+    # file_edited is false when there was no match in the whole file and thus no contents have changed.
+    if file_edited
+      backup_pathname = original_pathname + ".old"
+      File.copy(original_pathname, backup_pathname)
+      Tempfile.open("w") do |newfile|
+        contents.each do |line|
+          newfile.puts(line)
+        end
+        newfile.flush
+        FileUtils.mv(newfile.path, original_pathname)
+      end
+    end
+    self.file_edited = false
+
+  end
+  
+  private
+  
+  #helper method to do the match, replace, delete, and insert operations
+  #command is the switch of delete, replace, and insert ('d', 'r', 'i')
+  #method is to control operation on whole line or only the match (1 for line, 2 for match)
+  def search_match(regex, replace, command, method)
+    
+    #convert regex to a Regexp object (if not already is one) and store it in exp.
+    exp = Regexp.new(regex)
+
+    #loop through contents and do the appropriate operation depending on 'command' and 'method'
+    new_contents = []
+    
+    contents.each do |line|
+      if line.match(exp) 
+        self.file_edited = true
+        case
+        when command == 'r'
+          new_contents << ((method == 1) ? replace : line.gsub!(exp, replace))
+        when command == 'd'
+          if method == 2
+            new_contents << line.gsub!(exp, "")
+          end
+        when command == 'i'
+          new_contents << line
+          new_contents << replace
+        end
+      else
+        new_contents << line
+      end
+    end
+
+    self.contents = new_contents
+  end
+end

data/lib/chef/webui_user.rb

@@ -0,0 +1,231 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Nuo Yan (<nuo@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/config'
+require 'chef/mixin/params_validate'
+require 'chef/couchdb'
+require 'digest/sha1'
+require 'rubygems'
+require 'json'
+
+class Chef
+  class WebUIUser
+    
+    attr_accessor :name, :salt, :validated, :password, :couchdb_rev, :admin, :openid
+    
+    include Chef::Mixin::ParamsValidate
+    
+    DESIGN_DOCUMENT = {
+      "version" => 3,
+      "language" => "javascript",
+      "views" => {
+        "all" => {
+          "map" => <<-EOJS
+            function(doc) {
+              if (doc.chef_type == "webui_user") {
+                emit(doc.name, doc);
+              }
+            }
+          EOJS
+        },
+        "all_id" => {
+          "map" => <<-EOJS
+          function(doc) {
+            if (doc.chef_type == "webui_user") {
+              emit(doc.name, doc.name);
+            }
+          }
+          EOJS
+        },
+      },
+    }
+    
+    # Create a new Chef::WebUIUser object.
+    def initialize()
+      @name = nil
+      @salt = nil
+      @password = nil
+      @admin = false
+      @openid = nil
+      @couchdb_rev = nil
+      @couchdb = Chef::CouchDB.new
+    end
+    
+    def name=(n)
+      @name = n.gsub(/\./, '_')
+    end
+    
+    # Set the password for this object.
+    def set_password(password, confirm_password=password) 
+      raise ArgumentError, "Passwords do not match" unless password == confirm_password
+      raise ArgumentError, "Password cannot be blank" if (password.nil? || password.length==0)
+      raise ArgumentError, "Password must be a minimum of 6 characters" if password.length < 6
+      @salt = generate_salt
+      @password = encrypt_password(@salt, password)      
+    end
+    
+    def set_openid(given_openid)
+      @openid = given_openid
+    end 
+    
+    def verify_password(given_password)
+      encrypt_password(@salt, given_password) == @password ? true : false
+    end 
+    
+    # Serialize this object as a hash 
+    def to_json(*a)
+      attributes = Hash.new
+      recipes = Array.new
+      result = {
+        'name' => @name,
+        'json_class' => self.class.name,
+        'salt' => @salt,
+        'password' => @password,
+        'openid' => @openid,
+        'admin' => @admin,
+        'chef_type' => 'webui_user',
+      }
+      result["_rev"] = @couchdb_rev if @couchdb_rev
+      result.to_json(*a)
+    end
+    
+    # Create a Chef::WebUIUser from JSON
+    def self.json_create(o)
+      me = new
+      me.name = o["name"]
+      me.salt = o["salt"]
+      me.password = o["password"]
+      me.openid = o["openid"]
+      me.admin = o["admin"]
+      me.couchdb_rev = o["_rev"] if o.has_key?("_rev")
+      me
+    end
+    
+    # List all the Chef::WebUIUser objects in the CouchDB.  If inflate is set to true, you will get
+    # the full list of all registration objects.  Otherwise, you'll just get the IDs
+    def self.cdb_list(inflate=false)
+      rs = Chef::CouchDB.new.list("users", inflate)
+      if inflate
+        rs["rows"].collect { |r| r["value"] }
+      else
+        rs["rows"].collect { |r| r["key"] }
+      end
+    end
+    
+    def self.list(inflate=false)
+      r = Chef::REST.new(Chef::Config[:chef_server_url])
+      if inflate
+        response = Hash.new
+        Chef::Search::Query.new.search(:user) do |n|
+          response[n.name] = n unless n.nil?
+        end
+        response
+      else
+        r.get_rest("users")
+      end
+    end
+    
+    # Load an WebUIUser by name from CouchDB
+    def self.cdb_load(name)
+      Chef::CouchDB.new.load("webui_user", name)
+    end
+    
+    # Load a User by name
+    def self.load(name)
+      r = Chef::REST.new(Chef::Config[:chef_server_url])
+      r.get_rest("users/#{name}")
+    end
+    
+    
+    # Whether or not there is an WebUIUser with this key.
+    def self.has_key?(name)
+      Chef::CouchDB.new.has_key?("webui_user", name)
+    end
+    
+    # Remove this WebUIUser from the CouchDB
+    def cdb_destroy
+      @couchdb.delete("webui_user", @name, @couchdb_rev)
+    end
+    
+    # Remove this WebUIUser via the REST API
+    def destroy
+      r = Chef::REST.new(Chef::Config[:chef_server_url])
+      r.delete_rest("users/#{@name}")
+    end
+    
+    # Save this WebUIUser to the CouchDB
+    def cdb_save
+      results = @couchdb.store("webui_user", @name, self)
+      @couchdb_rev = results["rev"]
+    end
+    
+    # Save this WebUIUser via the REST API
+    def save
+      r = Chef::REST.new(Chef::Config[:chef_server_url])
+      begin
+        r.put_rest("users/#{@name}", self)
+      rescue Net::HTTPServerException => e
+        if e.response.code == "404"
+          r.post_rest("users", self)
+        else
+          raise e
+        end
+      end
+      self
+    end
+    
+    # Create the WebUIUser via the REST API
+    def create
+      r = Chef::REST.new(Chef::Config[:chef_server_url])
+      r.post_rest("users", self)
+      self
+    end
+    
+    # Set up our CouchDB design document
+    def self.create_design_document
+      Chef::CouchDB.new.create_design_document("users", DESIGN_DOCUMENT)
+    end
+    
+    #return true if an admin user exists. this is pretty expensive (O(n)), should think of a better way (nuo)
+    def self.admin_exist
+      users = self.cdb_list
+      users.each do |u|
+        user = self.cdb_load(u)
+        if user.admin
+          return user.name
+        end
+      end
+      nil
+    end
+    
+    protected
+    
+      def generate_salt
+        salt = Time.now.to_s
+        chars = ("a".."z").to_a + ("A".."Z").to_a + ("0".."9").to_a
+        1.upto(30) { |i| salt << chars[rand(chars.size-1)] }
+        salt
+      end
+    
+      def encrypt_password(salt, password)
+        Digest::SHA1.hexdigest("--#{salt}--#{password}--")
+      end
+    
+  end
+end