#!/usr/bin/env python
'''
Run this script to update all the copyright headers of files
that were changed this year.

For example:

// Copyright (c) 2009-2012 The Bitcoin Unlimited developers

it will change it to

// Copyright (c) 2009-2015 The Bitcoin Unlimited developers
// Copyright (c) 2015-2017 The Bitcoin Unlimited developers
'''
import os
import time
import re

year = time.gmtime()[0]
CMD_GIT_DATE = 'git log --format=@%%at -1 %s | date +"%%Y" -u -f -'
CMD_REGEX= "perl -pi -e 's/(20\d\d)(?:-20\d\d)? The Bitcoin/$1-%s The Bitcoin/' %s"
REGEX_CURRENT= re.compile("%s The Bitcoin" % year)
CMD_LIST_FILES= "find %s | grep %s"

FOLDERS = ["./qa", "./src"]
EXTENSIONS = [".cpp",".h", ".py"]

def get_git_date(file_path):
  r = os.popen(CMD_GIT_DATE % file_path)
  for l in r:
    # Result is one line, so just return
    return l.replace("\n","")
  return ""

n=1
for folder in FOLDERS:
  for extension in EXTENSIONS:
    for file_path in os.popen(CMD_LIST_FILES % (folder, extension)):
      file_path = os.getcwd() + file_path[1:-1]
      if file_path.endswith(extension):
        git_date = get_git_date(file_path)
        if str(year) == git_date:
          # Only update if current year is not found
          if REGEX_CURRENT.search(open(file_path, "r").read()) is None:
            print n,"Last git edit", git_date, "-", file_path
            os.popen(CMD_REGEX % (year,file_path))
            n = n + 1
