#  a small python script to test my page centering algorithm

import sys
from termcolor import colored, cprint

def main(curr_page, max_page):
  # default page settings
  start = 1
  end = max_page

  # initialize for later printing
  start_diff = 0
  end_diff = 0

  # handle centering
  if max_page >= 5:
      start = max(1, curr_page - 2)
      end = min(curr_page + 2, max_page)
      end_diff = max_page - curr_page
      if end_diff < 2:
          start = start - (2 - end_diff)
      start_diff = curr_page - 1
      if start_diff < 2:
          end = end + (2 - start_diff)

  print(f'max_page={max_page}')
  print(f'curr_page={curr_page}')
  print(f'start={start}')
  print(f'start_diff={start_diff}')
  print(f'end_diff={end_diff}')
  print(f'end={end}')
  print()

  i = start
  text = colored('')
  while i <= end:
      if i == curr_page:
          text = text + colored(f'{i}', 'grey', 'on_red') + colored(' ')
      else:
          text = text + colored(f'{i} ')
      i = i + 1
  cprint(text)


# executing script will call main()
if __name__ == "__main__":
  err_msg = 'usage: ./script <curr_page> <max_page>'
  try:
      arg1 = int(sys.argv[1])
      arg2 = int(sys.argv[2])
      # start the program
      main(arg1, arg2)
  except IndexError as e:
      print(e)
      print(err_msg)
  except Exception as e:
      print(e)