#!/usr/bin/env python

# -*- coding: utf-8 -*-

from gevent import monkey; monkey.patch_all()

# Importer d'abord, sinon exception

import anyjson, gevent, psycopg2

from sqlalchemy import (Boolean, Column, BigInteger, Integer, String, 

    ForeignKey, DateTime, create_engine, func)

from sqlalchemy.orm import relationship, sessionmaker

from sqlalchemy.ext.declarative import declarative_base

from gevent.pywsgi import WSGIServer

from urlparse import parse_qs

Base = declarative_base()

engine = create_engine('postgresql://postgres:doiteshimashite@localhost/tweet_live')

Session = sessionmaker(bind=engine)

class Tweet(Base):

	__tablename__ = 'tweet_tweet'

	id = Column(BigInteger, primary_key=True, autoincrement=False)

	created_at = Column(DateTime)

	text = Column(String)

	user_id = Column(Integer, ForeignKey('tweet_user.id'))

	user = relationship("User", backref="tweets")

	def __repr__(self):

		return anyjson.serialize({

			"id" : str(self.id),

			"created_at" : str(self.created_at),

			"text" : self.text,

			"user_id" : self.user_id,

			"screen_name" : self.user.screen_name

		})

class User(Base):

	__tablename__ = "tweet_user"

	id = Column(BigInteger, primary_key=True, autoincrement=False)

	created_at = Column(DateTime)

	screen_name = Column(String, index=True)

class TweetCast(object):

	def __init__(self):

		self.session = Session()

		a = self.session.query(func.max(Tweet.id))

		self.lastid = long(a[0][0])

		print self.lastid

		self.tweets = []

		self.clients = []

		self.clientpos = []

		print "__init__"

		gevent.spawn_later(1,self.getLastTweets)

	def getLastTweets(self):

		result = self.session.query(Tweet).filter(Tweet.id > self.lastid).all()

		if result:

			self.lastid = result[len(result)-1].id

			for tweet in result:

				self.tweets.append(tweet)

		print "getLastTweets - LastId = %d"%self.lastid

		gevent.spawn_later(1,self.getLastTweets)

	def webserver(self, env, start_response):

		print env

		if env['PATH_INFO'] == '/':

			query = parse_qs(env['QUERY_STRING'])

			start_response('200 OK', [('Content-Type', 'text/html')])

			cookie = env['HTTP_COOKIE']

			if cookie not in self.clients:

				self.clients.append(cookie)

				self.clientpos.append(0)

			clientid = self.clients.index(cookie)

			pos = self.clientpos[clientid]

			self.clientpos[clientid] = len(self.tweets)

			return ["%s[%s]%s"%(

				"%s("%query["callback"][0] if "callback" in query else "",

				",".join([str(tweet) for tweet in self.tweets[pos:]]),

				")" if "callback" in query else ""

			)]

		else:

			start_response('404 Not Found', [('Content-Type', 'text/html')])

			return ['<h1>Not Found</h1>']

tc = TweetCast()

WSGIServer(('', 8000), tc.webserver).serve_forever()