import hammock
import os
import requests
import sys
import json

# this is temporary
username = os.getenv("ACOM_USER","admin")
password = os.getenv("ACOM_PASS","admin")
server   = os.getenv("ACOM_SERVER","127.0.0.1:8000")

# TODO: error handling/output/etc
# TODO: format into actual command line

PARAMS = {
   'format'           : 'json',
}
HEADERS = {
   'Content-Type' : 'application/json'
}
AUTH = (username, password)

handle = hammock.Hammock("http://%s/api/v1" % server, auth=AUTH, append_slash=True, params=PARAMS, headers=HEADERS)

class Collection(object):

   def __init__(self, handle):
       self.handle   = handle
       self.response = self.accessor().GET(auth=AUTH, headers=HEADERS)
       assert self.response.status_code == 200
       # TODO: error handling on non-200
       print "RESPONSE=%s" % self.response.text
       self.data     = json.loads(self.response.text)
       self.meta     = self.data['meta']
       self.objects  = self.data['objects']
       self.meta     = self.data['meta']
       self.objects  = self.data['objects']
        
   def accessor(self):
       return exceptions.NotImplementedError()

   def add(self, data):
       # TODO: error handling
       json_data = json.dumps(data)
       response = self.accessor().POST(data=json_data)
       print response.status_code
       print response.text
       assert response.status_code == 201
       # FIXME: error handling
       data2 = response.text
       print data2
       return Entry(data)

   def __iter__(self):
       for x in self.objects:
           yield Entry(x)

class Entry(object):

   def __init__(self, data):
       self.data = data
       self.resource_uri = data.get('resource_uri', None)
       print "LOADING"
       self.accessor = hammock.Hammock(self.resource_uri, auth=AUTH, append_slash=True, params=PARAMS, headers=HEADERS)

   def __repr__(self):
       return repr(self.data)

class Organizations(Collection):

   def accessor(self):
       return self.handle.organizations

#(Epdb) got.text
#u'{"meta": {"limit": 20, "next": null, "offset": 0, "previous": null, "total_count": 1}, "objects": [{"active": true, "creation_date": "2013-03-15", "description": "testorg!", "id": 1, "name": "testorg", "resource_uri": "/api/v1/organizations/1/"}]}'
#

try:
    print "---"
    orgs = Organizations(handle)
    for x in orgs:
       print x
    print "---"
    orgs.add(dict(description="new org?", name="new org"))
    print "---"
    
    print "---"
    orgs = Organizations(handle)
    for x in orgs:
       print x
    

except requests.exceptions.ConnectionError:
    print "connect failed"
    sys.exit(1)