From f24f20948cf7e6cc0e14c2b1fc41a61d8d2fa34c Mon Sep 17 00:00:00 2001
From: Devin Carlen <devin.carlen@gmail.com>
Date: Sat, 11 Sep 2010 11:19:22 -0700
Subject: Security Group API layer cleanup

---
 nova/db/api.py            |  5 +++
 nova/db/sqlalchemy/api.py | 11 +++++++
 nova/endpoint/cloud.py    | 84 ++++++++++++++++++++++++-----------------------
 3 files changed, 59 insertions(+), 41 deletions(-)

diff --git a/nova/db/api.py b/nova/db/api.py
index cdbd15486..cf39438c2 100644
--- a/nova/db/api.py
+++ b/nova/db/api.py
@@ -471,6 +471,11 @@ def security_group_get_by_instance(context, instance_id):
     return IMPL.security_group_get_by_instance(context, instance_id)
 
 
+def securitygroup_exists(context, project_id, group_name):
+    """Indicates if a group name exists in a project"""
+    return IMPL.securitygroup_exists(context, project_id, group_name)
+
+
 def security_group_create(context, values):
     """Create a new security group"""
     return IMPL.security_group_create(context, values)
diff --git a/nova/db/sqlalchemy/api.py b/nova/db/sqlalchemy/api.py
index f3d4b68c4..513b47bc9 100644
--- a/nova/db/sqlalchemy/api.py
+++ b/nova/db/sqlalchemy/api.py
@@ -667,8 +667,19 @@ def security_group_get_by_instance(_context, instance_id):
                       ).all()
 
 
+def securitygroup_exists(_context, project_id, group_name):
+    try:
+        group = securitygroup_get_by_name(_context, project_id, group_name)
+        return group != None
+    except exception.NotFound:
+        return False
+        
+
 def security_group_create(_context, values):
     security_group_ref = models.SecurityGroup()
+    # FIXME(devcamcar): Unless I do this, rules fails with lazy load exception
+    # once save() is called.  This will get cleaned up in next orm pass.
+    security_group_ref.rules 
     for (key, value) in values.iteritems():
         security_group_ref[key] = value
     security_group_ref.save()
diff --git a/nova/endpoint/cloud.py b/nova/endpoint/cloud.py
index a26f90753..7408e02e9 100644
--- a/nova/endpoint/cloud.py
+++ b/nova/endpoint/cloud.py
@@ -214,43 +214,40 @@ class CloudController(object):
         return True
 
     @rbac.allow('all')
-    def describe_security_groups(self, context, **kwargs):
-        groups = []
-        for group in db.security_group_get_by_project(context,
-                                                      context.project.id):
-            group_dict = {}
-            group_dict['groupDescription'] = group.description
-            group_dict['groupName'] = group.name
-            group_dict['ownerId'] = context.user.id
-            group_dict['ipPermissions'] = []
-            for rule in group.rules:
-                rule_dict = {}
-                rule_dict['ipProtocol'] = rule.protocol
-                rule_dict['fromPort'] = rule.from_port
-                rule_dict['toPort'] = rule.to_port
-                rule_dict['groups'] = []
-                rule_dict['ipRanges'] = []
-                if rule.group_id:
-                    source_group = db.security_group_get(context, rule.group_id)
-                    rule_dict['groups'] += [ { 'groupName': source_group.name,
-                                              'userId': source_group.user_id } ]
-                else:
-                    rule_dict['ipRanges'] += [ { 'cidrIp': rule.cidr } ]
-                group_dict['ipPermissions'] += [ rule_dict ]
-            groups += [ group_dict ]
+    def describe_security_groups(self, context, group_name=None, **kwargs):
+        if context.user.is_admin():
+            groups = db.security_group_get_all(context)
+        else:
+            groups = db.security_group_get_by_project(context,
+                                                      context.project.id)
+        groups = [self._format_security_group(context, g) for g in groups]
+        if not group_name is None:
+            groups = [g for g in groups if g.name in group_name]
 
         return {'securityGroupInfo': groups }
-#
-#                 [{ 'groupDescription': group.description,
-#                     'groupName' : group.name,
-#                     'ownerId': context.user.id,
-#                     'ipPermissions' : [
-#                        { 'ipProtocol' : rule.protocol,
-#                          'fromPort' : rule.from_port,
-#                          'toPort' : rule.to_port,
-#                          'ipRanges' : [ { 'cidrIp' : rule.cidr } ] } for rule in group.rules ] } for group in \
-#
-#        return groups
+
+    def _format_security_group(self, context, group):
+        g = {}
+        g['groupDescription'] = group.description
+        g['groupName'] = group.name
+        g['ownerId'] = context.user.id
+        g['ipPermissions'] = []
+        for rule in group.rules:
+            r = {}
+            r['ipProtocol'] = rule.protocol
+            r['fromPort'] = rule.from_port
+            r['toPort'] = rule.to_port
+            r['groups'] = []
+            r['ipRanges'] = []
+            if rule.group_id:
+                source_group = db.security_group_get(context, rule.group_id)
+                r['groups'] += [{'groupName': source_group.name,
+                                 'userId': source_group.user_id}]
+            else:
+                r['ipRanges'] += [{'cidrIp': rule.cidr}]
+            g['ipPermissions'] += [r]
+        return g
+
 
     @rbac.allow('netadmin')
     def revoke_security_group_ingress(self, context, group_name,
@@ -354,12 +351,17 @@ class CloudController(object):
 
     @rbac.allow('netadmin')
     def create_security_group(self, context, group_name, group_description):
-        db.security_group_create(context,
-                                 values = { 'user_id' : context.user.id,
-                                            'project_id': context.project.id,
-                                            'name': group_name,
-                                            'description': group_description })
-        return True
+        if db.securitygroup_exists(context, context.project.id, group_name):
+            raise exception.ApiError('group %s already exists' % group_name)
+        
+        group = {'user_id' : context.user.id,
+                 'project_id': context.project.id,
+                 'name': group_name,
+                 'description': group_description}
+        group_ref = db.security_group_create(context, group)
+
+        return {'securityGroupSet': [self._format_security_group(context,
+                                                                 group_ref)]}
 
     @rbac.allow('netadmin')
     def delete_security_group(self, context, group_name, **kwargs):
-- 
cgit