types.go

package main

import (
	"database/sql"
	"encoding/json"
	"log"
	"os"

	_ "github.com/lib/pq"
	"github.com/thedadams/telegram-bot-api"
)

// CAHBot inherits from tgbotapi.
type CAHBot struct {
	*tgbotapi.BotAPI
	DBConn           *sql.DB
	AllQuestionCards []QuestionCard `json:"all_question_cards"`
	AllAnswerCards   []AnswerCard   `json:"all_answer_cards"`
	Settings         []Setting      `json:"settings"`
}

// NewCAHBot creates a new CAHBot.
func NewCAHBot(token string) (*CAHBot, error) {
	GenericBot, err := tgbotapi.NewBotAPI(os.Getenv("TOKEN"))
	if err != nil {
		log.Printf("Error initializing bot: %v", err)
		return nil, err
	}
	// Need to get the card data
	var AllQuestionCards []QuestionCard
	err = json.Unmarshal(AllQuestions, &AllQuestionCards)
	if err != nil {
		log.Printf("%v", err)
		return nil, err
	}
	var AllAnswerCards []AnswerCard
	err = json.Unmarshal(AllAnswers, &AllAnswerCards)
	if err != nil {
		log.Printf("%v", err)
		return nil, err
	}
	var Settings []Setting
	err = json.Unmarshal(AllSettings, &Settings)
	if err != nil {
		log.Printf("%v", err)
		return nil, err
	}
	db, err := sql.Open("postgres", os.Getenv("DATABASE_URL"))
	if err != nil {
		log.Printf("%v", err)
	}
	return &CAHBot{GenericBot, db, AllQuestionCards, AllAnswerCards, Settings}, err
}

// QuestionCard represents a white card in CAH.
type QuestionCard struct {
	ID         int    `json:"id"`
	Text       string `json:"text"`
	NumAnswers int    `json:"numAnswers"`
	Expansion  string `json:"expansion"`
}

// AnswerCard represents a black card in CAH.
type AnswerCard struct {
	ID        int    `json:"id"`
	Text      string `json:"text"`
	Expansion string `json:"expansion"`
}

// Setting represents a setting in the game that can be changed.
type Setting struct {
	Name    string    `json:"name"`
	CData   string    `json:"cdata"`
	Options []Setting `json:"options"` // optional
}