From 869a284bae731478ae2d276b29653fe26180493b Mon Sep 17 00:00:00 2001
From: Mika Haulo <mika@seravo.fi>
Date: Wed, 3 May 2017 14:35:03 +0300
Subject: [PATCH 1/2] Merge WP-palvelu Instance Switcher plugin to this plugin

---
 js/instance_switcher.js       |  17 ++++
 modules/instance_switcher.php | 172 ++++++++++++++++++++++++++++++++++
 seravo-plugin.php             |   7 ++
 style/instance_switcher.css   |   4 +
 4 files changed, 200 insertions(+)
 create mode 100644 js/instance_switcher.js
 create mode 100644 modules/instance_switcher.php
 create mode 100644 style/instance_switcher.css

diff --git a/js/instance_switcher.js b/js/instance_switcher.js
new file mode 100644
index 00000000..f9f50abe
--- /dev/null
+++ b/js/instance_switcher.js
@@ -0,0 +1,17 @@
+(function(){
+  var links = document.querySelectorAll('#wp-admin-bar-wpis li > a');
+  var listener = function(e){
+    e.preventDefault();
+    var instance = e.target.getAttribute('href').substr(1);
+    if (instance === 'exit') {
+      document.cookie = "wpp_shadow=;path=/";
+    } else {
+      document.cookie = "wpp_shadow=" + instance + ";path=/";
+    }
+    location.reload();
+  };
+  
+  for(var i = 0; i < links.length; i++){
+    links[i].addEventListener('click', listener);
+  }
+})();
diff --git a/modules/instance_switcher.php b/modules/instance_switcher.php
new file mode 100644
index 00000000..9ad739e1
--- /dev/null
+++ b/modules/instance_switcher.php
@@ -0,0 +1,172 @@
+<?php
+/*
+ * Plugin name: InstanceSwitcher
+ * Description: Enable users to manage their Seravo WordPress updates
+ * Version: 1.0
+ */
+
+namespace Seravo;
+
+if (!class_exists('InstanceSwitcher')) {
+  class InstanceSwitcher {
+
+    public static function load() {
+      // only run the instance switcher when in a container environment
+      if( ! getenv('CONTAINER') ) {
+        return;
+      }
+
+      // admin ajax action
+      add_action( 'wp_ajax_wpis_change_container', array( 'Seravo\InstanceSwitcher', 'change_wp_container' ) );
+      add_action( 'wp_ajax_nopriv_wpis_change_container', array( 'Seravo\InstanceSwitcher', 'change_wp_container' ) );
+      
+      // styles and scripts for the switcher
+      add_action( 'admin_enqueue_scripts', array( 'Seravo\InstanceSwitcher', 'assets' ), 999);
+      add_action( 'wp_enqueue_scripts', array( 'Seravo\InstanceSwitcher', 'assets' ), 999);
+      
+      // add the instance switcher menu
+      add_action( 'admin_bar_menu', array( 'Seravo\InstanceSwitcher', 'add_switcher' ), 999 );
+
+      // display a notice at the bottom of the window when in a shadow
+      if ( getenv('WP_ENV') && getenv('WP_ENV') != 'production' ) {
+        add_action('admin_footer', array( 'Seravo\InstanceSwitcher', 'render_shadow_indicator' ) );
+        add_action('wp_footer', array( 'Seravo\InstanceSwitcher', 'render_shadow_indicator' ) );
+        add_action('login_footer', array( 'Seravo\InstanceSwitcher', 'render_shadow_indicator' ) );
+        add_action('admin_notices', array( 'Seravo\InstanceSwitcher', 'render_shadow_admin_notice' ) );
+      }
+    }
+
+    public static function register_updates_page() {
+      if ( getenv('SERAVO_API_KEY') != '' ) {
+        add_submenu_page( 'tools.php', 'Instance Switcher', 'Instance Switcher', 'manage_options', 'updates_page', array(__CLASS__, 'load_updates_page') );
+      }
+    }
+
+    public static function load_updates_page() {
+      require_once(dirname( __FILE__ ) . '/../lib/updates-page.php');
+    }
+
+    /**
+    * Load javascript and stylesheets for the switcher
+    */
+    public static function assets(){
+      if ( !function_exists( 'is_admin_bar_showing' ) ) {
+        return;
+      }
+
+      // use this within the admin bar
+      if ( !is_admin_bar_showing() ) {
+        return;
+      }
+      
+      wp_enqueue_script( 'wpisjs', plugins_url( '../js/instance_switcher.js' , __FILE__), null, null, true );
+      wp_enqueue_style( 'wpisjs', plugins_url( '../style/instance_switcher.css' , __FILE__), null, null, 'all' );
+    }
+
+    /**
+    * Create the menu itself
+    */
+    public static function add_switcher(  $wp_admin_bar ){
+      
+      if ( ! function_exists( 'is_admin_bar_showing' ) ) {
+        return;
+      }
+
+      // use this within the admin bar
+      if ( ! is_admin_bar_showing() ) {
+        return;
+      }
+
+      // check permissions
+      if( ! current_user_can( 'activate_plugins' )){
+        return;
+      }
+      
+      
+      $instances = get_defined_constants();
+
+      // get the wpis specific constants
+      foreach( $instances as $key => $constant ) {
+        if( ! preg_match( '#WPIS-#', $key ) ) {
+          unset( $instances[$key] );
+        }
+      }
+      
+
+      if( empty( $instances ) ) {
+        return;
+      }
+
+      $id = 'wpis';
+      $current_instance = getenv('CONTAINER');
+
+      $instance_index = strpos($current_instance,'_') + 1;
+      for( $x = 0 ; $x < $instance_index; ++$x ) {
+        $current_instance = substr($current_instance, 1);
+      }
+
+      // define the name of the current instance to be shown in the bar
+      foreach( $instances as $key => $instance ) {
+        if($current_instance == $instance){
+          $current_instance = substr($key, 5);
+        }
+      }
+
+      $domain = ""; //$this->get_domain( $_SERVER['HTTP_HOST'] );
+
+      if ( getenv('WP_ENV') && getenv('WP_ENV') != 'production' ) {
+        $menuclass = 'wpis-warning';
+      }
+
+      // create the parent menu here
+      $wp_admin_bar->add_menu([
+        'id' => $id,
+        'title' => $current_instance,
+        'href' => '#',
+        'meta' => [
+          'class' => $menuclass,
+        ],
+      ]);
+
+      // add menu entries for each shadow
+      foreach($instances as $key => $instance) {
+        $wp_admin_bar->add_menu([
+          'parent' => $id,
+          'title' => substr($key, 5),
+          'id' => $instance,
+          'href' => "#$instance",
+        ]);
+      }
+
+      // Last item is always to exit shadow
+      $wp_admin_bar->add_menu(array(
+        'parent' => $id,
+        'title' => __('Exit Shadow', 'seravo-plugin'),
+        'id' => 'exit-shadow',
+        'href' => "#exit",
+      ));
+    }
+    
+    public static function render_shadow_indicator() {
+?>
+      <style>#shadow-indicator { font-family: Arial, sans-serif; position: fixed; bottom: 0; left: 0; right: 0; width: 100%; color: #fff; background: #cc0000; z-index: 3000; font-size:16px; line-height: 1; text-align: center; padding: 5px } #shadow-indicator a.clearlink { text-decoration: underline; color: #fff; }</style>
+      <div id="shadow-indicator">
+      <?php echo wp_sprintf( __('You are currently in %s.', 'seravo-plugin'), getenv( 'WP_ENV' ) ); ?> <a class="clearlink" href="/?wpp_shadow=clear"><?php _e('Exit', 'seravo-plugin'); ?></a>
+      </div>
+<?php
+    }
+    
+    /**
+    * Let plugins or themes display admin notice when inside a shadow
+    */
+    public static function render_shadow_admin_notice( $current_screen ) {
+      $current_screen = get_current_screen();
+      $admin_notice_content = apply_filters( 'seravo_instance_switcher_admin_notice', '', $current_screen );
+      if(!empty($admin_notice_content)) {
+        echo $admin_notice_content;
+      }
+    }
+  }
+
+  InstanceSwitcher::load();
+}
diff --git a/seravo-plugin.php b/seravo-plugin.php
index 84aa2d09..b516affc 100644
--- a/seravo-plugin.php
+++ b/seravo-plugin.php
@@ -167,7 +167,14 @@ public static function loadAllModules() {
      */
     if (apply_filters('seravo_show_domains_page',true)) {
       require_once(dirname( __FILE__ ) . '/modules/domains.php');
+    
+    /*
+     * Instance switcher
+     */
+    if ( apply_filters('seravo_show_instance_switcher', true) ) {
+      require_once(dirname( __FILE__ ) . '/modules/instance_switcher.php');
     }
+
   }
 }
 
