feat: Add streaming to YouTube and Twitch

 This feature adds the ability to stream the screen to YouTube and Twitch. It also includes several improvements and fixes to the
  existing recording functionality.

  Key changes:
   - Implement streaming to YouTube and Twitch with hotkey support to stop.
   - Add a "Twitch (Test Stream)" option to allow users to test their connection.
   - Improve ffmpeg arguments for better streaming quality and performance.
   - Add a default audio device to the recording options.
   - Update the README with the new features and usage instructions.
   - Bump the version to v2.0.0.

Author: ServerDeveloper9447

main.go

@@ -28,6 +28,7 @@ var (
 		RecordingOpts: RecordingOptions{
 			FPS:          30,
 			CaptureMouse: true,
+			AudioDevice:  "Stereo Mix (Realtek(R) Audio)",
 		},
 		HotkeyConfig: hotkeyConfig{
 			Modkeys:  []string{"ctrl", "shift"},
@@ -81,7 +82,7 @@ type streamConfig struct {
 type Config struct {
 	SaveLocation  string           `json:"save_location"`
 	RecordFunc    bool             `json:"record_func_enabled"`
-	RecordingOpts RecordingOptions `json:"recording_options,omitempty"`
+	RecordingOpts RecordingOptions `json:"recording_options"`
 	HotkeyConfig  hotkeyConfig     `json:"hotkey_config"`
 	StreamConfig  streamConfig     `json:"stream_config"`
 }
@@ -257,7 +258,6 @@ func init() {
 	}
 	initConfig()
 	configMode, reset, hotkeyConfigMode := flag.Bool("config", false, "Configure Captr"), flag.Bool("reset", false, "Reset Captr and delete appdata"), flag.Bool("hotkey", false, "Register a hotkey for stopping recording")
-	flag.Parse()
 	if *configMode {
 		cmd := exec.Command("notepad.exe", configFilePath)
 		if err := cmd.Start(); err != nil {
@@ -315,7 +315,7 @@ ________/\\\\\\\\\__________________________________________________________
        ____\////\\\\\\\\\_\//\\\\\\\\/\\_\/\\\____________\//\\\\\___\/\\\_________ 
         _______\/////////___\////////\//__\///______________\/////____\///__________
 
-v1.0.2
+v2.0.0
 
 `)
 	fmt.Println("Open config file by passing the --config flag")

readme.md

@@ -4,13 +4,15 @@
 (Windows OS Only)
 
 > **Status:** Release  
-> Currently supports full-screen, window-specific screenshots and full-screen recording. Window specific recording is still in development.
+> Currently supports full-screen, window-specific screenshots and full-screen recording. Supports streaming to youtube and twitch. Window specific recording is partially implemented.
 
 ## Features
 
 - 📸 Capture full-screen screenshots  
 - 🖼️ Capture specific window screenshots  
 - 🎥 Screen recording
+- 🎥 Window specific recording (partially implemented)
+- 🎥 Stream display to twitch or youtube
 
 ## Installation
 **It's a portable executable file. Doesn't need any installation.**<br><br>
@@ -19,7 +21,8 @@
 ## Usage
 Download the exe from the releases or [build yourself](#build-yourself).<br>
 Run the exe to use it.<br>
-For recording functionaility, you must have `ffmpeg` installed and added to path. If you don't have it, the app has the prebuilt prompt to download ffmpeg to the `%appdata%\captr\bin` folder in appdata to use.
+For recording and streaming functionaility, you must have `ffmpeg` installed and added to path. If you don't have it, the app has the prebuilt prompt to download ffmpeg to the `%appdata%\captr\bin` folder in appdata to use.<br>
+**For audio, please  manually configure the config file or enable "Stereo Mix" device in windows settings.**
 
 ### Flags
 - `--config`: Opens the config file in the notepad for manual edits.
@@ -30,7 +33,8 @@ For recording functionaility, you must have `ffmpeg` installed and added to path
 - [x] Full-Screen Screenshots
 - [x] Window-Specific Screenshots
 - [x] Screen Recording
-- [ ] Window Recording [DELAYED]
+- [x] Stream to Youtube and Twitch
+- [x] Window Recording [Partial]
 
 ## Build Yourself
 ### Prerequisites

recwindow.go

@@ -2,7 +2,6 @@ package main
 
 import (
 	"fmt"
-	"os"
 	"os/exec"
 	"path/filepath"
 	"strconv"
@@ -35,8 +34,6 @@ func RecordWindow() {
 	}
 
 	cmd := exec.Command(getFfmpegPath(), args...)
-	cmd.Stderr = os.Stderr
-	cmd.Stdout = os.Stdout
 	stdin, _ := cmd.StdinPipe()
 	var start time.Time
 	var err error

streamdisplay.go

@@ -5,13 +5,17 @@ import (
 	"os"
 	"os/exec"
 	"strconv"
+	"strings"
+	"time"
 
 	"github.com/AlecAivazis/survey/v2"
+	"github.com/go-toast/toast"
 	"github.com/go-vgo/robotgo"
+	"golang.design/x/hotkey"
 )
 
 const (
-	YOUTUBE_RTMP = "rtmp://a.rtmp.youtube.com/live2"
+	YOUTUBE_RTMP = "rtmp://x.rtmp.youtube.com/live2"
 	TWITCH_RTMP  = "rtmp://ingest.global-contribute.live-video.net/app"
 )
 
@@ -35,13 +39,31 @@ func StreamDisp() {
 	var service string
 	err = survey.AskOne(&survey.Select{
 		Message: "Select Service",
-		Options: []string{"Youtube", "Twitch"},
+		Options: []string{"Youtube", "Twitch", "Twitch (Test Stream)"},
 	}, &service, survey.WithValidator(survey.Required))
 	if err != nil {
 		fmt.Println("Some error occurred")
 		return
 	}
 
+	var modkeys []hotkey.Modifier
+	pressedKeys := map[string]bool{}
+	for _, key := range config.HotkeyConfig.Modkeys {
+		pressedKeys[key] = true
+	}
+	for _, mod := range []string{"ctrl", "alt", "shift"} {
+		if pressedKeys[mod] {
+			switch mod {
+			case "ctrl":
+				modkeys = append(modkeys, hotkey.ModCtrl)
+			case "alt":
+				modkeys = append(modkeys, hotkey.ModAlt)
+			case "shift":
+				modkeys = append(modkeys, hotkey.ModShift)
+			}
+		}
+	}
+
 	switch service {
 	case "Youtube":
 		key := config.StreamConfig.YoutubeStreamKey
@@ -59,35 +81,74 @@ func StreamDisp() {
 			setConfig("stream_config", stream_config)
 		}
 		args := []string{
-			"-filter_complex", fmt.Sprintf("ddagrab=output_idx=%d:framerate=%d,hwdownload,format=bgra", display, config.RecordingOpts.FPS),
+			"-filter_complex", fmt.Sprintf("ddagrab=output_idx=%d:framerate=%d:draw_mouse=%d,hwdownload,format=bgra", display, config.RecordingOpts.FPS, ternary(config.RecordingOpts.CaptureMouse, 1, 0)),
 			"-f", "dshow",
-			"-i", "audio=\"Stereo Mix (Realtek(R) Audio)\"",
+			"-i", fmt.Sprintf("audio=%s", config.RecordingOpts.AudioDevice),
 			"-c:v", "libx264",
 			"-preset", "veryfast",
 			"-b:v", "3000k",
 			"-c:a", "aac",
 			"-f", "flv",
 			fmt.Sprintf("%s/%s", YOUTUBE_RTMP, key),
 		}
+		fmt.Println("./ffmpeg", strings.Join(args, " "))
 		cmd := exec.Command(getFfmpegPath(), args...)
-		cmd.Stdout = os.Stdout
-		cmd.Stderr = os.Stderr
-		if err := cmd.Start(); err != nil {
+		stdin, _ := cmd.StdinPipe()
+		var start time.Time
+		if err, start = cmd.Start(), time.Now(); err != nil {
 			fmt.Println("Cannot start ffmpeg")
 			os.Exit(0)
 		}
+		fmt.Printf("Streaming started on youtube. Press %s to stop\n", strings.Join(append(config.HotkeyConfig.Modkeys, config.HotkeyConfig.Finalkey), "+"))
+		tickStop := make(chan struct{})
+		ticker := time.NewTicker(time.Second)
+		defer ticker.Stop()
+		i := 0
+		last := []string{"🔴", "⚫"}
+		go func() {
+			for {
+				select {
+				case <-ticker.C:
+					fmt.Printf("\r%s Streaming time elapsed: %02d:%02d", last[i], int(time.Since(start).Minutes()), int(time.Since(start).Seconds())%60)
+					i = (i + 1) % 2
+				case <-tickStop:
+				}
+			}
+		}()
 		defer func() {
 			if r := recover(); r != nil {
 				fmt.Println("\nUnexpected error:", r)
 			}
 			exec.Command("taskkill", "/T", "/F", "/PID", strconv.Itoa(cmd.Process.Pid)).Run()
 		}()
-		err = cmd.Wait()
+		go func() {
+			err = cmd.Wait()
+			if err != nil {
+				fmt.Println("Error waiting for ffmpeg to exit:", err)
+			}
+		}()
+		hk := hotkey.New(modkeys, keys[config.HotkeyConfig.Finalkey])
+		err = hk.Register()
 		if err != nil {
-			fmt.Println("Error waiting for ffmpeg to exit:", err)
+			fmt.Println("Error registering hotkey:", err)
+			return
+		}
+		defer hk.Unregister()
+		keyChan := hk.Keydown()
+		for range keyChan {
+			tickStop <- struct{}{}
+			stdin.Write([]byte("q"))
+			stdin.Close()
+			fmt.Println("\nStopping streaming...")
+			notif := toast.Notification{
+				AppID: "Captr",
+				Title: "Streaming stopped",
+			}
+			notif.Push()
+			break
 		}
 		fmt.Println("\nStreaming stopped")
-	case "Twitch":
+	case "Twitch", "Twitch (Test Stream)":
 		key := config.StreamConfig.TwitchStreamKey
 		if key == "" {
 			fmt.Println("Stream key not set")
@@ -103,31 +164,71 @@ func StreamDisp() {
 			setConfig("stream_config", stream_config)
 		}
 		args := []string{
-			"-filter_complex", fmt.Sprintf("ddagrab=output_idx=%d:framerate=%d,hwdownload,format=bgra", display, config.RecordingOpts.FPS),
-			"-draw_mouse", ternary(config.RecordingOpts.CaptureMouse, "1", "0"),
+			"-filter_complex", fmt.Sprintf("ddagrab=output_idx=%d:framerate=%d:draw_mouse=%d,hwdownload,format=bgra,format=yuv420p", display, config.RecordingOpts.FPS, ternary(config.RecordingOpts.CaptureMouse, 1, 0)),
+			"-f", "dshow",
+			"-i", fmt.Sprintf("audio=%s", config.RecordingOpts.AudioDevice),
 			"-c:v", "libx264",
-			"-preset", "ultrafast",
-			"-c:a", "aac",
+			"-preset", "veryfast",
 			"-b:v", "3000k",
+			"-c:a", "aac",
+			"-pix_fmt", "yuv420p",
 			"-f", "flv",
-			fmt.Sprintf("%s/%s", TWITCH_RTMP, key),
+			fmt.Sprintf("%s/%s", TWITCH_RTMP, ternary(service == "Twitch (Test Stream)", fmt.Sprintf("%s?bandwidthtest=true", key), key)),
 		}
 		cmd := exec.Command(getFfmpegPath(), args...)
-		cmd.Stdout = os.Stdout
-		cmd.Stderr = os.Stderr
-		if err := cmd.Start(); err != nil {
+		stdin, _ :=  cmd.StdinPipe()
+		var start time.Time
+		if err, start = cmd.Start(), time.Now(); err != nil {
 			fmt.Println("Cannot start ffmpeg")
 			os.Exit(0)
 		}
+		fmt.Printf("Streaming started on twitch. Press %s to stop\n", strings.Join(append(config.HotkeyConfig.Modkeys, config.HotkeyConfig.Finalkey), "+"))
+		tickStop := make(chan struct{})
+		ticker := time.NewTicker(time.Second)
+		defer ticker.Stop()
+		i := 0
+		last := []string{"🔴", "⚫"}
+		go func() {
+			for {
+				select {
+				case <-ticker.C:
+					fmt.Printf("\r%s Streaming time elapsed: %02d:%02d", last[i], int(time.Since(start).Minutes()), int(time.Since(start).Seconds())%60)
+					i = (i + 1) % 2
+				case <-tickStop:
+				}
+			}
+		}()
 		defer func() {
 			if r := recover(); r != nil {
 				fmt.Println("\nUnexpected error:", r)
 			}
 			exec.Command("taskkill", "/T", "/F", "/PID", strconv.Itoa(cmd.Process.Pid)).Run()
 		}()
-		err = cmd.Wait()
+		go func() {
+			err = cmd.Wait()
+			if err != nil {
+				fmt.Println("Error waiting for ffmpeg to exit:", err)
+			}
+		}()
+		hk := hotkey.New(modkeys, keys[config.HotkeyConfig.Finalkey])
+		err = hk.Register()
 		if err != nil {
-			fmt.Println("Error waiting for ffmpeg to exit:", err)
+			fmt.Println("Error registering hotkey:", err)
+			return
+		}
+		defer hk.Unregister()
+		keyChan := hk.Keydown()
+		for range keyChan {
+			tickStop <- struct{}{}
+			stdin.Write([]byte("q"))
+			stdin.Close()
+			fmt.Println("\nStopping streaming...")
+			notif := toast.Notification{
+				AppID: "Captr",
+				Title: "Streaming stopped",
+			}
+			notif.Push()
+			break
 		}
 		fmt.Println("\nStreaming stopped")
 	}

utils.go

@@ -239,8 +239,8 @@ func chooseWindow() w32.HWND {
 	}, &result, survey.WithValidator(survey.Required))
 
 	if err != nil {
-		fmt.Printf("Prompt failed %v\n", err)
-		return w32.HWND(0)
+		fmt.Printf("Prompt failed")
+		os.Exit(0)
 	}
 
 	return windows[result]
@@ -446,11 +446,11 @@ func getFfmpegPath() string {
 	return filepath.Join(appdataDir, "bin", "ffmpeg.exe")
 }
 
-func ternary[T any](cond bool, a, b T) T {
+func ternary[T any](cond bool, trueval, falseval T) T {
 	if cond {
-		return a
+		return trueval
 	}
-	return b
+	return falseval
 }
 
 func RegisterHotkey() ([]string, string) {