llvm-project/llvm/utils/wciia.py

#!/usr/bin/env python
"""
wciia - Whose Code Is It Anyway

Determines code owner of the file/folder relative to the llvm source root.
Code owner is determined from the content of the CODE_OWNERS.TXT 
by parsing the D: field

usage:

utils/wciia.py  path

limitations:
- must be run from llvm source root
- very simplistic algorithm
- only handles * as a wildcard
- not very user friendly 
- does not handle the proposed F: field

"""

from __future__ import print_function
import os

code_owners = {}


def process_files_and_folders(owner):
    filesfolders = owner['filesfolders']
    # paths must be in ( ... ) so strip them
    lpar = filesfolders.find('(')
    rpar = filesfolders.rfind(')')
    if rpar <= lpar:
        # give up
        return
    paths = filesfolders[lpar + 1:rpar]
    # split paths
    owner['paths'] = []
    for path in paths.split():
        owner['paths'].append(path)


def process_code_owner(owner):
    if 'filesfolders' in owner:
        filesfolders = owner['filesfolders']
    else:
        #		print "F: field missing, using D: field"
        owner['filesfolders'] = owner['description']
    process_files_and_folders(owner)
    code_owners[owner['name']] = owner


# process CODE_OWNERS.TXT first
code_owners_file = open("CODE_OWNERS.TXT", "r").readlines()
code_owner = {}
for line in code_owners_file:
    for word in line.split():
        if word == "N:":
            name = line[2:].strip()
            if code_owner:
                process_code_owner(code_owner)
                code_owner = {}
            # reset the values
            code_owner['name'] = name
        if word == "E:":
            email = line[2:].strip()
            code_owner['email'] = email
        if word == "D:":
            description = line[2:].strip()
            code_owner['description'] = description
        if word == "F:":
            filesfolders = line[2:].strip()
            code_owner['filesfolders'].append(filesfolders)


def find_owners(fpath):
    onames = []
    lmatch = -1
    #  very simplistic way of findning the best match
    for name in code_owners:
        owner = code_owners[name]
        if 'paths' in owner:
            for path in owner['paths']:
                #				print "searching (" + path + ")"
                # try exact match
                if fpath == path:
                    return name
                # see if path ends with a *
                rstar = path.rfind('*')
                if rstar > 0:
                    # try the longest match,
                    rpos = -1
                    if len(fpath) < len(path):
                        rpos = path.find(fpath)
                    if rpos == 0:
                        onames.append(name)
    onames.append('Chris Lattner')
    return onames


# now lest try to find the owner of the file or folder
import sys

if len(sys.argv) < 2:
    print("usage " + sys.argv[0] + " file_or_folder")
    exit(-1)

# the path we are checking
path = str(sys.argv[1])

# check if this is real path
if not os.path.exists(path):
    print("path (" + path + ") does not exist")
    exit(-1)

owners_name = find_owners(path)

# be grammatically correct
print("The owner(s) of the (" + path + ") is(are) : " + str(owners_name))

exit(0)

# bottom up walk of the current .
# not yet used
root = "."
for dir, subdirList, fileList in os.walk(root, topdown=False):
    print("dir :", dir)
    for fname in fileList:
        print("-", fname)
    print()