|
|
@@ -0,0 +1,149 @@
|
|
|
+package main
|
|
|
+
|
|
|
+import (
|
|
|
+ "log"
|
|
|
+ "time"
|
|
|
+
|
|
|
+ "gorm.io/driver/sqlite"
|
|
|
+ "gorm.io/gorm"
|
|
|
+)
|
|
|
+
|
|
|
+type DataBase struct {
|
|
|
+ DB *gorm.DB
|
|
|
+}
|
|
|
+
|
|
|
+func OpenDB(filename string) (*DataBase, error) {
|
|
|
+ db, err := gorm.Open(sqlite.Open(filename), &gorm.Config{})
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ DB := &DataBase{DB: db}
|
|
|
+ err = DB.DB.AutoMigrate(&ShipModel{}, &Ship{}, &User{})
|
|
|
+ return DB, err
|
|
|
+}
|
|
|
+
|
|
|
+func (db *DataBase) Close() error {
|
|
|
+ sqldb, err := db.DB.DB()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ return sqldb.Close()
|
|
|
+}
|
|
|
+
|
|
|
+type Model struct {
|
|
|
+ ID uint64 `gorm:"primaryKey"`
|
|
|
+ CreatedAt time.Time
|
|
|
+ UpdatedAt time.Time
|
|
|
+}
|
|
|
+
|
|
|
+type ShipModel struct {
|
|
|
+ Model
|
|
|
+ Name string `gorm:"unique"`
|
|
|
+
|
|
|
+ BaseHull uint64 // Armor Plates bump this, typically by 4 points per plate
|
|
|
+ BaseShields uint64 // Capacitors bump this, typically by 3 points per cap
|
|
|
+
|
|
|
+ // Buying costs and limits
|
|
|
+ Purchaseable bool // Determines if the ShipModel can be bought
|
|
|
+ RequiredXp uint64 // Preventing Users from purchasing very powerful ships without doing any combat
|
|
|
+ CostMetal uint64 // Normally selling a ship in metal is 40%, if it can't be bought however it's 60%
|
|
|
+ CostCircuit uint64 // Noormally selling a ship in circuits is 50%, if it can't be bought however it's 75%
|
|
|
+
|
|
|
+ // Limits
|
|
|
+ MaxGuns uint64
|
|
|
+ MaxRocketLaunchers uint64
|
|
|
+ MaxArmorPlates uint64
|
|
|
+ MaxRepairBays uint64
|
|
|
+ MaxShieldCapacitors uint64
|
|
|
+ MaxShieldGenerators uint64
|
|
|
+ MaxEngines uint64
|
|
|
+
|
|
|
+ // Starting values, assigned to Ship once bought/created
|
|
|
+ StartingGuns uint64
|
|
|
+ StartingRocketLaunchers uint64
|
|
|
+ StartingArmorPlates uint64
|
|
|
+ StartingRepairBays uint64
|
|
|
+ StartingRepairDrones uint64
|
|
|
+ StartingShieldCapacitors uint64
|
|
|
+ StartingShieldGenerators uint64
|
|
|
+ StartingEngines uint64
|
|
|
+}
|
|
|
+
|
|
|
+func (db *DataBase) SaveShipModel(model *ShipModel) error {
|
|
|
+ result := db.DB.Save(model)
|
|
|
+ return result.Error
|
|
|
+}
|
|
|
+
|
|
|
+func (db *DataBase) GetShipModelById(id uint64) *ShipModel {
|
|
|
+ var model *ShipModel
|
|
|
+ result := db.DB.Find(model, id)
|
|
|
+ if result.Error != nil {
|
|
|
+ log.Printf("DataBase.GetShipModelById(id=%d) > %v", id, result.Error)
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ return model
|
|
|
+}
|
|
|
+
|
|
|
+func (db *DataBase) GetShipModelByName(name string) *ShipModel {
|
|
|
+ var model *ShipModel
|
|
|
+ result := db.DB.Find(model, ShipModel{Name: name})
|
|
|
+ if result.Error != nil {
|
|
|
+ log.Printf("DataBase.GetShipModelByName(name='%s') > %v", name, result.Error)
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ return model
|
|
|
+}
|
|
|
+
|
|
|
+type Ship struct {
|
|
|
+ Model
|
|
|
+ UserID uint64 // Who owns this ship
|
|
|
+ ShipModel uint64 // ShipModel.ID, Limits of the ship
|
|
|
+
|
|
|
+ HullPoints uint64 // Current HP
|
|
|
+ ShieldPoints uint64 // Current SP
|
|
|
+
|
|
|
+ Guns uint64 // Guns deal 0-2 damage per gun
|
|
|
+ RocketLaunchers uint64 // A "special" attack, Each rocket deals 0-10 damage per rocket
|
|
|
+
|
|
|
+ ArmorPlates uint64 // Bumps HP up
|
|
|
+ RepairBays uint64 // Allows you to hold Repair Drones to repair HP in combat
|
|
|
+ RepairDrones uint64 // Each repair drone can repair 0-1 HP per turn in combat (but if you take damage there is a chance to loose a drone instead)
|
|
|
+ ShieldCapacitors uint64 // Bumps SP up, each Capacitor also adds a resistance to the amout of SP regained in combat
|
|
|
+ ShieldGenerators uint64 // Regains SP of 0-1 SP per turn, it is calculated together then applies a resistance rating based on the number of capacitors installed
|
|
|
+ Engines uint64 // Affects dodge and escape chances, "mass" is the number of installed parts
|
|
|
+}
|
|
|
+
|
|
|
+func (db *DataBase) SaveShip(ship *Ship) error {
|
|
|
+ result := db.DB.Save(ship)
|
|
|
+ return result.Error
|
|
|
+}
|
|
|
+
|
|
|
+type User struct {
|
|
|
+ Model
|
|
|
+ Name string `gorm:"unique"` // Nickname, commonly used
|
|
|
+ RealName string `gorm:"UniqueIndex,colate:nocase"` // BBS Realname used to identify User, typically not used/displayed
|
|
|
+ // Currency-likes
|
|
|
+ Xp uint64 // How experienced the User is, determines how tough random opponents will be
|
|
|
+ Metal uint64 // Common Currency
|
|
|
+ Circuit uint64 // Rare Currency
|
|
|
+ Rockets uint64 // Not a currency, a weapon, but there is technically no limit except the number that can be fired per "special" attack
|
|
|
+ Fuel uint64 // Not a currency, amount of fights left (A normal fight will consume 2, a fight where you run away is 3, and a fight where you died consumes all but 1 and kicks you off)
|
|
|
+ // Stuff
|
|
|
+ Ship *Ship
|
|
|
+ // Use UpdatedAt as LastOn
|
|
|
+}
|
|
|
+
|
|
|
+func (db *DataBase) SaveUser(u *User) error {
|
|
|
+ result := db.DB.Save(u)
|
|
|
+ return result.Error
|
|
|
+}
|
|
|
+
|
|
|
+func (db *DataBase) GetUserByName(name string) *User {
|
|
|
+ var user *User
|
|
|
+ result := db.DB.Find(user, User{RealName: name})
|
|
|
+ if result.Error != nil {
|
|
|
+ log.Printf("DataBase.GetUserByName(name='%s') > %v", name, result.Error)
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ return user
|
|
|
+}
|
