funmow/object.go

package funmow

import (
	"fmt"
	"sort"
	"strings"
	"github.com/mgutz/ansi"
)

const (
	MatchNone = iota
	MatchOne
	MatchMany
)

type PlayerMeta struct {
	ID       DBRef  `json:"id"`
	Password string `json:"password"`
}

type Object struct {
	ID          DBRef             `json:"id"`
	Type        string            `json:"type"`
	Owner       DBRef             `json:"owner"`
	Next        DBRef             `json:"next"`
	Name        string            `json:"name"`
	Description string            `json:"description"`
	Flags       map[string]bool   `json:"flags"`
	Properties  map[string]string `json:"properties"`

	db *DB
}

type ObjectFactory struct {
	db *DB
}

func NewObjectFactory(db *DB) *ObjectFactory {
	return &ObjectFactory{db: db}
}

func (f *ObjectFactory) NewObject() Object {
	o := Object{}
	o.ID, _ = f.db.Allocate()
	o.db = f.db
	return o
}

func (f *ObjectFactory) NewRawObject() Object {
	o := Object{}
	o.db = f.db
	return o
}

func (f *ObjectFactory) NewExit(name string, next DBRef, owner DBRef) Object {
	o := Object{
		Type:  "exit",
		Name:  name,
		Next:  next,
		Owner: owner,
		db:    f.db,
	}
	o.ID, _ = f.db.Allocate()
	return o
}

func (f *ObjectFactory) NewRoom() Object {
	o := Object{}
	o.ID, _ = f.db.Allocate()
	o.Type = "room"
	o.Flags = map[string]bool{"jump_ok": true}
	o.db = f.db
	return o
}

func (f *ObjectFactory) NewThing() Object {
	o := Object{}
	o.ID, _ = f.db.Allocate()
	o.Type = "thing"
	o.db = f.db
	return o
}

func (o *Object) Commit() error {
	return o.db.StoreObject(*o, o.ID)
}

func (o *Object) Refresh() bool {
	refreshed, found := o.db.RetrieveObject(o.ID)
	if found {
		*o = refreshed
	}
	return found
}

func (o Object) String() string {
	return fmt.Sprintf("id %d: %s", o.ID, o.Name)
}

func (o *Object) Remove(c *Object) error {
	return o.db.Unlink(o.ID, c.ID)
}

func (o *Object) Delete() error {
	err := o.db.Delete(o.ID)
	if err == nil {
		*o = Object{}
	}
	return err
}

func (o *Object) Contains(c *Object) error {
	return o.db.Link(o.ID, c.ID, c.Type)
}

func (o *Object) MatchExitNames(name string) ObjectList { // so much copypasta
	r := make(ObjectList, 0)

	children := o.db.GetChildren(o.ID) // map[DBRef]string

	for childID, childType := range children {
		if childType != "exit" {
			continue
		}
		o, found := o.db.Fetch(childID)
		if found {
			idName := fmt.Sprintf("#%d", childID)

			if strings.EqualFold(o.Name, name) || name == idName {
				r = append(r, o)
				break // have a match. Don't need to look for any more.
			}

			aliases := strings.Split(o.Name, ";")

			for _, v := range aliases {
				if v == name {
					r = append(r, o)
					break // no need to look at other aliases
				}
			}
		}
	}

	return r
}

func (o *Object) DetailedName() string {

	var label = fmt.Sprintf("%s <#%d>", o.Name, o.ID)

	colour := o.GetProp("color")

	if colour != "" {
		label = ansi.Color(label, colour)
	}

	return label
}

func (o *Object) ColorName() string {

	colour := o.GetProp("color")

	if colour != "" {
		return ansi.Color(o.Name, colour)
	}

	return o.Name
}

func (o *Object) GetContents(exclude DBRef) []string {
	r := make([]string, 0)
	children := o.db.GetChildren(o.ID)
	for childID, linkType := range children {
		if childID == exclude {
			continue
		}
		if !(linkType == "player" || linkType == "thing") {
			continue
		}
		o, found := o.db.Fetch(childID)
		if !found {
			continue
		}
		if linkType == "player" && !o.GetFlag("online") {
			continue
		}
		r = append(r, o.DetailedName())
	}
	sort.Strings(r)
	return r
}

func (o *Object) GetExits(parseAliases bool) []string {
	r := make([]string, 0)
	children := o.db.GetChildren(o.ID)
	for exitID, linkType := range children {
		if linkType != "exit" {
			continue
		}
		exit, found := o.db.Fetch(exitID)
		if !found {
			continue
		}
		if parseAliases {
			r = append(r, exit.FirstAlias())
		} else {
			r = append(r, exit.DetailedName())
		}
	}

	sort.Strings(r)
	return r
}

func (o *Object) FirstAlias() string {
	
	aliases := strings.Split(o.Name, ";")
	colour := o.GetProp("color")

	if colour != "" {
		return ansi.Color(aliases[0], colour)
	} else {
		return aliases[0]
	}
}


func (o *Object) SetFlag(flag string, value bool) {
	if o.Flags == nil {
		o.Flags = make(map[string]bool)
	}
	o.Flags[flag] = value
}

func (o *Object) GetFlag(flag string) bool {

	if o.Flags == nil {
		return false
	}
	value, ok := o.Flags[flag]
	if !ok {
		return false
	}

	return value
}

func (o *Object) SetProp(key string, value string) {
	if o.Properties == nil {
		o.Properties = make(map[string]string)
	}
	o.Properties[key] = value
}

func (o *Object) GetProp(key string) string {

	if o.Properties == nil {
		return ""
	}
	value, ok := o.Properties[key]
	if !ok {
		return ""
	}

	return value
}

type ObjectList []Object

func (l ObjectList) First() Object {
	if len(l) > 0 {
		return l[0]
	} else {
		return Object{}
	}
}

func (l ObjectList) ExactlyOne() (Object, int) {
	c := len(l)

	if c == 0 {
		return Object{}, MatchNone
	} else if c == 1 {
		return l[0], MatchOne
	} else {
		return Object{}, MatchMany
	}
}