#!/usr/bin/env python
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
# 
#   http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

import sys, httplib, getopt, socket

def usage(exitarg):
    print 'usage: %s [-q] host:port' % sys.argv[0]
    print 
    print '   -q: quiet mode'
    print
    print 'e.g. %s localhost:8080' % sys.argv[0]
    print
    sys.exit(exitarg)

gpfdist = ''
quiet = 0
uri = ''

try:
    (options, args) = getopt.getopt(sys.argv[1:], 'q')
except Exception, e:
    usage('Error: ' + str(e))

for (switch, val) in options:
    if (switch == '-q'):  quiet = 1

if len(args) != 1:
    usage('Error: please specify uri.')

host_port = args[0]

try:
    conn = httplib.HTTPConnection(host_port)
    conn.request('GET', '/')
    r = conn.getresponse()
    gpfdist = r.getheader('X-GPFDIST-VERSION', '')
except socket.error:
    if not quiet:
	print 'Error: gpfdist is not running (reason: socket error)'
	print 'Exit: 1'
    sys.exit(1)
     
if not gpfdist:
    if not quiet:
	print 'Error: gpfdist port is taken by some other programs'
	print 'Exit: 2'
    sys.exit(2)

if not quiet:
    print 'Okay, gpfdist version "%s" is running on %s.' % (gpfdist, host_port)
sys.exit(0)