website.go

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT license.

package object

type Website struct {
	Id   string `xorm:"varchar(100) notnull pk" json:"id"`
	Name string `xorm:"varchar(100)" json:"name"`

	Url         string `xorm:"varchar(100)" json:"url"`
	TrackConfig string `xorm:"mediumtext" json:"trackConfig"`

	State        string `xorm:"varchar(100)" json:"state"`
	SessionCount int    `json:"sessionCount"`
}

func countWebsites(websites []*Website) {
	for _, website := range websites {
		countWebsiteSessions(website)
	}
}

func countWebsiteSessions(website *Website) {
	sessionCount, err := ormManager.engine.Where("website_id = ?", website.Id).Count(&Session{})
	if err != nil {
		panic(err)
	}

	website.SessionCount = int(sessionCount)
	// TODO: Do we need to update `session_count` column?
	// _, err := ormManager.engine.Id(website.Id).Cols("session_count").Update(website)
	// if err != nil {
	// 	panic(err)
	// }
}

func GetWebsites() []*Website {
	websites := []*Website{}
	err := ormManager.engine.Asc("id").Find(&websites)
	if err != nil {
		panic(err)
	}

	countWebsites(websites)

	return websites
}

func GetWebsite(id string) *Website {
	website := Website{Id: id}
	existed, err := ormManager.engine.Get(&website)
	if err != nil {
		panic(err)
	}

	if existed {
		countWebsiteSessions(&website)
		return &website
	} else {
		return nil
	}
}

func UpdateWebsite(id string, website *Website) bool {
	if GetWebsite(id) == nil {
		return false
	}

	_, err := ormManager.engine.Id(id).AllCols().Update(website)
	if err != nil {
		panic(err)
	}

	//return affected != 0
	return true
}

func AddWebsite(website *Website) bool {
	affected, err := ormManager.engine.Insert(website)
	if err != nil {
		panic(err)
	}

	return affected != 0
}

func DeleteWebsite(id string) bool {
	affected, err := ormManager.engine.Id(id).Delete(&Website{})
	if err != nil {
		panic(err)
	}

	return affected != 0
}