Add install/uninstall extension points

Author: gonzalocasas

docs/plugins.rst

@@ -24,9 +24,13 @@ Category: ``booleans``
 Category: ``install``
 ^^^^^^^^^^^^^^^^^^^^^
 
-.. currentmodule:: compas_rhino.install
+.. currentmodule::
 
 * :func:`installable_rhino_packages`
+* :func:`after_rhino_install`
+
+.. currentmodule:: compas_rhino.uninstall
+* :func:`after_rhino_uninstall`
 
 
 Category: ``intersections``

src/compas/plugins.py

@@ -285,12 +285,12 @@ def wrapper(*args, **kwargs):
 
             # Collect all matching plugins
             elif selector == 'collect_all':
-                results = []
-
                 for plugin_impl in _collect_plugins(extension_point_url):
-                    results.append(plugin_impl.method(*args, **kwargs))
-
-                return results
+                    try:
+                        result = plugin_impl.method(*args, **kwargs)
+                        yield result
+                    except Exception as e:
+                        yield e
             else:
                 raise ValueError('Unexpected selector type. Must be either: first_match or collect_all')
 

src/compas_rhino/__init__.py

@@ -211,4 +211,5 @@ def _try_remove_bootstrapper(path):
 __all_plugins__ = [
     'compas_rhino.geometry.booleans',
     'compas_rhino.install',
+    'compas_rhino.uninstall',
 ]

src/compas_rhino/install.py

@@ -12,7 +12,12 @@
 import compas._os
 import compas.plugins
 
-__all__ = ['install']
+__all__ = [
+    'install',
+    'installable_rhino_packages',
+    'after_rhino_install',
+    'after_rhino_uninstall',
+]
 
 
 def install(version=None, packages=None):
@@ -84,8 +89,13 @@ def install(version=None, packages=None):
     symlinks = [(link['source_path'], link['link']) for link in symlinks_to_install]
     install_results = compas._os.create_symlinks(symlinks)
 
+    installed_packages = []
     for install_data, success in zip(symlinks_to_install, install_results):
-        result = 'OK' if success else 'ERROR: Cannot create symlink, try to run as administrator.'
+        if success:
+            installed_packages.append(install_data['name'])
+            result = 'OK'
+        else:
+            result = 'ERROR: Cannot create symlink, try to run as administrator.'
         results.append((install_data['name'], result))
 
     if not all(install_results):
@@ -106,10 +116,38 @@ def install(version=None, packages=None):
         if status != 'OK':
             exit_code = -1
 
+    if len(installed_packages):
+        print()
+        print('Running post-installation steps...')
+        print()
+        _run_post_execution_steps(after_rhino_install(installed_packages))
+
     print('\nCompleted.')
     if exit_code != 0:
         sys.exit(exit_code)
 
+def _run_post_execution_steps(steps_generator):
+    post_execution_errors = []
+    for result in steps_generator:
+        if isinstance(result, Exception):
+            post_execution_errors.append(result)
+            continue
+
+        for item in result:
+            try:
+                package, message, success = item
+                status = 'OK' if success else 'ERROR'
+                print('   {} {}: {}'.format(package.ljust(20), status, message))
+            except ValueError:
+                post_execution_errors.append(ValueError('Step successful, but result is wrongly formatted: {}'.format(str(result))))
+
+    if post_execution_errors:
+        print()
+        print('One or more errors occurred:')
+        print()
+        for error in post_execution_errors:
+            print('   - {}'.format(repr(error)))
+
 
 @compas.plugins.plugin(category='install', pluggable_name='installable_rhino_packages', tryfirst=True)
 def default_installable_rhino_packages():
@@ -143,6 +181,35 @@ def installable_rhino_packages():
     pass
 
 
+@compas.plugins.pluggable(category='install', selector='collect_all')
+def after_rhino_install(installed_packages):
+    """Allows extensions to execute actions after install to Rhino is done.
+
+    Extensions providing Rhino or Grasshopper features
+    can implement this pluggable interface to perform
+    additional steps after an installation to Rhino has
+    been completed.
+
+    Parameters
+    ----------
+    installed_packages : :obj:`list` of :obj:`str`
+        List of packages that have been installed successfully.
+
+    Examples
+    --------
+    >>> import compas.plugins
+    >>> @compas.plugins.plugin(category='install')
+    ... def after_rhino_install(installed_packages):
+    ...    # Do something after package is installed to Rhino, eg, copy components, etc
+    ...    return [('compas_ghpython', 'GH Components installed', True)]
+
+    Returns
+    -------
+    :obj:`list` of 3-tuple (str, str, bool)
+        List containing a 3-tuple with component name, message and ``True``/``False`` success flag.
+    """
+    pass
+
 def _update_bootstrapper(install_path, packages):
     # Take either the CONDA environment directory or the current Python executable's directory
     python_directory = os.environ.get('CONDA_PREFIX', None) or os.path.dirname(sys.executable)

src/compas_rhino/uninstall.py

@@ -6,11 +6,12 @@
 import os
 import sys
 
+import compas._os
+import compas.plugins
 import compas_rhino
+from compas_rhino.install import _run_post_execution_steps
 from compas_rhino.install import installable_rhino_packages
 
-import compas._os
-
 __all__ = ['uninstall']
 
 
@@ -68,8 +69,13 @@ def uninstall(version=None, packages=None):
     symlinks = [link['link'] for link in symlinks_to_uninstall]
     uninstall_results = compas._os.remove_symlinks(symlinks)
 
+    uninstalled_packages = []
     for uninstall_data, success in zip(symlinks_to_uninstall, uninstall_results):
-        result = 'OK' if success else 'ERROR: Cannot remove symlink, try to run as administrator.'
+        if success:
+            uninstalled_packages.append(uninstall_data['name'])
+            result = 'OK'
+        else:
+            result = 'ERROR: Cannot remove symlink, try to run as administrator.'
         results.append((uninstall_data['name'], result))
 
     if not all(uninstall_results):
@@ -92,6 +98,12 @@ def uninstall(version=None, packages=None):
         if status != 'OK':
             exit_code = -1
 
+    if len(uninstalled_packages):
+        print()
+        print('Running post-uninstallation steps...')
+        print()
+        _run_post_execution_steps(after_rhino_uninstall(uninstalled_packages))
+
     print('\nUninstall completed.')
     if exit_code != 0:
         sys.exit(exit_code)
@@ -126,6 +138,36 @@ def _filter_installed_packages(version, packages):
     return packages
 
 
+@compas.plugins.pluggable(category='install', selector='collect_all')
+def after_rhino_uninstall(uninstalled_packages):
+    """Allows extensions to execute actions after uninstall from Rhino is done.
+
+    Extensions providing Rhino or Grasshopper features
+    can implement this pluggable interface to perform
+    additional steps after the uninstall from Rhino has
+    been completed.
+
+    Parameters
+    ----------
+    uninstalled_packages : :obj:`list` of :obj:`str`
+        List of packages that have been uninstalled.
+
+    Examples
+    --------
+    >>> import compas.plugins
+    >>> @compas.plugins.plugin(category='install')
+    ... def after_rhino_uninstall(uninstalled_packages):
+    ...    # Do something cleanup, eg remove copied files.
+    ...    return [('compas_ghpython', 'GH Components uninstalled', True)]
+
+    Returns
+    -------
+    :obj:`list` of 3-tuple (str, str, bool)
+        List containing a 3-tuple with component name, message and ``True``/``False`` success flag.
+    """
+    pass
+
+
 # ==============================================================================
 # Main
 # ==============================================================================