diff --git a/style/instance_switcher.css b/style/instance_switcher.css
new file mode 100644
index 00000000..69b04215
--- /dev/null
+++ b/style/instance_switcher.css
@@ -0,0 +1,4 @@
+#wp-admin-bar-wpis.wpis-warning > a {
+  background-color: #cc0000;
+}
+

From 48413cd269087bd27961a65c73e69fbdd96678f1 Mon Sep 17 00:00:00 2001
From: Mika Haulo <mika@seravo.fi>
Date: Fri, 5 May 2017 01:34:18 +0300
Subject: [PATCH 2/2] Get shadow list from SWD API

---
 modules/instance_switcher.php | 63 +++++++++++++++++++----------------
 seravo-plugin.php             |  1 +
 2 files changed, 36 insertions(+), 28 deletions(-)

diff --git a/modules/instance_switcher.php b/modules/instance_switcher.php
index 9ad739e1..01cefbe2 100644
--- a/modules/instance_switcher.php
+++ b/modules/instance_switcher.php
@@ -62,6 +62,28 @@ public static function assets(){
       wp_enqueue_script( 'wpisjs', plugins_url( '../js/instance_switcher.js' , __FILE__), null, null, true );
       wp_enqueue_style( 'wpisjs', plugins_url( '../style/instance_switcher.css' , __FILE__), null, null, 'all' );
     }
