mirror of
https://github.com/onyx-and-iris/gobs-cli.git
synced 2026-04-18 07:03:37 +00:00
Compare commits
15 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 90aa5d4423 | |||
| da010d67a0 | |||
| 0c695298fd | |||
| 2f77fa1c54 | |||
| eafc3312a5 | |||
| 02541f9915 | |||
| 7fa43eb35c | |||
| 8aeb7cb183 | |||
| 6e25927bc1 | |||
| dd0bbfc0da | |||
| c04324d173 | |||
| 36d0753bd9 | |||
| 3095c0c49d | |||
| 53bbb58cfb | |||
| 5f2fe05caa |
3
.gitignore
vendored
3
.gitignore
vendored
@@ -35,6 +35,3 @@ gobs-cli.1
|
||||
|
||||
# Config files
|
||||
config.yaml
|
||||
|
||||
# Test files
|
||||
*_test.go
|
||||
@@ -50,3 +50,6 @@ issues:
|
||||
exclude:
|
||||
# gosec: Duplicated errcheck checks
|
||||
- G104
|
||||
exclude-files:
|
||||
# Exclude vendor directory
|
||||
- main_test.go
|
||||
|
||||
23
CHANGELOG.md
23
CHANGELOG.md
@@ -5,7 +5,28 @@ All notable changes to this project will be documented in this file.
|
||||
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
||||
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
||||
|
||||
# [0.3.1] - 2024-05-02
|
||||
# [0.6.0] - 2025-05-25
|
||||
|
||||
### Added
|
||||
|
||||
- filter commands, see [Filter](https://github.com/onyx-and-iris/gobs-cli?tab=readme-ov-file#filter)
|
||||
|
||||
# [0.5.0] - 2025-05-22
|
||||
|
||||
### Added
|
||||
|
||||
- hotkey commands, see [Hotkey](https://github.com/onyx-and-iris/gobs-cli?tab=readme-ov-file#hotkeycmd)
|
||||
|
||||
# [0.4.2] - 2025-05-08
|
||||
|
||||
### Added
|
||||
|
||||
- replaybuffer toggle command
|
||||
- studiomode enable/disable now print output to console
|
||||
- stream start/stop now print output to console
|
||||
- Unit tests
|
||||
|
||||
# [0.3.1] - 2025-05-02
|
||||
|
||||
### Added
|
||||
|
||||
|
||||
78
README.md
78
README.md
@@ -373,6 +373,12 @@ gobs-cli replaybuffer start
|
||||
gobs-cli replaybuffer stop
|
||||
```
|
||||
|
||||
- toggle: Toggle replay buffer.
|
||||
|
||||
```console
|
||||
gobs-cli replaybuffer toggle
|
||||
```
|
||||
|
||||
- status: Get replay buffer status.
|
||||
|
||||
```console
|
||||
@@ -437,4 +443,76 @@ gobs-cli virtualcam toggle
|
||||
gobs-cli virtualcam status
|
||||
```
|
||||
|
||||
### HotkeyCmd
|
||||
|
||||
- list: List all hotkeys.
|
||||
|
||||
```console
|
||||
gobs-cli hotkey list
|
||||
```
|
||||
|
||||
- trigger: Trigger a hotkey by name.
|
||||
|
||||
```console
|
||||
gobs-cli hotkey trigger OBSBasic.StartStreaming
|
||||
|
||||
gobs-cli hotkey trigger OBSBasic.StopStreaming
|
||||
```
|
||||
|
||||
- trigger-sequence: Trigger a hotkey by sequence.
|
||||
- flags:
|
||||
|
||||
*optional*
|
||||
- --shift: Press shift.
|
||||
- --ctrl: Press control.
|
||||
- --alt: Press alt.
|
||||
- --cmd: Press command (mac).
|
||||
|
||||
- args: keyID
|
||||
- Check [obs-hotkeys.h][obs-keyids] for a full list of OBS key ids.
|
||||
|
||||
```console
|
||||
gobs-cli hotkey trigger-sequence OBS_KEY_F1 --ctrl
|
||||
|
||||
gobs-cli hotkey trigger-sequence OBS_KEY_F1 --shift --ctrl
|
||||
```
|
||||
|
||||
### FilterCmd
|
||||
|
||||
- list: List all filters.
|
||||
|
||||
```console
|
||||
gobs-cli filter list
|
||||
```
|
||||
|
||||
- enable: Enable filter.
|
||||
- args: SourceName FilterName
|
||||
|
||||
```console
|
||||
gobs-cli enable 'Mic/Aux' 'Gain'
|
||||
```
|
||||
|
||||
- disable: Disable filter.
|
||||
- args: SourceName FilterName
|
||||
|
||||
```console
|
||||
gobs-cli disable 'Mic/Aux' 'Gain'
|
||||
```
|
||||
|
||||
- toggle: Toggle filter.
|
||||
- args: SourceName FilterName
|
||||
|
||||
```console
|
||||
gobs-cli toggle 'Mic/Aux' 'Gain'
|
||||
```
|
||||
|
||||
- status: Get filter status.
|
||||
- args: SourceName FilterName
|
||||
|
||||
```console
|
||||
gobs-cli status 'Mic/Aux' 'Gain'
|
||||
```
|
||||
|
||||
|
||||
[userconfigdir]: https://pkg.go.dev/os#UserConfigDir
|
||||
[obs-keyids]: https://github.com/obsproject/obs-studio/blob/master/libobs/obs-hotkeys.h
|
||||
149
filter.go
Normal file
149
filter.go
Normal file
@@ -0,0 +1,149 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/andreykaipov/goobs/api/requests/filters"
|
||||
)
|
||||
|
||||
// FilterCmd provides commands to manage filters in OBS Studio.
|
||||
type FilterCmd struct {
|
||||
List FilterListCmd `cmd:"" help:"List all filters." aliases:"ls"`
|
||||
Enable FilterEnableCmd `cmd:"" help:"Enable filter." aliases:"on"`
|
||||
Disable FilterDisableCmd `cmd:"" help:"Disable filter." aliases:"off"`
|
||||
Toggle FilterToggleCmd `cmd:"" help:"Toggle filter." aliases:"tg"`
|
||||
Status FilterStatusCmd `cmd:"" help:"Get filter status." aliases:"ss"`
|
||||
}
|
||||
|
||||
// FilterListCmd provides a command to list all filters in a scene.
|
||||
type FilterListCmd struct {
|
||||
SourceName string `arg:"" help:"Name of the source to list filters from."`
|
||||
}
|
||||
|
||||
// Run executes the command to list all filters in a scene.
|
||||
func (cmd *FilterListCmd) Run(ctx *context) error {
|
||||
filters, err := ctx.Client.Filters.GetSourceFilterList(
|
||||
filters.NewGetSourceFilterListParams().WithSourceName(cmd.SourceName),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
for _, filter := range filters.Filters {
|
||||
fmt.Fprintf(ctx.Out, "Name: %s\n Kind: %s\n Enabled: %t\n Settings: %+v\n",
|
||||
filter.FilterName, filter.FilterKind, filter.FilterEnabled, filter.FilterSettings)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// FilterEnableCmd provides a command to enable a filter in a scene.
|
||||
type FilterEnableCmd struct {
|
||||
SourceName string `arg:"" help:"Name of the source to enable filter from."`
|
||||
FilterName string `arg:"" help:"Name of the filter to enable."`
|
||||
}
|
||||
|
||||
// Run executes the command to enable a filter in a scene.
|
||||
func (cmd *FilterEnableCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.Filters.SetSourceFilterEnabled(
|
||||
filters.NewSetSourceFilterEnabledParams().
|
||||
WithSourceName(cmd.SourceName).
|
||||
WithFilterName(cmd.FilterName).
|
||||
WithFilterEnabled(true),
|
||||
)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to enable filter %s on source %s: %w",
|
||||
cmd.FilterName, cmd.SourceName, err)
|
||||
}
|
||||
fmt.Fprintf(ctx.Out, "Filter %s enabled on source %s.\n",
|
||||
cmd.FilterName, cmd.SourceName)
|
||||
return nil
|
||||
}
|
||||
|
||||
// FilterDisableCmd provides a command to disable a filter in a scene.
|
||||
type FilterDisableCmd struct {
|
||||
SourceName string `arg:"" help:"Name of the source to disable filter from."`
|
||||
FilterName string `arg:"" help:"Name of the filter to disable."`
|
||||
}
|
||||
|
||||
// Run executes the command to disable a filter in a scene.
|
||||
func (cmd *FilterDisableCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.Filters.SetSourceFilterEnabled(
|
||||
filters.NewSetSourceFilterEnabledParams().
|
||||
WithSourceName(cmd.SourceName).
|
||||
WithFilterName(cmd.FilterName).
|
||||
WithFilterEnabled(false),
|
||||
)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to disable filter %s on source %s: %w",
|
||||
cmd.FilterName, cmd.SourceName, err)
|
||||
}
|
||||
fmt.Fprintf(ctx.Out, "Filter %s disabled on source %s.\n",
|
||||
cmd.FilterName, cmd.SourceName)
|
||||
return nil
|
||||
}
|
||||
|
||||
// FilterToggleCmd provides a command to toggle a filter in a scene.
|
||||
type FilterToggleCmd struct {
|
||||
SourceName string `arg:"" help:"Name of the source to toggle filter from."`
|
||||
FilterName string `arg:"" help:"Name of the filter to toggle."`
|
||||
}
|
||||
|
||||
// Run executes the command to toggle a filter in a scene.
|
||||
func (cmd *FilterToggleCmd) Run(ctx *context) error {
|
||||
filter, err := ctx.Client.Filters.GetSourceFilter(
|
||||
filters.NewGetSourceFilterParams().
|
||||
WithSourceName(cmd.SourceName).
|
||||
WithFilterName(cmd.FilterName),
|
||||
)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get filter %s on source %s: %w",
|
||||
cmd.FilterName, cmd.SourceName, err)
|
||||
}
|
||||
|
||||
newStatus := !filter.FilterEnabled
|
||||
_, err = ctx.Client.Filters.SetSourceFilterEnabled(
|
||||
filters.NewSetSourceFilterEnabledParams().
|
||||
WithSourceName(cmd.SourceName).
|
||||
WithFilterName(cmd.FilterName).
|
||||
WithFilterEnabled(newStatus),
|
||||
)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to toggle filter %s on source %s: %w",
|
||||
cmd.FilterName, cmd.SourceName, err)
|
||||
}
|
||||
|
||||
if newStatus {
|
||||
fmt.Fprintf(ctx.Out, "Filter %s on source %s is now enabled.\n",
|
||||
cmd.FilterName, cmd.SourceName)
|
||||
} else {
|
||||
fmt.Fprintf(ctx.Out, "Filter %s on source %s is now disabled.\n",
|
||||
cmd.FilterName, cmd.SourceName)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// FilterStatusCmd provides a command to get the status of a filter in a scene.
|
||||
type FilterStatusCmd struct {
|
||||
SourceName string `arg:"" help:"Name of the source to get filter status from."`
|
||||
FilterName string `arg:"" help:"Name of the filter to get status."`
|
||||
}
|
||||
|
||||
// Run executes the command to get the status of a filter in a scene.
|
||||
func (cmd *FilterStatusCmd) Run(ctx *context) error {
|
||||
filter, err := ctx.Client.Filters.GetSourceFilter(
|
||||
filters.NewGetSourceFilterParams().
|
||||
WithSourceName(cmd.SourceName).
|
||||
WithFilterName(cmd.FilterName),
|
||||
)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get status of filter %s on source %s: %w",
|
||||
cmd.FilterName, cmd.SourceName, err)
|
||||
}
|
||||
if filter.FilterEnabled {
|
||||
fmt.Fprintf(ctx.Out, "Filter %s on source %s is enabled.\n",
|
||||
cmd.FilterName, cmd.SourceName)
|
||||
} else {
|
||||
fmt.Fprintf(ctx.Out, "Filter %s on source %s is disabled.\n",
|
||||
cmd.FilterName, cmd.SourceName)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
130
group_test.go
Normal file
130
group_test.go
Normal file
@@ -0,0 +1,130 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestGroupList(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmd := &GroupListCmd{
|
||||
SceneName: "Scene",
|
||||
}
|
||||
err := cmd.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to list groups: %v", err)
|
||||
}
|
||||
if !strings.Contains(out.String(), "test_group") {
|
||||
t.Fatalf("Expected output to contain 'test_group', got '%s'", out.String())
|
||||
}
|
||||
}
|
||||
|
||||
func TestGroupShow(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmd := &GroupShowCmd{
|
||||
SceneName: "Scene",
|
||||
GroupName: "test_group",
|
||||
}
|
||||
err := cmd.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to show group: %v", err)
|
||||
}
|
||||
if out.String() != "Group test_group is now shown.\n" {
|
||||
t.Fatalf("Expected output to be 'Group test_group is now shown.', got '%s'", out.String())
|
||||
}
|
||||
}
|
||||
|
||||
func TestGroupToggle(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmdStatus := &GroupStatusCmd{
|
||||
SceneName: "Scene",
|
||||
GroupName: "test_group",
|
||||
}
|
||||
err := cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get group status: %v", err)
|
||||
}
|
||||
var enabled bool
|
||||
if strings.Contains(out.String(), "Group test_group is shown.") {
|
||||
enabled = true
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdToggle := &GroupToggleCmd{
|
||||
SceneName: "Scene",
|
||||
GroupName: "test_group",
|
||||
}
|
||||
err = cmdToggle.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to toggle group: %v", err)
|
||||
}
|
||||
if enabled {
|
||||
if out.String() != "Group test_group is now hidden.\n" {
|
||||
t.Fatalf("Expected output to be 'Group test_group is now hidden.', got '%s'", out.String())
|
||||
}
|
||||
} else {
|
||||
if out.String() != "Group test_group is now shown.\n" {
|
||||
t.Fatalf("Expected output to be 'Group test_group is now shown.', got '%s'", out.String())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGroupStatus(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmdShow := &GroupShowCmd{
|
||||
SceneName: "Scene",
|
||||
GroupName: "test_group",
|
||||
}
|
||||
err := cmdShow.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to show group: %v", err)
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdStatus := &GroupStatusCmd{
|
||||
SceneName: "Scene",
|
||||
GroupName: "test_group",
|
||||
}
|
||||
err = cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get group status: %v", err)
|
||||
}
|
||||
if out.String() != "Group test_group is shown.\n" {
|
||||
t.Fatalf("Expected output to be 'Group test_group is shown.', got '%s'", out.String())
|
||||
}
|
||||
}
|
||||
74
hotkey.go
Normal file
74
hotkey.go
Normal file
@@ -0,0 +1,74 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/andreykaipov/goobs/api/requests/general"
|
||||
"github.com/andreykaipov/goobs/api/typedefs"
|
||||
)
|
||||
|
||||
// HotkeyCmd provides commands to manage hotkeys in OBS Studio.
|
||||
type HotkeyCmd struct {
|
||||
List HotkeyListCmd `cmd:"" help:"List all hotkeys." aliases:"ls"`
|
||||
Trigger HotkeyTriggerCmd `cmd:"" help:"Trigger a hotkey by name." aliases:"tr"`
|
||||
TriggerSequence HotkeyTriggerSequenceCmd `cmd:"" help:"Trigger a hotkey by sequence." aliases:"trs"`
|
||||
}
|
||||
|
||||
// HotkeyListCmd provides a command to list all hotkeys.
|
||||
type HotkeyListCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to list all hotkeys.
|
||||
func (cmd *HotkeyListCmd) Run(ctx *context) error {
|
||||
resp, err := ctx.Client.General.GetHotkeyList()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, hotkey := range resp.Hotkeys {
|
||||
fmt.Fprintln(ctx.Out, hotkey)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// HotkeyTriggerCmd provides a command to trigger a hotkey.
|
||||
type HotkeyTriggerCmd struct {
|
||||
Hotkey string `help:"Hotkey name to trigger." arg:""`
|
||||
}
|
||||
|
||||
// Run executes the command to trigger a hotkey.
|
||||
func (cmd *HotkeyTriggerCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.General.TriggerHotkeyByName(
|
||||
general.NewTriggerHotkeyByNameParams().WithHotkeyName(cmd.Hotkey),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// HotkeyTriggerSequenceCmd provides a command to trigger a hotkey sequence.
|
||||
type HotkeyTriggerSequenceCmd struct {
|
||||
Shift bool `flag:"" help:"Shift modifier."`
|
||||
Ctrl bool `flag:"" help:"Control modifier."`
|
||||
Alt bool `flag:"" help:"Alt modifier."`
|
||||
Cmd bool `flag:"" help:"Command modifier."`
|
||||
KeyID string ` help:"Key ID to trigger." arg:""`
|
||||
}
|
||||
|
||||
// Run executes the command to trigger a hotkey sequence.
|
||||
func (cmd *HotkeyTriggerSequenceCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.General.TriggerHotkeyByKeySequence(
|
||||
general.NewTriggerHotkeyByKeySequenceParams().
|
||||
WithKeyId(cmd.KeyID).
|
||||
WithKeyModifiers(&typedefs.KeyModifiers{
|
||||
Shift: cmd.Shift,
|
||||
Control: cmd.Ctrl,
|
||||
Alt: cmd.Alt,
|
||||
Command: cmd.Cmd,
|
||||
}),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
8
main.go
8
main.go
@@ -24,9 +24,9 @@ type ObsConfig struct {
|
||||
Timeout int `flag:"timeout" help:"Timeout in seconds." default:"5" env:"OBS_TIMEOUT"`
|
||||
}
|
||||
|
||||
// cli is the main command line interface structure.
|
||||
// CLI is the main command line interface structure.
|
||||
// It embeds the ObsConfig struct to inherit its fields and flags.
|
||||
type cli struct {
|
||||
type CLI struct {
|
||||
ObsConfig `embed:"" help:"OBS WebSocket configuration."`
|
||||
|
||||
Man mangokong.ManFlag `help:"Print man page."`
|
||||
@@ -43,6 +43,8 @@ type cli struct {
|
||||
Replaybuffer ReplayBufferCmd `help:"Manage replay buffer." cmd:"" aliases:"rb"`
|
||||
Studiomode StudioModeCmd `help:"Manage studio mode." cmd:"" aliases:"sm"`
|
||||
Virtualcam VirtualCamCmd `help:"Manage virtual camera." cmd:"" aliases:"vc"`
|
||||
Hotkey HotkeyCmd `help:"Manage hotkeys." cmd:"" aliases:"hk"`
|
||||
Filter FilterCmd `help:"Manage filters." cmd:"" aliases:"f"`
|
||||
}
|
||||
|
||||
type context struct {
|
||||
@@ -57,7 +59,7 @@ func main() {
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
var cli cli
|
||||
var cli CLI
|
||||
ctx := kong.Parse(
|
||||
&cli,
|
||||
kong.Name("GOBS-CLI"),
|
||||
|
||||
101
main_test.go
Normal file
101
main_test.go
Normal file
@@ -0,0 +1,101 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/andreykaipov/goobs"
|
||||
"github.com/andreykaipov/goobs/api/requests/config"
|
||||
"github.com/andreykaipov/goobs/api/requests/inputs"
|
||||
"github.com/andreykaipov/goobs/api/requests/scenes"
|
||||
typedefs "github.com/andreykaipov/goobs/api/typedefs"
|
||||
)
|
||||
|
||||
func getClient(t *testing.T) (*goobs.Client, func()) {
|
||||
t.Helper()
|
||||
client, err := connectObs(ObsConfig{
|
||||
Host: os.Getenv("OBS_HOST"),
|
||||
Port: 4455,
|
||||
Password: os.Getenv("OBS_PASSWORD"),
|
||||
Timeout: 5,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to connect to OBS: %v", err)
|
||||
}
|
||||
return client, func() {
|
||||
if err := client.Disconnect(); err != nil {
|
||||
t.Fatalf("Failed to disconnect from OBS: %v", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
client, err := connectObs(ObsConfig{
|
||||
Host: os.Getenv("OBS_HOST"),
|
||||
Port: 4455,
|
||||
Password: os.Getenv("OBS_PASSWORD"),
|
||||
Timeout: 5,
|
||||
})
|
||||
if err != nil {
|
||||
os.Exit(1)
|
||||
}
|
||||
defer client.Disconnect()
|
||||
|
||||
setup(client)
|
||||
|
||||
// Run the tests
|
||||
exitCode := m.Run()
|
||||
|
||||
teardown(client)
|
||||
|
||||
// Exit with the appropriate code
|
||||
os.Exit(exitCode)
|
||||
}
|
||||
|
||||
func setup(client *goobs.Client) {
|
||||
client.Config.SetStreamServiceSettings(config.NewSetStreamServiceSettingsParams().
|
||||
WithStreamServiceType("rtmp_common").
|
||||
WithStreamServiceSettings(&typedefs.StreamServiceSettings{
|
||||
Server: "auto",
|
||||
Key: os.Getenv("OBS_STREAM_KEY"),
|
||||
}))
|
||||
|
||||
client.Config.SetCurrentSceneCollection(config.NewSetCurrentSceneCollectionParams().
|
||||
WithSceneCollectionName("test-collection"))
|
||||
|
||||
client.Scenes.CreateScene(scenes.NewCreateSceneParams().
|
||||
WithSceneName("gobs-test"))
|
||||
client.Inputs.CreateInput(inputs.NewCreateInputParams().
|
||||
WithSceneName("gobs-test").
|
||||
WithInputName("gobs-test-input").
|
||||
WithInputKind("color_source_v3").
|
||||
WithInputSettings(map[string]any{
|
||||
"color": 3279460728,
|
||||
"width": 1920,
|
||||
"height": 1080,
|
||||
"visible": true,
|
||||
}).
|
||||
WithSceneItemEnabled(true))
|
||||
client.Inputs.CreateInput(inputs.NewCreateInputParams().
|
||||
WithSceneName("gobs-test").
|
||||
WithInputName("gobs-test-input-2").
|
||||
WithInputKind("color_source_v3").
|
||||
WithInputSettings(map[string]any{
|
||||
"color": 1789347616,
|
||||
"width": 720,
|
||||
"height": 480,
|
||||
"visible": true,
|
||||
}).
|
||||
WithSceneItemEnabled(true))
|
||||
}
|
||||
|
||||
func teardown(client *goobs.Client) {
|
||||
client.Scenes.RemoveScene(scenes.NewRemoveSceneParams().
|
||||
WithSceneName("gobs-test"))
|
||||
|
||||
client.Config.SetCurrentSceneCollection(config.NewSetCurrentSceneCollectionParams().
|
||||
WithSceneCollectionName("default"))
|
||||
|
||||
client.Stream.StopStream()
|
||||
client.Record.StopRecord()
|
||||
}
|
||||
40
profile.go
40
profile.go
@@ -9,18 +9,18 @@ import (
|
||||
|
||||
// ProfileCmd provides commands to manage profiles in OBS Studio.
|
||||
type ProfileCmd struct {
|
||||
List ListProfileCmd `help:"List profiles." cmd:"" aliases:"ls"`
|
||||
Current CurrentProfileCmd `help:"Get current profile." cmd:"" aliases:"c"`
|
||||
Switch SwitchProfileCmd `help:"Switch profile." cmd:"" aliases:"sw"`
|
||||
Create CreateProfileCmd `help:"Create profile." cmd:"" aliases:"new"`
|
||||
Remove RemoveProfileCmd `help:"Remove profile." cmd:"" aliases:"rm"`
|
||||
List ProfileListCmd `help:"List profiles." cmd:"" aliases:"ls"`
|
||||
Current ProfileCurrentCmd `help:"Get current profile." cmd:"" aliases:"c"`
|
||||
Switch ProfileSwitchCmd `help:"Switch profile." cmd:"" aliases:"sw"`
|
||||
Create ProfileCreateCmd `help:"Create profile." cmd:"" aliases:"new"`
|
||||
Remove ProfileRemoveCmd `help:"Remove profile." cmd:"" aliases:"rm"`
|
||||
}
|
||||
|
||||
// ListProfileCmd provides a command to list all profiles.
|
||||
type ListProfileCmd struct{} // size = 0x0
|
||||
// ProfileListCmd provides a command to list all profiles.
|
||||
type ProfileListCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to list all profiles.
|
||||
func (cmd *ListProfileCmd) Run(ctx *context) error {
|
||||
func (cmd *ProfileListCmd) Run(ctx *context) error {
|
||||
profiles, err := ctx.Client.Config.GetProfileList()
|
||||
if err != nil {
|
||||
return err
|
||||
@@ -33,11 +33,11 @@ func (cmd *ListProfileCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// CurrentProfileCmd provides a command to get the current profile.
|
||||
type CurrentProfileCmd struct{} // size = 0x0
|
||||
// ProfileCurrentCmd provides a command to get the current profile.
|
||||
type ProfileCurrentCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to get the current profile.
|
||||
func (cmd *CurrentProfileCmd) Run(ctx *context) error {
|
||||
func (cmd *ProfileCurrentCmd) Run(ctx *context) error {
|
||||
profiles, err := ctx.Client.Config.GetProfileList()
|
||||
if err != nil {
|
||||
return err
|
||||
@@ -47,13 +47,13 @@ func (cmd *CurrentProfileCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// SwitchProfileCmd provides a command to switch to a different profile.
|
||||
type SwitchProfileCmd struct {
|
||||
// ProfileSwitchCmd provides a command to switch to a different profile.
|
||||
type ProfileSwitchCmd struct {
|
||||
Name string `arg:"" help:"Name of the profile to switch to." required:""`
|
||||
}
|
||||
|
||||
// Run executes the command to switch to a different profile.
|
||||
func (cmd *SwitchProfileCmd) Run(ctx *context) error {
|
||||
func (cmd *ProfileSwitchCmd) Run(ctx *context) error {
|
||||
profiles, err := ctx.Client.Config.GetProfileList()
|
||||
if err != nil {
|
||||
return err
|
||||
@@ -74,13 +74,13 @@ func (cmd *SwitchProfileCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// CreateProfileCmd provides a command to create a new profile.
|
||||
type CreateProfileCmd struct {
|
||||
// ProfileCreateCmd provides a command to create a new profile.
|
||||
type ProfileCreateCmd struct {
|
||||
Name string `arg:"" help:"Name of the profile to create." required:""`
|
||||
}
|
||||
|
||||
// Run executes the command to create a new profile.
|
||||
func (cmd *CreateProfileCmd) Run(ctx *context) error {
|
||||
func (cmd *ProfileCreateCmd) Run(ctx *context) error {
|
||||
profiles, err := ctx.Client.Config.GetProfileList()
|
||||
if err != nil {
|
||||
return err
|
||||
@@ -100,13 +100,13 @@ func (cmd *CreateProfileCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// RemoveProfileCmd provides a command to remove an existing profile.
|
||||
type RemoveProfileCmd struct {
|
||||
// ProfileRemoveCmd provides a command to remove an existing profile.
|
||||
type ProfileRemoveCmd struct {
|
||||
Name string `arg:"" help:"Name of the profile to delete." required:""`
|
||||
}
|
||||
|
||||
// Run executes the command to remove an existing profile.
|
||||
func (cmd *RemoveProfileCmd) Run(ctx *context) error {
|
||||
func (cmd *ProfileRemoveCmd) Run(ctx *context) error {
|
||||
profiles, err := ctx.Client.Config.GetProfileList()
|
||||
if err != nil {
|
||||
return err
|
||||
|
||||
35
record.go
35
record.go
@@ -9,6 +9,7 @@ type RecordCmd struct {
|
||||
Start RecordStartCmd `cmd:"" help:"Start recording." aliases:"s"`
|
||||
Stop RecordStopCmd `cmd:"" help:"Stop recording." aliases:"st"`
|
||||
Toggle RecordToggleCmd `cmd:"" help:"Toggle recording." aliases:"tg"`
|
||||
Status RecordStatusCmd `cmd:"" help:"Show recording status." aliases:"ss"`
|
||||
Pause RecordPauseCmd `cmd:"" help:"Pause recording." aliases:"p"`
|
||||
Resume RecordResumeCmd `cmd:"" help:"Resume recording." aliases:"r"`
|
||||
}
|
||||
@@ -44,25 +45,39 @@ type RecordToggleCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to toggle recording.
|
||||
func (cmd *RecordToggleCmd) Run(ctx *context) error {
|
||||
// Check if recording is in progress
|
||||
status, err := ctx.Client.Record.ToggleRecord()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if status.OutputActive {
|
||||
fmt.Fprintln(ctx.Out, "Recording started successfully.")
|
||||
} else {
|
||||
fmt.Fprintln(ctx.Out, "Recording stopped successfully.")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// RecordStatusCmd shows the recording status.
|
||||
type RecordStatusCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to show recording status.
|
||||
func (cmd *RecordStatusCmd) Run(ctx *context) error {
|
||||
status, err := ctx.Client.Record.GetRecordStatus()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if status.OutputActive {
|
||||
_, err = ctx.Client.Record.StopRecord()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
fmt.Fprintln(ctx.Out, "Recording stopped successfully.")
|
||||
if status.OutputPaused {
|
||||
fmt.Fprintln(ctx.Out, "Recording is paused.")
|
||||
} else {
|
||||
_, err = ctx.Client.Record.StartRecord()
|
||||
if err != nil {
|
||||
return err
|
||||
fmt.Fprintln(ctx.Out, "Recording is in progress.")
|
||||
}
|
||||
fmt.Fprintln(ctx.Out, "Recording started successfully.")
|
||||
} else {
|
||||
fmt.Fprintln(ctx.Out, "Recording is not in progress.")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
105
record_test.go
Normal file
105
record_test.go
Normal file
@@ -0,0 +1,105 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestRecordStartStatusStop(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmdStart := &RecordStartCmd{}
|
||||
err := cmdStart.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start recording: %v", err)
|
||||
}
|
||||
if out.String() != "Recording started successfully.\n" {
|
||||
t.Fatalf("Expected output to be 'Recording started successfully.', got '%s'", out.String())
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
time.Sleep(1 * time.Second) // Wait for a second to ensure recording has started
|
||||
|
||||
cmdStatus := &RecordStatusCmd{}
|
||||
err = cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get recording status: %v", err)
|
||||
}
|
||||
if out.String() != "Recording is in progress.\n" {
|
||||
t.Fatalf("Expected output to be 'Recording is in progress.', got '%s'", out.String())
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdStop := &RecordStopCmd{}
|
||||
err = cmdStop.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to stop recording: %v", err)
|
||||
}
|
||||
if out.String() != "Recording stopped successfully.\n" {
|
||||
t.Fatalf("Expected output to be 'Recording stopped successfully.', got '%s'", out.String())
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
time.Sleep(1 * time.Second) // Wait for a second to ensure recording has stopped
|
||||
|
||||
cmdStatus = &RecordStatusCmd{}
|
||||
err = cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get recording status: %v", err)
|
||||
}
|
||||
if out.String() != "Recording is not in progress.\n" {
|
||||
t.Fatalf("Expected output to be 'Recording is not in progress.', got '%s'", out.String())
|
||||
}
|
||||
}
|
||||
|
||||
func TestRecordToggle(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmdStatus := &RecordStatusCmd{}
|
||||
err := cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get recording status: %v", err)
|
||||
}
|
||||
var active bool
|
||||
if out.String() == "Recording is in progress.\n" {
|
||||
active = true
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdToggle := &RecordToggleCmd{}
|
||||
err = cmdToggle.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to toggle recording: %v", err)
|
||||
}
|
||||
|
||||
time.Sleep(1 * time.Second) // Wait for a second to ensure toggle has taken effect
|
||||
|
||||
if active {
|
||||
if out.String() != "Recording stopped successfully.\n" {
|
||||
t.Fatalf("Expected output to be 'Recording stopped successfully.', got '%s'", out.String())
|
||||
}
|
||||
} else {
|
||||
if out.String() != "Recording started successfully.\n" {
|
||||
t.Fatalf("Expected output to be 'Recording started successfully.', got '%s'", out.String())
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -8,6 +8,7 @@ import (
|
||||
type ReplayBufferCmd struct {
|
||||
Start ReplayBufferStartCmd `help:"Start replay buffer." cmd:"" aliases:"s"`
|
||||
Stop ReplayBufferStopCmd `help:"Stop replay buffer." cmd:"" aliases:"st"`
|
||||
Toggle ReplayBufferToggleCmd `help:"Toggle replay buffer." cmd:"" aliases:"tg"`
|
||||
Status ReplayBufferStatusCmd `help:"Get replay buffer status." cmd:"" aliases:"ss"`
|
||||
Save ReplayBufferSaveCmd `help:"Save replay buffer." cmd:"" aliases:"sv"`
|
||||
}
|
||||
@@ -30,6 +31,24 @@ func (cmd *ReplayBufferStopCmd) Run(ctx *context) error {
|
||||
return err
|
||||
}
|
||||
|
||||
// ReplayBufferToggleCmd toggles the replay buffer state.
|
||||
type ReplayBufferToggleCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to toggle the replay buffer.
|
||||
func (cmd *ReplayBufferToggleCmd) Run(ctx *context) error {
|
||||
status, err := ctx.Client.Outputs.ToggleReplayBuffer()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if status.OutputActive {
|
||||
fmt.Fprintln(ctx.Out, "Replay buffer started successfully.")
|
||||
} else {
|
||||
fmt.Fprintln(ctx.Out, "Replay buffer stopped successfully.")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// ReplayBufferStatusCmd retrieves the status of the replay buffer.
|
||||
type ReplayBufferStatusCmd struct{} // size = 0x0
|
||||
|
||||
|
||||
58
scene_test.go
Normal file
58
scene_test.go
Normal file
@@ -0,0 +1,58 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestSceneList(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmd := &SceneListCmd{}
|
||||
err := cmd.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to list scenes: %v", err)
|
||||
}
|
||||
if !strings.Contains(out.String(), "gobs-test") {
|
||||
t.Fatalf("Expected output to contain 'gobs-test', got '%s'", out.String())
|
||||
}
|
||||
}
|
||||
|
||||
func TestSceneCurrent(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
// Set the current scene to "gobs-test"
|
||||
cmdSwitch := &SceneSwitchCmd{
|
||||
NewScene: "gobs-test",
|
||||
}
|
||||
err := cmdSwitch.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to switch to scene: %v", err)
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdCurrent := &SceneCurrentCmd{}
|
||||
err = cmdCurrent.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get current scene: %v", err)
|
||||
}
|
||||
if out.String() != "gobs-test\n" {
|
||||
t.Fatalf("Expected output to contain 'gobs-test', got '%s'", out.String())
|
||||
}
|
||||
}
|
||||
@@ -8,17 +8,17 @@ import (
|
||||
|
||||
// SceneCollectionCmd provides commands to manage scene collections in OBS Studio.
|
||||
type SceneCollectionCmd struct {
|
||||
List ListSceneCollectionCmd `help:"List scene collections." cmd:"" aliases:"ls"`
|
||||
Current CurrentSceneCollectionCmd `help:"Get current scene collection." cmd:"" aliases:"c"`
|
||||
Switch SwitchSceneCollectionCmd `help:"Switch scene collection." cmd:"" aliases:"sw"`
|
||||
Create CreateSceneCollectionCmd `help:"Create scene collection." cmd:"" aliases:"new"`
|
||||
List SceneCollectionListCmd `help:"List scene collections." cmd:"" aliases:"ls"`
|
||||
Current SceneCollectionCurrentCmd `help:"Get current scene collection." cmd:"" aliases:"c"`
|
||||
Switch SceneCollectionSwitchCmd `help:"Switch scene collection." cmd:"" aliases:"sw"`
|
||||
Create SceneCollectionCreateCmd `help:"Create scene collection." cmd:"" aliases:"new"`
|
||||
}
|
||||
|
||||
// ListSceneCollectionCmd provides a command to list all scene collections.
|
||||
type ListSceneCollectionCmd struct{} // size = 0x0
|
||||
// SceneCollectionListCmd provides a command to list all scene collections.
|
||||
type SceneCollectionListCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to list all scene collections.
|
||||
func (cmd *ListSceneCollectionCmd) Run(ctx *context) error {
|
||||
func (cmd *SceneCollectionListCmd) Run(ctx *context) error {
|
||||
collections, err := ctx.Client.Config.GetSceneCollectionList()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get scene collection list: %w", err)
|
||||
@@ -31,11 +31,11 @@ func (cmd *ListSceneCollectionCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// CurrentSceneCollectionCmd provides a command to get the current scene collection.
|
||||
type CurrentSceneCollectionCmd struct{} // size = 0x0
|
||||
// SceneCollectionCurrentCmd provides a command to get the current scene collection.
|
||||
type SceneCollectionCurrentCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to get the current scene collection.
|
||||
func (cmd *CurrentSceneCollectionCmd) Run(ctx *context) error {
|
||||
func (cmd *SceneCollectionCurrentCmd) Run(ctx *context) error {
|
||||
collections, err := ctx.Client.Config.GetSceneCollectionList()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get scene collection list: %w", err)
|
||||
@@ -45,13 +45,13 @@ func (cmd *CurrentSceneCollectionCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// SwitchSceneCollectionCmd provides a command to switch to a different scene collection.
|
||||
type SwitchSceneCollectionCmd struct {
|
||||
// SceneCollectionSwitchCmd provides a command to switch to a different scene collection.
|
||||
type SceneCollectionSwitchCmd struct {
|
||||
Name string `arg:"" help:"Name of the scene collection to switch to." required:""`
|
||||
}
|
||||
|
||||
// Run executes the command to switch to a different scene collection.
|
||||
func (cmd *SwitchSceneCollectionCmd) Run(ctx *context) error {
|
||||
func (cmd *SceneCollectionSwitchCmd) Run(ctx *context) error {
|
||||
collections, err := ctx.Client.Config.GetSceneCollectionList()
|
||||
if err != nil {
|
||||
return err
|
||||
@@ -74,13 +74,13 @@ func (cmd *SwitchSceneCollectionCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// CreateSceneCollectionCmd provides a command to create a new scene collection.
|
||||
type CreateSceneCollectionCmd struct {
|
||||
// SceneCollectionCreateCmd provides a command to create a new scene collection.
|
||||
type SceneCollectionCreateCmd struct {
|
||||
Name string `arg:"" help:"Name of the scene collection to create." required:""`
|
||||
}
|
||||
|
||||
// Run executes the command to create a new scene collection.
|
||||
func (cmd *CreateSceneCollectionCmd) Run(ctx *context) error {
|
||||
func (cmd *SceneCollectionCreateCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.Config.CreateSceneCollection(
|
||||
config.NewCreateSceneCollectionParams().WithSceneCollectionName(cmd.Name),
|
||||
)
|
||||
|
||||
32
sceneitem_test.go
Normal file
32
sceneitem_test.go
Normal file
@@ -0,0 +1,32 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestSceneItemList(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmd := &SceneItemListCmd{
|
||||
SceneName: "gobs-test",
|
||||
}
|
||||
err := cmd.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to list scene items: %v", err)
|
||||
}
|
||||
if !strings.Contains(out.String(), "gobs-test-input") {
|
||||
t.Fatalf("Expected output to contain 'gobs-test-input', got '%s'", out.String())
|
||||
}
|
||||
if !strings.Contains(out.String(), "gobs-test-input-2") {
|
||||
t.Fatalf("Expected output to contain 'gobs-test-input-2', got '%s'", out.String())
|
||||
}
|
||||
}
|
||||
40
stream.go
40
stream.go
@@ -17,10 +17,22 @@ type StreamStartCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to start streaming.
|
||||
func (cmd *StreamStartCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.Stream.StartStream()
|
||||
// Check if the stream is already active
|
||||
status, err := ctx.Client.Stream.GetStreamStatus()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if status.OutputActive {
|
||||
fmt.Fprintln(ctx.Out, "Stream is already active.")
|
||||
return nil
|
||||
}
|
||||
|
||||
_, err = ctx.Client.Stream.StartStream()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
fmt.Fprintln(ctx.Out, "Streaming started successfully.")
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -29,10 +41,22 @@ type StreamStopCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to stop streaming.
|
||||
func (cmd *StreamStopCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.Stream.StopStream()
|
||||
// Check if the stream is already inactive
|
||||
status, err := ctx.Client.Stream.GetStreamStatus()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !status.OutputActive {
|
||||
fmt.Fprintln(ctx.Out, "Stream is already inactive.")
|
||||
return nil
|
||||
}
|
||||
|
||||
_, err = ctx.Client.Stream.StopStream()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
fmt.Fprintln(ctx.Out, "Streaming stopped successfully.")
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -41,19 +65,15 @@ type StreamToggleCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to toggle streaming.
|
||||
func (cmd *StreamToggleCmd) Run(ctx *context) error {
|
||||
status, err := ctx.Client.Stream.GetStreamStatus()
|
||||
status, err := ctx.Client.Stream.ToggleStream()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if status.OutputActive {
|
||||
_, err = ctx.Client.Stream.StopStream()
|
||||
fmt.Fprintf(ctx.Out, "Stopping stream...\n")
|
||||
fmt.Fprintln(ctx.Out, "Streaming started successfully.")
|
||||
} else {
|
||||
_, err = ctx.Client.Stream.StartStream()
|
||||
fmt.Fprintf(ctx.Out, "Starting stream...\n")
|
||||
}
|
||||
if err != nil {
|
||||
return err
|
||||
fmt.Fprintln(ctx.Out, "Streaming stopped successfully.")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
131
stream_test.go
Normal file
131
stream_test.go
Normal file
@@ -0,0 +1,131 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestStreamStart(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmdStatus := &StreamStatusCmd{}
|
||||
err := cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get stream status: %v", err)
|
||||
}
|
||||
var active bool
|
||||
if strings.Contains(out.String(), "Output active: true") {
|
||||
active = true
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdStart := &StreamStartCmd{}
|
||||
err = cmdStart.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start stream: %v", err)
|
||||
}
|
||||
|
||||
time.Sleep(1 * time.Second) // Wait for the stream to start
|
||||
|
||||
if active {
|
||||
if out.String() != "Stream is already active.\n" {
|
||||
t.Fatalf("Expected 'Stream is already active.', got: %s", out.String())
|
||||
}
|
||||
} else {
|
||||
if out.String() != "Streaming started successfully.\n" {
|
||||
t.Fatalf("Expected 'Streaming started successfully.', got: %s", out.String())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestStreamStop(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmdStatus := &StreamStatusCmd{}
|
||||
err := cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get stream status: %v", err)
|
||||
}
|
||||
var active bool
|
||||
if strings.Contains(out.String(), "Output active: true") {
|
||||
active = true
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdStop := &StreamStopCmd{}
|
||||
err = cmdStop.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to stop stream: %v", err)
|
||||
}
|
||||
|
||||
time.Sleep(1 * time.Second) // Wait for the stream to stop
|
||||
|
||||
if active {
|
||||
if out.String() != "Streaming stopped successfully.\n" {
|
||||
t.Fatalf("Expected 'Streaming stopped successfully.', got: %s", out.String())
|
||||
}
|
||||
} else {
|
||||
if out.String() != "Stream is already inactive.\n" {
|
||||
t.Fatalf("Expected 'Stream is already inactive.', got: %s", out.String())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestStreamToggle(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmdStatus := &StreamStatusCmd{}
|
||||
err := cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to get stream status: %v", err)
|
||||
}
|
||||
var active bool
|
||||
if strings.Contains(out.String(), "Output active: true") {
|
||||
active = true
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdToggle := &StreamToggleCmd{}
|
||||
err = cmdToggle.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to toggle stream: %v", err)
|
||||
}
|
||||
|
||||
time.Sleep(1 * time.Second) // Wait for the stream to toggle
|
||||
|
||||
if active {
|
||||
if out.String() != "Streaming stopped successfully.\n" {
|
||||
t.Fatalf("Expected 'Streaming stopped successfully.', got: %s", out.String())
|
||||
}
|
||||
} else {
|
||||
if out.String() != "Streaming started successfully.\n" {
|
||||
t.Fatalf("Expected 'Streaming started successfully.', got: %s", out.String())
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -23,6 +23,8 @@ func (cmd *StudioModeEnableCmd) Run(ctx *context) error {
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to enable studio mode: %w", err)
|
||||
}
|
||||
|
||||
fmt.Fprintln(ctx.Out, "Studio mode is now enabled")
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -35,6 +37,8 @@ func (cmd *StudioModeDisableCmd) Run(ctx *context) error {
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to disable studio mode: %w", err)
|
||||
}
|
||||
|
||||
fmt.Fprintln(ctx.Out, "Studio mode is now disabled")
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
68
studiomode_test.go
Normal file
68
studiomode_test.go
Normal file
@@ -0,0 +1,68 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestStudioModeEnable(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmdEnable := &StudioModeEnableCmd{}
|
||||
err := cmdEnable.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to enable studio mode: %v", err)
|
||||
}
|
||||
if out.String() != "Studio mode is now enabled\n" {
|
||||
t.Fatalf("expected 'Studio mode is now enabled', got: %s", out.String())
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdStatus := &StudioModeStatusCmd{}
|
||||
err = cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to get studio mode status: %v", err)
|
||||
}
|
||||
if out.String() != "Studio mode is enabled\n" {
|
||||
t.Fatalf("expected 'Studio mode is enabled', got: %s", out.String())
|
||||
}
|
||||
}
|
||||
|
||||
func TestStudioModeDisable(t *testing.T) {
|
||||
client, disconnect := getClient(t)
|
||||
defer disconnect()
|
||||
|
||||
var out bytes.Buffer
|
||||
context := &context{
|
||||
Client: client,
|
||||
Out: &out,
|
||||
}
|
||||
|
||||
cmdDisable := &StudioModeDisableCmd{}
|
||||
err := cmdDisable.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to disable studio mode: %v", err)
|
||||
}
|
||||
if out.String() != "Studio mode is now disabled\n" {
|
||||
t.Fatalf("expected 'Studio mode is now disabled', got: %s", out.String())
|
||||
}
|
||||
// Reset output buffer for the next command
|
||||
out.Reset()
|
||||
|
||||
cmdStatus := &StudioModeStatusCmd{}
|
||||
err = cmdStatus.Run(context)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to get studio mode status: %v", err)
|
||||
}
|
||||
if out.String() != "Studio mode is disabled\n" {
|
||||
t.Fatalf("expected 'Studio mode is disabled', got: %s", out.String())
|
||||
}
|
||||
}
|
||||
@@ -6,17 +6,17 @@ import (
|
||||
|
||||
// VirtualCamCmd handles the virtual camera commands.
|
||||
type VirtualCamCmd struct {
|
||||
Start StartVirtualCamCmd `help:"Start virtual camera." cmd:"" aliases:"s"`
|
||||
Stop StopVirtualCamCmd `help:"Stop virtual camera." cmd:"" aliases:"st"`
|
||||
Toggle ToggleVirtualCamCmd `help:"Toggle virtual camera." cmd:"" aliases:"tg"`
|
||||
Status StatusVirtualCamCmd `help:"Get virtual camera status." cmd:"" aliases:"ss"`
|
||||
Start VirtualCamStartCmd `help:"Start virtual camera." cmd:"" aliases:"s"`
|
||||
Stop VirtualCamStopCmd `help:"Stop virtual camera." cmd:"" aliases:"st"`
|
||||
Toggle VirtualCamToggleCmd `help:"Toggle virtual camera." cmd:"" aliases:"tg"`
|
||||
Status VirtualCamStatusCmd `help:"Get virtual camera status." cmd:"" aliases:"ss"`
|
||||
}
|
||||
|
||||
// StartVirtualCamCmd starts the virtual camera.
|
||||
type StartVirtualCamCmd struct{} // size = 0x0
|
||||
// VirtualCamStartCmd starts the virtual camera.
|
||||
type VirtualCamStartCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to start the virtual camera.
|
||||
func (c *StartVirtualCamCmd) Run(ctx *context) error {
|
||||
func (c *VirtualCamStartCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.Outputs.StartVirtualCam()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to start virtual camera: %w", err)
|
||||
@@ -25,11 +25,11 @@ func (c *StartVirtualCamCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// StopVirtualCamCmd stops the virtual camera.
|
||||
type StopVirtualCamCmd struct{} // size = 0x0
|
||||
// VirtualCamStopCmd stops the virtual camera.
|
||||
type VirtualCamStopCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to stop the virtual camera.
|
||||
func (c *StopVirtualCamCmd) Run(ctx *context) error {
|
||||
func (c *VirtualCamStopCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.Outputs.StopVirtualCam()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to stop virtual camera: %w", err)
|
||||
@@ -38,11 +38,11 @@ func (c *StopVirtualCamCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// ToggleVirtualCamCmd toggles the virtual camera.
|
||||
type ToggleVirtualCamCmd struct{} // size = 0x0
|
||||
// VirtualCamToggleCmd toggles the virtual camera.
|
||||
type VirtualCamToggleCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to toggle the virtual camera.
|
||||
func (c *ToggleVirtualCamCmd) Run(ctx *context) error {
|
||||
func (c *VirtualCamToggleCmd) Run(ctx *context) error {
|
||||
_, err := ctx.Client.Outputs.ToggleVirtualCam()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to toggle virtual camera: %w", err)
|
||||
@@ -50,11 +50,11 @@ func (c *ToggleVirtualCamCmd) Run(ctx *context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// StatusVirtualCamCmd retrieves the status of the virtual camera.
|
||||
type StatusVirtualCamCmd struct{} // size = 0x0
|
||||
// VirtualCamStatusCmd retrieves the status of the virtual camera.
|
||||
type VirtualCamStatusCmd struct{} // size = 0x0
|
||||
|
||||
// Run executes the command to get the status of the virtual camera.
|
||||
func (c *StatusVirtualCamCmd) Run(ctx *context) error {
|
||||
func (c *VirtualCamStatusCmd) Run(ctx *context) error {
|
||||
status, err := ctx.Client.Outputs.GetVirtualCamStatus()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get virtual camera status: %w", err)
|
||||
|
||||
Reference in New Issue
Block a user