Matteo Scandolo | 920e8fd | 2017-08-08 13:05:24 -0700 | [diff] [blame] | 1 | |
| 2 | # Copyright 2017-present Open Networking Foundation |
| 3 | # |
| 4 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | # you may not use this file except in compliance with the License. |
| 6 | # You may obtain a copy of the License at |
| 7 | # |
| 8 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | # |
| 10 | # Unless required by applicable law or agreed to in writing, software |
| 11 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | # See the License for the specific language governing permissions and |
| 14 | # limitations under the License. |
| 15 | |
| 16 | |
Matteo Scandolo | 21dde41 | 2017-07-11 18:54:12 -0700 | [diff] [blame] | 17 | from grpc_client.main import GRPC_Client |
| 18 | from klein import Klein |
Matteo Scandolo | 09d469c | 2017-07-07 11:46:48 -0700 | [diff] [blame] | 19 | import os |
| 20 | from tosca.parser import TOSCA_Parser |
| 21 | from tosca.default import TOSCA_DEFS_DIR |
| 22 | import json |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 23 | |
| 24 | BANNER = """ |
| 25 | _ ______ _____ __________ _____ _________ |
| 26 | | |/ / __ \/ ___/ /_ __/ __ \/ ___// ____/ | |
| 27 | | / / / /\__ \ / / / / / /\__ \/ / / /| | |
| 28 | / / /_/ /___/ / / / / /_/ /___/ / /___/ ___ | |
| 29 | /_/|_\____//____/ /_/ \____//____/\____/_/ |_| |
| 30 | """ |
| 31 | |
| 32 | class TOSCA_WebServer: |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 33 | |
Matteo Scandolo | 09d469c | 2017-07-07 11:46:48 -0700 | [diff] [blame] | 34 | current_dir = os.path.dirname(os.path.realpath(__file__)) |
| 35 | template_dir = os.path.join(current_dir, 'templates/') |
| 36 | |
Matteo Scandolo | 21dde41 | 2017-07-11 18:54:12 -0700 | [diff] [blame] | 37 | app = Klein() |
| 38 | |
| 39 | def execute_tosca(self, recipe): |
Matteo Scandolo | 1fedfae | 2017-10-09 13:57:00 -0700 | [diff] [blame] | 40 | self.parser.execute() |
| 41 | response_text = "Created models: %s" % str(self.parser.ordered_models_name) |
| 42 | return response_text |
| 43 | |
| 44 | def errorCallback(self, failure, request): |
| 45 | request.setResponseCode(500) |
Matteo Scandolo | 21dde41 | 2017-07-11 18:54:12 -0700 | [diff] [blame] | 46 | try: |
Matteo Scandolo | b2eff2c | 2018-03-02 12:58:10 -0800 | [diff] [blame^] | 47 | f = failure.getErrorMessage() |
| 48 | print "[XOS-TOSCA] Error while loading TOSCA: \n\n", f |
| 49 | return f |
Matteo Scandolo | 1fedfae | 2017-10-09 13:57:00 -0700 | [diff] [blame] | 50 | except: |
| 51 | print failure |
| 52 | return "Internal server error, please report this along with the failed recipe." |
Matteo Scandolo | 21dde41 | 2017-07-11 18:54:12 -0700 | [diff] [blame] | 53 | |
| 54 | @app.route('/', methods=['GET']) |
| 55 | def index(self, request): |
Matteo Scandolo | 09d469c | 2017-07-07 11:46:48 -0700 | [diff] [blame] | 56 | request.responseHeaders.addRawHeader(b"content-type", b"application/json") |
| 57 | tosca_defs = [f for f in os.listdir(TOSCA_DEFS_DIR) if not f.startswith('.')] |
| 58 | |
| 59 | response = {} |
| 60 | for d in tosca_defs: |
| 61 | name = d.replace('.yaml', '') |
| 62 | response[name] = "/custom_type/%s" % name |
| 63 | return json.dumps(response) |
Matteo Scandolo | 21dde41 | 2017-07-11 18:54:12 -0700 | [diff] [blame] | 64 | |
Matteo Scandolo | 78ca3eb | 2017-07-13 16:58:22 -0700 | [diff] [blame] | 65 | @app.route("/custom_type/<name>") |
| 66 | def custom_type(self, request, name): |
| 67 | request.responseHeaders.addRawHeader(b"content-type", b"text/plain") |
| 68 | custom_type = open(TOSCA_DEFS_DIR + '/' + name + '.yaml').read() |
| 69 | return custom_type |
| 70 | |
Matteo Scandolo | 21dde41 | 2017-07-11 18:54:12 -0700 | [diff] [blame] | 71 | @app.route('/run', methods=['POST']) |
Matteo Scandolo | 78ca3eb | 2017-07-13 16:58:22 -0700 | [diff] [blame] | 72 | def run(self, request): |
Matteo Scandolo | 21dde41 | 2017-07-11 18:54:12 -0700 | [diff] [blame] | 73 | recipe = request.content.read() |
| 74 | headers = request.getAllHeaders() |
| 75 | username = headers['xos-username'] |
| 76 | password = headers['xos-password'] |
| 77 | |
| 78 | d = GRPC_Client().create_secure_client(username, password, recipe) |
| 79 | self.parser = TOSCA_Parser(recipe, username, password) |
Matteo Scandolo | 1fedfae | 2017-10-09 13:57:00 -0700 | [diff] [blame] | 80 | tosca_execution = d.addCallback(self.execute_tosca) |
| 81 | tosca_execution.addErrback(self.errorCallback, request) |
Matteo Scandolo | 21dde41 | 2017-07-11 18:54:12 -0700 | [diff] [blame] | 82 | return d |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 83 | |
Matteo Scandolo | 78ca3eb | 2017-07-13 16:58:22 -0700 | [diff] [blame] | 84 | @app.route('/delete', methods=['POST']) |
| 85 | def delete(self, request): |
| 86 | recipe = request.content.read() |
| 87 | headers = request.getAllHeaders() |
| 88 | username = headers['xos-username'] |
| 89 | password = headers['xos-password'] |
| 90 | |
| 91 | d = GRPC_Client().create_secure_client(username, password, recipe) |
| 92 | self.parser = TOSCA_Parser(recipe, username, password, delete=True) |
Matteo Scandolo | 1fedfae | 2017-10-09 13:57:00 -0700 | [diff] [blame] | 93 | tosca_execution = d.addCallback(self.execute_tosca) |
| 94 | tosca_execution.addErrback(self.errorCallback, request) |
Matteo Scandolo | 78ca3eb | 2017-07-13 16:58:22 -0700 | [diff] [blame] | 95 | return d |
Matteo Scandolo | 09d469c | 2017-07-07 11:46:48 -0700 | [diff] [blame] | 96 | |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 97 | def __init__(self): |
Matteo Scandolo | 5c0af1b | 2017-07-05 14:51:21 -0700 | [diff] [blame] | 98 | self.app.run('0.0.0.0', '9102') |