+    
+    public static function load_shadow_list(){
+      if ( ( $shadow_list = get_transient( 'shadow_list' ) ) === false ) {
+        $site = getenv('USER');
+        $ch = curl_init('http://localhost:8888/v1/site/' . $site . '/shadows');
+        curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
+        curl_setopt($ch, CURLOPT_HTTPHEADER, array('X-Api-Key: ' . getenv('SERAVO_API_KEY')));
+        $response = curl_exec($ch);
+        $httpcode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
+      
+        if (curl_error($ch) || $httpcode != 200) {
+          error_log('SWD API error '. $httpcode .': '. curl_error($ch));
+          die('API call failed: ' . curl_error($ch));
+        }
+      
+        curl_close($ch);
+        $shadow_list = json_decode($response, true);
+        set_transient( 'shadow_list', $shadow_list, 10 * MINUTE_IN_SECONDS );
+      }
+      
+      return $shadow_list;
+    }
 
     /**
     * Create the menu itself
@@ -82,46 +104,25 @@ public static function add_switcher(  $wp_admin_bar ){
         return;
       }
       
-      
-      $instances = get_defined_constants();
-
-      // get the wpis specific constants
-      foreach( $instances as $key => $constant ) {
-        if( ! preg_match( '#WPIS-#', $key ) ) {
-          unset( $instances[$key] );
-        }
-      }
-      
+      $instances = InstanceSwitcher::load_shadow_list();
 
       if( empty( $instances ) ) {
         return;
       }
 
       $id = 'wpis';
-      $current_instance = getenv('CONTAINER');
-
-      $instance_index = strpos($current_instance,'_') + 1;
-      for( $x = 0 ; $x < $instance_index; ++$x ) {
-        $current_instance = substr($current_instance, 1);
-      }
-
-      // define the name of the current instance to be shown in the bar
-      foreach( $instances as $key => $instance ) {
-        if($current_instance == $instance){
-          $current_instance = substr($key, 5);
-        }
-      }
-
       $domain = ""; //$this->get_domain( $_SERVER['HTTP_HOST'] );
 
       if ( getenv('WP_ENV') && getenv('WP_ENV') != 'production' ) {
         $menuclass = 'wpis-warning';
       }
+      
+      $current_title = getenv('WP_ENV');
 
       // create the parent menu here
       $wp_admin_bar->add_menu([
         'id' => $id,
-        'title' => $current_instance,
+        'title' => $current_title,
         'href' => '#',
         'meta' => [
           'class' => $menuclass,
@@ -130,11 +131,17 @@ public static function add_switcher(  $wp_admin_bar ){
 
       // add menu entries for each shadow
       foreach($instances as $key => $instance) {
+        $title = $instance["env"];
+      
+        if ( strlen( $instance["info"] ) > 0 ) {
+          $title .= " (" . $instance["info"] . ")";
+        }
+      
         $wp_admin_bar->add_menu([
           'parent' => $id,
-          'title' => substr($key, 5),
-          'id' => $instance,
-          'href' => "#$instance",
+          'title' => $title,
+          'id' => $instance["name"],
+          'href' => "#" . substr($instance["name"], 6),
         ]);
       }
 
diff --git a/seravo-plugin.php b/seravo-plugin.php
index b516affc..ef61a4a4 100644
--- a/seravo-plugin.php
+++ b/seravo-plugin.php
@@ -167,6 +167,7 @@ public static function loadAllModules() {
      */
     if (apply_filters('seravo_show_domains_page',true)) {
       require_once(dirname( __FILE__ ) . '/modules/domains.php');
+    }
     
     /*
      * Instance switcher