#       Licensed to the Apache Software Foundation (ASF) under one
#       or more contributor license agreements.  See the NOTICE file
#       distributed with this work for additional information
#       regarding copyright ownership.  The ASF licenses this file
#       to you under the Apache License, Version 2.0 (the
#       "License"); you may not use this file except in compliance
#       with the License.  You may obtain a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#       Unless required by applicable law or agreed to in writing,
#       software distributed under the License is distributed on an
#       "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
#       KIND, either express or implied.  See the License for the
#       specific language governing permissions and limitations
#       under the License.

import sys
import json
import logging

from pylons import tmpl_context as c

from ming.orm import session, MappedClass

from allura import model as M

log = logging.getLogger(__name__)

def main():
    test = sys.argv[-1] == 'test'
    projects = M.Project.query.find().all()
    log.info('Restoring labels on projects')
    for p in projects:
        restore_labels(p, test)
    if not test:
        session(p).flush()
    log.info('Restoring labels on artifacts')
    for p in projects:
        if p.parent_id: continue
        c.project = p
        for name, cls in MappedClass._registry.iteritems():
            if not issubclass(cls, M.Artifact): continue
            if session(cls) is None: continue
            for a in cls.query.find():
                restore_labels(a, test)
        if not test:
            M.artifact_orm_session.flush()
        M.artifact_orm_session.clear()

def restore_labels(obj, test=True):
    if not obj.labels: return
    labels = obj.labels
    while True:
        if not labels or labels[0] != '[': return
        lbllen = map(len, labels)
        if max(lbllen) != 1: return
        if min(lbllen) != 1: return
        s = ''.join(labels)
        s = s.replace("u'", "'")
        s = s.replace('u"', '"')
        jobj = '{"obj":' + s.replace("'", '"') + '}'
        try:
            new_labels = json.loads(jobj)['obj']
        except ValueError:
            # some weird problem with json decoding, just erase the labels
            new_labels = []
        if not isinstance(new_labels, list): return
        for lbl in new_labels:
            if not isinstance(lbl, basestring): return
        log.info('%s: %s => %s', obj.__class__, labels, new_labels)
        labels = new_labels
        if not test:
            log.info('...actually restoring labels') 
            obj.labels = new_labels

if __name__ == '__main__':
    main()