Compare commits
No commits in common. "develop" and "v0.1.0" have entirely different histories.
@ -1,3 +1,3 @@
|
|||||||
SN_BASE_URL=
|
SN_BASE_URL=
|
||||||
SN_API_KEY=
|
SN_API_TOKEN=
|
||||||
SN_MEDIA_URL=
|
SN_MEDIA_URL=
|
||||||
|
1
.gitignore
vendored
1
.gitignore
vendored
@ -2,4 +2,3 @@
|
|||||||
!assets/*.png
|
!assets/*.png
|
||||||
.env
|
.env
|
||||||
*.sqlite3
|
*.sqlite3
|
||||||
chessbot
|
|
||||||
|
13
Makefile
13
Makefile
@ -1,13 +0,0 @@
|
|||||||
.PHONY: build test
|
|
||||||
|
|
||||||
SOURCES := main.go $(shell find chess -name '*.go')
|
|
||||||
|
|
||||||
build: chessbot
|
|
||||||
|
|
||||||
chessbot: $(SOURCES)
|
|
||||||
go build -o chessbot .
|
|
||||||
|
|
||||||
test:
|
|
||||||
## -count=1 is used to disable cache
|
|
||||||
## use -run <regexp> to only run tests that match <regexp>
|
|
||||||
go test ./chess -v -count=1
|
|
102
chess/board.go
102
chess/board.go
@ -12,19 +12,13 @@ import (
|
|||||||
|
|
||||||
"golang.org/x/image/font"
|
"golang.org/x/image/font"
|
||||||
"golang.org/x/image/font/basicfont"
|
"golang.org/x/image/font/basicfont"
|
||||||
"golang.org/x/image/font/opentype"
|
|
||||||
"golang.org/x/image/math/fixed"
|
"golang.org/x/image/math/fixed"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Board struct {
|
type Board struct {
|
||||||
tiles [8][8]*Piece
|
tiles [8][8]*Piece
|
||||||
turn Color
|
turn Color
|
||||||
Moves []string
|
Moves []string
|
||||||
moveIndicators []Tile
|
|
||||||
}
|
|
||||||
|
|
||||||
type Tile struct {
|
|
||||||
X, Y int
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewBoard() *Board {
|
func NewBoard() *Board {
|
||||||
@ -107,7 +101,7 @@ func (b *Board) Image() *image.RGBA {
|
|||||||
x := xi * 128
|
x := xi * 128
|
||||||
y := yi * 128
|
y := yi * 128
|
||||||
rect = image.Rect(x, y, x+128, y+128)
|
rect = image.Rect(x, y, x+128, y+128)
|
||||||
bg = image.NewUniform(getTileColor(b, xi, yi))
|
bg = image.NewUniform(getTileColor(xi, yi))
|
||||||
draw.Draw(img, rect, bg, p, draw.Src)
|
draw.Draw(img, rect, bg, p, draw.Src)
|
||||||
|
|
||||||
piece = b.tiles[xi][yi]
|
piece = b.tiles[xi][yi]
|
||||||
@ -192,23 +186,17 @@ func drawCoordinate(img *image.RGBA, x, y int, flipped bool) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
drawString := func(s string, origin fixed.Point26_6) {
|
drawString := func(s string, origin fixed.Point26_6) {
|
||||||
color := getTileColor(nil, x, y)
|
color := getTileColor(x, y)
|
||||||
if !flipped && color == Light {
|
if !flipped && color == Light {
|
||||||
color = Dark
|
color = Dark
|
||||||
} else if !flipped {
|
} else if !flipped {
|
||||||
color = Light
|
color = Light
|
||||||
}
|
}
|
||||||
|
// TODO: use SN font and make it bold
|
||||||
face, err := loadFontFace("lightningvolt.ttf")
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("error loading font: %v\n", err)
|
|
||||||
face = basicfont.Face7x13
|
|
||||||
}
|
|
||||||
|
|
||||||
d := &font.Drawer{
|
d := &font.Drawer{
|
||||||
Dst: img,
|
Dst: img,
|
||||||
Src: image.NewUniform(color),
|
Src: image.NewUniform(color),
|
||||||
Face: face,
|
Face: basicfont.Face7x13,
|
||||||
Dot: origin,
|
Dot: origin,
|
||||||
}
|
}
|
||||||
d.DrawString(s)
|
d.DrawString(s)
|
||||||
@ -221,36 +209,11 @@ func drawCoordinate(img *image.RGBA, x, y int, flipped bool) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if row != "" {
|
if row != "" {
|
||||||
origin = fixed.P((x+1)*128-20, y*128+23)
|
origin = fixed.P((x+1)*128-12, y*128+15)
|
||||||
drawString(row, origin)
|
drawString(row, origin)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func loadFontFace(path string) (font.Face, error) {
|
|
||||||
fontBytes, err := os.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
ttfFont, err := opentype.Parse(fontBytes)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
faceOptions := &opentype.FaceOptions{
|
|
||||||
Size: 32,
|
|
||||||
DPI: 72,
|
|
||||||
Hinting: font.HintingNone,
|
|
||||||
}
|
|
||||||
|
|
||||||
fontFace, err := opentype.NewFace(ttfFont, faceOptions)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return fontFace, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *Board) SetPiece(name PieceName, color Color, position string) error {
|
func (b *Board) SetPiece(name PieceName, color Color, position string) error {
|
||||||
var (
|
var (
|
||||||
piece *Piece
|
piece *Piece
|
||||||
@ -277,10 +240,6 @@ func (b *Board) SetPiece(name PieceName, color Color, position string) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Board) AlgebraicNotation() string {
|
func (b *Board) AlgebraicNotation() string {
|
||||||
if len(b.Moves) == 0 {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
var text string
|
var text string
|
||||||
for i, m := range b.Moves {
|
for i, m := range b.Moves {
|
||||||
if i%2 == 0 {
|
if i%2 == 0 {
|
||||||
@ -397,7 +356,7 @@ func (b *Board) Move(move string) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// make sure the move is marked as a check if it was
|
// make sure the move is marked as a check if it was
|
||||||
if b.InCheck() && !strings.HasSuffix(move, "+") && !strings.HasSuffix(move, "#") {
|
if b.InCheck() && !strings.HasSuffix(move, "+") {
|
||||||
move += "+"
|
move += "+"
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -416,7 +375,6 @@ func parseMove(move string) (string, int, int, string, error) {
|
|||||||
)
|
)
|
||||||
|
|
||||||
move = strings.TrimSuffix(move, "+")
|
move = strings.TrimSuffix(move, "+")
|
||||||
move = strings.TrimSuffix(move, "#")
|
|
||||||
|
|
||||||
if move == "O-O" {
|
if move == "O-O" {
|
||||||
return "K", 5, 7, "g1", nil
|
return "K", 5, 7, "g1", nil
|
||||||
@ -839,7 +797,6 @@ func (b *Board) movePawn(position string, fromX int, fromY int, promotion string
|
|||||||
if promotion != "" {
|
if promotion != "" {
|
||||||
return b.promotePawn(toX, toY, promotion)
|
return b.promotePawn(toX, toY, promotion)
|
||||||
}
|
}
|
||||||
b.moveIndicators = []Tile{{fromX, fromY}, {toX, toY}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -862,7 +819,6 @@ func (b *Board) movePawn(position string, fromX int, fromY int, promotion string
|
|||||||
if promotion != "" {
|
if promotion != "" {
|
||||||
return b.promotePawn(toX, toY, promotion)
|
return b.promotePawn(toX, toY, promotion)
|
||||||
}
|
}
|
||||||
b.moveIndicators = []Tile{{toX, yPrev}, {toX, toY}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -879,7 +835,6 @@ func (b *Board) movePawn(position string, fromX int, fromY int, promotion string
|
|||||||
if promotion != "" {
|
if promotion != "" {
|
||||||
return b.promotePawn(toX, toY, promotion)
|
return b.promotePawn(toX, toY, promotion)
|
||||||
}
|
}
|
||||||
b.moveIndicators = []Tile{{toX, yPrev}, {toX, toY}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -989,7 +944,6 @@ func (b *Board) moveRook(position string, queen bool, fromX int, fromY int) erro
|
|||||||
p = b.getPiece(xPrev, yPrev)
|
p = b.getPiece(xPrev, yPrev)
|
||||||
b.tiles[x][y] = p
|
b.tiles[x][y] = p
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1020,7 +974,6 @@ func (b *Board) moveBishop(position string, queen bool) error {
|
|||||||
if ((!queen && piece.Name == Bishop) || (queen && piece.Name == Queen)) && piece.Color == b.turn {
|
if ((!queen && piece.Name == Bishop) || (queen && piece.Name == Queen)) && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
} else {
|
} else {
|
||||||
// direction blocked by other piece
|
// direction blocked by other piece
|
||||||
@ -1039,7 +992,6 @@ func (b *Board) moveBishop(position string, queen bool) error {
|
|||||||
if ((!queen && piece.Name == Bishop) || (queen && piece.Name == Queen)) && piece.Color == b.turn {
|
if ((!queen && piece.Name == Bishop) || (queen && piece.Name == Queen)) && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
} else {
|
} else {
|
||||||
break
|
break
|
||||||
@ -1057,7 +1009,6 @@ func (b *Board) moveBishop(position string, queen bool) error {
|
|||||||
if ((!queen && piece.Name == Bishop) || (queen && piece.Name == Queen)) && piece.Color == b.turn {
|
if ((!queen && piece.Name == Bishop) || (queen && piece.Name == Queen)) && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
} else {
|
} else {
|
||||||
break
|
break
|
||||||
@ -1075,7 +1026,6 @@ func (b *Board) moveBishop(position string, queen bool) error {
|
|||||||
if ((!queen && piece.Name == Bishop) || (queen && piece.Name == Queen)) && piece.Color == b.turn {
|
if ((!queen && piece.Name == Bishop) || (queen && piece.Name == Queen)) && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
} else {
|
} else {
|
||||||
break
|
break
|
||||||
@ -1169,7 +1119,6 @@ func (b *Board) moveKnight(position string, fromX int, fromY int) error {
|
|||||||
p = b.getPiece(xPrev, yPrev)
|
p = b.getPiece(xPrev, yPrev)
|
||||||
b.tiles[x][y] = p
|
b.tiles[x][y] = p
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1241,8 +1190,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
b.tiles[5][y] = rook
|
b.tiles[5][y] = rook
|
||||||
b.tiles[7][y] = nil
|
b.tiles[7][y] = nil
|
||||||
|
|
||||||
b.moveIndicators = []Tile{{4, y}, {x, y}, {5, y}, {7, y}}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1276,8 +1223,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
b.tiles[3][y] = rook
|
b.tiles[3][y] = rook
|
||||||
b.tiles[0][y] = nil
|
b.tiles[0][y] = nil
|
||||||
|
|
||||||
b.moveIndicators = []Tile{{2, y}, {4, y}, {3, y}, {0, y}}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -1289,7 +1234,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1300,7 +1244,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1311,7 +1254,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1322,7 +1264,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1333,7 +1274,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1344,7 +1284,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1355,7 +1294,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1366,7 +1304,6 @@ func (b *Board) moveKing(position string, castle bool) error {
|
|||||||
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
if piece != nil && piece.Name == King && piece.Color == b.turn {
|
||||||
b.tiles[xPrev][yPrev] = nil
|
b.tiles[xPrev][yPrev] = nil
|
||||||
b.tiles[x][y] = piece
|
b.tiles[x][y] = piece
|
||||||
b.moveIndicators = []Tile{{xPrev, yPrev}, {x, y}}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1451,25 +1388,8 @@ func (b *Board) getCollision(position string) (*Piece, error) {
|
|||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func getTileColor(b *Board, x, y int) Color {
|
func getTileColor(x, y int) Color {
|
||||||
|
if x%2 == y%2 {
|
||||||
lightTile := x%2 == y%2
|
|
||||||
|
|
||||||
if b != nil {
|
|
||||||
// highlight move
|
|
||||||
// TODO: refactor using alpha channels
|
|
||||||
for _, t := range b.moveIndicators {
|
|
||||||
if t.X == x && t.Y == y {
|
|
||||||
if lightTile {
|
|
||||||
return LightGreen
|
|
||||||
} else {
|
|
||||||
return DarkGreen
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if lightTile {
|
|
||||||
return Light
|
return Light
|
||||||
} else {
|
} else {
|
||||||
return Dark
|
return Dark
|
||||||
|
@ -472,22 +472,6 @@ func TestBoardCheck(t *testing.T) {
|
|||||||
assertParse(t, b, "Kxf7")
|
assertParse(t, b, "Kxf7")
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestBoardCheckmate(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
b := chess.NewBoard()
|
|
||||||
|
|
||||||
assert.False(t, b.InCheck())
|
|
||||||
|
|
||||||
// fool's mate
|
|
||||||
assertParse(t, b, "f3 e6 g4 Qh4#")
|
|
||||||
|
|
||||||
assert.True(t, b.InCheck())
|
|
||||||
assert.True(t, strings.HasSuffix(b.Moves[len(b.Moves)-1], "#"), "checkmate move should end with #")
|
|
||||||
|
|
||||||
assertMoveError(t, b, "a3", "invalid move a3: king is in check")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBoardPin(t *testing.T) {
|
func TestBoardPin(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
|
@ -58,10 +58,8 @@ const (
|
|||||||
type Color color.Color
|
type Color color.Color
|
||||||
|
|
||||||
var (
|
var (
|
||||||
Light Color = color.RGBA{240, 217, 181, 255}
|
Light Color = color.RGBA{240, 217, 181, 255}
|
||||||
Dark Color = color.RGBA{181, 136, 99, 255}
|
Dark Color = color.RGBA{181, 136, 99, 255}
|
||||||
LightGreen Color = color.RGBA{205, 210, 106, 255}
|
|
||||||
DarkGreen Color = color.RGBA{170, 162, 58, 255}
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func NewPiece(name PieceName, color Color) (*Piece, error) {
|
func NewPiece(name PieceName, color Color) (*Piece, error) {
|
||||||
|
9
db/db.go
9
db/db.go
@ -53,15 +53,6 @@ func ItemHasReply(parentId int, userId int) (bool, error) {
|
|||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if count > 0 {
|
|
||||||
return true, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// check if parent already exists, this means we ignored it
|
|
||||||
if err = db.QueryRow(`SELECT COUNT(1) FROM items WHERE id = ?`, parentId).Scan(&count); err != nil {
|
|
||||||
return true, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return count > 0, nil
|
return count > 0, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
3
go.mod
3
go.mod
@ -3,7 +3,7 @@ module github.com/ekzyis/chessbot
|
|||||||
go 1.23.0
|
go 1.23.0
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/ekzyis/snappy v0.7.0
|
github.com/ekzyis/snappy v0.5.5-0.20240923014110-b880c1256e13
|
||||||
github.com/mattn/go-sqlite3 v1.14.23
|
github.com/mattn/go-sqlite3 v1.14.23
|
||||||
github.com/stretchr/testify v1.9.0
|
github.com/stretchr/testify v1.9.0
|
||||||
golang.org/x/image v0.20.0
|
golang.org/x/image v0.20.0
|
||||||
@ -12,7 +12,6 @@ require (
|
|||||||
require (
|
require (
|
||||||
github.com/davecgh/go-spew v1.1.1 // indirect
|
github.com/davecgh/go-spew v1.1.1 // indirect
|
||||||
github.com/pmezard/go-difflib v1.0.0 // indirect
|
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||||
golang.org/x/text v0.18.0 // indirect
|
|
||||||
gopkg.in/guregu/null.v4 v4.0.0 // indirect
|
gopkg.in/guregu/null.v4 v4.0.0 // indirect
|
||||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
)
|
)
|
||||||
|
6
go.sum
6
go.sum
@ -1,7 +1,7 @@
|
|||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/ekzyis/snappy v0.7.0 h1:RcFTUHdZFTBFOnh6cG9HCL/RPK6L13qdxDrjBNZFjEM=
|
github.com/ekzyis/snappy v0.5.5-0.20240923014110-b880c1256e13 h1:yxDqMo4HzCVzhxPNTBpST0KVVHVMwGouzwfXsb4WYw4=
|
||||||
github.com/ekzyis/snappy v0.7.0/go.mod h1:UksYI0dU0+cnzz0LQjWB1P0QQP/ghx47e4atP99a5Lk=
|
github.com/ekzyis/snappy v0.5.5-0.20240923014110-b880c1256e13/go.mod h1:UksYI0dU0+cnzz0LQjWB1P0QQP/ghx47e4atP99a5Lk=
|
||||||
github.com/mattn/go-sqlite3 v1.14.23 h1:gbShiuAP1W5j9UOksQ06aiiqPMxYecovVGwmTxWtuw0=
|
github.com/mattn/go-sqlite3 v1.14.23 h1:gbShiuAP1W5j9UOksQ06aiiqPMxYecovVGwmTxWtuw0=
|
||||||
github.com/mattn/go-sqlite3 v1.14.23/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
|
github.com/mattn/go-sqlite3 v1.14.23/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
|
||||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
@ -10,8 +10,6 @@ github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsT
|
|||||||
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
golang.org/x/image v0.20.0 h1:7cVCUjQwfL18gyBJOmYvptfSHS8Fb3YUDtfLIZ7Nbpw=
|
golang.org/x/image v0.20.0 h1:7cVCUjQwfL18gyBJOmYvptfSHS8Fb3YUDtfLIZ7Nbpw=
|
||||||
golang.org/x/image v0.20.0/go.mod h1:0a88To4CYVBAHp5FXJm8o7QbUl37Vd85ply1vyD8auM=
|
golang.org/x/image v0.20.0/go.mod h1:0a88To4CYVBAHp5FXJm8o7QbUl37Vd85ply1vyD8auM=
|
||||||
golang.org/x/text v0.18.0 h1:XvMDiNzPAl0jr17s6W9lcaIhGUfUORdGCNsuLmPG224=
|
|
||||||
golang.org/x/text v0.18.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
|
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
gopkg.in/guregu/null.v4 v4.0.0 h1:1Wm3S1WEA2I26Kq+6vcW+w0gcDo44YKYD7YIEJNHDjg=
|
gopkg.in/guregu/null.v4 v4.0.0 h1:1Wm3S1WEA2I26Kq+6vcW+w0gcDo44YKYD7YIEJNHDjg=
|
||||||
|
Binary file not shown.
155
main.go
155
main.go
@ -14,54 +14,20 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
c = sn.GetClient()
|
c = sn.GetClient()
|
||||||
me *sn.User
|
// TODO: fetch our id from SN API
|
||||||
|
meId = 21858
|
||||||
)
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
|
||||||
for {
|
for {
|
||||||
updateMe()
|
|
||||||
tickGameStart(c)
|
tickGameStart(c)
|
||||||
tickGameProgress(c)
|
tickGameProgress(c)
|
||||||
time.Sleep(15 * time.Second)
|
time.Sleep(15 * time.Second)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func updateMe() {
|
|
||||||
var (
|
|
||||||
oldMe *sn.User
|
|
||||||
err error
|
|
||||||
warnThreshold = 100
|
|
||||||
)
|
|
||||||
|
|
||||||
maybeWarn := func() {
|
|
||||||
if me.Privates.Sats < warnThreshold {
|
|
||||||
log.Printf("~~~ warning: low balance ~~~\n")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if me == nil {
|
|
||||||
// make sure first update is successful
|
|
||||||
if me, err = c.Me(); err != nil {
|
|
||||||
log.Fatalf("failed to fetch me: %v\n", err)
|
|
||||||
}
|
|
||||||
log.Printf("fetched me: id=%d name=%s balance=%d\n", me.Id, me.Name, me.Privates.Sats)
|
|
||||||
maybeWarn()
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
oldMe = me
|
|
||||||
if me, err = c.Me(); err != nil {
|
|
||||||
log.Printf("failed to update me: %v\n", err)
|
|
||||||
me = oldMe
|
|
||||||
} else {
|
|
||||||
log.Printf("updated me. balance: %d\n", me.Privates.Sats)
|
|
||||||
}
|
|
||||||
|
|
||||||
maybeWarn()
|
|
||||||
}
|
|
||||||
|
|
||||||
func tickGameStart(c *sn.Client) {
|
func tickGameStart(c *sn.Client) {
|
||||||
var (
|
var (
|
||||||
mentions []sn.Notification
|
mentions []sn.Notification
|
||||||
@ -73,26 +39,33 @@ func tickGameStart(c *sn.Client) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Printf("fetched %d mentions\n", len(mentions))
|
log.Printf("fetched %d mention(s)\n", len(mentions))
|
||||||
|
|
||||||
for _, n := range mentions {
|
for _, n := range mentions {
|
||||||
|
|
||||||
if !isRecent(n.Item.CreatedAt) {
|
if exists, err := db.ItemHasReply(n.Item.Id, meId); err != nil {
|
||||||
log.Printf("ignoring old mention %d\n", n.Item.Id)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if handled, err := alreadyHandled(n.Item.Id); err != nil {
|
|
||||||
log.Printf("failed to check for existing reply to game start in item %d: %v\n", n.Item.Id, err)
|
log.Printf("failed to check for existing reply to game start in item %d: %v\n", n.Item.Id, err)
|
||||||
continue
|
continue
|
||||||
} else if handled {
|
} else if exists {
|
||||||
// TODO: check if move changed
|
// TODO: check if move changed
|
||||||
log.Printf("reply to game start in item %d already exists\n", n.Item.Id)
|
log.Printf("reply to game start in item %d already exists\n", n.Item.Id)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
if err = handleGameStart(&n.Item); err != nil {
|
if err = handleGameStart(&n.Item); err != nil {
|
||||||
handleError(&n.Item, err)
|
|
||||||
|
// don't reply to mentions that we failed to parse as a game start
|
||||||
|
// to support unrelated mentions
|
||||||
|
if err.Error() == "failed to parse game start" {
|
||||||
|
log.Printf("ignoring error for item %d: %v\n", n.Item.Id, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err2 := createComment(n.Item.Id, fmt.Sprintf("`%v`", err)); err2 != nil {
|
||||||
|
log.Printf("failed to reply with error to item %d: %v\n", n.Item.Id, err2)
|
||||||
|
} else {
|
||||||
|
log.Printf("replied to game start in item %d with error: %v\n", n.Item.Id, err)
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
log.Printf("started new game via item %d\n", n.Item.Id)
|
log.Printf("started new game via item %d\n", n.Item.Id)
|
||||||
}
|
}
|
||||||
@ -114,30 +87,27 @@ func tickGameProgress(c *sn.Client) {
|
|||||||
|
|
||||||
for _, n := range replies {
|
for _, n := range replies {
|
||||||
|
|
||||||
if !isRecent(n.Item.CreatedAt) {
|
if exists, err := db.ItemHasReply(n.Item.Id, meId); err != nil {
|
||||||
log.Printf("ignoring old reply %d\n", n.Item.Id)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if handled, err := alreadyHandled(n.Item.Id); err != nil {
|
|
||||||
log.Printf("failed to check for existing reply to game update in item %d: %v\n", n.Item.Id, err)
|
log.Printf("failed to check for existing reply to game update in item %d: %v\n", n.Item.Id, err)
|
||||||
continue
|
continue
|
||||||
} else if handled {
|
} else if exists {
|
||||||
// TODO: check if move changed
|
// TODO: check if move changed
|
||||||
log.Printf("reply to game update in item %d already exists\n", n.Item.Id)
|
log.Printf("reply to game update in item %d already exists\n", n.Item.Id)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
if parent, err := c.Item(n.Item.ParentId); err != nil {
|
|
||||||
log.Printf("failed to fetch parent %d of %d\n", n.Item.ParentId, n.Item.Id)
|
|
||||||
continue
|
|
||||||
} else if parent.User.Id != me.Id {
|
|
||||||
log.Printf("ignoring nested reply %d\n", n.Item.Id)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if err = handleGameProgress(&n.Item); err != nil {
|
if err = handleGameProgress(&n.Item); err != nil {
|
||||||
handleError(&n.Item, err)
|
|
||||||
|
if err.Error() == "failed to parse game update" {
|
||||||
|
log.Printf("ignoring error for item %d: %v\n", n.Item.Id, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err2 := createComment(n.Item.Id, fmt.Sprintf("`%v`", err)); err2 != nil {
|
||||||
|
log.Printf("failed to reply with error to item %d: %v\n", n.Item.Id, err2)
|
||||||
|
} else {
|
||||||
|
log.Printf("replied to game start in item %d with error: %v\n", n.Item.Id, err)
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
log.Printf("updated game via item %d\n", n.Item.Id)
|
log.Printf("updated game via item %d\n", n.Item.Id)
|
||||||
}
|
}
|
||||||
@ -179,15 +149,8 @@ func handleGameStart(req *sn.Item) error {
|
|||||||
return fmt.Errorf("failed to upload image for item %d: %v\n", req.Id, err)
|
return fmt.Errorf("failed to upload image for item %d: %v\n", req.Id, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// reply with algebraic notation, image and info
|
// reply with algebraic notation and image
|
||||||
infoMove := "e4"
|
res = strings.Trim(fmt.Sprintf("%s\n\n%s", b.AlgebraicNotation(), imgUrl), " ")
|
||||||
if len(b.Moves) > 0 {
|
|
||||||
infoMove = "e5"
|
|
||||||
}
|
|
||||||
info := fmt.Sprintf("_A new chess game has been started!_\n\n"+
|
|
||||||
"_Reply with a move like `%s` to continue the game. "+
|
|
||||||
"See [here](https://stacker.news/chess#how-to-continue) for details._", infoMove)
|
|
||||||
res = strings.Trim(fmt.Sprintf("%s\n\n%s\n\n%s", b.AlgebraicNotation(), imgUrl, info), " ")
|
|
||||||
if _, err = createComment(req.Id, res); err != nil {
|
if _, err = createComment(req.Id, res); err != nil {
|
||||||
return fmt.Errorf("failed to reply to item %d: %v\n", req.Id, err)
|
return fmt.Errorf("failed to reply to item %d: %v\n", req.Id, err)
|
||||||
}
|
}
|
||||||
@ -216,7 +179,7 @@ func handleGameProgress(req *sn.Item) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, item := range thread {
|
for _, item := range thread {
|
||||||
if item.User.Id == me.Id {
|
if item.User.Id == meId {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -259,27 +222,6 @@ func handleGameProgress(req *sn.Item) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleError(req *sn.Item, err error) {
|
|
||||||
|
|
||||||
// don't reply to mentions that we failed to parse as a game start
|
|
||||||
// to support unrelated mentions
|
|
||||||
if err.Error() == "failed to parse game start" {
|
|
||||||
log.Printf("ignoring error for item %d: %v\n", req.Id, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if err.Error() == "failed to parse game update" {
|
|
||||||
log.Printf("ignoring error for item %d: %v\n", req.Id, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err2 := createComment(req.Id, fmt.Sprintf("`%v`", err)); err2 != nil {
|
|
||||||
log.Printf("failed to reply with error to item %d: %v\n", req.Id, err2)
|
|
||||||
} else {
|
|
||||||
log.Printf("replied to game start in item %d with error: %v\n", req.Id, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func createComment(parentId int, text string) (*sn.Item, error) {
|
func createComment(parentId int, text string) (*sn.Item, error) {
|
||||||
var (
|
var (
|
||||||
commentId int
|
commentId int
|
||||||
@ -296,7 +238,7 @@ func createComment(parentId int, text string) (*sn.Item, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if err = db.InsertItem(comment); err != nil {
|
if err = db.InsertItem(comment); err != nil {
|
||||||
return nil, fmt.Errorf("failed to insert item %d into db: %v\n", comment.Id, err)
|
return nil, fmt.Errorf("failed to insert item %d into db: %v\n", err, comment.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
return comment, nil
|
return comment, nil
|
||||||
@ -306,46 +248,31 @@ func parseGameStart(input string) (string, error) {
|
|||||||
for _, line := range strings.Split(input, "\n") {
|
for _, line := range strings.Split(input, "\n") {
|
||||||
line = strings.Trim(line, " ")
|
line = strings.Trim(line, " ")
|
||||||
|
|
||||||
var found bool
|
if !strings.HasPrefix(line, "@chess") {
|
||||||
if line, found = strings.CutPrefix(line, "@chess"); !found {
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
return strings.Trim(line, " "), nil
|
return strings.Trim(strings.ReplaceAll(line, "@chess", ""), " "), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return "", errors.New("failed to parse game start")
|
return "", errors.New("failed to parse game start")
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseGameProgress(input string) (string, error) {
|
func parseGameProgress(input string) (string, error) {
|
||||||
input = strings.Trim(input, " ")
|
|
||||||
|
|
||||||
lines := strings.Split(input, "\n")
|
lines := strings.Split(input, "\n")
|
||||||
words := strings.Split(input, " ")
|
words := strings.Split(input, " ")
|
||||||
|
|
||||||
if len(lines) == 1 && len(words) == 1 {
|
if len(lines) == 1 && len(words) == 1 {
|
||||||
return strings.Trim(strings.ReplaceAll(input, "@chess", ""), " "), nil
|
return strings.Trim(input, " "), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, line := range strings.Split(input, "\n") {
|
for _, line := range strings.Split(input, "\n") {
|
||||||
line = strings.Trim(line, " ")
|
if !strings.Contains(line, "@chess") {
|
||||||
|
|
||||||
var found bool
|
|
||||||
if line, found = strings.CutPrefix(line, "@chess"); !found {
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
return strings.Trim(line, " "), nil
|
return strings.Trim(strings.ReplaceAll(line, "@chess", ""), " "), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return "", errors.New("failed to parse game update")
|
return "", errors.New("failed to parse game update")
|
||||||
}
|
}
|
||||||
|
|
||||||
func isRecent(t time.Time) bool {
|
|
||||||
x := time.Now().Add(-30 * time.Second)
|
|
||||||
return t.After(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func alreadyHandled(id int) (bool, error) {
|
|
||||||
return db.ItemHasReply(id, me.Id)
|
|
||||||
}
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user