Early import project api

gns3server/controller/__init__.py

@@ -286,6 +286,20 @@ class Controller:
             yield from project.open()
         return project
 
+    def get_free_project_name(self, base_name):
+        """
+        Generate a free project name base on the base name
+        """
+        names = [ p.name for p in self._projects.values() ]
+        if base_name not in names:
+            return base_name
+        i = 1
+        while "{}-{}".format(base_name, i) in names:
+            i += 1
+            if i > 1000000:
+                raise aiohttp.web.HTTPConflict(text="A project name could not be allocated (node limit reached?)")
+        return "{}-{}".format(base_name, i)
+
     @property
     def projects(self):
         """

gns3server/controller/import_project.py

@@ -0,0 +1,69 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2016 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 <http://www.gnu.org/licenses/>.
+
+import os
+import json
+import uuid
+import asyncio
+import zipfile
+import aiohttp
+
+from ..config import Config
+
+
+"""
+Handle the import of project from a .gns3project
+"""
+
+@asyncio.coroutine
+def import_project(controller, project_id, stream, gns3vm=True):
+    """
+    Import a project contain in a zip file
+
+    You need to handle OSerror exceptions
+
+    :param stream: A io.BytesIO of the zipfile
+    :param gns3vm: True move Docker, IOU and Qemu to the GNS3 VM
+    :returns: Project
+    """
+    server_config = Config.instance().get_section_config("Server")
+    projects_path = os.path.expanduser(server_config.get("projects_path", "~/GNS3/projects"))
+    os.makedirs(projects_path, exist_ok=True)
+
+    with zipfile.ZipFile(stream) as myzip:
+
+        try:
+            topology = json.loads(myzip.read("project.gns3").decode())
+            # If the project name is already used we generate a new one
+            topology["name"] = controller.get_free_project_name(topology["name"])
+        except KeyError:
+            raise aiohttp.web.HTTPConflict(text="Can't import topology the .gns3 is corrupted or missing")
+
+        path = os.path.join(projects_path, topology["name"])
+        os.makedirs(path)
+        myzip.extractall(path)
+
+        dot_gns3_path = os.path.join(path, topology["name"] + ".gns3")
+        # We change the project_id to avoid erasing the project
+        topology["project_id"] = project_id
+        with open(dot_gns3_path, "w+") as f:
+            json.dump(topology, f, indent=4)
+        os.remove(os.path.join(path, "project.gns3"))
+
+    project = yield from controller.load_project(dot_gns3_path)
+    return project
+

gns3server/handlers/api/compute/project_handler.py

@@ -409,7 +409,7 @@ class ProjectHandler:
 
         # We write the content to a temporary location and after we extract it all.
         # It could be more optimal to stream this but it is not implemented in Python.
-        # Spooled means the file is temporary kept in memory until max_size is reached
+        # Spooled means the file is temporary kept in memory until max_size is reached
         try:
             with tempfile.SpooledTemporaryFile(max_size=10000) as temp:
                 while True:

gns3server/handlers/api/controller/project_handler.py

@@ -18,10 +18,12 @@
 import os
 import aiohttp
 import asyncio
-
+import tempfile
 
 from gns3server.web.route import Route
 from gns3server.controller import Controller
+from gns3server.controller.project import Project
+from gns3server.controller.import_project import import_project
 from gns3server.config import Config
 
 
@@ -251,6 +253,39 @@ class ProjectHandler:
 
         yield from response.write_eof()
 
+    @Route.post(
+        r"/projects/{project_id}/import",
+        description="Import a project from a portable archive",
+        parameters={
+            "project_id": "Project UUID",
+        },
+        raw=True,
+        output=PROJECT_OBJECT_SCHEMA,
+        status_codes={
+            200: "Project imported",
+            403: "Forbidden to import project"
+        })
+    def import_project(request, response):
+
+        controller = Controller.instance()
+
+        # We write the content to a temporary location and after we extract it all.
+        # It could be more optimal to stream this but it is not implemented in Python.
+        # Spooled means the file is temporary kept in memory until max_size is reached
+        try:
+            with tempfile.SpooledTemporaryFile(max_size=10000) as temp:
+                while True:
+                    packet = yield from request.content.read(512)
+                    if not packet:
+                        break
+                    temp.write(packet)
+                project = yield from import_project(controller, request.match_info["project_id"], temp, gns3vm=bool(int(request.GET.get("gns3vm", "1"))))
+        except OSError as e:
+            raise aiohttp.web.HTTPInternalServerError(text="Could not import the project: {}".format(e))
+
+        response.json(project)
+        response.set_status(201)
+
     @Route.get(
         r"/projects/{project_id}/files/{path:.+}",
         description="Get a file from a project. Beware you have warranty to be able to access only to file global to the project (for example README.txt)",
@@ -332,3 +367,5 @@ class ProjectHandler:
             raise aiohttp.web.HTTPNotFound()
         except PermissionError:
             raise aiohttp.web.HTTPForbidden()
+
+