111 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			111 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| require "json"
 | |
| require "db/exceptions/invalid_record"
 | |
| require "db/mongo/models/key"
 | |
| require "fileutils"
 | |
| 
 | |
| module Version2_0
 | |
|   class KeyRoutes < BaseRoutes
 | |
| 
 | |
|     def initialize wrapper
 | |
|       super wrapper
 | |
|       puts "Key routes initialized"
 | |
|     end
 | |
| 
 | |
|     before %r{\A/key(/[\w]+)?\z} do
 | |
|       if request.delete?
 | |
|         check_headers :accept
 | |
|       else
 | |
|         check_headers :accept, :content_type
 | |
|       end
 | |
|       check_privileges("key", "w")
 | |
|     end
 | |
| 
 | |
|     after %r{\A/key(/[\w]+)?\z} do
 | |
|       statistic
 | |
|     end
 | |
| 
 | |
|     # Get list of available ssh keys
 | |
|     #
 | |
|     # * *Request*
 | |
|     #   - method : GET
 | |
|     #   - headers :
 | |
|     #     - Accept: application/json
 | |
|     #
 | |
|     # * *Returns* : array of strings
 | |
|     #   [
 | |
|     #     {
 | |
|     #       "scope": "system", -> 'system' - key was added by server, 'user' - key was added by user
 | |
|     #       "id": "devops"
 | |
|     #     }
 | |
|     #   ]
 | |
|     get "/keys" do
 | |
|       check_headers :accept
 | |
|       check_privileges("key", "r")
 | |
|       keys = BaseRoutes.mongo.keys.map {|i| i.to_hash}
 | |
|       keys.each {|k| k.delete("path")}   # We should not return path to the key
 | |
|       json keys
 | |
|     end
 | |
| 
 | |
|     # Create ssh key on devops server
 | |
|     #
 | |
|     # * *Request*
 | |
|     #   - method : POST
 | |
|     #   - headers :
 | |
|     #     - Accept: application/json
 | |
|     #     - Content-Type: application/json
 | |
|     #   - body :
 | |
|     #     {
 | |
|     #       "file_name": "key file name",
 | |
|     #       "key_name": "key name",
 | |
|     #       "content": "key content"
 | |
|     #     }
 | |
|     #
 | |
|     # * *Returns* :
 | |
|     #   201 - Created
 | |
|     post "/key" do
 | |
|       key = create_object_from_json_body
 | |
|       fname = check_filename(key["file_name"], "Parameter 'file_name' must be a not empty string")
 | |
|       kname = check_string(key["key_name"], "Parameter 'key_name' should be a not empty string")
 | |
|       content = check_string(key["content"], "Parameter 'content' should be a not empty string")
 | |
|       file_name = File.join(DevopsService.config[:keys_dir], fname)
 | |
|       halt(400, "File '#{fname}' already exist") if File.exists?(file_name)
 | |
|       File.open(file_name, "w") do |f|
 | |
|         f.write(content)
 | |
|         f.chmod(0400)
 | |
|       end
 | |
| 
 | |
|       key = Key.new({"path" => file_name, "id" => kname})
 | |
|       BaseRoutes.mongo.key_insert key
 | |
|       create_response("Created", nil, 201)
 | |
|     end
 | |
| 
 | |
|     # Delete ssh key from devops server
 | |
|     #
 | |
|     # * *Request*
 | |
|     #   - method : DELETE
 | |
|     #   - headers :
 | |
|     #     - Accept: application/json
 | |
|     #
 | |
|     # * *Returns* :
 | |
|     #   200 - Deleted
 | |
|     delete "/key/:key" do
 | |
|       servers = BaseRoutes.mongo.servers_by_key params[:key]
 | |
|       unless servers.empty?
 | |
|         s_str = servers.map{|s| s.id}.join(", ")
 | |
|         raise DependencyError.new "Deleting is forbidden: Key is used in servers: #{s_str}"
 | |
|       end
 | |
| 
 | |
|       k = BaseRoutes.mongo.key params[:key]
 | |
|       begin
 | |
|         FileUtils.rm(k.path)
 | |
|       rescue
 | |
|         logger.error "Missing key file for #{params[:key]} - #{k.filename}"
 | |
|       end
 | |
|       r = BaseRoutes.mongo.key_delete params[:key]
 | |
|       return [500, r["err"].inspect] unless r["err"].nil?
 | |
|       create_response("Key '#{params[:key]}' removed")
 | |
|     end
 | |
| 
 | |
|   end
 | |
| end
 | 
