Wardle.py/src/g.py

"""
Copyright(c) 2022 SoftFish

Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files(the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and / or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
"""


import w
import c
import ui
import argparse
from time import sleep
from datetime import date
from signal import signal
from signal import SIGINT
from os import get_terminal_size

# Provide a clean getaway
def end():
   ui.clear()
   exit()

# Set up sigint handler
def handler(signum, frame):
   if signum == SIGINT:
      end()
signal(SIGINT, handler)

# Set up argument parser
parser = argparse.ArgumentParser(
   description='A barebones CLI Wordle clone, using official Wardle solutions. Also a Wordle cheating tool. You have been warned.'
   )
parser.add_argument('day',    nargs="?",           type=int, help="Wordle day number")
parser.add_argument('-l',     metavar="solution",            help="look up the day of a given solution")
parser.add_argument('-d',     metavar="day",       type=int, help="look up the solution of a given day")
parser.add_argument('-g',     metavar="guesses",   type=int, help="number of guesses")
parser.add_argument('--word', metavar="solution",            help="force a particular word")
parser.add_argument('--classic',  action='store_true',       help="use the classic Wordle word list")
parser.add_argument('--nonsense', action='store_true',       help="allow nonsensical guesses")
parser.add_argument('--hard',     action='store_true',       help="enable hard mode (wip)")
parser.add_argument('--clear',    action='store_true',       help="clear screen and reset cursor position on start")
# TODO: Implement hard mode properly (more than just a *)

# Parse the args
args = parser.parse_args()\

# Handle the args
#   Select the correct word list
if args.classic:
   import w
else:
   import w2 as w
#   Get today's day number
if args.day is not None:
   # Ahoy-hoy, time traveller!
   d = args.day
else:
   # find num. days since Wordle launch
   d = (date.today() - date(2021, 6, 19)).days
#   l is for "Find this word in the answers list"
if args.l:
   l = args.l.lower()
   # Look up the answer's index
   if l in w.Answers and w.Answers.index(l) < d:
      print ("Wordle {}: {}".format(w.Answers.index(l), l))
   else:

      print ("{} is not a Wordle (yet?)".format(args.l))
   exit(0)
#   d is for "Find this day's answer"
if args.d != None:
   # Look up the answer
   if args.d < len(w.Answers):
      print (w.Answers[args.d])
   else:
      print ("No Solution")
   exit(0)
#   g is for "maximum guesses"
if args.g:
   max_guesses = abs(args.g)
else:
   max_guesses = 6
#   Get today's word
if args.word:
   solution = args.word
   # Indicate a non-standard word
   d = args.word
   # If solution not in words list, enable nonsense
   if solution not in w.Words + w.Answers:
      args.nonsense = True
else:
   if d >= len(w.Answers):
      d = len(w.Answers) - 1
   solution = w.Answers[d]
# End arg parsing


# ! The game is below this point

# Good data structures to have
#   Box characters!
boxes = ["⬛", "🟨", "🟩"]
#   Colors!
#        gray bg    yellow bg  green bg   white fg  black bg
colors= [0x13a3a3c, 0x1b59f3b, 0x1538d4e, 0xd7dadc, 0x1121213]
#   Guesses go here
guesses = []
letters = [4] * 27


# Letter is in boxes
def wordbox(word, showword = 0):
   output = ""
   line = [0] * len(word)
   guess = list(word)
   sol   = list(solution)
   # Green pass
   for i in range(len(word)):
      if guess[i] == sol[i]:
         # Mark the letter 'green' (2)
         line[i] = 2
         letters[letter_num(word[i])] = 2
         # Remove letter from solution and guess
         sol[i] = guess[i] = 0
   # Yellow pass
   for i in range(len(word)):
      if guess[i] and word[i] in sol:
         # Mark the letter 'yellow' (1)
         line[i] = 1
         letters[letter_num(word[i])] = 1
         # Remove letter from solution and guess
         sol[sol.index(word[i])] = guess[i] = 0
   # Gray pass
   for i in range(len(word)):
      if line[i] == 0:
         letters[letter_num(word[i])] = 0
   # Turn blocks into a string, and print it
   if showword:
      # Move up to replace the input box
      output += ui.m(0,-1)
      for i in range(len(word)):
         #output += c.c(7,0,1) + (c.c(2,1) if line[i] == 2 else c.c(3,1) if line[i] == 1 else c.c(0,1,1)) + word[i].upper() + " " + c.reset
         output += c.c24(colors[3]) + c.c24(colors[line[i]]) + " " + word[i].upper() + " " + c.RESET
   else:
      for i in line:
         #output += c.c(7, 0, 1) + (c.c(2, 1) if i == 2 else c.c(3, 1) if i == 1 else c.c(0, 1, 1)) + boxes[i] + c.reset
         output += c.c24(colors[3]) + c.c24(colors[i]) + boxes[i] + c.RESET
   return output

def letter_num(char):
   if ord('a') <= ord(char.lower()) <= ord('z'):
      return ord(char.lower()) - ord('a')
   return 26

def keeb(x, y):
   def colorify(string):
      s = ""
      for char in string:
         s += c.c24(colors[letters[letter_num(char)]]) + char + c.RESET
      return s
   S = ["qwertyuiop","asdfghjkl","zxcvbnm"]
   ui.uprint(x, y,   colorify(" " + S[0] + " "))
   ui.uprint(x, y+1, colorify("  " + S[1] + " "))
   ui.uprint(x, y+2, colorify("  " + S[2] + "   "))


#   intro: Print the intro text
def intro():
   title  = "Wordle {}{}".format(d, "*" if args.hard else "")
   spaces = '  '
   # spaces = ' ' * ((13 - len(title)) // 2)
   intro  = c.RESET          # Reset the text printing colors
   if args.clear:
      intro += c.clear(False) # Clear the screen
      intro += ui.m_abs(0,0)   # Move the cursor to 0,0 (top left)
                                          # [  Wordle 100* ]
   intro += spaces + title + "\n"
   return intro

def game():
   x = get_terminal_size().columns//2 - 8
   ui.uprint(x, 0, intro())
   words = w.Words + w.Answers
   guess = 0
   while guess < max_guesses:
      # lol i should probably use curses
      keeb(x, 10)
      user_input = ui.uinput(x, len(guesses) + 2, c.clear() + "     ").lower()
      if len(user_input) == 5 and (user_input in words or args.nonsense):
         # Add guesses to the guessed list, and put the word in boxes
         guesses.append(user_input)
         ui.uprint(x, len(guesses) + 2, wordbox(user_input, 1))
         # win condition
         if (user_input == solution):
            win(True)
            return
         else:
            guess += 1
      else:
         # Allow the user to make it stop
         if user_input == "exit":
            end()
         # Alert the user that the word was not found in the list, and wait for them to read it
         ui.uprint (x, len(guesses) + 2, "not in word list.")
         sleep(1.5)
   # lose
   win(False)

def win(won):
   board = "Wordle {} {}/{}{}\n".format(d, len(guesses) if won else "X", max_guesses, "*" if args.hard else "")
   for gi in guesses:
      board += "\n" + wordbox(gi)

   print(ui.m(0,12) + board)


ui.init(15,12)
game()