add function to take claim

go.mod

@@ -2,4 +2,4 @@ module go.wperron.io/themis
 
 go 1.19
 
-require github.com/mattn/go-sqlite3 v1.14.15 // indirect
+require github.com/mattn/go-sqlite3 v1.14.15

store.go

@@ -1,6 +1,7 @@
 package themis
 
 import (
+	"context"
 	"database/sql"
 	_ "embed"
 	"fmt"
@@ -11,6 +12,30 @@ import (
 //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
 }
@@ -21,7 +46,53 @@ func NewStore(conn string) (*Store, error) {
 		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
+}