#!/usr/bin/env python3

## BEEM is a videoconference application on the Android Platform.

## Copyright (C) 2009-2012 by Frederic-Charles Barthelery,

##			   Nikita Kozlov,

##			   Vincent Veronis.

##

## This file is part of BEEM.

##

## BEEM 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 3 of the License, or

## (at your option) any later version.

##

## BEEM 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 BEEM.  If not, see <http://www.gnu.org/licenses/>.

##

## Please send bug reports with examples or suggestions to

## contact@beem-project.com or http://www.beem-project.com/

import urllib.request

import argparse, sys

from xml.dom.minidom import parse, getDOMImplementation

def getJidsFromUrl(url):

    f = urllib.request.urlopen(url)

    indoc = parse(f)

    jids = []

    query = indoc.documentElement

    if query.localName == "query" :

        for item in query.getElementsByTagName("item"):

            jid = item.getAttribute("jid")

            jids.append(jid)

    return jids

def createDocument():

    impl = getDOMImplementation()

    return impl.createDocument(None, "resources", None)

def purge(elems):

    res = []

    for i in elems:

        if i not in res:

            res.append(i)

    return res

def appendStringElem(doc, node, jids):

    for i in jids:

        item = doc.createElement("item")

        text = doc.createTextNode(i)

        item.appendChild(text)

        node.appendChild(item)

parser = argparse.ArgumentParser(description='Collect some free xmpp services')

parser.add_argument('url', metavar="url",

        default=['http://xmpp.net/services.xml', "https://list.jabber.at/api/?format=services.xml"] , nargs='*',

        help='url to get the services')

parser.add_argument('-o', metavar="FILE", type=argparse.FileType('bw'),

        default = sys.stdout.buffer,

        help='send output to FILE')

args = parser.parse_args()

# collect the servers jid

jids = []

for url in args.url:

    jids += getJidsFromUrl(url)

jids = purge(jids)

# create the xml output document

outdoc = createDocument()

res_element = outdoc.documentElement

string_array_elem = outdoc.createElement('string-array')

res_element.appendChild(string_array_elem)

string_array_elem.setAttribute("name", "xmpp_server_list")

appendStringElem(outdoc, string_array_elem, jids)

# print result

f = args.o

f.write(outdoc.toprettyxml(encoding="utf-8"))