#!/bin/sh
#
# command line interface to initiate Altavista search
# requires external "queryencode"
# 
# 1999-05-09  naddy

browser=${BROWSER:-'lynx -nocolor -cookies'}
url=http://www.altavista.com/cgi-bin/query
me=$(basename "$0")

usage() {
  cat <<EOF
usage: $me [options] query
  -a		advanced query
  -c		only count matches (requires -a)
  -f date	date from (requires -a)
  -h		help
  -l langcode	limit query to language (not with -U)
  -L		list languages codes and exit
  -r keywords	ranking keywords (requires -a)
  -R		refine search
  -t date	date to (requires -a)
  -U		search Usenet rather than the Web
EOF
}

# supported languages
listlang() {
  column <<EOF
zh Chinese
cs Czech
da Danish
nl Dutch
en English
et Estonian
fi Finnish
fr French
de German
el Greek
he Hebrew
hu Hungarian
is Icelandic
it Italian
ja Japanese
ko Korean
lv Latvian
lt Lithuanian
no Norwegian
pl Polish
pt Portuguese
ro Romanian
ru Russian
es Spanish
sv Swedish
EOF
}

checklang() {
  case $1 in
    cs|da|de|el|en|es|et|fi|fr|he|hu|is|it) return 0 ;;
    ja|ko|lt|lv|nl|no|pl|pt|ro|ru|sv|zh)    return 0 ;;
    *)                                      return 1 ;;
  esac
}

# defaults
pg=q		# simply query
text=yes	# text mode, no crap (fixed)
what=web	# search Web
kl=XX		# any language
r=		# ranking keywords
act=search	# full search
d0=		# from date
d1=		# to date
fmt=		# don't count only

while getopts ':acf:hl:Lr:Rt:U' flag; do
  case $flag in
    a) pg=aq
       ;;
    c) fmt=n
       ;;
    f) d0=$OPTARG
       ;;
    h) usage;
       exit
       ;;
    l) kl=$OPTARG
       if ! checklang "$kl"; then
         echo "$me: unsupported language $kl" >&2
         exit 2
       fi
       ;;
    L) listlang
       exit
       ;;
    r) r=$OPTARG
       ;;
    R) act=refine
       ;;
    t) d1=$OPTARG
       ;;
    U) what=news
       ;;
    ?) echo "$me: unknown option -$OPTARG" >&2
       usage >&2
       exit 2
       ;;
  esac
done
shift $(($OPTIND-1))

if [ $# -eq 0 ]; then
  echo "$me: missing query" >&2
  usage >&2
  exit 2
else
  q=$*
fi

if [ $pg = q ]; then
  if [ -n "$r" ]; then
    echo "$me: -r requires advanced query" >&2
    exit 2
  fi
  if [ -n "$d0" ]; then
    echo "$me: -f requires advanced query" >&2
    exit 2
  fi
  if [ -n "$d1" ]; then
    echo "$me: -t requires advanced query" >&2
    exit 2
  fi
  if [ -n "$fmt" ]; then
    echo "$me: -c requires advanced query" >&2
    exit 2
  fi
fi

if [ $what = news -a "$kl" != XX ]; then
  echo "$me: can't specify language for Usenet search" >&2
  exit 2
fi

url="$url?pg=$pg"
url="$url&text=$text"
url="$url&what=$what"
test $what = web && url="$url&kl=$kl"
test $pg = aq && url="$url&r=$(queryencode "$r")"
url="$url&act=$act"
url="$url&q=$(queryencode "$q")"
if [ $pg = aq ]; then
  url="$url&d0=$(queryencode "$d0")"
  url="$url&d0=$(queryencode "$d1")"
  test -n "$fmt" && url="$url&fmt=$fmt"
fi

exec $browser "$url"
#echo "$url"

## EOF ##