|  | @@ -0,0 +1,131 @@
 | 
	
		
			
				|  |  | +#!/usr/bin/python
 | 
	
		
			
				|  |  | +#============================================================================
 | 
	
		
			
				|  |  | +# This file is part of Pwman3.
 | 
	
		
			
				|  |  | +#
 | 
	
		
			
				|  |  | +# Pwman3 is free software; you can redistribute it and/or modify
 | 
	
		
			
				|  |  | +# it under the terms of the GNU General Public License, version 2
 | 
	
		
			
				|  |  | +# as published by the Free Software Foundation;
 | 
	
		
			
				|  |  | +#
 | 
	
		
			
				|  |  | +# Pwman3 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 Pwman3; if not, write to the Free Software
 | 
	
		
			
				|  |  | +# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 | 
	
		
			
				|  |  | +#============================================================================
 | 
	
		
			
				|  |  | +# Copyright (C) 2006 Ivan Kelly <ivan@ivankelly.net>
 | 
	
		
			
				|  |  | +#============================================================================
 | 
	
		
			
				|  |  | +from pwman.util.crypto import CryptoEngine
 | 
	
		
			
				|  |  | +import getopt
 | 
	
		
			
				|  |  | +import sys
 | 
	
		
			
				|  |  | +if 'darwin' in sys.platform:
 | 
	
		
			
				|  |  | +    from pwman.ui.cli import PwmanCliMac
 | 
	
		
			
				|  |  | +else:
 | 
	
		
			
				|  |  | +    from pwman.ui.cli import PwmanCli
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +import pwman.util.config as config
 | 
	
		
			
				|  |  | +import pwman.data.factory
 | 
	
		
			
				|  |  | +import os
 | 
	
		
			
				|  |  | +import os.path
 | 
	
		
			
				|  |  | +import subprocess as sp
 | 
	
		
			
				|  |  | +_saveconfig = True
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def print_help():
 | 
	
		
			
				|  |  | +    print """Syntax: %s [options]
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | + -c, --config FILE      Read configuration from FILE
 | 
	
		
			
				|  |  | + -d, --database FILE    Use FILE as database
 | 
	
		
			
				|  |  | + -e, --encryption ALGO  Use ALGO to encrypt data
 | 
	
		
			
				|  |  | +                        Possible options are:
 | 
	
		
			
				|  |  | +                         AES, ARC2, ARC4, Blowfish(default),
 | 
	
		
			
				|  |  | +                         CAST, DES, DES3, IDEA, RC5
 | 
	
		
			
				|  |  | + -h, --help             Display this help and exit
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +Please report bugs at https://github.com/pwman3/pwman3
 | 
	
		
			
				|  |  | +""" % (sys.argv[0])
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +def which(cmd):
 | 
	
		
			
				|  |  | +    _, cmdname = os.path.split(cmd)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    for path in os.environ["PATH"].split(os.pathsep):
 | 
	
		
			
				|  |  | +        cmd = os.path.join(path, cmdname)
 | 
	
		
			
				|  |  | +        if os.path.isfile(cmd) and os.access(cmd, os.X_OK):
 | 
	
		
			
				|  |  | +            return cmd
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    return None
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +try:
 | 
	
		
			
				|  |  | +    config_dir = os.path.expanduser("~/.pwman")
 | 
	
		
			
				|  |  | +    if not os.path.isdir(config_dir):
 | 
	
		
			
				|  |  | +        os.mkdir(config_dir)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    config_file = os.path.join(config_dir, "config")
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    default_config = {'Global': {'umask': '0100', 'colors': 'yes'},
 | 
	
		
			
				|  |  | +                      'Database': {'type': 'SQLite',
 | 
	
		
			
				|  |  | +                                   'filename': os.path.join(config_dir,
 | 
	
		
			
				|  |  | +                                                            "pwman.db")},
 | 
	
		
			
				|  |  | +                      'Encryption': {'algorithm': 'Blowfish'},
 | 
	
		
			
				|  |  | +                      'Readline': {'history': os.path.join(config_dir,
 | 
	
		
			
				|  |  | +                                                           "history")}
 | 
	
		
			
				|  |  | +                      }
 | 
	
		
			
				|  |  | +    config.set_defaults(default_config)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    opts, args = getopt.getopt(sys.argv[1:], 'c:d:e:h',
 | 
	
		
			
				|  |  | +                               ["config=", "database=",
 | 
	
		
			
				|  |  | +                               "encryption=", "help"])
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    for o in opts:
 | 
	
		
			
				|  |  | +        if o[0] == '-c' or o[0] == "--config":
 | 
	
		
			
				|  |  | +            config_file = os.path.expanduser(o[1])
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +        if o[0] == '-h' or o[0] == "--help":
 | 
	
		
			
				|  |  | +            print_help()
 | 
	
		
			
				|  |  | +            sys.exit(0)
 | 
	
		
			
				|  |  | +    # if no config exists yet, look for xsel
 | 
	
		
			
				|  |  | +    if os.path.exists(config_file):
 | 
	
		
			
				|  |  | +        config.load(config_file)
 | 
	
		
			
				|  |  | +        xselpath = config.get_value("Global", "xselpath")
 | 
	
		
			
				|  |  | +    else:
 | 
	
		
			
				|  |  | +        xselpath = which("xsel")
 | 
	
		
			
				|  |  | +        config.set_value("Global", "xsel", xselpath)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    for o in opts:
 | 
	
		
			
				|  |  | +        if o[0] == '-d' or o[0] == "--database":
 | 
	
		
			
				|  |  | +            config.set_value("Database", "filename",
 | 
	
		
			
				|  |  | +                             os.path.expanduser(o[1]))
 | 
	
		
			
				|  |  | +            _saveconfig = False
 | 
	
		
			
				|  |  | +        if o[0] == '-e' or o[0] == "--encryption":
 | 
	
		
			
				|  |  | +            config.set_value("Encryption", "algorithm", o[1])
 | 
	
		
			
				|  |  | +            _saveconfig = False
 | 
	
		
			
				|  |  | +    # set umask before creating/opening any files
 | 
	
		
			
				|  |  | +    umask = int(config.get_value("Global", "umask"))
 | 
	
		
			
				|  |  | +    os.umask(umask)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    enc = CryptoEngine.get()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    dbtype = config.get_value("Database", "type")
 | 
	
		
			
				|  |  | +    db = pwman.data.factory.create(dbtype)
 | 
	
		
			
				|  |  | +    cli = PwmanCli(db, xselpath)
 | 
	
		
			
				|  |  | +except SystemExit, e:
 | 
	
		
			
				|  |  | +    sys.exit(e)
 | 
	
		
			
				|  |  | +except Exception, e:
 | 
	
		
			
				|  |  | +    print "Error: %s" % (e)
 | 
	
		
			
				|  |  | +    sys.exit(-1)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +try:
 | 
	
		
			
				|  |  | +    try:
 | 
	
		
			
				|  |  | +        cli.cmdloop()
 | 
	
		
			
				|  |  | +    except KeyboardInterrupt, e:
 | 
	
		
			
				|  |  | +        print
 | 
	
		
			
				|  |  | +finally:
 | 
	
		
			
				|  |  | +    try:
 | 
	
		
			
				|  |  | +        if _saveconfig:
 | 
	
		
			
				|  |  | +            config.save(config_file)
 | 
	
		
			
				|  |  | +    except Exception, e:
 | 
	
		
			
				|  |  | +        print "Error: %s" % (e)
 | 
	
		
			
				|  |  | +        sys.exit(-1)
 |