serverdensity-heroku 0.0.3

data/bin/plugins.py

@@ -0,0 +1,315 @@
+#!/usr/bin/env python
+
+"""
+Classes for plugin download, installation, and registration.
+
+"""
+
+
+import ConfigParser
+import os
+import platform
+import urllib, urllib2
+from optparse import OptionParser
+from zipfile import ZipFile
+
+
+BASE_URL = 'http://plugins.serverdensity.com/'
+
+python_version = platform.python_version_tuple()
+
+if int(python_version[1]) >= 6:
+    import json
+else:
+    import minjson
+    json = None
+
+
+class App(object):
+    """
+    Class for collecting arguments and options for the plugin
+    download and installation process.
+
+    """
+    def __init__(self):
+        usage = 'usage: %prog [options] key'
+        self.parser = OptionParser(usage=usage)
+        self.parser.add_option('-v', '--verbose', action='store_true', dest='verbose',
+                               default=False, help='run in verbose mode')
+        self.parser.add_option('-r', '--remove', action='store_true', dest='remove',
+                               default=False, help='remove plugin')
+        self.parser.add_option('-u', '--update', action='store_true', dest='update',
+                               default=False, help='update installed plugins')
+
+    def run(self):
+        """
+        Entry point to the plugin helper application.
+
+        """
+        (options, args) = self.parser.parse_args()
+        if len(args) != 1:
+            if options.update:
+                updater = PluginUpdater(verbose=options.verbose)
+                updater.start()
+                return
+            else:
+                self.parser.error('incorrect number of arguments')
+        if options.remove:
+            remover = PluginRemover(key=args[0], verbose=options.verbose)
+            remover.start()
+        else:
+            downloader = PluginDownloader(key=args[0], verbose=options.verbose)
+            downloader.start()
+
+class PluginMetadata(object):
+    def __init__(self, downloader=None):
+        self.downloader = downloader
+
+    def get(self):
+        raise Exception, 'sub-classes to provide implementation.'
+
+    def json(self):
+        metadata = self.get()
+        if self.downloader.verbose:
+            print metadata
+        if json:
+            return json.loads(metadata)
+        else:
+            return minjson.safeRead(metadata)
+
+class FilePluginMetadata(PluginMetadata):
+    """
+    File-based metadata provider, for testing purposes.
+
+    """
+    def get(self):
+        path = os.path.join(os.path.dirname(__file__), 'tests/plugin.json')
+        if self.downloader.verbose:
+            print 'reading plugin data from %s' % path
+        f = open(path, 'r')
+        data = f.read()
+        f.close()
+        return data
+
+class WebPluginMetadata(PluginMetadata):
+    """
+    Web-based metadata provider.
+
+    """
+    def __init__(self, downloader=None, agent_key=None):
+        super(WebPluginMetadata, self).__init__(downloader=downloader)
+        self.agent_key = agent_key
+
+    def get(self):
+        url = '%sinstall/' % BASE_URL
+        data = {
+            'installId': self.downloader.key,
+            'agentKey': self.agent_key
+        }
+        if self.downloader.verbose:
+            print 'sending %s to %s' % (data, url)
+        request = urllib2.urlopen(url, urllib.urlencode(data))
+        response = request.read()
+        return response
+
+class Action(object):
+    def __init__(self, key=None, verbose=True):
+        self.key = key
+        self.verbose = verbose
+
+    def start(self):
+        raise Exception, 'sub-classes to provide implementation.'
+
+class PluginUpdater(Action):
+    def __init__(self, verbose=True):
+        super(PluginUpdater, self).__init__(verbose=verbose)
+
+    def __get_installs(self):
+        url = '%supdate/' % BASE_URL
+        data = {
+            'agentKey': self.config.agent_key
+        }
+        request = urllib2.urlopen(url, urllib.urlencode(data))
+        response = request.read()
+        if json:
+            return json.loads(response)
+        else:
+            return minjson.safeRead(response)
+
+    def start(self):
+        self.config = AgentConfig(action=self)
+        if self.verbose:
+            print 'updating plugins'
+        installs = self.__get_installs()
+        for install_id in installs['installIds']:
+            PluginDownloader(key=install_id, verbose=self.verbose).start()
+
+class PluginRemover(Action):
+    """
+    Class for removing a plugin.
+
+    """
+    def __init__(self, key=None, verbose=True):
+        super(PluginRemover, self).__init__(key=key, verbose=verbose)
+
+    def __send_removal(self):
+        url = '%suninstall/' % BASE_URL
+        data = {
+            'installId': self.key,
+            'agentKey': self.config.agent_key
+        }
+        request = urllib2.urlopen(url, urllib.urlencode(data))
+        response = request.read()
+        if json:
+            return json.loads(response)
+        else:
+            return minjson.safeRead(response)
+
+    def __remove_file(self, name):
+        name = '%s.py' % name
+        path = os.path.join(self.config.plugin_path, name)
+        if self.verbose:
+            print 'removing %s' % path
+        os.remove(path)
+
+    def start(self):
+        self.config = AgentConfig(action=self)
+        if self.verbose:
+            print 'removing plugin with install key:', self.key
+        response = self.__send_removal()
+        if self.verbose:
+            print 'retrieved remove response.'
+        assert 'status' in response, 'response is not valid.'
+        if 'status' in response and response['status'] == 'error':
+            raise Exception, response['msg']
+        self.__remove_file(response['name'])
+        print 'plugin removed successfully.'
+
+class PluginDownloader(Action):
+    """
+    Class for downloading a plugin.
+
+    """
+    def __init__(self, key=None, verbose=True):
+        super(PluginDownloader, self).__init__(key=key, verbose=verbose)
+
+    def __prepare_plugin_directory(self):
+        if not os.path.exists(self.config.plugin_path):
+            if self.verbose:
+                print '%s does not exist, creating' % self.config.plugin_path
+            os.mkdir(self.config.plugin_path)
+            if self.verbose:
+                print '%s created' % self.config.plugin_path
+        elif self.verbose:
+            print '%s exists' % self.config.plugin_path
+
+    def __download(self):
+        self.url = '%sdownload/%s/%s/' % (BASE_URL, self.key, self.config.agent_key)
+        if self.verbose:
+            print 'downloading for agent %s: %s' % (self.config.agent_key, self.url)
+        request = urllib2.urlopen(self.url)
+        data = request.read()
+        path = os.path.join(self.config.plugin_path, '%s.zip' % self.key)
+        f = open(path, 'w')
+        f.write(data)
+        f.close()
+        z = ZipFile(path, 'r')
+        
+        try:
+            if json:
+                if self.verbose:
+                    print 'extract all: %s' % (os.path.dirname(path))
+                z.extractall(os.path.dirname(path))
+            else:
+                for name in z.namelist():
+                    if self.verbose:
+                        print 'extract loop: %s' % (os.path.join(os.path.dirname(path), name))
+                    data = z.read(name)
+                    f = open(os.path.join(os.path.dirname(path), name), 'w')
+                    f.write(data)
+                    f.close()
+        
+        except Exception, ex:
+            print ex
+
+        z.close()
+        os.remove(path)
+
+    def start(self):
+        self.config = AgentConfig(action=self)
+        metadata = WebPluginMetadata(self, agent_key=self.config.agent_key).json()
+        if self.verbose:
+            print 'retrieved metadata.'
+        assert 'configKeys' in metadata or 'status' in metadata, 'metadata is not valid.'
+        if 'status' in metadata and metadata['status'] == 'error':
+            raise Exception, metadata['msg']
+        self.__prepare_plugin_directory()
+        self.__download()
+        self.config.prompt(metadata['configKeys'])
+        print 'plugin installed; please restart your agent'
+
+class AgentConfig(object):
+    """
+    Class for writing new config options to sd-agent config.
+
+    """
+    def __init__(self, action=None):
+        self.action = action
+        self.path = self.__get_config_path()
+        assert self.path, 'no config path found.'
+        self.config = self.__parse()
+        if self.config.get('Main', 'plugin_directory'):
+            self.plugin_path = self.config.get('Main', 'plugin_directory')
+        self.agent_key = self.config.get('Main', 'agent_key')
+        assert self.agent_key, 'no agent key.'
+
+    def __get_config_path(self):
+        paths = (
+            '/etc/sd-agent/config.cfg',
+            os.path.join(os.path.dirname(__file__), 'config.cfg')
+        )
+        for path in paths:
+            if os.path.exists(path):
+                if self.action.verbose:
+                    print 'found config at %s' % path
+                return path
+
+    def __parse(self):
+        if os.access(self.path, os.R_OK) == False:
+            if self.action.verbose:
+                print 'cannot access config'
+            raise Exception, 'cannot access config'
+        if self.action.verbose:
+            print 'found config, parsing'
+        config = ConfigParser.ConfigParser()
+        config.read(self.path)
+        if self.action.verbose:
+            print 'parsed config'
+        return config
+
+    def __write(self, values):
+        for key in values.keys():   
+            self.config.set('Main', key, values[key])
+        try:
+            f = open(self.path, 'w')
+            self.config.write(f)
+            f.close()
+        except Exception, ex:
+            print ex
+            sys.exit(1)
+
+    def prompt(self, options):
+        if not options:
+            return
+        values = {}
+        for option in options:
+            if not self.config.has_option('Main', option):
+                values[option] = raw_input('value for %s: ' % option)
+        self.__write(values)
+
+if __name__ == '__main__':
+    try:
+        app = App()
+        app.run()
+    except Exception, ex:
+        print 'error: %s' % ex

