Adds Credential Manipulation

[pharos-tools.git] / laas-fog / pharoslaas / actions / vpn_make_user.py

diff --git a/laas-fog/pharoslaas/actions/vpn_make_user.py b/laas-fog/pharoslaas/actions/vpn_make_user.py
new file mode 100644 (file)
index 0000000..56cf9e6
--- /dev/null
+++ b/laas-fog/pharoslaas/actions/vpn_make_user.py
@@ -0,0 +1,61 @@
+##############################################################################
+# Copyright 2017 Parker Berberian and Others                                 #
+#                                                                            #
+# 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.                                             #
+##############################################################################
+
+import datetime
+from vpnAction import VPNAction
+import json
+
+
+class Make_VPN_User(VPNAction):
+
+    def __init__(self, config=None):
+        super(Make_VPN_User, self).__init__(config=config)
+
+    def run(self, booking=None, user=None, passwd=None):
+        if user == "None":
+            user = None
+        if passwd == "None":
+            passwd = None
+        name, passwd, dn = self.makeNewUser(name=user, passwd=passwd)
+        vpn_info = {}
+        vpn_info['dn'] = dn
+        vpn_info['username'] = name
+        vpn_info['password'] = passwd
+        now = datetime.date.today()
+        vpn_info['created'] = now.isoformat()  # 'YYYY-MM-DD' today
+        self.action_service.set_value(
+                name='vpn_'+name,
+                value=json.dumps(vpn_info),
+                local=False,
+                encrypt=True
+                )
+        if booking is not None:
+            self.addUserToBooking(vpn_info, booking)
+
+    def addUserToBooking(self, vpn_info, booking):
+        name = "booking_" + str(booking)
+        booking = json.loads(
+                self.action_service.get_value(
+                    name=name,
+                    local=False
+                    )
+                )
+        booking['vpn_key'] = "vpn_" + vpn_info['username']
+        self.action_service.set_value(
+                name=name,
+                value=json.dumps(booking),
+                local=False
+                )