Get --help and --version options working on all scripts

Added the autoconf option std-options to the scripts/ directory, which
checks to ensure all programs have both --help and --version options. A
few things needed cleaning up to get this working. To test these types
of options, use the 'make distcheck' target.

Signed-off-by: Dan McGee <dan@archlinux.org>

1 files changed, 178 insertions, 0 deletions

diff --git a/scripts/rankmirrors.in b/scripts/rankmirrors.in
new file mode 100644
index 00000000..da0811c0
--- /dev/null
+++ b/scripts/rankmirrors.in
@@ -0,0 +1,178 @@
+#! /usr/bin/python
+#
+#   rankmirrors : read a list of mirrors from a file and rank them by speed
+#
+#   Original Idea copyright (c) 2006 R.G. <chesercat>
+#   Modified 2006 by Dan McGee <dan@archlinux.org>
+#
+#   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., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
+#   USA.
+#
+import os, sys, datetime, time, socket, urllib2
+from optparse import OptionParser
+
+def createOptParser():
+    usage = "usage: %prog [options] MIRRORFILE | URL"
+    version = '%prog (pacman) @PACKAGE_VERSION@'
+    description = "Ranks pacman mirrors by their connection and opening " \
+            "speed. Pacman mirror files are located in /etc/pacman.d/. It " \
+            "can also rank one mirror if the URL is provided."
+    parser = OptionParser(usage = usage, version = version,
+                          description = description)
+    parser.add_option("-n", type = "int", dest = "num", default = 0,
+                      help = "number of servers to output, 0 for all")
+    parser.add_option("-t", "--times", action = "store_true",
+                      dest = "times", default = False,
+                      help = "only output mirrors and their response times")
+    parser.add_option("-u", "--url", action = "store_true", dest = "url",
+                      default = False, help = "test a specific url")
+    parser.add_option("-v", "--verbose", action = "store_true",
+                      dest = "verbose", default = False,
+                      help = "be verbose in ouptut")
+    # The following two options should be automatic
+    #parser.add_option("-h", "--help", action = "help")
+    #parser.add_option("-V", "--version", action = "version")
+    return parser
+
+def timeCmd(cmd):
+    before = time.time()
+    try:
+        cmd()
+    except KeyboardInterrupt, ki:
+        raise ki
+    except socket.timeout, ioe:
+        return 'timeout'
+    except Exception, e:
+        return 'unreachable'
+    return time.time() - before
+
+def talkToServer(serverUrl):
+    opener = urllib2.build_opener()
+    # retrieve first 50,000 bytes only
+    tmp = opener.open(serverUrl).read(50000)
+
+def getFuncToTime(serverUrl):
+    return lambda : talkToServer(serverUrl)
+
+def cmpPairBySecond(p1, p2):
+    if p1[1] == p2[1]:
+        return 0
+    if p1[1] < p2[1]:
+        return -1
+    return 1
+
+def printResults(servers, time, verbose, num):
+    items = servers.items()
+    items.sort(cmpPairBySecond)
+    itemsLen = len(items)
+    numToShow = num
+    if numToShow > itemsLen or numToShow == 0:
+        numToShow = itemsLen
+    if itemsLen > 0:
+        if time:
+            print
+            print ' Servers sorted by time (seconds):'
+            for i in items[0:numToShow]:
+                if i[1] == 'timeout' or i[1] == 'unreachable':
+                    print i[0], ':', i[1]
+                else:
+                    print i[0], ':', "%.2f" % i[1]
+        else:
+            for i in items[0:numToShow]:
+                print 'Server =', i[0]
+
+if __name__ == "__main__":
+    parser = createOptParser()
+    (options, args) = parser.parse_args()
+
+    if len(args) != 1:
+        parser.print_help(sys.stderr)
+        sys.exit(0)
+
+    # allows connections to time out if they take too long
+    socket.setdefaulttimeout(10)
+
+    if options.url:
+        if options.verbose:
+            print 'Testing', args[0] + '...'
+        try:
+            serverToTime = timeCmd(getFuncToTime(args[0]))
+        except KeyboardInterrupt, ki:
+            sys.exit(1)
+        if serverToTime == 'timeout' or serverToTime == 'unreachable':
+            print args[0], ':', serverToTime
+        else:
+            print args[0], ':', "%.2f" % serverToTime
+        sys.exit(0)
+
+    if not os.path.isfile(args[0]):
+        print >>sys.stderr, 'rankmirrors: file', args[0], 'does not exist.'
+        sys.exit(1)
+
+    fl = open(args[0], 'r')
+    serverToTime = {}
+    if options.times:
+        print 'Querying servers, this may take some time...'
+    else:
+        print "# Server list generated by rankmirrors on",
+        print datetime.date.today()
+    for ln in fl.readlines():
+        splitted = ln.split('=')
+        if splitted[0].strip() != 'Server':
+            if not options.times:
+                print ln,
+            continue
+
+        serverUrl = splitted[1].strip()
+        if serverUrl[-1] == '\n':
+            serverUrl = serverUrl[0:-1]
+        if options.verbose and options.times:
+            print serverUrl, '...',
+        elif options.verbose:
+            print '#', serverUrl, '...',
+        elif options.times:
+            print ' * ',
+        sys.stdout.flush()
+
+        # add *.db.tar.gz to server name. the repo name is parsed
+        # from the mirror url; it is the third (or fourth) dir
+        # from the end, where the url is http://foo/bar/REPO/os/arch
+        try:
+            splitted2 = serverUrl.split('/')
+            if serverUrl[-1] != '/':
+                repoName = splitted2[-3]
+                dbFileName = '/' + repoName + '.db.tar.gz'
+            else:
+                repoName = splitted2[-4]
+                dbFileName = repoName + '.db.tar.gz'
+        except:
+            dbFileName = ''
+
+        try:
+            serverToTime[serverUrl] = timeCmd(getFuncToTime(serverUrl + dbFileName))
+            if options.verbose:
+                try:
+                    print "%.2f" % serverToTime[serverUrl]
+                except:
+                    print serverToTime[serverUrl]
+        except:
+            print
+            printResults(serverToTime, options.times, options.verbose,
+                         options.num)
+            sys.exit(0)
+
+    printResults(serverToTime, options.times, options.verbose, options.num)
+
+# vim: set ts=4 sw=4 et: