"os/exec"
"os/signal"
"path"
+ "sort"
"strings"
"syscall"
"time"
var (
Newwin = flag.String("newwin", "cmd/newwin", "Path to newwin command")
- DebugFifo = flag.String("debug", "debug", "Path to debug FIFO to be created")
+ DebugFifo = flag.String("debug", "", "Path to debug FIFO to be created")
DebugFd *os.File
UmaskCur int
)
entrypoint := flag.String("entrypoint", mmc.GetEntrypoint(), "Entrypoint")
notifyCmd := flag.String("notify", "cmd/notify", "Path to notification handler")
heartbeatCh := flag.String("heartbeat-ch", "town-square", "Channel for heartbeating")
+ userStatusFifo := flag.String("user-status", "", "Path to FIFO for user statuses")
flag.Parse()
log.SetFlags(log.Lshortfile)
log.SetOutput(os.Stdout)
var updateQueue []string
LastSent := time.Now()
+ Chans := make(map[string]*model.Channel)
+ time.Sleep(mmc.SleepTime)
+ page, resp, err := c.GetChannelsForTeamForUser(Team.Id, me.Id, false, "")
+ if err != nil {
+ if DebugFd != nil {
+ spew.Fdump(DebugFd, resp)
+ }
+ log.Fatalln(err)
+ }
+ if DebugFd != nil {
+ spew.Fdump(DebugFd, page)
+ }
+ for _, ch := range page {
+ if ch.Type == "D" {
+ continue
+ }
+ Chans[ch.Name] = ch
+ pth := path.Join("chans", strings.ReplaceAll(ch.Name, ".", "_"))
+ updateQueue = append(updateQueue, pth, ch.Id)
+ os.MkdirAll(pth, 0777)
+ rewriteIfChanged(path.Join(pth, "id"), ch.Id+"\n")
+ rewriteIfChanged(path.Join(pth, "info"), fmt.Sprintf(
+ "%s\n%s\n%s\n",
+ ch.DisplayName,
+ ch.Header,
+ ch.Purpose,
+ ))
+ if _, err := os.Stat(path.Join(pth, mmc.OutRec)); err != nil &&
+ errors.Is(err, fs.ErrNotExist) {
+ if _, err = os.OpenFile(
+ path.Join(pth, mmc.OutRec), os.O_WRONLY|os.O_CREATE, 0o666,
+ ); err != nil {
+ log.Fatalln(err)
+ }
+ }
+
+ usersPth := path.Join(pth, "users")
+ os.Remove(usersPth)
+ if err := syscall.Mkfifo(usersPth, 0666); err != nil {
+ log.Fatalln(err)
+ }
+ go func(ch *model.Channel) {
+ for {
+ time.Sleep(mmc.SleepTime)
+ fd, err := os.OpenFile(
+ usersPth, os.O_WRONLY|os.O_APPEND, os.FileMode(0666),
+ )
+ if err != nil {
+ log.Println("OpenFile:", usersPth, err)
+ continue
+ }
+ for n := 0; ; n++ {
+ users, resp, err := c.GetUsersInChannel(ch.Id, n, mmc.PerPage, "")
+ if err != nil {
+ if DebugFd != nil {
+ spew.Fdump(DebugFd, resp)
+ }
+ log.Println("GetUsersInChannel:", err)
+ fd.Close()
+ continue
+ }
+ if DebugFd != nil {
+ spew.Fdump(DebugFd, users)
+ }
+ for _, u := range users {
+ fmt.Fprintf(fd, "%s\n", u.Username)
+ }
+ if len(users) < mmc.PerPage {
+ break
+ }
+ }
+ fd.Close()
+ }
+ }(ch)
+
+ pth = path.Join(pth, "in")
+ os.Remove(pth)
+ if err := syscall.Mkfifo(pth, 0666); err != nil {
+ log.Fatalln(err)
+ }
+ go func(ch *model.Channel) {
+ for {
+ fd, err := os.OpenFile(pth, os.O_RDONLY, os.FileMode(0666))
+ if err != nil {
+ continue
+ }
+ data, err := io.ReadAll(fd)
+ fd.Close()
+ if err != nil {
+ continue
+ }
+ if _, err = makePost(c, ch.Id, string(data)); err != nil {
+ log.Println("makePost:", err)
+ }
+ LastSent = time.Now()
+ }
+ }(ch)
+ }
+
Users, err := mmc.GetUsers(c, DebugFd)
if err != nil {
log.Fatalln(err)
}
+
UsersDC := make(map[string]*model.Channel, len(Users))
+ userIds := make([]string, 0, len(Users))
for _, u := range Users {
+ userIds = append(userIds, u.Id)
pth := path.Join("users", strings.ReplaceAll(u.Username, ".", "_"))
os.MkdirAll(pth, 0777)
rewriteIfChanged(
}(u)
}
- Chans := make(map[string]*model.Channel)
- time.Sleep(mmc.SleepTime)
- page, resp, err := c.GetChannelsForTeamForUser(Team.Id, me.Id, false, "")
- if err != nil {
- if DebugFd != nil {
- spew.Fdump(DebugFd, resp)
- }
- log.Fatalln(err)
- }
- if DebugFd != nil {
- spew.Fdump(DebugFd, page)
- }
- for _, ch := range page {
- if ch.Type == "D" {
- continue
- }
- Chans[ch.Name] = ch
- pth := path.Join("chans", strings.ReplaceAll(ch.Name, ".", "_"))
- updateQueue = append(updateQueue, pth, ch.Id)
- os.MkdirAll(pth, 0777)
- rewriteIfChanged(path.Join(pth, "id"), ch.Id+"\n")
- rewriteIfChanged(path.Join(pth, "info"), fmt.Sprintf(
- "%s\n%s\n%s\n",
- ch.DisplayName,
- ch.Header,
- ch.Purpose,
- ))
- if _, err := os.Stat(path.Join(pth, mmc.OutRec)); err != nil &&
- errors.Is(err, fs.ErrNotExist) {
- if _, err = os.OpenFile(
- path.Join(pth, mmc.OutRec), os.O_WRONLY|os.O_CREATE, 0o666,
- ); err != nil {
- log.Fatalln(err)
+ UserStatus := make(map[string]string)
+ updateUserStatus := func() {
+ statuses, resp, err := c.GetUsersStatusesByIds(userIds)
+ if err != nil {
+ if DebugFd != nil {
+ spew.Fdump(DebugFd, resp)
}
- }
-
- usersPth := path.Join(pth, "users")
- os.Remove(usersPth)
- if err := syscall.Mkfifo(usersPth, 0666); err != nil {
log.Fatalln(err)
}
- go func(ch *model.Channel) {
+ if DebugFd != nil {
+ spew.Fdump(DebugFd, teams)
+ }
+ for _, s := range statuses {
+ UserStatus[Users[s.UserId].Username] = s.Status
+ }
+ }
+ if *userStatusFifo != "" {
+ updateUserStatus()
+ go func() {
for {
time.Sleep(mmc.SleepTime)
fd, err := os.OpenFile(
- usersPth, os.O_WRONLY|os.O_APPEND, os.FileMode(0666),
+ *userStatusFifo, os.O_WRONLY|os.O_APPEND, os.FileMode(0666),
)
if err != nil {
- log.Println("OpenFile:", usersPth, err)
+ log.Println("OpenFile:", *userStatusFifo, err)
continue
}
- for n := 0; ; n++ {
- users, resp, err := c.GetUsersInChannel(ch.Id, n, mmc.PerPage, "")
- if err != nil {
- if DebugFd != nil {
- spew.Fdump(DebugFd, resp)
- }
- log.Println("GetUsersInChannel:", err)
- fd.Close()
+ statuses := make(map[string][]string)
+ for name, status := range UserStatus {
+ statuses[status] = append(statuses[status], name)
+ }
+ for status := range statuses {
+ if status == "offline" {
continue
}
- if DebugFd != nil {
- spew.Fdump(DebugFd, users)
- }
- for _, u := range users {
- fmt.Fprintf(fd, "%s\n", u.Username)
- }
- if len(users) < mmc.PerPage {
- break
- }
- }
- fd.Close()
- }
- }(ch)
-
- pth = path.Join(pth, "in")
- os.Remove(pth)
- if err := syscall.Mkfifo(pth, 0666); err != nil {
- log.Fatalln(err)
- }
- go func(ch *model.Channel) {
- for {
- fd, err := os.OpenFile(pth, os.O_RDONLY, os.FileMode(0666))
- if err != nil {
- continue
+ sort.Strings(statuses[status])
+ fmt.Fprintln(fd, status+":", strings.Join(statuses[status], " "))
}
- data, err := io.ReadAll(fd)
fd.Close()
- if err != nil {
- continue
- }
- if _, err = makePost(c, ch.Id, string(data)); err != nil {
- log.Println("makePost:", err)
- }
- LastSent = time.Now()
}
- }(ch)
+ }()
}
log.Println("syncing", len(updateQueue)/2, "rooms")
for {
select {
case <-t.C:
+ updateUserStatus()
if time.Now().Before(LastSent.Add(time.Minute)) {
continue
}
if e == nil || !e.IsValid() {
continue
}
- data := e.GetData()
if DebugFd != nil {
- spew.Fdump(DebugFd, e.EventType(), data)
+ spew.Fdump(DebugFd, e)
}
+ data := e.GetData()
var user *model.User
if userId, ok := data["user_id"]; ok && userId.(string) != "" {
user = Users[userId.(string)]
case model.WebsocketEventPostEdited,
model.WebsocketEventPostDeleted,
model.WebsocketEventPosted:
- chName := data["channel_name"].(string)
+ chName, ok := data["channel_name"].(string)
+ if !ok {
+ continue
+ }
var post model.Post
if err = json.NewDecoder(
strings.NewReader(data["post"].(string)),
log.Fatalln(err)
}
case model.WebsocketEventStatusChange:
- log.Println("status change:", user.Username, "->", data["status"].(string))
+ status := data["status"].(string)
+ UserStatus[user.Username] = status
if *notifyCmd != "" {
exec.Command(*notifyCmd, fmt.Sprintf(
- "status: %s -> %s",
- user.Username, data["status"].(string),
+ "status: %s -> %s", user.Username, status,
)).Run()
}
case model.WebsocketEventHello: