"""
Find connections between two words
"""

import click
import simplejson as json

from app.thesaurus.api import Thesaurus

@click.command()
@click.option('-a', '--a', 'opt_word_a', required=True,
  help='Starting word')
@click.option('-b', '--b', 'opt_word_b', required=True,
  help='Ending word')
@click.pass_context
def cli(ctx, opt_word_a, opt_word_b):
  """Find connections between two words
  """
  thesaurus = Thesaurus()
  print(f"Starting word: {opt_word_a}")
  print(f"Ending word: {opt_word_b}")
  results_a = thesaurus.search(opt_word_a)
  results_b = thesaurus.search(opt_word_b)
  # use sets
  # make set of results_a
  # find overlap with results_b
  # if there's no match...
  #   search for first word in results_a
  #   loop over results...
  # print(json.dumps(results, indent=2))