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()