From 409b565dc8067da25b0003dde51e6e2d427d322a Mon Sep 17 00:00:00 2001
From: Daniel Molina <danmolin@fdi.ucm.es>
Date: Fri, 14 Oct 2011 15:53:32 +0200
Subject: [PATCH] feature #873: Add UserOCCI and UserPoolOCCI classes

---
 src/cloud/occi/lib/UserOCCI.rb     | 64 ++++++++++++++++++++++++++++++
 src/cloud/occi/lib/UserPoolOCCI.rb | 43 ++++++++++++++++++++
 2 files changed, 107 insertions(+)
 create mode 100644 src/cloud/occi/lib/UserOCCI.rb
 create mode 100644 src/cloud/occi/lib/UserPoolOCCI.rb

diff --git a/src/cloud/occi/lib/UserOCCI.rb b/src/cloud/occi/lib/UserOCCI.rb
new file mode 100644
index 0000000000..e12b2cb7f7
--- /dev/null
+++ b/src/cloud/occi/lib/UserOCCI.rb
@@ -0,0 +1,64 @@
+# -------------------------------------------------------------------------- #
+# Copyright 2002-2011, OpenNebula Project Leads (OpenNebula.org)             #
+#                                                                            #
+# 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 'OpenNebula'
+
+include OpenNebula
+
+require 'quota'
+
+class UserOCCI < User
+    FORCE_USAGE = true
+
+    OCCI_USER = %q{
+        <USER href="<%= base_url %>/user/<%= self.id.to_s  %>">
+            <ID><%= self.id.to_s %></ID>
+            <NAME><%= self.name %></NAME>
+            <QUOTA>
+            <% user_quota.each { |key,value|
+                key_s = key.to_s.upcase
+                value_i = value.to_i %>
+                <<%= key_s %>><%= value_i %></<%= key_s %>>
+            <% } %>
+            </QUOTA>
+            <USAGE>
+            <% user_usage.each { |key,value|
+                key_s = key.to_s.upcase
+                value_i = value.to_i %>
+                <<%= key_s %>><%= value_i %></<%= key_s %>>
+            <% } %>
+            </USAGE>
+        </USER>
+    }
+
+    # Class constructor
+    def initialize(xml, client)
+        super(xml, client)
+    end
+
+    # Creates the OCCI representation of a User
+    def to_occi(base_url)
+        quota = Quota.new
+        user_usage = quota.get_usage(self.id, nil, FORCE_USAGE)
+        user_usage.delete(:uid)
+
+        user_quota = quota.get_quota(self.id)
+        user_quota.delete(:uid)
+
+        occi = ERB.new(OCCI_USER)
+        return occi.result(binding).gsub(/\n\s*/,'')
+    end
+end
diff --git a/src/cloud/occi/lib/UserPoolOCCI.rb b/src/cloud/occi/lib/UserPoolOCCI.rb
new file mode 100644
index 0000000000..acb745a5f7
--- /dev/null
+++ b/src/cloud/occi/lib/UserPoolOCCI.rb
@@ -0,0 +1,43 @@
+# -------------------------------------------------------------------------- #
+# Copyright 2002-2011, OpenNebula Project Leads (OpenNebula.org)             #
+#                                                                            #
+# 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 'OpenNebula'
+
+include OpenNebula
+
+class UserPoolOCCI < UserPool
+    OCCI_USER_POOL = %q{
+        <USER_COLLECTION>
+            <% self.each{ |user|  %>
+            <USER href="<%= base_url %>/user/<%= user.id.to_s  %>" name="<%= user.name  %>"/>
+            <% } %>
+        </USER_COLLECTION>
+    }
+
+
+    # Creates the OCCI representation of a User Pool
+    def to_occi(base_url)
+        begin
+            occi = ERB.new(OCCI_USER_POOL)
+            occi_text = occi.result(binding)
+        rescue Exception => e
+            error = OpenNebula::Error.new(e.message)
+            return error
+        end
+
+        return occi_text.gsub(/\n\s*/,'')
+    end
+end
\ No newline at end of file