diff --git a/gns3server/handlers/index_handler.py b/gns3server/handlers/index_handler.py index 082e13ba..6dfd6c15 100644 --- a/gns3server/handlers/index_handler.py +++ b/gns3server/handlers/index_handler.py @@ -14,12 +14,15 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see . +import os +import aiohttp from gns3server.web.route import Route from gns3server.controller import Controller from gns3server.compute.port_manager import PortManager from gns3server.compute.project_manager import ProjectManager from gns3server.version import __version__ +from gns3server.utils.static import get_static_path class IndexHandler: @@ -64,6 +67,33 @@ class IndexHandler: response.template("project.html", project=controller.get_project(request.match_info["project_id"])) + @Route.get( + r"/static/web-ui/{filename:.+}", + parameters={ + "filename": "Static filename" + }, + status_codes={ + 200: "Static file returned", + 404: "Static cannot be found", + }, + raw=True, + description="Get static resource") + def webui(request, response): + filename = request.match_info["filename"] + filename = os.path.normpath(filename).strip("/") + filename = os.path.join('web-ui', filename) + + # Raise error if user try to escape + if filename[0] == ".": + raise aiohttp.web.HTTPForbidden() + + static = get_static_path(filename) + + if not os.path.exists(static): + static = get_static_path('web-ui/index.html') + + yield from response.file(static) + @Route.get( r"/v1/version", description="Old 1.0 API" diff --git a/gns3server/static/.gitkeep b/gns3server/static/.gitkeep new file mode 100644 index 00000000..e69de29b diff --git a/gns3server/templates/index.html b/gns3server/templates/index.html index e013d37d..1aef0dad 100644 --- a/gns3server/templates/index.html +++ b/gns3server/templates/index.html @@ -6,6 +6,7 @@

If you are looking for uploading the IOU. You can since 1.4 upload them directly from the client see: this documentation.

