#!/usr/bin/python
# Copyright (c) 2005-2014 Red Hat, Inc.
#
#    Koji is free software; you can redistribute it and/or
#    modify it under the terms of the GNU Lesser General Public
#    License as published by the Free Software Foundation;
#    version 2.1 of the License.
#
#    This software 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
#    Lesser General Public License for more details.
#
#    You should have received a copy of the GNU Lesser General Public
#    License along with this software; if not, write to the Free Software
#    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
#
# Authors:
#       Mike McLean <mikem@redhat.com>

# This modules provides a thread-safe way of passing
# request context around in a global way
#    - db connections
#    - request data
#    - auth data

from __future__ import absolute_import
import six.moves._thread
from six.moves import range
import six

class _data(object):
    pass

class ThreadLocal(object):
    def __init__(self):
        object.__setattr__(self, '_tdict', {})

    # should probably be getattribute, but easier to debug this way
    def __getattr__(self, key):
        id = six.moves._thread.get_ident()
        tdict = object.__getattribute__(self, '_tdict')
        if id not in tdict:
            raise AttributeError(key)
        data = tdict[id]
        return object.__getattribute__(data, key)

    def __setattr__(self, key, value):
        id = six.moves._thread.get_ident()
        tdict = object.__getattribute__(self, '_tdict')
        if id not in tdict:
            tdict[id] = _data()
        data = tdict[id]
        return object.__setattr__(data, key, value)

    def __delattr__(self, key):
        id = six.moves._thread.get_ident()
        tdict = object.__getattribute__(self, '_tdict')
        if id not in tdict:
            raise AttributeError(key)
        data = tdict[id]
        ret = object.__delattr__(data, key)
        if len(data.__dict__) == 0:
            del tdict[id]
        return ret

    def __str__(self):
        id = six.moves._thread.get_ident()
        tdict = object.__getattribute__(self, '_tdict')
        return "(current thread: %s) {" % id  + \
            ", ".join(["%s : %s" %(k, v.__dict__) for (k, v) in six.iteritems(tdict)]) + \
            "}"

    def _threadclear(self):
        id = six.moves._thread.get_ident()
        tdict = object.__getattribute__(self, '_tdict')
        if id not in tdict:
            return
        del tdict[id]


context = ThreadLocal()


if __name__ == '__main__':

    #testing

    #context.foo = 1
    #context.bar = 2
    print(context)
    #del context.bar
    print(context)

    import random
    import time
    def test():
        context.foo = random.random()
        time.sleep(1.5+random.random())
        context._threadclear()
        print(context)

    for x in range(1, 10):
        six.moves._thread.start_new_thread(test, ())

    time.sleep(4)
    print('')
    print(context)

    context.foo = 1
    context.bar = 2
    print(context.foo, context.bar)
    print(context)
    context._threadclear()
    print(context)