gosora/common/widget_wol.go

package common

import (
	"bytes"
	//"log"
	"net/http/httptest"

	p "git.tuxpa.in/a/gosora/common/phrases"
	min "git.tuxpa.in/a/gosora/common/templates"
)

type wolUsers struct {
    *Header
    Name      string
    Users     []*User
    UserCount int
}

func wolInit(w *Widget, sched *WidgetScheduler) error {
    sched.Add(w)
    return nil
}

func wolGetUsers() ([]*User, int) {
    ucount := WsHub.UserCount()
    // We don't want a ridiculously long list, so we'll show the number if it's too high and only show staff individually
    var users []*User
    if ucount < 30 {
        users = WsHub.AllUsers()
        if len(users) >= 30 {
            users = nil
        }
    }
    return users, ucount
}

func wolBuild(w *Widget, hvars interface{}) (string, error) {
    users, ucount := wolGetUsers()
    wol := &wolUsers{hvars.(*Header), p.GetTmplPhrase("widget.online_name"), users, ucount}
    err := wol.Header.Theme.RunTmpl("widget_online", wol, wol.Header.Writer)
    return "", err
}

func wolRender(w *Widget, hvars interface{}) (string, error) {
    iTickMask := w.TickMask.Load()
    if iTickMask != nil {
        tickMask := iTickMask.(*Widget)
        if tickMask != nil {
            return tickMask.Body, nil
        }
    }
    return wolBuild(w, hvars)
}

var wolLastUsers []*User

func wolTick(widget *Widget) error {
    w := httptest.NewRecorder()
    users, ucount := wolGetUsers()
    inOld := func(id int) bool {
        for _, user := range wolLastUsers {
            if id == user.ID {
                return true
            }
        }
        return false
    }

    // Avoid rebuilding the widget, if the users are exactly the same as on the last tick
    if len(users) == len(wolLastUsers) {
        diff := false
        for _, user := range users {
            if !inOld(user.ID) {
                diff = true
                break
            }
        }
        if !diff {
            iTickMask := widget.TickMask.Load()
            if iTickMask != nil {
                tickMask := iTickMask.(*Widget)
                if tickMask != nil {
                    return nil
                }
            }
        }
    }
    //log.Printf("users: %+v\n", users)
    //log.Printf("wolLastUsers: %+v\n", wolLastUsers)

    wol := &wolUsers{SimpleDefaultHeader(w), p.GetTmplPhrase("widget.online_name"), users, ucount}
    err := wol.Header.Theme.RunTmpl("widget_online", wol, wol.Header.Writer)
    if err != nil {
        return err
    }

    buf := new(bytes.Buffer)
    buf.ReadFrom(w.Result().Body)
    bs := buf.String()
    if Config.MinifyTemplates {
        bs = min.Minify(bs)
    }

    twidget := &Widget{}
    *twidget = *widget
    twidget.Body = bs
    widget.TickMask.Store(twidget)
    wolLastUsers = users

    hTbl := GetHookTable()
    _, _ = hTbl.VhookSkippable("tasks_tick_widget_wol", widget, bs)

    return nil
}