// wordle.go

package main

import (
    "fmt"
    "os"
    "bufio"
    "time"
    "math/rand"
    "strings"
    "strconv"

    tea "github.com/charmbracelet/bubbletea"
    lg "github.com/charmbracelet/lipgloss"
)

type model struct {
    curTarget   []string
    curGuess    []string
    oldGuesses  [][]string
    curLetter   int
    curTry      int
    found       bool
    msg         string
    letters     map[string]int
}

var wordList []string
var wordSet map[string]struct{}

func initialModel() model {

    wordSet = make(map[string]struct{})

    fin, _ := os.Open(os.Args[1]); defer fin.Close()

    fs := bufio.NewScanner(fin)
 
    fs.Split(bufio.ScanLines)
  
    for fs.Scan() {
        w := strings.ToUpper(strings.TrimSpace(fs.Text()))
        if len(w) == 5 {
            wordList = append(wordList, w)
            wordSet[w] = struct{}{}
        }
    }

    t := strings.Split(wordList[rand.Intn(len(wordList))], "")

    letters := make(map[string]int)

    for c := 'A'; c <= 'Z'; c++ {
        letters[string(c)] = 0
    }

    return model{
        []string{t[0], t[1], t[2], t[3], t[4]}, 
        []string{" ", " ", " ", " ", " "},
        [][]string{},
        0, 0, false, "", letters,
    }
}

func (m model) Init() tea.Cmd {
    return nil
}

func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
    switch msg := msg.(type) {
    case tea.KeyMsg:

        mstr := strings.ToLower(msg.String())
        isLetter := len(mstr) == 1 && mstr >= "a" && mstr <= "z"

        if (isLetter) {

            mstr = strings.ToUpper(mstr)
            m.curGuess[m.curLetter] = mstr
            if (m.curLetter < 4) { m.curLetter++ }

        } else {
            switch mstr {
            case "left", "backspace":
                if (m.curLetter > 0) { m.curLetter-- }
            case "right":
                if (m.curLetter < 4 && m.curGuess[m.curLetter] != " ") { m.curLetter++ }
            case "ctrl+c", "esc":
                return m, tea.Quit
            case "enter":

                g := strings.TrimSpace(strings.Join(m.curGuess, ""))
                t := strings.Join(m.curTarget, "")
                _, exists := wordSet[g]

                if exists {

                    wd := wordDiff(t, g)

                    for i, c := range m.curGuess {
                        if wd[i] == 0 {
                            m.letters[c] = 3
                        } else if wd[i] == 2 {
                            m.letters[c] = 2
                        } else if m.letters[c] != 2 {
                            m.letters[c] = 1
                        }
                    }

                    m.found = m.found || g == t
                    m.oldGuesses = append(m.oldGuesses, m.curGuess[:])
                    m.curGuess = []string{" ", " ", " ", " ", " "}
                    m.curTry++
                    m.curLetter = 0
                    
                    if m.found {
                        m.msg = "congrats!"
                    } else if m.curTry > 5 {
                        m.msg = fmt.Sprintf("target word was '%v'", t)
                    } else {
                        m.msg = ""
                    }

                } else if (len(g) == 5) {
                    m.msg = fmt.Sprintf("word '%v' not in dictionary", g)
                    m.curGuess = []string{" ", " ", " ", " ", " "}
                    m.curLetter = 0    
                }
            }
        }
    }

    if (m.found || m.curTry == 6) {
        return m, tea.Quit
    } else {
        return m, nil
    }
}

