489 lines
11 KiB
Go
489 lines
11 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"runtime/pprof"
|
|
"strings"
|
|
|
|
"git.sr.ht/~tslocum/netris/pkg/event"
|
|
"git.sr.ht/~tslocum/netris/pkg/game"
|
|
"github.com/gdamore/tcell"
|
|
"github.com/tslocum/tview"
|
|
)
|
|
|
|
type Keybinding struct {
|
|
k tcell.Key
|
|
r rune
|
|
m tcell.ModMask
|
|
|
|
a event.GameAction
|
|
}
|
|
|
|
var keybindings = []*Keybinding{
|
|
{r: 'z', a: event.ActionRotateCCW},
|
|
{r: 'Z', a: event.ActionRotateCCW},
|
|
{r: 'x', a: event.ActionRotateCW},
|
|
{r: 'X', a: event.ActionRotateCW},
|
|
{k: tcell.KeyLeft, a: event.ActionMoveLeft},
|
|
{r: 'h', a: event.ActionMoveLeft},
|
|
{r: 'H', a: event.ActionMoveLeft},
|
|
{k: tcell.KeyDown, a: event.ActionSoftDrop},
|
|
{r: 'j', a: event.ActionSoftDrop},
|
|
{r: 'J', a: event.ActionSoftDrop},
|
|
{k: tcell.KeyUp, a: event.ActionHardDrop},
|
|
{r: 'k', a: event.ActionHardDrop},
|
|
{r: 'K', a: event.ActionHardDrop},
|
|
{k: tcell.KeyRight, a: event.ActionMoveRight},
|
|
{r: 'l', a: event.ActionMoveRight},
|
|
{r: 'L', a: event.ActionMoveRight},
|
|
}
|
|
|
|
var draftKeybindings []*Keybinding
|
|
|
|
func scrollMessages(direction int) {
|
|
var scroll int
|
|
if showLogLines > 3 {
|
|
scroll = (showLogLines - 2) * direction
|
|
} else {
|
|
scroll = showLogLines * direction
|
|
}
|
|
|
|
r, _ := recent.GetScrollOffset()
|
|
r += scroll
|
|
if r < 0 {
|
|
r = 0
|
|
}
|
|
recent.ScrollTo(r, 0)
|
|
|
|
draw <- event.DrawAll
|
|
}
|
|
|
|
// Render functions called here don't need to be queued (Draw is called when nil is returned)
|
|
func handleKeypress(ev *tcell.EventKey) *tcell.EventKey {
|
|
k := ev.Key()
|
|
r := ev.Rune()
|
|
|
|
if capturingKeybind {
|
|
capturingKeybind = false
|
|
if k == tcell.KeyEscape {
|
|
draftKeybindings = nil
|
|
|
|
app.SetRoot(gameSettingsContainerGrid, true)
|
|
updateGameSettings()
|
|
|
|
return nil
|
|
}
|
|
|
|
for i, bind := range draftKeybindings {
|
|
if (bind.k != 0 && bind.k != k) || (bind.r != 0 && bind.r != r) || (bind.m != 0 && bind.m != ev.Modifiers()) {
|
|
continue
|
|
}
|
|
|
|
draftKeybindings = append(draftKeybindings[:i], draftKeybindings[i+1:]...)
|
|
break
|
|
}
|
|
|
|
var action event.GameAction
|
|
switch gameSettingsSelectedButton {
|
|
case 0:
|
|
action = event.ActionRotateCCW
|
|
case 1:
|
|
action = event.ActionRotateCW
|
|
case 2:
|
|
action = event.ActionMoveLeft
|
|
case 3:
|
|
action = event.ActionMoveRight
|
|
case 4:
|
|
action = event.ActionSoftDrop
|
|
case 5:
|
|
action = event.ActionHardDrop
|
|
default:
|
|
log.Fatal("setting keybind for unknown action")
|
|
}
|
|
|
|
draftKeybindings = append(draftKeybindings, &Keybinding{k: k, r: r, m: ev.Modifiers(), a: action})
|
|
|
|
app.SetRoot(gameSettingsContainerGrid, true)
|
|
updateGameSettings()
|
|
return nil
|
|
} else if titleVisible {
|
|
if titleScreen > 1 {
|
|
switch k {
|
|
case tcell.KeyEscape:
|
|
if titleScreen == 5 {
|
|
titleScreen = 4
|
|
gameListSelected = 0
|
|
titleSelectedButton = 0
|
|
app.SetRoot(gameListContainerGrid, true)
|
|
renderGameList()
|
|
updateTitle()
|
|
return nil
|
|
} else if titleScreen == 4 {
|
|
titleScreen = 0
|
|
} else {
|
|
titleScreen = 1
|
|
}
|
|
titleSelectedButton = 0
|
|
|
|
app.SetRoot(titleContainerGrid, true)
|
|
updateTitle()
|
|
return nil
|
|
}
|
|
|
|
if titleScreen == 3 {
|
|
switch k {
|
|
case tcell.KeyTab:
|
|
gameSettingsSelectedButton++
|
|
if gameSettingsSelectedButton > 7 {
|
|
gameSettingsSelectedButton = 7
|
|
}
|
|
|
|
updateGameSettings()
|
|
return nil
|
|
case tcell.KeyBacktab:
|
|
gameSettingsSelectedButton--
|
|
if gameSettingsSelectedButton < 0 {
|
|
gameSettingsSelectedButton = 0
|
|
}
|
|
|
|
updateGameSettings()
|
|
return nil
|
|
case tcell.KeyEnter:
|
|
if gameSettingsSelectedButton == 6 || gameSettingsSelectedButton == 7 {
|
|
if gameSettingsSelectedButton == 7 {
|
|
keybindings = make([]*Keybinding, len(draftKeybindings))
|
|
copy(keybindings, draftKeybindings)
|
|
}
|
|
draftKeybindings = nil
|
|
|
|
titleScreen = 1
|
|
titleSelectedButton = 0
|
|
|
|
app.SetRoot(titleContainerGrid, true)
|
|
updateTitle()
|
|
return nil
|
|
}
|
|
|
|
modal := tview.NewModal().SetText("Press desired key(s) to set keybinding or press Escape to cancel.").ClearButtons()
|
|
app.SetRoot(modal, true)
|
|
|
|
capturingKeybind = true
|
|
return nil
|
|
}
|
|
} else if titleScreen == 4 {
|
|
switch k {
|
|
case tcell.KeyUp:
|
|
if titleSelectedButton == 0 {
|
|
if gameListSelected > 0 {
|
|
gameListSelected--
|
|
}
|
|
renderGameList()
|
|
}
|
|
return nil
|
|
case tcell.KeyBacktab:
|
|
previousTitleButton()
|
|
updateTitle()
|
|
renderGameList()
|
|
return nil
|
|
case tcell.KeyDown:
|
|
if titleSelectedButton == 0 {
|
|
if gameListSelected < len(gameList)-1 {
|
|
gameListSelected++
|
|
}
|
|
renderGameList()
|
|
}
|
|
return nil
|
|
case tcell.KeyTab:
|
|
nextTitleButton()
|
|
updateTitle()
|
|
renderGameList()
|
|
return nil
|
|
case tcell.KeyEnter:
|
|
if titleSelectedButton == 0 {
|
|
if gameListSelected >= 0 && gameListSelected < len(gameList) {
|
|
joinGame <- gameList[gameListSelected].ID
|
|
}
|
|
} else if titleSelectedButton == 1 {
|
|
titleScreen = 5
|
|
titleSelectedButton = 0
|
|
|
|
resetNewGameInputs()
|
|
app.SetRoot(newGameContainerGrid, true).SetFocus(nil)
|
|
updateTitle()
|
|
} else if titleSelectedButton == 2 {
|
|
titleScreen = 5
|
|
titleSelectedButton = 0
|
|
|
|
modal := tview.NewModal().SetText("Joining another server by IP via GUI is not yet implemented.\nPlease re-launch netris with the --connect argument instead.\n\nPress Escape to go back").ClearButtons()
|
|
app.SetRoot(modal, true)
|
|
} else if titleSelectedButton == 3 {
|
|
titleScreen = 0
|
|
titleSelectedButton = 0
|
|
|
|
app.SetRoot(titleContainerGrid, true)
|
|
updateTitle()
|
|
}
|
|
|
|
return nil
|
|
default:
|
|
if titleSelectedButton == 0 {
|
|
switch r {
|
|
case 'j', 'J':
|
|
if gameListSelected < len(gameList)-1 {
|
|
gameListSelected++
|
|
}
|
|
renderGameList()
|
|
return nil
|
|
case 'k', 'K':
|
|
if gameListSelected > 0 {
|
|
gameListSelected--
|
|
}
|
|
renderGameList()
|
|
return nil
|
|
case 'r', 'R':
|
|
refreshGameList()
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
} else if titleScreen == 5 {
|
|
switch k {
|
|
case tcell.KeyBacktab:
|
|
previousTitleButton()
|
|
updateTitle()
|
|
return nil
|
|
case tcell.KeyTab:
|
|
nextTitleButton()
|
|
updateTitle()
|
|
return nil
|
|
case tcell.KeyEnter:
|
|
if titleSelectedButton == 3 {
|
|
titleScreen = 4
|
|
gameListSelected = 0
|
|
titleSelectedButton = 0
|
|
app.SetRoot(gameListContainerGrid, true)
|
|
renderGameList()
|
|
updateTitle()
|
|
} else if titleSelectedButton == 4 {
|
|
joinGame <- event.GameIDNewCustom
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return ev
|
|
}
|
|
|
|
switch k {
|
|
case tcell.KeyEnter:
|
|
if titleScreen == 1 {
|
|
switch titleSelectedButton {
|
|
case 0:
|
|
resetPlayerSettingsForm()
|
|
|
|
titleScreen = 2
|
|
titleSelectedButton = 0
|
|
|
|
app.SetRoot(playerSettingsContainerGrid, true).SetFocus(playerSettingsForm)
|
|
return nil
|
|
case 1:
|
|
titleScreen = 3
|
|
titleSelectedButton = 0
|
|
gameSettingsSelectedButton = 0
|
|
|
|
draftKeybindings = make([]*Keybinding, len(keybindings))
|
|
copy(draftKeybindings, keybindings)
|
|
|
|
app.SetRoot(gameSettingsContainerGrid, true).SetFocus(buttonKeybindRotateCCW)
|
|
return nil
|
|
case 2:
|
|
titleScreen = 0
|
|
titleSelectedButton = 0
|
|
|
|
updateTitle()
|
|
return nil
|
|
}
|
|
} else {
|
|
if joinedGame {
|
|
switch titleSelectedButton {
|
|
case 0:
|
|
setTitleVisible(false)
|
|
return nil
|
|
case 1:
|
|
titleScreen = 1
|
|
titleSelectedButton = 0
|
|
|
|
updateTitle()
|
|
return nil
|
|
case 2:
|
|
done <- true
|
|
return nil
|
|
}
|
|
} else {
|
|
switch titleSelectedButton {
|
|
case 0:
|
|
titleScreen = 4
|
|
titleSelectedButton = 0
|
|
gameListSelected = 0
|
|
|
|
refreshGameList()
|
|
renderGameList()
|
|
|
|
app.SetRoot(gameListContainerGrid, true).SetFocus(nil)
|
|
updateTitle()
|
|
return nil
|
|
case 1:
|
|
joinGame <- event.GameIDNewLocal
|
|
return nil
|
|
case 2:
|
|
titleScreen = 1
|
|
titleSelectedButton = 0
|
|
|
|
updateTitle()
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
case tcell.KeyUp, tcell.KeyBacktab:
|
|
previousTitleButton()
|
|
updateTitle()
|
|
return nil
|
|
case tcell.KeyDown, tcell.KeyTab:
|
|
nextTitleButton()
|
|
updateTitle()
|
|
return nil
|
|
case tcell.KeyEscape:
|
|
if titleScreen == 1 {
|
|
titleScreen = 0
|
|
titleSelectedButton = 0
|
|
updateTitle()
|
|
} else if joinedGame {
|
|
setTitleVisible(false)
|
|
} else {
|
|
done <- true
|
|
}
|
|
return nil
|
|
default:
|
|
switch r {
|
|
case 'k', 'K':
|
|
previousTitleButton()
|
|
updateTitle()
|
|
return nil
|
|
case 'j', 'J':
|
|
nextTitleButton()
|
|
updateTitle()
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return ev
|
|
}
|
|
|
|
if inputActive {
|
|
switch k {
|
|
case tcell.KeyEnter:
|
|
msg := inputView.GetText()
|
|
if msg != "" {
|
|
if strings.HasPrefix(msg, "/cpu") {
|
|
if profileCPU == nil {
|
|
if len(msg) < 5 {
|
|
logMessage("Profile name must be specified")
|
|
} else {
|
|
profileName := strings.TrimSpace(msg[5:])
|
|
|
|
var err error
|
|
profileCPU, err = os.Create(profileName)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = pprof.StartCPUProfile(profileCPU)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
logMessage(fmt.Sprintf("Started profiling CPU usage as %s", profileName))
|
|
}
|
|
} else {
|
|
pprof.StopCPUProfile()
|
|
profileCPU.Close()
|
|
profileCPU = nil
|
|
|
|
logMessage("Stopped profiling CPU usage")
|
|
}
|
|
} else if strings.HasPrefix(msg, "/version") {
|
|
v := game.Version
|
|
if v == "" {
|
|
v = "unknown"
|
|
}
|
|
|
|
logMessage(fmt.Sprintf("netris version %s", v))
|
|
} else if strings.HasPrefix(msg, "/ping") {
|
|
if activeGame != nil {
|
|
activeGame.ProcessAction(event.ActionPing)
|
|
}
|
|
} else if strings.HasPrefix(msg, "/stats") {
|
|
if activeGame != nil {
|
|
activeGame.ProcessAction(event.ActionStats)
|
|
}
|
|
} else {
|
|
if activeGame != nil {
|
|
activeGame.Event <- &event.MessageEvent{Message: msg}
|
|
} else {
|
|
logMessage("Message not sent - not currently connected to any game")
|
|
}
|
|
}
|
|
}
|
|
|
|
setInputStatus(false)
|
|
return nil
|
|
case tcell.KeyPgUp:
|
|
scrollMessages(-1)
|
|
return nil
|
|
case tcell.KeyPgDn:
|
|
scrollMessages(1)
|
|
return nil
|
|
case tcell.KeyEscape:
|
|
setInputStatus(false)
|
|
return nil
|
|
}
|
|
|
|
return ev
|
|
}
|
|
|
|
switch k {
|
|
case tcell.KeyEnter:
|
|
setInputStatus(!inputActive)
|
|
return nil
|
|
case tcell.KeyTab:
|
|
setShowDetails(!showDetails)
|
|
return nil
|
|
case tcell.KeyPgUp:
|
|
scrollMessages(-1)
|
|
return nil
|
|
case tcell.KeyPgDn:
|
|
scrollMessages(1)
|
|
return nil
|
|
case tcell.KeyEscape:
|
|
setTitleVisible(true)
|
|
return nil
|
|
}
|
|
|
|
for _, bind := range keybindings {
|
|
if (bind.k != 0 && bind.k != k) || (bind.r != 0 && bind.r != r) || (bind.m != 0 && bind.m != ev.Modifiers()) {
|
|
continue
|
|
} else if activeGame == nil {
|
|
break
|
|
}
|
|
|
|
activeGame.ProcessAction(bind.a)
|
|
return nil
|
|
}
|
|
|
|
return ev
|
|
}
|