// cents retrieves the cents out of the payload of an event.
func payloadCents(event cells.Event) int {
cents, ok := event.Payload().Get(cells.DefaultPayload)
if !ok {
return -1
}
return cents.(int)
}
// Locked represents the locked state receiving coins.
func (m *lockMachine) Locked(ctx cells.Context, event cells.Event) (behaviors.FSMState, error) {
switch event.Topic() {
case "cents?":
return m.Locked, event.Respond(m.cents)
case "info?":
info := fmt.Sprintf("state 'locked' with %d cents", m.cents)
return m.Locked, event.Respond(info)
case "coin!":
cents := payloadCents(event)
if cents < 1 {
return nil, fmt.Errorf("do not insert buttons")
}
m.cents += cents
if m.cents > 100 {
m.cents -= 100
return m.Unlocked, nil
}
return m.Locked, nil
case "button-press!":
if m.cents > 0 {
ctx.Environment().EmitNew("restorer", "drop!", m.cents, event.Scene())
m.cents = 0
}
return m.Locked, nil
case "screwdriver!":
// Allow a screwdriver to bring the lock into an undefined state.
return nil, nil
}
return m.Locked, fmt.Errorf("illegal topic in state 'locked': %s", event.Topic())
}
// ProcessEvent executes the state function and stores
// the returned new state.
func (b *fsmBehavior) ProcessEvent(event cells.Event) error {
switch event.Topic() {
case cells.StatusTopic:
status := FSMStatus{
Done: b.done,
Error: b.err,
}
if err := event.Respond(status); err != nil {
return err
}
default:
if b.done {
return nil
}
state, err := b.state(b.ctx, event)
if err != nil {
b.done = true
b.err = err
} else if state == nil {
b.done = true
}
b.state = state
}
return nil
}
// newEventData returns the passed event as event data to collect.
func newEventData(event cells.Event) EventData {
data := EventData{
Topic: event.Topic(),
Payload: event.Payload(),
}
return data
}
// ProcessEvent calls a callback functions with the event data.
func (b *callbackBehavior) ProcessEvent(event cells.Event) error {
for _, callbackFunc := range b.callbackFuncs {
if err := callbackFunc(event.Topic(), event.Payload()); err != nil {
return err
}
}
return nil
}
// PrecessEvent emits a ticker event each time the
// defined duration elapsed.
func (b *tickerBehavior) ProcessEvent(event cells.Event) error {
if event.Topic() == TickerTopic {
pvs := cells.PayloadValues{
TickerIDPayload: b.ctx.ID(),
TickerTimePayload: time.Now(),
}
b.ctx.EmitNew(TickerTopic, pvs, nil)
}
return nil
}
func (b *restorerBehavior) ProcessEvent(event cells.Event) error {
switch event.Topic() {
case "grab!":
cents := b.cents
b.cents = 0
return event.Respond(cents)
case "drop!":
b.cents += payloadCents(event)
}
return nil
}
func (c *censor) filter(event cells.Event) bool {
message, _ := event.Payload().Get("message")
user, _ := event.Payload().Get("from")
for _, w := range strings.Split(message.(string), " ") {
if _, ok := c.words[w]; ok {
log.Infof("%s said a bad word: %s", user, w)
c.warnings[user.(string)]++
return false
}
}
return true
}
// Configuration returns the configuration payload
// of the passed event or an empty configuration.
func Configuration(event cells.Event) configuration.Configuration {
payload, ok := event.Payload().Get(ConfigurationPayload)
if !ok {
logger.Warningf("event does not contain configuration payload")
config, _ := configuration.ReadString("{config}")
return config
}
config, ok := payload.(configuration.Configuration)
if !ok {
logger.Warningf("configuration payload has illegal type")
config, _ := configuration.ReadString("{config}")
return config
}
return config
}
// ProcessEvent collects and re-emits events.
func (b *collectorBehavior) ProcessEvent(event cells.Event) error {
switch event.Topic() {
case cells.CollectedTopic:
response := make([]EventData, len(b.collected))
copy(response, b.collected)
if err := event.Respond(response); err != nil {
return err
}
case cells.ResetTopic:
b.collected = []EventData{}
default:
b.collected = append(b.collected, newEventData(event))
if len(b.collected) > b.max {
b.collected = b.collected[1:]
}
b.ctx.Emit(event)
}
return nil
}
func (b *collectBehavior) ProcessEvent(event cells.Event) error {
switch event.Topic() {
case cells.ProcessedTopic:
processed := make([]string, len(b.processed))
copy(processed, b.processed)
err := event.Respond(processed)
if err != nil {
return err
}
case cells.ResetTopic:
b.processed = []string{}
case cells.PingTopic:
err := event.Respond(cells.PongResponse)
if err != nil {
return err
}
case iterateTopic:
err := b.ctx.SubscribersDo(func(s cells.Subscriber) error {
return s.ProcessNewEvent("love", b.ctx.ID()+" loves "+s.ID(), event.Scene())
})
if err != nil {
return err
}
case panicTopic:
panic("Ouch!")
case subscribersTopic:
var ids []string
b.ctx.SubscribersDo(func(s cells.Subscriber) error {
ids = append(ids, s.ID())
return nil
})
err := event.Respond(ids)
if err != nil {
return err
}
default:
b.processed = append(b.processed, fmt.Sprintf("%v", event))
return b.ctx.Emit(event)
}
return nil
}
// ProcessEvent reads, validates and emits a configuration.
func (b *configuratorBehavior) ProcessEvent(event cells.Event) error {
switch event.Topic() {
case ReadConfigurationTopic:
// Read configuration
filename, ok := event.Payload().GetString(ConfigurationFilenamePayload)
if !ok {
logger.Errorf("cannot read configuration without filename payload")
return nil
}
logger.Infof("reading configuration from %q", filename)
config, err := configuration.ReadFile(filename)
if err != nil {
return errors.Annotate(err, ErrCannotReadConfiguration, errorMessages)
}
// If wanted then validate it.
if b.validate != nil {
err = b.validate(config)
if err != nil {
return errors.Annotate(err, ErrCannotValidateConfiguration, errorMessages)
}
}
// All done, emit it.
pvs := cells.PayloadValues{
ConfigurationPayload: config,
}
b.ctx.EmitNew(ConfigurationTopic, pvs, event.Scene())
}
return nil
}
// ProcessEvent counts the event for the return value of the counter func
// and emits this value.
func (b *counterBehavior) ProcessEvent(event cells.Event) error {
switch event.Topic() {
case cells.CountersTopic:
response := b.copyCounters()
if err := event.Respond(response); err != nil {
return err
}
case cells.ResetTopic:
b.counters = make(map[string]int64)
default:
cids := b.counterFunc(b.ctx.ID(), event)
if cids != nil {
for _, cid := range cids {
v, ok := b.counters[cid]
if ok {
b.counters[cid] = v + 1
} else {
b.counters[cid] = 1
}
topic := "counter:" + cid
b.ctx.EmitNew(topic, b.counters[cid], event.Scene())
}
}
}
return nil
}
func (u *user) ProcessEvent(event cells.Event) error {
switch event.Topic() {
case SAYS_TO:
to, ok := event.Payload().Get("to")
if !ok || to.(string) != makeUserID(u.name) {
return nil
}
log.Info(event.Payload())
case SAY:
env := u.ctx.Environment()
roomID, _ := event.Payload().Get("room")
if !env.HasCell(roomID.(string)) {
log.Panicf("%s not found", roomID)
return fmt.Errorf("room %s not found", roomID)
}
ok, err := env.Request(roomID.(string), IN_ROOM, cells.PayloadValues{"user": makeUserID(u.name)}, nil, time.Second)
if err != nil {
log.Panic(err)
return err
}
if !ok.(bool) {
return fmt.Errorf("%s is not in %s", u.name, roomID)
}
payload := event.Payload()
payload = payload.Apply(cells.PayloadValues{"from": makeUserID(u.name)})
env.EmitNew(roomID.(string), SAYS_ALL, payload, nil)
case USER_DISCONNECTED:
log.Info(USER_DISCONNECTED)
}
return nil
}
func (lu *logUser) ProcessEvent(event cells.Event) (err error) {
// TODO: would be nice to have an event.Source()
if event.Topic() == SAYS_ALL {
user, _ := event.Payload().Get("from")
msg, _ := event.Payload().Get("message")
log.Infof("%s heard %v say \"%v\"", lu.name, user, msg)
}
return
}
// ProcessEvent stores and flags the event in the event scene.
// So other code parts using the same scene can wait for the signal.
func (b *sceneBehavior) ProcessEvent(event cells.Event) error {
scn := event.Scene()
if scn != nil {
err := scn.StoreAndFlag(event.Topic(), event.Payload())
if err != nil {
return err
}
}
return nil
}
func (b *building) ProcessEvent(event cells.Event) (err error) {
switch event.Topic() {
case ROOM_ADDED:
room, _ := event.Payload().Get("room")
b.rooms[room.(string)] = true
log.Infof("added %v", room)
case LIST_ROOMS:
rooms := []string{}
for room := range b.rooms {
rooms = append(rooms, room)
}
event.Respond(rooms)
case SAYS_ALL:
user, ok := event.Payload().Get("from")
if !ok || user.(string) != PublicAddressUserID {
return fmt.Errorf("invalid public address request")
}
b.ctx.Emit(event)
}
return
}
func (wsb *webSocketBehavior) ProcessEvent(event cells.Event) (err error) {
switch event.Topic() {
case SAYS_TO:
to, ok := event.Payload().Get("to")
if !ok || to.(string) != makeUserID(wsb.user) {
return
}
msg, err := json.Marshal(payloadToValues(event.Payload()))
if err != nil {
log.Fatal(err)
}
wsb.output <- msg
case SAYS_ALL:
msg, err := json.Marshal(payloadToValues(event.Payload()))
if err != nil {
log.Fatal(err)
}
wsb.output <- msg
}
return
}
func (c *censor) ProcessEvent(event cells.Event) (err error) {
switch event.Topic() {
case CENSOR:
ok := c.filter(event)
userID, _ := event.Payload().Get("from")
event.Respond(ok)
if _, ok := c.warnings[userID.(string)]; !ok {
c.warnings[userID.(string)] = 0
}
if !ok {
return c.ctx.EmitNew(SAYS_TO, cells.PayloadValues{
"message": fmt.Sprintf("You can't say that! You've been warned %d times.", c.warnings[userID.(string)]),
"from": c.id,
"to": userID.(string),
}, nil)
}
}
return
}
func (r *room) ProcessEvent(event cells.Event) (err error) {
switch event.Topic() {
case SAYS_ALL, SAYS_TO:
user, _ := event.Payload().Get("from")
switch user.(string) {
case PublicAddressUserID, r.censorID:
r.ctx.Emit(event)
default:
env := r.ctx.Environment()
ok, err := env.Request(r.censorID, CENSOR, event.Payload(), nil, time.Second)
if err != nil {
log.Error(err)
return err
}
if ok.(bool) {
r.ctx.Emit(event)
}
}
case IN_ROOM:
userID, _ := event.Payload().Get("user")
ok, _ := r.users[userID.(string)]
event.Respond(ok)
case USER_ADDED:
userID, _ := event.Payload().Get("user")
if ok, _ := r.users[userID.(string)]; !ok {
r.users[userID.(string)] = true
r.ctx.Environment().EmitNew(r.ctx.ID(), SAYS_ALL, cells.PayloadValues{
"message": fmt.Sprintf("%s has entered the room", userID),
"from": PublicAddressUserID,
}, nil)
}
case USER_COUNT:
event.Respond(len(r.users))
}
return
}
// Unlocked represents the unlocked state receiving coins.
func (m *lockMachine) Unlocked(ctx cells.Context, event cells.Event) (behaviors.FSMState, error) {
switch event.Topic() {
case "cents?":
return m.Unlocked, event.Respond(m.cents)
case "info?":
info := fmt.Sprintf("state 'unlocked' with %d cents", m.cents)
return m.Unlocked, event.Respond(info)
case "coin!":
cents := payloadCents(event)
ctx.EmitNew("return", cents, event.Scene())
return m.Unlocked, nil
case "button-press!":
ctx.Environment().EmitNew("restorer", "drop!", m.cents, event.Scene())
m.cents = 0
return m.Locked, nil
}
return m.Unlocked, fmt.Errorf("illegal topic in state 'unlocked': %s", event.Topic())
}
func (b *emitBehavior) ProcessEvent(event cells.Event) error {
return b.ctx.EmitNew(sleepTopic, event.Payload(), nil)
}
