initial commit

3 files changed, 488 insertions, 0 deletions

diff --git a/server/channel/channel.go b/server/channel/channel.go
new file mode 100644
index 0000000..50b43b6
--- /dev/null
+++ b/server/channel/channel.go
@@ -0,0 +1,148 @@
+package channel
+
+import (
+	"citrons.xyz/talk/proto"
+	"citrons.xyz/talk/server/object"
+	"citrons.xyz/talk/server/session"
+	"citrons.xyz/talk/server/user"
+)
+
+type ChannelStore struct {
+	world *object.World
+	byName map[string]*Channel
+}
+
+type Channel struct {
+	store *ChannelStore
+	id string
+	name string
+	members map[string]Membership
+	messages []proto.Object
+	byId map[string]int
+	defaultMembership Membership
+	Stream session.Stream
+}
+
+func NewStore(world *object.World) *ChannelStore {
+	return &ChannelStore {world, make(map[string]*Channel)}
+}
+
+func (cs *ChannelStore) CreateChannel(name string) (*Channel, *proto.Fail) {
+	if cs.byName[name] != nil {
+		return nil, &proto.Fail {
+			"name-taken", "", map[string]string {"": name},
+		}
+	}
+	var c Channel
+	c.store = cs
+	c.name = name
+	c.members = make(map[string]Membership)
+	c.byId = make(map[string]int)
+	c.defaultMembership = DefaultMembership
+
+	cs.byName[name] = &c
+	c.id = cs.world.NewObject(&c)
+	return &c, nil
+}
+
+func (cs *ChannelStore) ByName(name string) *Channel {
+	return cs.byName[name]
+}
+
+func (c *Channel) Name() string {
+	return c.name
+}
+
+func (c *Channel) Id() string {
+	return c.id
+}
+
+func (c *Channel) Rename(name string) *proto.Fail {
+	if c.store.byName[name] != nil {
+		return &proto.Fail {
+			"name-taken", "", map[string]string {"": name},
+		}
+	}
+	c.store.byName[c.name] = nil
+	c.store.byName[name] = c
+	c.name = name
+	return nil
+}
+
+func (c *Channel) Put(m proto.Object) proto.Object {
+	m.Id = proto.GenId()
+	m.Fields["t"] = proto.Timestamp()
+	c.byId[m.Id] = len(c.messages)
+	c.messages = append(c.messages, m)
+	for s, _ := range c.Stream.Subscribers() {
+		if m.Fields["f"] == s.UserId {
+			continue
+		}
+		if c.members[s.UserId].See {
+			s.Event(proto.NewCmd("p", c.id, m))
+		}
+	}
+	return m
+}
+
+func (c *Channel) prune() {
+	for m, _ := range c.members {
+		switch c.store.world.GetObject(m).(type) {
+		case *user.User:
+		default:
+			delete(c.members, m)
+		}
+	}
+}
+
+func (c *Channel) Join(u *user.User) *proto.Fail {
+	if c.members[u.Id()].Yes {
+		return nil
+	}
+	c.members[u.Id()] = c.defaultMembership
+	u.Channels[c.id] = true
+	c.Put(proto.Object{"join", "", map[string]string {"": u.Id()}})
+	return nil
+}
+
+func (c *Channel) Leave(u *user.User) *proto.Fail {
+	if !c.members[u.Id()].Yes {
+		return nil
+	}
+	delete(c.members, u.Id())
+	delete(u.Channels, c.id)
+	c.Put(proto.Object{"leave", "", map[string]string {"": u.Id()}})
+	return nil
+}
+
+func (c *Channel) Members() map[string]Membership {
+	c.prune()
+	return c.members
+}
+
+func (c *Channel) SetMembership(u *user.User, m Membership) {
+	if c.members[u.Id()].Yes {
+		c.members[u.Id()] = m
+	}
+}
+
+func (c *Channel) Delete() {
+	c.Stream.Event(proto.NewCmd("delete", c.id))
+	c.Stream.UnsubscribeAll()
+
+	for m, _ := range c.members {
+		switch u := c.store.world.GetObject(m).(type) {
+		case *user.User:
+			u.Channels[c.id] = false
+		default:
+		}
+	}
+	delete(c.store.byName, c.name)
+	c.store.world.RemoveObject(c.id)
+}
+
+func (c *Channel) GetInfo() proto.Object {
+	return proto.Object {
+		"channel", c.id, map[string]string {"": c.name},
+	}
+}
diff --git a/server/channel/command.go b/server/channel/command.go
new file mode 100644
index 0000000..94cf38f
--- /dev/null
+++ b/server/channel/command.go
@@ -0,0 +1,253 @@
+package channel
+
+import (
+	"citrons.xyz/talk/server/session"
+	"citrons.xyz/talk/server/user"
+	"citrons.xyz/talk/proto"
+)
+
+func (c *Channel) SendRequest(r session.Request) {
+	switch r.Cmd.Kind {
+
+	case "p":
+		if len(r.Cmd.Args) != 1 {
+			r.ReplyInvalid()
+			return
+		}
+		m := r.Cmd.Args[0]
+		switch m.Kind {
+		case "m":
+			for k, _ := range m.Fields {
+				switch k {
+				case "":
+				default:
+					r.ReplyInvalid()
+					return
+				}
+			}
+		default:
+			r.ReplyInvalid()
+			return
+		}
+
+		if !c.members[r.From.UserId].Put {
+			r.Reply(proto.Fail{"forbidden", "", nil}.Cmd())
+			return
+		}
+
+		m.Fields["f"] = r.From.UserId
+		err := c.Put(m)
+		if err != nil {
+			r.Reply(err.Cmd())
+			return
+		}
+		r.Reply(proto.NewCmd("p", c.id, m))
+
+	case "s":
+		r.From.Subscribe(&c.Stream)
+		r.Reply(proto.NewCmd("i", "", c.GetInfo()))
+
+	case "u":
+		r.From.Unsubscribe(&c.Stream)
+		r.Reply(proto.NewCmd("ok", ""))
+
+	case "join":
+		u := c.store.world.GetObject(r.From.UserId).(*user.User)
+		err := c.Join(u)
+		if err != nil {
+			r.Reply(err.Cmd())
+		} else {
+			r.Reply(proto.NewCmd("ok", ""))
+		}
+
+	case "leave":
+		u := c.store.world.GetObject(r.From.UserId).(*user.User)
+		err := c.Leave(u)
+		if err != nil {
+			r.Reply(err.Cmd())
+		} else {
+			r.Reply(proto.NewCmd("ok", ""))
+		}
+
+	case "delete":
+		if !c.members[r.From.UserId].Op {
+			r.Reply(proto.Fail{"forbidden", "", nil}.Cmd())
+			return
+		}
+		c.Delete()
+
+	case "update":
+		if len(r.Cmd.Args) != 1 {
+			r.ReplyInvalid()
+			return
+		}
+		upd := r.Cmd.Args[0]
+		if upd.Kind != "channel" {
+			r.ReplyInvalid()
+			return
+		}
+		name := c.name
+		for k, v := range upd.Fields {
+			switch k {
+			case "":
+				name = v
+			default:
+				r.ReplyInvalid()
+				return
+			}
+		}
+
+		if !c.members[r.From.UserId].Update {
+			r.Reply(proto.Fail{"forbidden", "", nil}.Cmd())
+			return
+		}
+		if name != c.name {
+			err := c.Rename(name)
+			if err != nil {
+				r.Reply(err.Cmd())
+				return
+			}
+		}
+		c.Stream.Event(r.Cmd)
+		r.Reply(proto.NewCmd("ok", ""))
+
+	case "list":
+		if !c.members[r.From.UserId].Yes {
+			r.Reply(proto.Fail{"forbidden", "", nil}.Cmd())
+			return
+		}
+		cmd := proto.NewCmd("list", c.Id())
+		for m, _ := range c.Members() {
+			u := c.store.world.GetObject(m).(*user.User)
+			cmd.Args = append(cmd.Args, u.GetInfo())
+		}
+		r.Reply(cmd)
+
+	case "history":
+		if len(r.Cmd.Args) != 1 {
+			r.ReplyInvalid()
+			return
+		}
+		h := r.Cmd.Args[0]
+		var min int
+		var max int
+		switch h.Kind {
+		case "latest":
+			max = len(c.messages)
+			min = max - 20
+		case "before", "around", "at":
+			var id string
+			for k, v := range h.Fields {
+				switch k {
+				case "":
+					id = v
+				default:
+					r.ReplyInvalid()
+					return
+				}
+			}
+			i, ok := c.byId[id]
+			if !ok {
+				r.Reply(proto.Fail{"bad-target", "", nil}.Cmd())
+				return
+			}
+			switch h.Kind {
+			case "before":
+				max = i
+				min = max - 20
+			case "around":
+				min = i - 9
+				max = i + 11
+			case "at":
+				min = i
+				max = i + 1
+			}
+		default:
+			r.ReplyInvalid()
+			return
+		}
+		if min < 0 {
+			min = 0
+		}
+		if max > len(c.messages) {
+			max = len(c.messages)
+		}
+
+		p := c.members[r.From.UserId]
+		if !p.History || !p.See {
+			r.Reply(proto.Fail{"forbidden", "", nil}.Cmd())
+			return
+		}
+		cmd := proto.NewCmd("history", c.id)
+		cmd.Args = c.messages[min:max]
+		r.Reply(cmd)
+
+	case "membership":
+		if len(r.Cmd.Args) != 1 {
+			r.ReplyInvalid()
+			return
+		}
+		m := r.Cmd.Args[0]
+		if m.Kind != "u" {
+			r.ReplyInvalid()
+			return
+		}
+
+		if !c.members[r.From.UserId].Yes {
+			r.Reply(proto.Fail{"forbidden", "", nil}.Cmd())
+			return
+		}
+
+		if !c.members[m.Id].Yes {
+			r.Reply(proto.Fail{
+				"not-in-channel", "", map[string]string {"": m.Id},
+			}.Cmd())
+			return
+		}
+		i := c.members[m.Id].GetInfo()
+		i.Fields[""] = m.Id
+		r.Reply(proto.NewCmd("i", "", i))
+
+	case "config":
+		if len(r.Cmd.Args) != 1 {
+			r.ReplyInvalid()
+			return
+		}
+		o := r.Cmd.Args[0]
+		switch o.Kind {
+		case "membership":
+			id := o.Fields[""]
+			if id == "" {
+				r.ReplyInvalid()
+				return
+			}
+			new, err := c.members[id].Change(o)
+			if err != nil {
+				r.Reply(err.Cmd())
+				return
+			}
+
+			if !c.members[r.From.UserId].Op {
+				r.Reply(proto.Fail{"forbidden", "", nil}.Cmd())
+				return
+			}
+			if !c.members[id].Yes {
+				r.Reply(proto.Fail{
+					"not-in-channel", "", map[string]string {"": id},
+				}.Cmd())
+				return
+			}
+			c.members[id] = new
+			c.Put(o)
+
+			i := new.GetInfo()
+			i.Fields[""] = id
+			r.Reply(proto.NewCmd("i", "", i))
+		default:
+			r.ReplyInvalid()
+		}
+
+	default:
+		r.ReplyInvalid()
+	}
+}
diff --git a/server/channel/membership.go b/server/channel/membership.go
new file mode 100644
index 0000000..3a44517
--- /dev/null
+++ b/server/channel/membership.go
@@ -0,0 +1,87 @@
+package channel
+
+import (
+	"citrons.xyz/talk/proto"
+)
+
+type Membership struct {
+	Yes bool
+	See bool
+	Put bool
+	History bool
+	Moderate bool
+	Update bool
+	Invite bool
+	Op bool
+}
+
+var DefaultMembership = Membership {
+	Yes: true,
+	See: true,
+	Put: true,
+	History: true,
+	Moderate: false,
+	Update: false,
+	Op: false,
+}
+
+var CreatorMembership = Membership {
+	Yes: true,
+	See: true,
+	Put: true,
+	History: true,
+	Moderate: true,
+	Update: true,
+	Op: true,
+}
+
+func (m Membership) Change(spec proto.Object) (Membership, *proto.Fail) {
+	new := m
+	for k, v := range spec.Fields {
+		var field *bool
+		switch k {
+		case "see":
+			field = &new.See
+		case "put":
+			field = &new.Put
+		case "history":
+			field = &new.History
+		case "moderate":
+			field = &new.Moderate
+		case "update":
+			field = &new.Update
+		case "":
+			continue
+		default:
+			return new, &proto.Fail{"invalid", "", nil}
+		}
+		switch v {
+		case "yes":
+			*field = true
+		case "no":
+			*field = false
+		default:
+			return new, &proto.Fail{"invalid", "", nil}
+		}
+	}
+	return new, nil
+}
+
+func bstr(b bool) string {
+	if b {
+		return "yes"
+	} else {
+		return "no"
+	}
+}
+
+func (m Membership) GetInfo() proto.Object {
+	return proto.Object {"membership", "", map[string]string {
+		"see": bstr(m.See),
+		"put": bstr(m.Put),
+		"history": bstr(m.History),
+		"moderate": bstr(m.Moderate),
+		"update": bstr(m.Update),
+		"op": bstr(m.Op),
+	}}
+}