X-Git-Url: https://scripts.mit.edu/gitweb/wizard.git/blobdiff_plain/a8b52be70700da2453a925a8f4cf7523e713aa92..715c070af45346766a9ab78e7cb76faa14da57bf:/plugins/scripts/wizard_scripts.py

diff --git a/plugins/scripts/wizard_scripts.py b/plugins/scripts/wizard_scripts.py
index 23fb1ee..4c6540a 100644
--- a/plugins/scripts/wizard_scripts.py
+++ b/plugins/scripts/wizard_scripts.py
@@ -9,23 +9,35 @@ import logging
 import urlparse
 import time
 import errno
+import sqlalchemy
 
 import wizard
-from wizard import shell, util, user
+from wizard import deploy, shell, install, util, user
+
+def dummy_apps():
+    return [
+        "joomla", "e107", "gallery2", "advancedbook", "phpical",
+        "trac", "turbogears", "django", "rails",
+        # these are technically deprecated
+        "advancedpoll", "gallery",
+    ]
 
 def deploy_web(dir):
     # try the directory
     homedir, _, web_path = dir.partition("/web_scripts")
     if web_path:
+        # XXX: To be truly correct, we should emulate the logic of suexec; but
+        # looking at the home directory is a pretty good stopgap for now
+        name = homedir.rpartition("/")[2]
         yield urlparse.ParseResult(
                 "http",
-                util.get_dir_owner(homedir) + ".scripts.mit.edu",
+                name + ".scripts.mit.edu",
                 web_path.rstrip('/'),
                 "", "", "")
         yield urlparse.ParseResult(
                 "http",
                 "scripts.mit.edu",
-                "/~" + util.get_dir_owner(homedir) + web_path.rstrip('/'),
+                "/~" + name + web_path.rstrip('/'),
                 "", "", "")
     else:
         logging.info("Directory location did not contain web_scripts: %s", dir)
@@ -109,10 +121,23 @@ def sql_auth(url):
             return url
         except shell.CallError:
             pass
+        except OSError:
+            pass
+    return None
+
+def sql_drop(url):
+    if url.host == "sql.mit.edu":
+        try:
+            shell.call("/mit/scripts/sql/bin/drop-database", url.database)
+            return True
+        except shell.CallError:
+            pass
     return None
 
 def user_email(name):
-    # XXX: simplistic strategy which doesn't work most of the time
+    # XXX: simplistic heuristic which doesn't work most of the time
+    # See Scripts #224 (this issue) and Scripts #193 (tracking a contact
+    # address)
     return "%s@scripts.mit.edu" % name
 
 def user_operator():
@@ -143,9 +168,65 @@ def user_passwd(dir, uid):
     if not dir.startswith("/afs/"):
         return None
     try:
-        result = shell.eval("hesinfo %d uid", uid)
+        result = shell.eval("hesinfo", str(uid), "uid")
     except shell.CallError:
         return None
-    name, password, uid, gid, gecos, homedir, shell = result.split(":")
+    name, password, uid, gid, gecos, homedir, console = result.split(":")
     realname = gecos.split(",")[0]
-    return user.Info(name, uid, gid, realname, homedir, shell)
+    return user.Info(name, uid, gid, realname, homedir, console)
+
+class MysqlStrategy(install.Strategy):
+    """
+    Performs scripts specific guesses for MySQL variables.  This
+    may create an appropriate database for the user.
+    """
+    side_effects = True
+    provides = frozenset(["dsn"])
+    def prepare(self):
+        """Uses the SQL programs in the scripts locker"""
+        logging.info("Attempting wizard_scripts MySQL strategy")
+        if self.application.database != "mysql":
+            raise install.StrategyFailed
+        try:
+            self._triplet = shell.eval("/mit/scripts/sql/bin/get-password").split()
+        except OSError:
+            raise install.StrategyFailed
+        except shell.CallError:
+            raise install.StrategyFailed
+        if len(self._triplet) != 3:
+            raise install.StrategyFailed
+        self._username = os.getenv('USER')
+        if self._username is None:
+            raise install.StrategyFailed
+    def execute(self, options):
+        """Creates a new database for the user using ``get-next-database`` and ``create-database``."""
+        host, username, password = self._triplet
+        # race condition
+        name = shell.eval("/mit/scripts/sql/bin/get-next-database", os.path.basename(self.dir))
+        database = shell.eval("/mit/scripts/sql/bin/create-database", name)
+        if not database:
+            raise CreateDatabaseError
+        options.dsn = sqlalchemy.engine.url.URL("mysql", username=username, password=password, host=host, database=database)
+
+class EmailStrategy(install.Strategy):
+    """Performs script specific guess for email."""
+    provides = frozenset(["email"])
+    def prepare(self):
+        """Uses :envvar:`USER` and assumes you are an MIT affiliate."""
+        # XXX: This might be buggy, because locker might be set to USER
+        self._user = os.getenv("USER")
+        if self._user is None:
+            raise install.StrategyFailed
+    def execute(self, options):
+        """No-op."""
+        options.email = self._user + "@mit.edu"
+
+class CreateDatabaseError(wizard.Error):
+    """Could not create a database with the create-database script."""
+    def __str__(self):
+        return """
+
+ERROR: We were unable to create a database for you: this may indicate
+that you have exceeded your quota of databases or indicate a problem
+with the sql.mit.edu service.  Please mail scripts@mit.edu with this
+error message."""