From 9218601b522ca3b0dd40aa1c35b1e6b4e339bf09 Mon Sep 17 00:00:00 2001
From: adam <adam@adbraun.lol>
Date: Wed, 29 Jan 2025 13:19:30 +0000
Subject: [PATCH] fix logic for hard mode and strict hard mode

Reviewed-on: https://gitea.adbraun.lol/adam/wordle/pulls/2
---
 wordle | 38 +++++++++++++++++++++-----------------
 1 file changed, 21 insertions(+), 17 deletions(-)

diff --git a/wordle b/wordle
index 82dd492..d042102 100755
--- a/wordle
+++ b/wordle
@@ -41,8 +41,7 @@ function main() {
   for (( row=0; row<HEIGHT; row++ )); do
     declare GUESS
     GUESS=$(retrive_guess)
-    check_guess "${GUESS}" "${row}"
-    win=$?
+    check_guess "${GUESS}" "${row}" ; win=$?
 
     reveal_row "${row}"
     [[ $win == 5 ]] && echo -e "${CLEAR_DOWN}YOU WIN" && exit 0
@@ -157,6 +156,9 @@ function retrive_guess() {
 
   [[ ${valid_guess} == 1 ]] && GUESS=$(retrive_guess "Not A Valid Word")
   [[ ${valid_guess} == 2 ]] && GUESS=$(retrive_guess "Not A Valid Guess")
+  [[ ${valid_guess} == 3 ]] && GUESS=$(retrive_guess "GREEN Issue")
+  [[ ${valid_guess} == 4 ]] && GUESS=$(retrive_guess "YELLOW Issue")
+  [[ ${valid_guess} == 5 ]] && GUESS=$(retrive_guess "POOL Issue")
 
   echo -n "${GUESS^^}"
 }
@@ -183,14 +185,15 @@ function hard_mode_check() {
 
   for (( i=0; i<WIDTH; i++ )); do
     if [[ "${STRICT_HARD_MODE}" == 1 ]]; then
-      [[ "${POOL}" == *${GUESS:i:1}* ]] || return 2
+      [[ "${POOL}" == *${GUESS:i:1}* ]] || return 5
     fi
 
-    color_check "${GUESS:i:1}" $i && continue
-
-    if [[ "${yellows}" == *${GUESS:i:1}* ]]; then
-      yellows="${yellows/${GUESS:i:1}/}"
-      color_check "${GUESS:i:1}" $i && return 2
+    if [[ -n "${greens[i]}" ]]; then
+      [[ "${greens[i]}" != "${GUESS:i:1}" ]] && return 3
+    fi
+    
+    if [[ -n "${yellows[i]}" ]]; then
+      [[ "${yellows[i]}" == "${GUESS:i:1}" ]] && return 4
     fi
 
   done
@@ -213,10 +216,9 @@ function check_guess() {
   declare -r _guess=$1
   declare -r _row=$2
   declare -r _answer="${ANSWER}"
-  declare green_count=0
   declare chars=""
-  declare -g greens=""
-  declare -g yellows=""
+
+  declare -ga yellows=()
 
   for (( i=0; i<WIDTH; i++ )); do
     [[ "${_answer:i:1}" != "${_guess:i:1}" ]] && chars+=${_answer:i:1}
@@ -228,21 +230,23 @@ function check_guess() {
     declare _char=${_guess:$c:1}
 
     if [[ "${_char}" == "${_answer:$c:1}" ]]; then
-      green_count=$(( green_count+1 ))
-      greens+="${_char}"
+      greens[c]="${_char}"
       BOARD[$_row,$c]="${BOLD}${LETTER}${GREEN} ${_char} "
-      [[ $green_count == 5 ]] && return 5
       continue
     fi
 
     if [[ "${chars}" == *${_char}* ]]; then
       chars="${chars/${_char}/}"
-      yellows+="${_char}"
+      yellows[c]="${_char}"
       BOARD[$_row,$c]="${BOLD}${LETTER}${YELLOW} ${_char} "
-    else
-      POOL="${POOL/${_char}/}"
+      continue
     fi
+    [[ " ${yellows[*]} " =~ ${_char} ]] && continue
+    [[ " ${greens[*]} " =~ ${_char} ]] && continue
+    POOL="${POOL/${_char}/_}"
   done
+
+  [[ "${#greens[*]}" == 5 ]] && return 5
 }
 
 function reveal_row() {