data/bin/sd-agent-pkg.init

@@ -0,0 +1,85 @@
+#!/bin/sh
+#
+###############################################################################
+# sd-agent
+#
+# Written by Boxed Ice <hello@boxedice.com>
+# A server monitoring daemon for www.serverdensity.com
+#
+# Licensed under Simplified BSD License (see LICENSE)
+#
+###############################################################################
+#
+# chkconfig: 345 85 15
+# description: Server Density Monitoring Agent
+
+### BEGIN INIT INFO
+# Provides: sd-agent
+# Short-Description: Start and start sd-agent
+# Description: sd-agent is the monitoring agent component for Server Density
+# Required-Start:
+# Required-Stop:
+# Default-Start: 2 3 4 5
+# Default-Stop: 0 1 6
+### END INIT INFO
+
+AGENTPATH="/usr/bin/sd-agent/agent.py"
+AGENTUSER="sd-agent"
+PIDPATH="/var/run/sd-agent/"
+
+[ -f $AGENTPATH ] || echo "/usr/bin/sd-agent not found"
+
+# Source function library.
+if [ -f /etc/init.d/functions ]; then
+        . /etc/init.d/functions
+fi
+
+if [ -f /etc/SuSE-release ]; then
+        . /etc/rc.status
+        rc_reset
+fi
+
+# Action to take
+case "$1" in
+  start)
+        if [ ! -d $PIDPATH ]; then
+            mkdir -p $PIDPATH
+            chown sd-agent:sd-agent $PIDPATH
+        fi
+
+        su $AGENTUSER -c "python $AGENTPATH stop init"
+        su $AGENTUSER -c "python $AGENTPATH start init --clean"
+
+        if [ -f /etc/SuSE-release ]; then
+                rc_status -v
+        elif [ -f /etc/debian_version ] || [ -f /etc/lsb-release ] || [ -f /etc/gentoo-release ]; then
+                echo " Started"
+        else
+                success
+                echo
+        fi
+        echo
+    ;;
+  stop)
+        su $AGENTUSER -c "python $AGENTPATH stop init"
+
+        if [ -f /etc/SuSE-release ]; then
+                rc_status -v
+        elif [ -f /etc/debian_version ] || [ -f /etc/lsb-release ] || [ -f /etc/gentoo-release ]; then
+                echo " Stopped"
+        else
+                success
+                echo
+        fi
+        echo
+    ;;
+  restart)
+        $0 stop
+        $0 start
+    ;;
+  *)
+        echo "Usage: /etc/init.d/sd-agent start|stop|restart"
+        exit 1
+esac
+
+exit 0

