#! /usr/bin/python
#-*- coding: utf-8 -*-

import os

# abrt_v4 TABLE columns:
UUID         = 0
UID          = 1
INFORMALL    = 2
DUMPDIR_PATH = 3
COUNT        = 4
REPORTED     = 5
TIME         = 6
MESSAGE      = 7

# abrt_v4_reportresult columns:
#UUID     = 0
#UID      = 1
REPORTER = 2
RESULT_MESSAGE  = 3

def get_db_path():
    path = "/var/spool/abrt/abrt-db"
    try:
        with open("/etc/abrt/plugins/SQLite3.conf") as f:
            for line in f:
                line = line.split('=', 1)
                if len(line) == 2 and line[0].strip() == "DBPath":
                    path = line[1].strip()
    except Exception, ex:
        pass
    return path

def get_url_from_text(text):
    url_marks = ["http://", "https://", "ftp://", "ftps://", "file://"]
    lines = text.split('\n')
    url = ""
    for mark in url_marks:
        for line in lines:
            last_mark = line.find(mark)
            if last_mark != -1:
                url_end = line.find(' ',last_mark)
                if url_end == -1:
                    url_end = len(line)
                url = "URL=" + line[last_mark:url_end]
    return url

def format_reported_to(reported_to):
    reporter = reported_to[REPORTER]
    url = get_url_from_text(reported_to[RESULT_MESSAGE])
    if not url:
        url = reported_to[RESULT_MESSAGE]
    return reporter + ": " + url

if __name__ == "__main__":
    try:
        from sqlite3 import dbapi2 as sqlite
        db = sqlite.connect(get_db_path())
        crashes = db.execute("SELECT * FROM abrt_v4")
        # abrt_v4 TABLE columns:
        # UUID | UID | INFORMALL | DUMPDIR_PATH | COUNT | REPORTED | TIME | MESSAGE
        for crash in crashes:
            # abrt_v4_reportresult columns:
            # UUID | UID | REPORTER | MESSAGE
            report_results = db.execute("SELECT * FROM abrt_v4_reportresult WHERE UUID='%s'" % crash[UUID])

            # save count from db to file
            count_file = "%s/count" % crash[DUMPDIR_PATH]
            # don't overwrite
            if not os.path.exists(count_file):
                try:
                    fout = open(count_file, "w")
                    fout.write(str(crash[COUNT]))
                    fout.close()
                except Exception, ex:
                    # silently ignore errors -> probably stalled db, but we can't
                    # do much about it, so it's better to not polute the rpm output
                    pass

            # save uuid from db to file
            uuid_file = "%s/uuid" % crash[DUMPDIR_PATH]
            # don't overwrite
            if not os.path.exists(uuid_file):
                try:
                    fout = open(uuid_file, "w")
                    fout.write(str(crash[UUID]))
                    fout.close()
                except Exception, ex:
                    # silently ignore errors -> probably stalled db, but we can't
                    # do much about it, so it's better to not polute the rpm output
                    pass

            results = report_results.fetchall()
            if results:
                # save report results from db to file
                reported_to_file = "%s/reported_to" % crash[DUMPDIR_PATH]
                if not os.path.exists(reported_to_file):
                    try:
                        fout = open(reported_to_file, "w")
                    except Exception, ex:
                        # silently ignore errors -> probably stalled db, but we can't
                        # do much about it, so it's better to not polute the rpm output
                        continue

                    for report_result in results:
                        # print "\t", format_reported_to(report_result)
                        # I know, it adds a '\n' to the end, but it's not a problem
                        fout.write("%s\n" % format_reported_to(report_result))
                    fout.close()
        db.close()
    except Exception, ex:
        # in case of any unhandled error, just ignore it, the worst, what
        # can happen is that the old reports are marked as unreported
        #print ex
        pass