filestash/server/plugin/plg_widget_chat/handler.go

package plg_widget_chat

import (
	"net/http"
	"strings"
	"time"

	. "github.com/mickael-kerjean/filestash/server/common"
)

func listMessages(ctx *App, w http.ResponseWriter, r *http.Request) {
	path, err := PathBuilder(ctx, r.URL.Query().Get("path"))
	if err != nil {
		SendErrorResult(w, err)
		return
	}
	rows, err := db.QueryContext(r.Context(), `
		SELECT path, author, message, creation_date
			FROM messages
			WHERE path GLOB ?
			ORDER BY creation_date ASC
	`, globAll(path))
	if err != nil {
		SendErrorResult(w, err)
		return
	}
	defer rows.Close()

	out := []Message{}
	for rows.Next() {
		var m Message
		if err := rows.Scan(
			&m.Path,
			&m.Author,
			&m.Message,
			&m.CreatedAt,
		); err != nil {
			SendErrorResult(w, err)
			return
		}
		if ctx.Session["path"] != "" {
			m.Path = strings.TrimPrefix(m.Path, strings.TrimSuffix(ctx.Session["path"], "/"))
		}
		out = append(out, m)
	}
	SendSuccessResults(w, out)
}

func createMessage(ctx *App, w http.ResponseWriter, r *http.Request) {
	path, err := PathBuilder(ctx, r.URL.Query().Get("path"))
	if err != nil {
		SendErrorResult(w, err)
		return
	}
	msg, ok := ctx.Body["message"].(string)
	if !ok {
		SendErrorResult(w, NewError("Invalid parameters", 400))
		return
	}
	author := getUser(ctx.Session)
	_, err = db.ExecContext(ctx.Context, `
		INSERT INTO messages(id, path, author, message, creation_date)
		VALUES(?,?,?,?,?)
	`, newID(), path, author, msg, time.Now().Unix())
	if err != nil {
		SendErrorResult(w, err)
		return
	}

	extractMentions := func(message string) []string {
		matches := mention_re.FindAllStringSubmatch(message, -1)
		out := make([]string, 0, len(matches))
		for _, m := range matches {
			name := strings.TrimSpace(m[1])
			if name != "" {
				out = append(out, name)
			}
		}
		return out
	}
	for _, name := range extractMentions(msg) {
		go processMention(map[string]string{
			"path":    path,
			"author":  author,
			"mention": name,
			"message": msg,
		})
	}
	SendSuccessResult(w, nil)
}

func lookupUsers(ctx *App, w http.ResponseWriter, r *http.Request) {
	if ctx.Share.Id != "" {
		SendSuccessResults(w, []DirectoryUser{})
		return
	}
	dir := Hooks.Get.DirectoryService()
	if dir == nil {
		SendSuccessResults(w, []DirectoryUser{})
		return
	}
	results, err := dir.Search(r.URL.Query().Get("q"))
	if err != nil {
		SendErrorResult(w, err)
		return
	}
	SendSuccessResults(w, results)
}