simple cli with room APIs

This commit is contained in:
David Zhao
2020-11-03 22:01:54 -08:00
parent 925d1c653e
commit 9ebfebe2dd
8 changed files with 626 additions and 4 deletions
+11
View File
@@ -0,0 +1,11 @@
package commands
import (
"encoding/json"
"fmt"
)
func PrintJSON(obj interface{}) {
txt, _ := json.Marshal(obj)
fmt.Println(string(txt))
}
+98
View File
@@ -0,0 +1,98 @@
package commands
import (
"context"
"fmt"
"net/http"
"github.com/urfave/cli/v2"
"github.com/livekit/livekit-server/proto/livekit"
)
var (
roomFlag = &cli.StringFlag{
Name: "room-id",
Required: true,
}
hostFlag = &cli.StringFlag{
Name: "host",
Value: "http://localhost:7880",
}
RoomCommands = []*cli.Command{
{
Name: "create-room",
Before: createClient,
Action: createRoom,
Flags: []cli.Flag{
roomFlag,
hostFlag,
},
},
{
Name: "get-room",
Before: createClient,
Action: getRoom,
Flags: []cli.Flag{
roomFlag,
hostFlag,
},
},
{
Name: "delete-room",
Before: createClient,
Action: deleteRoom,
Flags: []cli.Flag{
roomFlag,
hostFlag,
},
},
}
roomClient livekit.RoomService
)
func createClient(c *cli.Context) error {
host := c.String("host")
roomClient = livekit.NewRoomServiceJSONClient(host, &http.Client{})
return nil
}
func createRoom(c *cli.Context) error {
roomId := c.String("room-id")
room, err := roomClient.CreateRoom(context.Background(), &livekit.CreateRoomRequest{
RoomId: roomId,
})
if err != nil {
return err
}
PrintJSON(room)
return nil
}
func getRoom(c *cli.Context) error {
roomId := c.String("room-id")
room, err := roomClient.GetRoom(context.Background(), &livekit.GetRoomRequest{
RoomId: roomId,
})
if err != nil {
return err
}
PrintJSON(room)
return nil
}
func deleteRoom(c *cli.Context) error {
roomId := c.String("room-id")
_, err := roomClient.DeleteRoom(context.Background(), &livekit.DeleteRoomRequest{
RoomId: roomId,
})
if err != nil {
return err
}
fmt.Println("deleted room", roomId)
return nil
}
+22
View File
@@ -1,3 +1,25 @@
package main
import (
"fmt"
"os"
"github.com/urfave/cli/v2"
"github.com/livekit/livekit-server/cmd/cli/commands"
"github.com/livekit/livekit-server/pkg/logger"
)
// command line util that tests server
func main() {
app := &cli.App{
Name: "livekit-cli",
}
app.Commands = append(app.Commands, commands.RoomCommands...)
logger.InitDevelopment()
if err := app.Run(os.Args); err != nil {
fmt.Println(err)
}
}