func (m model) View() string {

    gs := []string{" ", " ", " ", " ", " ", " "}

    stDefault := lg.NewStyle()
    stRed := lg.NewStyle().Foreground(lg.Color("#ff0000"))
    stGreen := lg.NewStyle().Foreground(lg.Color("#00ff00"))
    stYellow := lg.NewStyle().Foreground(lg.Color("#ffff00"))
    stCursor := lg.NewStyle().Background(lg.Color("#646464"))

    var st lg.Style

    var ls []string

    for c := 'A'; c <= 'Z'; c++ {

        sc := string(c)

        switch m.letters[sc] {
        case 0: st = stDefault
        case 1: st = stYellow
        case 2: st = stGreen
        case 3: st = stRed
        }

        ls = append(ls, st.Render(sc))
    }

    for i := 0; i < 6; i++ {

        var mg []string
        g := []string{" ", " ", " ", " ", " "}

        if i < m.curTry {
            mg = m.oldGuesses[i]
            wd := wordDiff(strings.Join(m.curTarget, ""), strings.Join(mg, ""))

            for j := 0; j < 5; j++ {

                st = stDefault

                switch wd[j] {
                case 0: st = stRed
                case 1: st = stYellow
                case 2: st = stGreen
                }

                g[j] = fmt.Sprintf("[%v]", st.Render(mg[j]))
            }

        } else if i == m.curTry && !m.found {
            mg = m.curGuess

            for j := 0; j < 5; j++ {
                
                st = stDefault

                if (m.curLetter == j) {
                    st = stCursor
                }
                
                g[j] = fmt.Sprintf("[%v]", st.Render(mg[j]))
            }

        }

        if i < m.curTry || (i == m.curTry && !m.found)  {
            switch i {
            case 0, 1, 2: st = stGreen
            case 3, 4: st = stYellow
            case 5: st = stRed
            }

            gs[i] = fmt.Sprintf("%v - %v", st.Render(strconv.Itoa(i + 1)), 
                                           strings.TrimSpace(strings.Join(g, "")))
        }    

    }

    return fmt.Sprintf("\n%v\n\n%v\n\n%v\n\n", 
                        strings.Join(ls, " "), 
                        strings.TrimSpace(strings.Join(gs, "\n")), m.msg)
}

func wordDiff(target, guess string) (ret [5]int) {
    
    for i := 0; i < 5; i++ {
        if guess[i] == target[i] {
            ret[i] = 2
        } else {
            for j := 0; j < 5; j++ {
                if guess[i] == target[j] {
                    ret[i] = 1
                    break
                }
            }
        }
    }

    return
}

func main() {
    rand.Seed(time.Now().UnixNano())
    p := tea.NewProgram(initialModel())
    if err := p.Start(); err != nil {
        fmt.Printf("Alas, there's been an error: %v", err)
        os.Exit(1)
    }
}

@echo off
cls
path C:\Users\Michael\bin;%PATH%
chcp 65001 > nul

if "%ConEmuDir%"=="" (
  prompt $E[33;1m$P$G$E[90m 
)

set DIRCMD=/ogn

:: Macro Aliases

doskey apath=for %%F in ($1) do @path %%~fF;%%PATH%%
doskey mpath=for %%A in ("%%PATH:;=";"%%") do @if not "%%~A"=="" @echo   %%~A
doskey rpath=for %%F in ($1) do @for /F "usebackq delims=" %%M in (`echo set "PATH=%%PATH:%%~fF=%%"`) do @%%M

doskey deltree=rd/s/q $*
doskey dira=dir/a $*
doskey dird=dir/a-hd $*
doskey mcd=@md $* $T @cd $*

doskey 7z="C:\Program Files (x86)\7-Zip\7z.exe" $*
doskey npp="C:\Program Files (x86)\Notepad++\Notepad++" $*

... and so on.

::if 0 {
  @echo off
  tclsh "%~f0" %*
  goto :EOF
}

set f [open "|doskey /macros"]
set s [read $f]
if {[catch {close $f} err]} {
  puts $err
  exit
}

set lines []
set max 0
foreach line [lsort -dictionary [split $s \n]] {
  if {$line eq {}} continue
  lassign [split $line "="] alias command
  lappend lines [list $alias $command]
  set max [expr {max($max, [string length $alias])}]
}

foreach alias_command $lines {
  lassign $alias_command alias command
  puts "[format %${max}s $alias] = $command"
}

::if 0 {
  @echo off
  :: 
  :: Report the number of cmd.exe processes running in this Windows Console window
  :: (You can 'exit' n-1 times before this Windows Console closes.)
  ::
  set source=%~f0
  if exist %source%.bat set source=%source%.bat
  tclsh %source% %*
  goto :EOF
}

set pss [exec {*}[auto_execok wmic] process get processid,parentprocessid,executablepath]
foreach ps [split $pss \n] {
  lassign [lreverse [string map {\\ /} $ps]] pid ppid exename
  set PARENTS($pid) $ppid
  set EXES($pid) [file tail $exename]
}

set p [pid]
set n 0
while {$p in [array names PARENTS]} {
  if {$EXES($PARENTS($p)) ne "cmd.exe"} break
  incr n
  set p $PARENTS($p)
}

puts $n

...
alias dir='ls -CF --color'
alias dira='ls -A'
alias npp='/mnt/c/Program\ Files\ \(x86\)/Notepad\+\+/Notepad\+\+.exe'
...

@echo off
path C:\Python37\Scripts;C:\Python37;%PATH%
python.exe %*