Initial commit. Download/upload and run commands after upload.

oobsync.py

@@ -0,0 +1,221 @@
+#! /opt/openoffice4/program/python
+# -*- coding: utf-8 -*-
+# OpenOffice Basic macros source synchronizer.
+#   by imacat <imacat@mail.imacat.idv.tw>, 2016-08-31
+
+import argparse
+import os
+import sys
+import time
+
+import uno
+from com.sun.star.connection import NoConnectException
+
+def main():
+    """ The main program. """
+    t_start = time.time()
+    global args
+    
+    # Parses the arguments
+    parse_args()
+    
+    
+    # Connects to the OpenOffice
+    oo = OpenOffice()
+    
+    # Synchronize the Basic macros.
+    sync_macros(oo)
+    
+    print >> sys.stderr, "Done.  %02d:%02d elapsed." % \
+        (int((time.time() - t_start) / 60), (time.time() - t_start) % 60)
+
+def parse_args():
+    """ Parses the arguments. """
+    global args
+    
+    parser = argparse.ArgumentParser(description="Synchronize the local Basic scripts with OpenOffice Basic.")
+    parser.add_argument("action", metavar="action", choices=["checkout", "co", "checkin", "ci"],
+        help="checkout (co) or checkin (ci) the source files.")
+    parser.add_argument("library", metavar="Library",
+        help="The Library to checkout/checkin the macros.")
+    parser.add_argument("projdir", metavar="dir", nargs="?",
+        default=os.getcwd(),
+        help="The project source directory (default to the current directory).")
+    parser.add_argument("--run", metavar="Module.Macro",
+        help="The macro to run after the checkin, if any.")
+    parser.add_argument("-v", "--version",
+        action="version", version="%(prog)s 1.0")
+    args = parser.parse_args()
+    
+    return
+
+def sync_macros(oo):
+    """ Synchronize the Basic macros. """
+    global args
+    
+    libraries = oo.service_manager.createInstance(
+        "com.sun.star.script.ApplicationScriptLibraryContainer")
+    # Checks-in the sources onto OpenOffice
+    if args.action in ["checkin", "ci"]:
+        modules = read_in_source_dir(args.projdir)
+        if len(modules) == 0:
+            print >> sys.stderr, "ERROR: Found no source macros in " + args.projdir
+            return
+        update_basic_modules(libraries, args.library, modules)
+        if args.run != None:
+            factory = oo.service_manager.DefaultContext.getByName(
+                "/singletons/com.sun.star.script.provider.theMasterScriptProviderFactory")
+            provider = factory.createScriptProvider("")
+            script = provider.getScript(
+                "vnd.sun.star.script:" + args.library + "." + args.run + "?language=Basic&location=application")
+            script.invoke((), (), ())
+    
+    # Checks-out the sources from OpenOffice
+    else:
+        modules = read_basic_modules(libraries, args.library)
+        if len(modules) == 0:
+            print >> sys.stderr, "ERROR: Library " + args.library + " does not exist"
+            return
+        update_source_dir(args.projdir, modules)
+    return
+
+def read_in_source_dir(projdir):
+    """ Reads-in the source macros. """
+    modules = {}
+    for entry in os.listdir(projdir):
+        path = os.path.join(projdir, entry)
+        if os.path.isfile(path) and entry.lower().endswith(".vb"):
+            modname = entry[0:-3]
+            f = open(path)
+            modules[modname] = f.read()
+            f.close()
+    return modules
+
+def update_source_dir(projdir, modules):
+    """ Updates the source macros. """
+    curmods = {}
+    for entry in os.listdir(projdir):
+        path = os.path.join(projdir, entry)
+        if os.path.isfile(path) and entry.lower().endswith(".vb"):
+            modname = entry[0:-3]
+            curmods[modname] = True
+            if not modules.has_key(modname):
+                os.remove(path)
+                print >> sys.stderr, modname + ".vb removed."
+            else:
+                f = open(path, "r+")
+                if modules[modname] != f.read():
+                    f.seek(0)
+                    f.write(modules[modname])
+                    print >> sys.stderr, modname + ".vb updated."
+                f.close()
+    for modname in [x for x in modules.keys() if not curmods.has_key(x)]:
+        path = os.path.join(projdir, modname + ".vb")
+        f = open(path, "w")
+        f.write(modules[modname])
+        f.close()
+        print >> sys.stderr, modname + ".vb added."
+    return
+
+def read_basic_modules(libraries, libname):
+    """ Reads the OpenOffice Basic macros from the macros storage. """
+    modules = {}
+    if not libraries.hasByName(libname):
+        return modules
+    library = libraries.getByName(libname)
+    for modname in library.getElementNames():
+        modules[modname] = library.getByName(modname).encode("utf-8")
+    return modules
+
+def update_basic_modules(libraries, libname, modules):
+    """ Updates the OpenOffice Basic macros storage. """
+    if not libraries.hasByName(libname):
+        libraries.createLibrary(libname)
+        print >> sys.stderr, "Library " + libname + " created."
+        library = libraries.getByName(libname)
+        for modname in modules.keys():
+            library.insertByName(modname, modules[modname])
+            print >> sys.stderr, "Module " + modname + " added."
+    else:
+        library = libraries.getByName(libname)
+        origmods = library.getElementNames()
+        for modname in modules.keys():
+            if not library.hasByName(modname):
+                library.insertByName(modname, modules[modname])
+                print >> sys.stderr, "Module " + modname + " added."
+            elif modules[modname] != library.getByName(modname).encode("utf-8"):
+                library.replaceByName(modname, modules[modname])
+                print >> sys.stderr, "Module " + modname + " updated."
+        for modname in origmods:
+            if not modules.has_key(modname):
+                library.removeByName(modname)
+                print >> sys.stderr, "Module " + modname + " removed."
+    return
+
+class OpenOffice:
+    """The OpenOffice connection."""
+    
+    def __init__(self):
+        """Initializes the object."""
+        self.bootstrap_context = None
+        self.service_manager = None
+        self.desktop = None
+        self.connect()
+    
+    def connect(self):
+        """Connects to the running OpenOffice process."""
+        # Obtains the local context
+        local_context = uno.getComponentContext()
+        # Obtains the local service manager
+        local_service_manager = local_context.getServiceManager()
+        # Obtains the URL resolver
+        url_resolver = local_service_manager.createInstanceWithContext(
+            "com.sun.star.bridge.UnoUrlResolver", local_context)
+        # Obtains the context
+        url = "uno:socket,host=localhost,port=2002;urp;StarOffice.ComponentContext"
+        while True:
+            try:
+                self.bootstrap_context = url_resolver.resolve(url)
+            except NoConnectException:
+                self.start_oo()
+            else:
+                break
+        # Obtains the service manager
+        self.service_manager = self.bootstrap_context.getServiceManager()
+        # Obtains the desktop service
+        self.desktop = self.service_manager.createInstanceWithContext(
+            "com.sun.star.frame.Desktop", self.bootstrap_context)
+    
+    def start_oo(self):
+        """Starts the OpenOffice in server listening mode"""
+        # For MS-Windows, which does not have fork()
+        if os.name == "nt":
+            from subprocess import Popen
+            ooexec = os.path.join(os.path.dirname(uno.__file__), "soffice.exe")
+            DETACHED_PROCESS = 0x00000008
+            Popen([ooexec, "-accept=socket,host=localhost,port=2002;urp;"],
+                close_fds=True, creationflags=DETACHED_PROCESS)
+            time.sleep(2)
+            return
+        
+        # For POSIX systems, including Linux
+        try:
+            pid = os.fork()
+        except OSError:
+            sys.stderr.write("failed to fork().\n")
+            sys.exit(1)
+        if pid != 0:
+            time.sleep(2)
+            return
+        os.setsid()
+        ooexec = os.path.join(os.path.dirname(uno.__file__), "soffice.bin")
+        try:
+            os.execl(ooexec, ooexec,
+                "-accept=socket,host=localhost,port=2002;urp;")
+        except OSError:
+            sys.stderr.write(ooexec
+                + ": Failed to run the OpenOffice server.\n")
+            sys.exit(1)
+
+if __name__ == "__main__":
+    main()