org-capture-server/pycl.py

#!/usr/bin/env python

# Copyright (C) 2009  David Hilley <davidhi@cc.gatech.edu>
# Copyright (C) 2010  Matt DeVuyst <mdevuyst@gmail.com>
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
#
import cgi, urlparse
import subprocess
import tempfile, time
import os, sys, re
import stat
import optparse
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer

_default_port = 9292
_default_editor = "emacsclient"

temp_has_delete=True
processes = {}

class Handler(BaseHTTPRequestHandler):
    global temp_has_delete

    def do_GET(self):
	if self.path == '/status':
		self.send_response(200)
		self.send_header('Content-Type', 'text/plain; charset=utf-8')
		self.end_headers()
		self.wfile.write('edit-server is running.\n')
		return
        self.send_error(404, "GET Not Found: %s" % self.path)

    def do_POST(self):
        global processes
        try:
            (content, params) = cgi.parse_header(self.headers.
                                                 getheader('content-type'))

            clength = 0
            cl = self.headers.getheader('content-length')

            if cl != None:
                clength = int(cl)
            else:
                self.send_response(411)
                self.end_headers()
                return

            body = self.rfile.read(clength)
            print body

            l = [s for s in self.path.split('/') if s]
            print l

            existing_file = self.headers.getheader('x-file')

            # write text into file
            if not existing_file or existing_file == "undefined":
                existing = False
                url = self.headers.getheader('x-url')
                print "url:", url
                prefix = "chrome_"
                if url:
                    prefix += re.sub("[^.\w]", "_", re.sub("^.*?//","",url))
                prefix += "_"
                if temp_has_delete==True:
                    f = tempfile.NamedTemporaryFile(
                            delete=False, prefix=prefix, suffix='.txt')
                    fname = f.name
                else:
                    tf = tempfile.mkstemp(prefix=prefix, suffix='.txt')
                    f = os.fdopen(tf[0],"w")
                    fname = tf[1]
                print "Opening new file ", fname
            else:
                existing = True
                p = processes[existing_file]
                print "Opening existing file ", existing_file
                f = open(existing_file, "w")
                fname = existing_file

            f.write(body)
            f.close()
            last_mod_time = os.stat(fname)[stat.ST_MTIME]

            if not existing:
                # spawn editor...
                print "Spawning editor... ", fname

                cmd = self.editor.split(",")
                cmd.append(fname)
                p = subprocess.Popen(cmd, close_fds=True)
                processes[fname] = p

            saved = False
            rc = None
            while (True):
                time.sleep(1)
                rc = p.poll()
                if rc != None: break
                mod_time = os.stat(fname)[stat.ST_MTIME]
                if mod_time != last_mod_time:
                    print "new mod time:", mod_time, " last:", last_mod_time
                    last_mod_time = mod_time
                    saved = True
                if saved: break

            if saved or not rc:
                    self.send_response(200)

                    f = file(fname, 'r')
                    s = f.read()
                    f.close()
            else:
                    if rc > 0:
                            msg = 'text editor returned %d' % rc
                    elif rc < 0:
                            msg = 'text editor died on signal %d' % -rc
                    self.send_error(404, msg)

            if saved:
                self.send_header('x-open', "true")
            else:
                try:
                    os.unlink(fname)
                except :
                    print "Unable to unlink:", fname
                    pass

            self.send_header('x-file', fname)
            self.end_headers()
            self.wfile.write(s)
        except :
            print "Error: ", sys.exc_info()[0]
            self.send_error(404, "Not Found: %s" % self.path)

def parse_options():
    parser = optparse.OptionParser()
    parser.add_option(
        "-p", "--port", type="int", dest="port", default=_default_port,
        help="port number to listen on (default: " + str(_default_port) + ")")
    parser.add_option(
        "-e", "--editor", dest="editor", default=_default_editor,
        help='text editor to spawn (default: "' + _default_editor + '")')
    return parser.parse_args()[0]

def main():
    global temp_has_delete
    import platform
    t = platform.python_version_tuple()
    if int(t[0]) == 2 and int(t[1]) < 6:
        temp_has_delete = False;
        print "Handling lack of delete for NamedTemporaryFile:", temp_has_delete
    options = parse_options()
    Handler.editor = options.editor
    try:
        httpserv = HTTPServer(('localhost', options.port), Handler)
        httpserv.table = {}
        httpserv.serve_forever()
    except KeyboardInterrupt:
        httpserv.socket.close()

if __name__ == '__main__':
    main()