# By Alexandre Passant
# Licenced under BSD licence

#!/usr/bin/python

import RDF
import sys

# Graph node color
# Besed on regexp from the sameAs label
colors = {
  'default':'ffffff',
  'flickr':'ff3399',
  'facebook':'003399',
  'twitter':'33ccff',
  'wordpress':'666666'
}

def run_query(file, query):
  model = RDF.Model()
  parser = RDF.Parser()
  parser.parse_into_model(model, file)
  query = """
PREFIX foaf: <http://xmlns.com/foaf/0.1/>
PREFIX owl: <http://www.w3.org/2002/07/owl#>
PREFIX rdfs: <http://www.w3.org/2000/01/rdf-schema#>
%s""" %query
  query = RDF.Query(query, query_language="sparql")
  return query.execute(model)

def get_same_as(file, uri) :
  sparql = """
SELECT DISTINCT ?sameAs ?seeAlso ?label
WHERE {
  <%s> owl:sameAs ?sameAs .
  ?sameAs rdfs:seeAlso ?seeAlso ;
    rdfs:label ?label
}
""" %uri
  return run_query(file, sparql)

def get_rel_from_uri(file, uri) :
  sparql = """
SELECT DISTINCT ?name ?knows ?seeAlso ?img
WHERE {
  <%s> foaf:knows ?knows .
  OPTIONAL { ?knows foaf:name ?name }
  OPTIONAL { ?knows foaf:img ?img }
  OPTIONAL { ?knows rdfs:seeAlso ?seeAlso }
}
""" %uri
  return run_query(file, sparql)

def get_name(file, uri) :
  sparql = """
SELECT DISTINCT ?name
WHERE {
  <%s> foaf:name ?name ;
} LIMIT 1
""" %uri
  res = run_query(file, sparql)
  for name in res :
    return name['name']

def get_color(label):
    l = label.lower()
    if colors.has_key(l):
      return colors[l]
    return colors['default']

def usage():
  print """Usage: foafgear.py foaf_URI"""

def foafgear(uri) :
  myname = get_name(uri, uri)
  n=net=1
  print """<?xml version="1.0"?>
<graph title="My graph title" bgcolor="ffffff" linecolor="cccccc" viewmode="display">
\t<node id="n1" text="%s" color="6000ff" textcolor="ffffff" scale="60"/>""" %myname
  res = get_same_as(uri, uri)
  for result in res:
    net+=1
    uri = result['sameAs']._get_uri()
    file = result['seeAlso']._get_literal_value()['string']
    label = result['label']._get_literal_value()['string']
    l = label.split()[0]
    color = get_color(l)
    print """\t<node id="net%s" text="%s" color="%s" textcolor="ffffff" scale="60"/>""" %(net, l, color)
    print """\t<edge sourceNode="n1" targetNode="net%s" label="" textcolor="555555"/>""" %net
    res = get_rel_from_uri(file, uri)
    done = []
    for result in res:
      n+=1
      knows = result['knows']
      name = result['name']
      img = result['img']
      seeAlso = result['seeAlso']
      if knows not in done:
        done.append(knows)
        if not name and seeAlso:
          knows = knows._get_uri()
          seeAlso = result['seeAlso']._get_uri()
          name = get_name(seeAlso, knows)
        if name:
          name = name._get_literal_value()['string'].encode('utf-8')
	  image = ''
	  if img:
	    image = "image=\"%s\"" %img._get_uri() 
          print """\t<node id="n%s" text="%s" color="%s" textcolor="ffffff" scale="60" %s/>""" %(n, name, color, image)
          print """\t<edge sourceNode="net%s" targetNode="n%s" label="" textcolor="555555"/>""" %(net, n)
  print "</graph>"

if __name__ == '__main__' :
  if len(sys.argv) != 2:
    usage()
  else: 
    foafgear(sys.argv[1])