From aa664f0eb71d47e5a841691085ac152a75a19ce4 Mon Sep 17 00:00:00 2001 From: Mark Haines Date: Tue, 14 Mar 2017 17:05:02 +0000 Subject: [PATCH] Tool for creating room events (#36) * Tool for creating room events --- .../dendrite/tools/create-room-events/main.go | 133 ++++++++++++++++++ 1 file changed, 133 insertions(+) create mode 100644 src/github.com/matrix-org/dendrite/tools/create-room-events/main.go diff --git a/src/github.com/matrix-org/dendrite/tools/create-room-events/main.go b/src/github.com/matrix-org/dendrite/tools/create-room-events/main.go new file mode 100644 index 00000000..18c4906e --- /dev/null +++ b/src/github.com/matrix-org/dendrite/tools/create-room-events/main.go @@ -0,0 +1,133 @@ +// Generate a list of matrix room events for load testing. +// Writes the events to stdout by default. +package main + +import ( + "encoding/base64" + "encoding/json" + "flag" + "fmt" + "github.com/matrix-org/dendrite/roomserver/api" + "github.com/matrix-org/gomatrixserverlib" + "golang.org/x/crypto/ed25519" + "os" + "strings" + "time" +) + +const usage = `Usage: %s + +Generate a list of matrix room events for load testing. +Writes the events to stdout separated by new lines + +Arguments: + +` + +var ( + serverName = flag.String("server-name", "localhost", "The name of the matrix server to generate events for") + keyID = flag.String("key-id", "ed25519:auto", "The ID of the key used to sign the events") + privateKeyString = flag.String("private-key", defaultKey, "Base64 encoded private key to sign events with") + roomID = flag.String("room-id", "!roomid:$SERVER_NAME", "The room ID to generate events in") + userID = flag.String("user-id", "@userid:$SERVER_NAME", "The user ID to use as the event sender") + messageCount = flag.Int("message-count", 10, "The number of m.room.messsage events to generate") + format = flag.String("Format", "InputRoomEvent", "The output format to use for the messages: InputRoomEvent or Event") +) + +func defaulting(value, defaultValue string) string { + if value == "" { + return defaultValue + } + return value +} + +// By default we use a private key of 0. +const defaultKey = "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA" + +var privateKey ed25519.PrivateKey +var emptyString = "" +var now time.Time +var b gomatrixserverlib.EventBuilder +var eventID int + +func main() { + flag.Usage = func() { + fmt.Fprintf(os.Stderr, usage, os.Args[0]) + flag.PrintDefaults() + } + + flag.Parse() + *userID = strings.Replace(*userID, "$SERVER_NAME", *serverName, 1) + *roomID = strings.Replace(*roomID, "$SERVER_NAME", *serverName, 1) + + // Decode the ed25519 private key. + privateKeyBytes, err := base64.RawStdEncoding.DecodeString(*privateKeyString) + if err != nil { + panic(err) + } + privateKey = ed25519.PrivateKey(privateKeyBytes) + + // Build a m.room.create event. + b.Sender = *userID + b.RoomID = *roomID + b.Type = "m.room.create" + b.StateKey = &emptyString + b.SetContent(map[string]string{"creator": *userID}) + create := buildAndOutput() + + // Build a m.room.member event. + b.Type = "m.room.member" + b.StateKey = userID + b.SetContent(map[string]string{"membership": "join"}) + b.AuthEvents = []gomatrixserverlib.EventReference{create} + member := buildAndOutput() + + // Build a number of m.room.message events. + b.Type = "m.room.message" + b.StateKey = nil + b.SetContent(map[string]string{"body": "Test Message"}) + b.AuthEvents = []gomatrixserverlib.EventReference{create, member} + for i := 0; i < *messageCount; i++ { + buildAndOutput() + } +} + +// Build an event and write the event to the output. +func buildAndOutput() gomatrixserverlib.EventReference { + eventID++ + id := fmt.Sprintf("$%d:%s", eventID, *serverName) + now = time.Unix(0, 0) + event, err := b.Build(id, now, *serverName, *keyID, privateKey) + if err != nil { + panic(err) + } + writeEvent(event) + reference := event.EventReference() + b.PrevEvents = []gomatrixserverlib.EventReference{reference} + b.Depth++ + return reference +} + +// Write an event to the output. +func writeEvent(event gomatrixserverlib.Event) { + encoder := json.NewEncoder(os.Stdout) + if *format == "InputRoomEvent" { + var ire api.InputRoomEvent + ire.Kind = api.KindNew + ire.Event = event.JSON() + authEventIDs := []string{} + for _, ref := range b.AuthEvents { + authEventIDs = append(authEventIDs, ref.EventID) + } + ire.AuthEventIDs = authEventIDs + if err := encoder.Encode(ire); err != nil { + panic(err) + } + } else if *format == "Event" { + if err := encoder.Encode(event); err != nil { + panic(err) + } + } else { + panic(fmt.Errorf("Format %q is not valid, must be %q or %q", format, "InputRoomEvent", "Event")) + } +}