aitestory/model/tag.go

package model

import (
	"errors"
	"fmt"

	"git.aiterp.net/AiteRP/aitestory/server"
	"git.aiterp.net/gisle/wrouter/generate"
)

// TagTypes are the allowed values for Tag.Type
var TagTypes = []string{
	"Location",
	"Character",
	"Event",
	"Organization",
	"Source",
}

// Tag describes a tag
type Tag struct {
	ID   string
	Type string
	Name string
}

// Insert adds the tag to the database, giving it a new unique ID
func (tag *Tag) Insert() error {
	db := server.Main.DB

	// Validate tag type
	validType := false
	for _, tagType := range TagTypes {
		if tagType == tag.Type {
			validType = true
			break
		}
	}
	if !validType {
		return fmt.Errorf("\"%s\" is not a valid tag type", tag.Type)
	}

	// Validate tag name
	if len(tag.Name) == 0 {
		return errors.New("Tag name is empty")
	}

	// Generate an ID if none exists
	if tag.ID == "" {
		tag.ID = generate.ID()
	}

	// Do the thing
	_, err := db.Exec("INSERT INTO `tag` (id,type,name,disabled) VALUES (?,?,?,false)", tag.ID, tag.Type, tag.Name)
	if err != nil {
		return err
	}

	return nil
}

// ListTags finds all the tags, without filter. If it hits
// the tag cache, it will copy it making it safe to modify
func ListTags() ([]Tag, error) {
	db := server.Main.DB

	// Read from the database
	rows, err := db.Query("SELECT id,type,name FROM `tag` WHERE disabled=false")
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	results := make([]Tag, 0, 64)
	for rows.Next() {
		tag := Tag{}
		rows.Scan(&tag.ID, &tag.Type, &tag.Name)
		results = append(results, tag)
	}

	return results, nil
}