{% endblock %} diff --git a/gns3server/utils/static.py b/gns3server/utils/static.py new file mode 100644 index 00000000..004794ef --- /dev/null +++ b/gns3server/utils/static.py @@ -0,0 +1,38 @@ +#!/usr/bin/env python +# +# Copyright (C) 2018 GNS3 Technologies Inc. +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + +import os + + +def get_static_path(filename): + """ + Returns full static path for given filename + :param filename: relative filename + :return: absolute path + """ + + static_directory = get_static_dir() + return os.path.join(static_directory, filename) + + +def get_static_dir(): + """ + Returns location of static directory + :return: absolute path + """ + current_dir = os.path.dirname(os.path.abspath(__file__)) + return os.path.abspath(os.path.join(current_dir, '..', 'static')) \ No newline at end of file diff --git a/gns3server/web/response.py b/gns3server/web/response.py index f2a77529..0367c60b 100644 --- a/gns3server/web/response.py +++ b/gns3server/web/response.py @@ -117,6 +117,9 @@ class Response(aiohttp.web.Response): """ Return a file as a response """ + if not os.path.exists(path): + raise aiohttp.web.HTTPNotFound() + ct, encoding = mimetypes.guess_type(path) if not ct: ct = 'application/octet-stream' diff --git a/gns3server/web/web_server.py b/gns3server/web/web_server.py index 7cad075b..5c9b3248 100644 --- a/gns3server/web/web_server.py +++ b/gns3server/web/web_server.py @@ -29,6 +29,7 @@ import functools import time import atexit +from gns3server.utils.static import get_static_dir from .route import Route from ..config import Config from ..compute import MODULES @@ -274,6 +275,13 @@ class WebServer: m = module.instance() m.port_manager = PortManager.instance() + # adding static route + self._app.router.add_static( + '/static/', + path=get_static_dir(), + name='static' + ) + log.info("Starting server on {}:{}".format(self._host, self._port)) self._handler = self._app.make_handler() diff --git a/scripts/update-bundled-web-ui.sh b/scripts/update-bundled-web-ui.sh new file mode 100755 index 00000000..e48e45d6 --- /dev/null +++ b/scripts/update-bundled-web-ui.sh @@ -0,0 +1,67 @@ +#!/usr/bin/env bash + +# +# Copyright (C) 2018 GNS3 Technologies Inc. +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + +# +# Syncs WebUI with gns3server +# +# For updating with fresh latest repo just type: +# $ sh update-bundled-web-ui.sh +# +# It's also possible to update with custom repo and branch by: +# $ sh update-bundled-web-ui.sh ../my-custom-web-ui-repo/ +# + +CURRENT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )" +GNS3SERVER_DIR=$(realpath "$CURRENT_DIR/..") +REPO_DIR="/tmp/gns3-web-ui" +CUSTOM_REPO=false + +if [ $# -eq 1 ]; then + PARAM="$1" + CUSTOM_REPO=true + REPO_DIR=$(realpath "$PWD/${PARAM%/}") + echo "Custom repo dir: $REPO_DIR" +fi + +echo "Removing: $GNS3SERVER_DIR/gns3server/static/web-ui/*" + +rm -rf $GNS3SERVER_DIR/gns3server/static/web-ui/* + +echo "Re-create: $GNS3SERVER_DIR/gns3server/static/web-ui" + +mkdir -p "$GNS3SERVER_DIR/gns3server/static/web-ui/" + +if [ "$CUSTOM_REPO" = false ] ; then + if [ ! -d /tmp/gns3-web-ui ]; then + git clone https://github.com/GNS3/gns3-web-ui.git "$REPO_DIR" + fi +fi + +echo "Current working dir $REPO_DIR" + +cd "$REPO_DIR" + +yarn install +yarn ng build -e prod --base-href /static/web-ui/ + +cp -R $REPO_DIR/dist/* "$GNS3SERVER_DIR/gns3server/static/web-ui/" + +cd "$GNS3SERVER_DIR" + +#git add . +#git commit -m "Sync WebUI" diff --git a/tests/handlers/test_index.py b/tests/handlers/test_index.py index 385940f0..7aec57d3 100644 --- a/tests/handlers/test_index.py +++ b/tests/handlers/test_index.py @@ -15,13 +15,12 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see . - -import aiohttp import os -from unittest.mock import patch + from gns3server.version import __version__ from gns3server.controller import Controller +from gns3server.utils.static import get_static_path def test_index(http_root): @@ -50,6 +49,21 @@ def test_project(http_root, async_run): assert response.status == 200 +def test_web_ui(http_root, tmpdir): + tmpfile = get_static_path('web-ui/testing.txt') + with open(tmpfile, 'w+') as f: + f.write('world') + response = http_root.get('/static/web-ui/testing.txt') + assert response.status == 200 + os.remove(tmpfile) + + +def test_web_ui_not_found(http_root, tmpdir): + response = http_root.get('/static/web-ui/not-found.txt') + # should serve web-ui/index.html + assert response.status == 200 + + def test_v1(http_root): """ The old api v1 raise a 429 diff --git a/tests/utils/test_path.py b/tests/utils/test_path.py index c0b1c3c3..8cf6ce57 100644 --- a/tests/utils/test_path.py +++ b/tests/utils/test_path.py @@ -21,7 +21,6 @@ import aiohttp from gns3server.utils.path import check_path_allowed, get_default_project_directory -from gns3server.utils import force_unix_path def test_check_path_allowed(config, tmpdir): diff --git a/tests/utils/test_static.py b/tests/utils/test_static.py new file mode 100644 index 00000000..0174275c --- /dev/null +++ b/tests/utils/test_static.py @@ -0,0 +1,22 @@ +#!/usr/bin/env python +# +# Copyright (C) 2018 GNS3 Technologies Inc. +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + +from gns3server.utils.static import get_static_path + + +def test_get_static_path(): + assert get_static_path('test').endswith('gns3server/static/test') diff --git a/tests/web/test_response.py b/tests/web/test_response.py new file mode 100644 index 00000000..ad3c2361 --- /dev/null +++ b/tests/web/test_response.py @@ -0,0 +1,44 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2018 GNS3 Technologies Inc. +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + +import pytest + +from unittest.mock import MagicMock +from aiohttp.web import HTTPNotFound + +from gns3server.web.response import Response + + +@pytest.fixture() +def response(): + request = MagicMock() + return Response(request=request) + + +def test_response_file(async_run, tmpdir, response): + filename = str(tmpdir / 'hello') + with open(filename, 'w+') as f: + f.write('world') + + async_run(response.file(filename)) + assert response.status == 200 + + +def test_response_file_not_found(async_run, tmpdir, response): + filename = str(tmpdir / 'hello-not-found') + + pytest.raises(HTTPNotFound, lambda: async_run(response.file(filename)))