data/bin/sd-agent.init

@@ -0,0 +1,73 @@
+#!/bin/sh
+#
+###############################################################################
+# sd-agent
+#
+# Written by Boxed Ice <hello@boxedice.com>
+# A server monitoring daemon for www.serverdensity.com
+#
+# Licensed under Simplified BSD License (see LICENSE)
+#
+###############################################################################
+#
+# chkconfig: 345 85 15
+# description: Server Density Monitoring Agent
+
+### BEGIN INIT INFO
+# Provides: sd-agent
+# Short-Description: Start and start sd-agent
+# Description: sd-agent is the monitoring agent component for Server Density
+### END INIT INFO
+
+AGENTPATH="/usr/bin/sd-agent/agent.py"
+
+[ -f $AGENTPATH ] || echo "/usr/bin/sd-agent not found"
+
+# Source function library.
+if [ -f /etc/init.d/functions ]; then
+        . /etc/init.d/functions
+fi
+
+if [ -f /etc/SuSE-release ]; then
+        . /etc/rc.status
+        rc_reset
+fi
+
+# Action to take
+case "$1" in
+  start)
+        python $AGENTPATH stop init
+        python $AGENTPATH start init --clean
+        if [ -f /etc/SuSE-release ]; then
+                rc_status -v
+        elif [ -f /etc/debian_version ] || [ -f /etc/lsb-release ] || [ -f /etc/gentoo-release ]; then
+                echo " Started"
+        else
+            	success
+                echo
+        fi
+        echo
+	;;
+  stop)
+       	python $AGENTPATH stop init
+       	
+        if [ -f /etc/SuSE-release ]; then
+                rc_status -v
+        elif [ -f /etc/debian_version ] || [ -f /etc/lsb-release ] || [ -f /etc/gentoo-release ]; then
+                echo " Stopped"
+        else
+                success
+                echo
+        fi
+        echo
+	;;
+  restart)
+        $0 stop
+        $0 start
+        ;;
+  *)
+    	echo "Usage: /etc/init.d/sd-agent start|stop|restart"
+        exit 1
+esac
+
+exit 0