You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
2.4 KiB
99 lines
2.4 KiB
package themis
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
_ "embed"
|
|
"fmt"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
//go:embed migrations/init.sql
|
|
var initScript string
|
|
|
|
const (
|
|
CLAIM_TYPE_AREA = iota
|
|
CLAIM_TYPE_REGION
|
|
CLAIM_TYPE_TRADE
|
|
)
|
|
|
|
var claimTypeEnum = map[string]int{
|
|
"area": CLAIM_TYPE_AREA,
|
|
"region": CLAIM_TYPE_REGION,
|
|
"trade": CLAIM_TYPE_TRADE,
|
|
}
|
|
|
|
var claimTypeEnumVals = map[int]string{
|
|
CLAIM_TYPE_AREA: "area",
|
|
CLAIM_TYPE_REGION: "region",
|
|
CLAIM_TYPE_TRADE: "trade",
|
|
}
|
|
|
|
var claimTypeToColumn = map[string]string{
|
|
"area": "area",
|
|
"region": "region",
|
|
"trade": "trade_node",
|
|
}
|
|
|
|
type Store struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewStore(conn string) (*Store, error) {
|
|
db, err := sql.Open("sqlite3", conn)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
|
|
_, err = db.Exec(initScript)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to run init script: %w", err)
|
|
}
|
|
|
|
return &Store{
|
|
db: db,
|
|
}, nil
|
|
}
|
|
|
|
func (s *Store) Claim(ctx context.Context, player, province string, typ int) error {
|
|
tx, err := s.db.Begin()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to begin transaction: %w", err)
|
|
}
|
|
defer tx.Commit()
|
|
|
|
// Check conflicts
|
|
stmt, err := s.db.PrepareContext(ctx, fmt.Sprintf(`SELECT COUNT(1) FROM provinces WHERE provinces.%s = ? and provinces.name in (
|
|
SELECT provinces.name FROM claims LEFT JOIN provinces ON claims.val = provinces.trade_node WHERE claims.claim_type = 'trade'
|
|
UNION SELECT provinces.name from claims LEFT JOIN provinces ON claims.val = provinces.region WHERE claims.claim_type = 'region'
|
|
UNION SELECT provinces.name from claims LEFT JOIN provinces ON claims.val = provinces.area WHERE claims.claim_type = 'area'
|
|
)`, claimTypeToColumn[claimTypeEnumVals[typ]]))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to prepare conflicts query: %w", err)
|
|
}
|
|
|
|
row := stmt.QueryRowContext(ctx, province)
|
|
var count int
|
|
err = row.Scan(&count)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get count of conflicting provinces: %w", err)
|
|
}
|
|
|
|
if count > 0 {
|
|
return fmt.Errorf("found %d conflicting provinces", count)
|
|
}
|
|
|
|
stmt, err = s.db.PrepareContext(ctx, "INSERT INTO claims (player, claim_type, val) VALUES (?, ?, ?)")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to prepare claim query: %w", err)
|
|
}
|
|
|
|
_, err = stmt.ExecContext(ctx, player, claimTypeEnumVals[typ], province)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to insert claim: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|