From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: (qmail 20296 invoked from network); 15 Sep 2004 06:42:27 -0000 Received: from news.dotsrc.org (HELO a.mx.sunsite.dk) (130.225.247.88) by ns1.primenet.com.au with SMTP; 15 Sep 2004 06:42:27 -0000 Received: (qmail 29026 invoked from network); 15 Sep 2004 06:42:19 -0000 Received: from sunsite.dk (130.225.247.90) by a.mx.sunsite.dk with SMTP; 15 Sep 2004 06:42:19 -0000 Received: (qmail 576 invoked by alias); 15 Sep 2004 06:41:34 -0000 Mailing-List: contact zsh-users-help@sunsite.dk; run by ezmlm Precedence: bulk X-No-Archive: yes X-Seq: 7993 Received: (qmail 561 invoked from network); 15 Sep 2004 06:41:33 -0000 Received: from unknown (HELO a.mx.sunsite.dk) (130.225.247.88) by sunsite.dk with SMTP; 15 Sep 2004 06:41:33 -0000 Received: (qmail 26891 invoked from network); 15 Sep 2004 06:40:34 -0000 Received: from happygiraffe.net (81.6.215.59) by a.mx.sunsite.dk with SMTP; 15 Sep 2004 06:40:32 -0000 Received: from localhost (localhost.happygiraffe.net [127.0.0.1]) by happygiraffe.net (Postfix) with ESMTP id C68A1B8C6 for ; Wed, 15 Sep 2004 07:40:31 +0100 (BST) Received: from happygiraffe.net ([127.0.0.1]) by localhost (ppe.happygiraffe.net [127.0.0.1]) (amavisd-new, port 10024) with LMTP id 62343-01-2 for ; Wed, 15 Sep 2004 07:40:31 +0100 (BST) Received: by happygiraffe.net (Postfix, from userid 1001) id 653AEB8C5; Wed, 15 Sep 2004 07:40:31 +0100 (BST) Date: Wed, 15 Sep 2004 07:40:31 +0100 To: zsh-users@sunsite.dk Subject: PostgreSQL completion Message-ID: <20040915064031.GA52150@ppe.happygiraffe.net> Mime-Version: 1.0 Content-Type: multipart/mixed; boundary="Qxx1br4bt0+wmkIi" Content-Disposition: inline User-Agent: Mutt/1.5.6i From: dom@happygiraffe.net (Dominic Mitchell) X-Virus-Scanned: by amavisd-new at happygiraffe.net X-Spam-Checker-Version: SpamAssassin 2.63 on a.mx.sunsite.dk X-Spam-Level: X-Spam-Status: No, hits=0.0 required=6.0 tests=none autolearn=no version=2.63 X-Spam-Hits: 0.0 --Qxx1br4bt0+wmkIi Content-Type: text/plain; charset=us-ascii Content-Disposition: inline I've got some way through a (hopefully) useful completion for PostgreSQL. It's useful as it stands, but I have a few questions about some bits of it. 1. I use psql to get information from the database. But what happens if psql spits out an error? At the moment, it just messes up the display... 2. How do you complete either a hostname or a path at the same point? PostgreSQL accepts either a hostname or a directory containing a socket after -h. Anyway, I hope this is useful to somebody. -Dom --Qxx1br4bt0+wmkIi Content-Type: text/plain; charset=us-ascii Content-Description: Zsh completion for PostgreSQL Content-Disposition: attachment; filename=_pgsql_utils #compdef psql pg_dump createdb dropdb # # zsh completion functions for PostgreSQL client programs. Based on # _mysql_utils. # # Dominic Mitchell # _pgsql_get_identity () { _pgsql_user=${(v)opt_args[(i)-U|--username]} _pgsql_port=${(v)opt_args[(i)-p|--port]} _pgsql_host=${(v)opt_args[(i)-h|--host]} _pgsql_params=( ${_pgsql_user:+"--username=$_pgsql_user"} ${_pgsql_port:+"--port=$_pgsql_port"} ${_pgsql_host:+"--host=$_pgsql_host"} ) } _pgsql_users () { local _pgsql_user _pgsql_port _pgsql_host _pgsql_params _pgsql_get_identity # We use _pgsql_port and _pgsql_host directly here instead of # _pgsql_params so as to not pick up a partially completed # username. _pgsql_params=( ${_pgsql_port:+"--port=$_pgsql_port"} ${_pgsql_host:+"--host=$_pgsql_host"} ) compadd "$@" - ${${(f)~~"$( psql $_pgsql_params[@] -At -c '\du' template1 )"}[@]%%|*} } _pgsql_tables () { local _pgsql_user _pgsql_port _pgsql_host _pgsql_params _pgsql_get_identity # Need to pull out the database name from the existing arguments. # This is going to vary between commands. Thankfully, it's only # used by pg_dump, which always has the dbname in arg1. If it's # not present it defaults to ${PGDATABASE:-$LOGNAME}, which # matches (I think) the PostgreSQL behaviour. local db db=${line[1]:-${PGDATABASE:-$LOGNAME}} # XXX In postgres 7.3 and above, the schema name is in the first # column. I'm not sure how best to work around that... It really # needs to be prepended with a "." to the table name. # Many thanks to Oliver Kiddle for pointing out how to get the 2nd # column out of this... compadd "$@" - \ ${${${(f)~~"$( psql $_pgsql_params[@] -At -c '\dt' $db )"}#*|}%%|*} } _pgsql_databases () { local _pgsql_user _pgsql_port _pgsql_host _pgsql_params _pgsql_get_identity # Should I grep out template0? compadd "$@" - ${${(f)~~"$( psql $_pgsql_params[@] -At -l)"}[@]%%|*} } ## ## The actual completion code for the commands ## _psql () { local curcontext="$curcontext" state line expl typeset -A opt_args _arguments -C -s \ "$_pgsql_common_opts[@]" \ {-V,--version}'[display client version]' \ {-a,--echo-all}'[print commands read]' \ {-A,--no-align}'[unaligned output mode]' \ {-c+,--command=}':execute SQL command:' \ {-d+,--dbname=}':database to connect to:_pgsql_databases' \ {-e,--echo-queries}'[display queries submitted]' \ {-E,--echo-hidden}'[display hidden queries]' \ {-f+,--file=}':SQL file to read:_files' \ {-F+,--field-separator=}':field separator char:' \ {-H,--html}'[HTML output]' \ {-l,--list}'[list databases]' \ {-o+,--output=}':query output:_files' \ {-P+,--pset=}':set psql variable:' \ {-q,--quiet}'[non verbose mode]' \ {-R+,--record-separator=}':record separator char:' \ {-s,--single-step}'[prompt before each query]' \ {-S,--single-line}'[newline sends query]' \ {-t,--tuples-only}'[dont display header/footer]' \ {-T+,--table-attr=}':HTML table options:' \ -u'[prompt for username/password]' \ {-v+,--set=,--variable=}':set SQL variable:' \ {-x,--expanded}'[one column per line]' \ {-X,--no-psqlrc}'[dont read ~/.psqlrc]' \ ':PostgreSQL database:_pgsql_databases' \ ':PostgreSQL user:_pgsql_users' } _pg_dump () { local curcontext="$curcontext" state line expl typeset -A opt_args _arguments -C -s \ "$_pgsql_common_opts[@]" \ {-a,--data-only}'[dump only data]' \ {-b,--blobs}'[dump blobs as well]' \ {-c,--clean}'[include clean cmds in dump]' \ {-C,--create}'[include createdb cmds in dump]' \ {-d,--inserts}'[use INSERT not COPY]' \ {-D,--{attribute,column}-inserts}'[use INSERT (cols) not COPY]' \ {-f+,--file=}':output file:_files' \ {-F+,--format=}':output format:_values "format" "p[plain text]" "t[tar]" "c[custom]"' \ {-i,--ignore-version}'[ignore version mismatch]' \ {-n+,--schema=}':schema to dump:' \ {-o,--oids}'[dump objects identifiers for every table]' \ {-O,--no-owner}'[dont recreate as same owner]' \ {-R,--no-reconnect}'[dont output connect]' \ {-s,--schema-only}'[no data, only schema]' \ {-S+,--superuser=}':superuser name:_pgsql_users' \ {-t+,--table=}':table to dump:_pgsql_tables' \ {-v,--verbose}'[verbose mode]' \ {-V,--version}'[display client version]' \ {-x,--no-{acl,privileges}}'[dont dump ACLs]' \ -X+':option:_values "option" use-set-session-authorization disable-triggers' \ {-Z+,--compress=}':compression level:_values "level" 9 8 7 6 5 4 3 2 1 0' \ ':PostgreSQL database:_pgsql_databases' } _createdb () { local curcontext="$curcontext" state line expl typeset -A opt_args _arguments -C -s \ "$_pgsql_common_opts[@]" \ {-e,--echo}'[display SQL queries]' \ {-q,--quiet}'[non verbose mode]' \ {-D+,--location=}':database location:_directories' \ {-T+,--template=}':database template:_pgsql_databases' \ {-E+,--encoding=}':database encoding:_values "encoding" $_pgsql_encodings[@]' \ ':PostgreSQL database:' \ ':comment:' } _dropdb () { local curcontext="$curcontext" state line expl typeset -A opt_args _arguments -C -s \ "$_pgsql_common_opts[@]" \ {-e,--echo}'[display SQL queries]' \ {-q,--quiet}'[non verbose mode]' \ {-i,--interactive}'[confirm before drop]' \ ':PostgreSQL database:_pgsql_databases' } _pgsql_utils () { local _pgsql_common_opts _pgsql_encodings _pgsql_common_opts=( {-\?,--help}'[display help]' # XXX This can also complete the path to a directory which # contains the postgresql socket (eg: /tmp). I'm not sure how # to say that though. {-h+,--host=}':database host:_hosts' # Postgres doesn't like service names here, which is why we # don't use _ports. {-p+,--port=}':database port number:' {-U+,--username=}':connect as user:_pgsql_users' {-W,--password}'[prompt for password]' ) # Taken from # . # It'd be real nice if postgres could tell us these... _pgsql_encodings=( SQL_ASCII EUC_{JP,CN,KR,TW} JOHAB UNICODE MULE_INTERNAL LATIN{1,2,3,4,5,6,7,8,9,10} ISO_8859_{5,6,7,8} KOI8 WIN ALT WIN1256 TCVN WIN874 ) case "$service" in psql) _psql "$@" ;; pg_dump) _pg_dump "$@" ;; createdb) _createdb "$@" ;; dropdb) _dropdb "$@" ;; esac } _pgsql_utils "$@" # vim: set ai et sw=4 syntax=zsh : --Qxx1br4bt0+wmkIi--