#!/usr/bin/env python3
# -*- coding: utf-8 -*-"
# Copyright 2021 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Update rebase database with new information from upstream and next database"""

from __future__ import print_function
import sqlite3
import re
from common import rebasedb, upstreamdb, nextdb

subject = re.compile(
    '(ANDROID: *|CHROMIUM: *|CHROMEOS: *|UPSTREAM: *|FROMGIT: *|FROMLIST: *|BACKPORT: *)*(.*)'
)


def findsha(uconn, sha, patchid, desc):
    """Try to find matching SHA in provided database.

    Return updated SHA, or None if not found.
    """

    c = uconn.cursor()

    if sha is not None:
        c.execute("select sha from commits where sha is '%s'" % sha)
        usha = c.fetchone()
        if usha:
            # print("  Found SHA %s in upstream database" % usha)
            return usha[0]

    # Now look for patch id if provided
    if patchid is not None:
        c.execute("select sha from commits where patchid is '%s'" % patchid)
        usha = c.fetchone()
        if usha:
            # print("  Found SHA %s in upstream database based on patch ID" % usha)
            return usha[0]

    # The SHA is not upstream, or not known at all.
    # See if we can find the commit subject.
    s = subject.search(desc)
    if s:
        sdesc = s.group(2).replace("'", "''")
        c.execute("select sha from commits where subject is '%s'" % sdesc)
        usha = c.fetchone()
        if usha:
            # print("  Found upstream SHA '%s' based on subject line" % usha)
            return usha[0]

    return None


def update_commits():
    """Validate 'usha' field in rebase database.

    Verify if the upstream SHA actually exists by looking it up in the upstream
    database. If it doesn't exist, and if a matching commit is not found either,
    remove it.
    """

    conn = sqlite3.connect(rebasedb)
    uconn = sqlite3.connect(upstreamdb)
    nconn = sqlite3.connect(nextdb) if nextdb else None
    c = conn.cursor()

    c.execute('select sha, usha, patchid, subject from commits')
    for (sha, usha, patchid, desc) in c.fetchall():
        uusha = findsha(uconn, usha, patchid, desc)
        # if it is not in the upstream database, maybe it is in -next.
        # Try to pick it up from there.
        if uusha is None and nconn:
            uusha = findsha(nconn, usha, None, desc)
        if not uusha:
            uusha = ''
        if usha != uusha:
            print("SHA '%s': Updating usha '%s' with '%s'" % (sha, usha, uusha))
            c.execute("UPDATE commits set usha='%s' where sha='%s'" %
                      (uusha, sha))

    conn.commit()
    conn.close()
    uconn.close()
    if nconn:
        nconn.close()


if __name__ == '__main__':
    update_commits()