logging.getLogger().setLevel(logging.INFO)

args = parse_cmdline_args()

new_api_file = os.path.join(os.path.expanduser(args.pr_branch),

api_info.API_INFO_FILE_NAME)

old_api_file = os.path.join(os.path.expanduser(args.base_branch),

api_info.API_INFO_FILE_NAME)

if os.path.exists(new_api_file):

with open(new_api_file) as f:

new_api_json = json.load(f)

else:

new_api_json = {}

if os.path.exists(old_api_file):

with open(old_api_file) as f:

old_api_json = json.load(f)

else:

old_api_json = {}

diff = generate_diff_json(new_api_json, old_api_json)

if diff:

logging.info(f'json diff: \n{json.dumps(diff, indent=2)}')

logging.info(f'plain text diff report: \n{generate_text_report(diff)}')

report = generate_markdown_report(diff)

logging.info(f'markdown diff report: \n{report}')

else:

logging.info('No API Diff Detected.')

report = ""

output_dir = os.path.expanduser(args.output_dir)

if not os.path.exists(output_dir):

os.makedirs(output_dir)

api_report_path = os.path.join(output_dir, API_DIFF_FILE_NAME)

logging.info(f'Writing API diff report to {api_report_path}')

with open(api_report_path, 'w') as f:

"""diff_json only contains module & api that has a change.

NEXT_LEVEL = {'module': 'api_types', 'api_types': 'apis', 'apis': 'sub_apis'}

next_level = NEXT_LEVEL.get(level)

diff = {}

for key in set(new_api.keys()).union(old_api.keys()):

# Added API

if key not in old_api:

diff[key] = new_api[key]

diff[key]['status'] = STATUS_ADD

if diff[key].get('declaration'):

diff[key]['declaration'] = [STATUS_ADD] + diff[key]['declaration']

# Removed API

elif key not in new_api:

diff[key] = old_api[key]

diff[key]['status'] = STATUS_REMOVED

if diff[key].get('declaration'):

diff[key]['declaration'] = [STATUS_REMOVED] + diff[key]['declaration']

# Module Build Error. If a "module" exist but have no

# content (e.g. doc_path), it must have a build error.

elif level == 'module' and (not new_api[key]['path']

or not old_api[key]['path']):

diff[key] = {'status': STATUS_ERROR}

# Check diff in child level and diff in declaration

else:

child_diff = generate_diff_json(new_api[key][next_level],

old_api[key][next_level],

level=next_level) if next_level else {}

declaration_diff = new_api[key].get('declaration') != old_api[key].get(

'declaration') if level in ['apis', 'sub_apis'] else False

# No diff

if not child_diff and not declaration_diff:

continue

diff[key] = new_api[key]

# Changes at child level

if child_diff:

diff[key][next_level] = child_diff

# Modified API (changes in API declaration)

if declaration_diff:

diff[key]['status'] = STATUS_MODIFIED

diff[key]['declaration'] = [STATUS_ADD] + \

new_api[key]['declaration'] + \

[STATUS_REMOVED] + \

old_api[key]['declaration']

report = ''

indent_str = ' ' * level

for key, value in diff.items():

# filter out ["path", "api_type_link", "api_link", "declaration", "status"]

if isinstance(value, dict):

if key in ['api_types', 'apis', 'sub_apis']:

report += generate_text_report(value, level=level)

else:

status_text = f"{value.get('status', '')}:" if 'status' in value else ''

if status_text:

if print_key:

report += f'{indent_str}{status_text} {key}\n'

else:

report += f'{indent_str}{status_text}\n'

if value.get('declaration'):

for d in value.get('declaration'):

report += f'{indent_str}{d}\n'

else:

report += f'{indent_str}{key}\n'

report += generate_text_report(value, level=level + 1)

report = ''

header_str = '#' * (level + 3)

for key, value in diff.items():

if isinstance(value, dict):

if key in ['api_types', 'apis', 'sub_apis']:

report += generate_markdown_report(value, level=level)

else:

current_status = value.get('status')

if current_status:

# Module level: Always print out module name and class name as title

if level in [0, 2]:

report += f'{header_str} [{current_status}] {key}\n'

if current_status != STATUS_ERROR: # ADDED,REMOVED,MODIFIED

report += '<details>\n<summary>\n'

report += f'[{current_status}] {key}\n'

report += '</summary>\n\n'

declarations = value.get('declaration', [])

sub_report = generate_text_report(value, level=1, print_key=False)

detail = process_declarations(current_status, declarations,

sub_report)

report += f'```diff\n{detail}\n```\n\n</details>\n\n'

else: # no diff at current level

report += f'{header_str} {key}\n'

report += generate_markdown_report(value, level=level + 1)

# Module level: Always print out divider in the end

if level == 0:

report += '-----\n'

"""Diff syntax highlighting in Github Markdown."""

detail = ''

if current_status == STATUS_MODIFIED:

for line in (declarations + sub_report.split('\n')):

if STATUS_ADD in line:

prefix = '+ '

continue

elif STATUS_REMOVED in line:

prefix = '- '

continue

if line:

detail += f'{prefix}{line}\n'

else:

prefix = '+ ' if current_status == STATUS_ADD else '- '

for line in (declarations + sub_report.split('\n')):

if line:

detail += f'{prefix}{line}\n'

"""Categorize API info by Swift and Objective-C."""

lines = detail.split('\n')

swift_lines = [line.replace('Swift', '') for line in lines if 'Swift' in line]

objc_lines = [

line.replace('Objective-C', '') for line in lines if 'Objective-C' in line

]

swift_detail = 'Swift:\n' + '\n'.join(swift_lines) if swift_lines else ''

objc_detail = 'Objective-C:\n' + '\n'.join(objc_lines) if objc_lines else ''

if not swift_detail and not objc_detail:

return detail

else:

parser = argparse.ArgumentParser()

parser.add_argument('-p', '--pr_branch')

parser.add_argument('-b', '--base_branch')

parser.add_argument('-o', '--output_dir', default='output_dir')

args = parser.parse_args()