Comments and formatting
[streamdesktop.git] / streamdesktop
1 #!/bin/bash
2
3 set -e
4
5 if [ "$BASH_VERSION" = "" ];then
6 echo "This sctipt must be run with Bash, try 'bash $0'";
7 exit 1
8 fi
9
10 # Copyright (C) 2020 Ruben Rodriguez <ruben@gnu.org>
11 #
12 # This program is free software; you can redistribute it and/or modify
13 # it under the terms of the GNU General Public License as published by
14 # the Free Software Foundation; either version 2 of the License, or
15 # (at your option) any later version.
16 #
17 # This program is distributed in the hope that it will be useful,
18 # but WITHOUT ANY WARRANTY; without even the implied warranty of
19 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 # GNU General Public License for more details.
21 #
22 # You should have received a copy of the GNU General Public License
23 # along with this program; if not, write to the Free Software
24 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
25
26 # A simple script to capture your desktop and microphone into a file and an icecast2 streaming server
27
28 # Streamer side dependencies: wmctrl gstreamer1.0-tools gstreamer1.0-plugins-good gstreamer1.0-pulseaudio pulseaudio ffmpeg
29 # sudo apt-get install wmctrl gstreamer1.0-tools gstreamer1.0-plugins-good gstreamer1.0-pulseaudio pulseaudio ffmpeg
30
31 # Server side dependencies: icecast2
32
33 # Server connection settings
34 SERVER=my.icecast2server.org
35 PORT=80 # port to stream to
36 PASSWORD=hackme
37
38 # Stream settings
39 BITRATE=1000000 # Bitrate of the video, in bps
40 SOUNDQUALITY=0.4 # OGG compression quality, from 0 to 1
41 FRAMERATE=20 # frames per second
42 SHOWPOINTER="true" # show or hide the mouse pointer
43 # Width and height are upper limits. Video will not be upscaled or stretched, it would be proportionally downscaled to fit the WIDTH and HEIGHT dimensions
44 WIDTH=1280
45 HEIGHT=720
46
47 DATE=$(date +%Y-%m-%d-%H_%M_%S)
48 MOUNT=$1
49 [ "$MOUNT" = "" ] && MOUNT=recording
50
51 cat << EOF
52 Usage: $0 [title]
53 The title parameter will be used for the name of the mount, and the name of the file saved:
54 http://$SERVER/$MOUNT.webm
55 File saved locally to ./$MOUNT-${DATE}.webm
56 Log saved to $MOUNT-${DATE}.log
57
58 AUDIO INPUT: this script will record and stream the audio from the default input selection in Pulseaudio. Click on the volume icon in your desktop environment, go to settings, then input, and select the appropriate device and input level.
59
60 SCREEN SELECTION. Note: broadcasting a single window may crash if you minimize or resize the screen after the streaming starts! If you have issues with this, select 0 to broadcast the full screen, but remember to close or hide any privacy-sensitive windows." If the stream is just an empty black square, make sure your desktop is running X11, not Wayland or others.
61
62 0: Capture FULL SCREEN
63 EOF
64
65 # Window selector
66 WINDOWLIST="$(wmctrl -l)"
67 i=1
68 echo "$WINDOWLIST" |while read line
69 do
70 echo -n "$i: "
71 name=$(echo $line| cut -d' ' -f 4-)
72 echo $name
73 i=$(expr $i + 1)
74 done
75
76 echo
77 echo "Select the window to capture, or press enter to capture the whole screen"
78 read -s -n1 SELECTION
79
80 XID=$(echo "$WINDOWLIST" | sed -n "${SELECTION}p" | sed 's/ .*//')
81
82 if [ "$SELECTION" = 0 ] || [ "$SELECTION" = "" ] ; then
83 XPARAM=""
84 else
85 XPARAM="xid=$XID"
86 fi
87
88
89 # The video needs to be processed after recording
90 trap ctrl_c INT
91 ctrl_c() {
92 echo
93 echo -n "Remuxing streamed video... "
94 mv $MOUNT-${DATE}.webm $MOUNT-${DATE}.webm.bak
95 ffmpeg -i $MOUNT-${DATE}.webm.bak -acodec copy -vcodec copy $MOUNT-${DATE}.webm >> $MOUNT-${DATE}.log 2>&1
96 echo done.
97 echo
98 echo Use $MOUNT-${DATE}.webm for publishing, you can delete $MOUNT-${DATE}.webm.bak
99 echo
100 echo "You can trim the beginning and end of the video with this command (replace the timestamps, HH:MM:SS):"
101 echo ffmpeg -i $MOUNT-${DATE}.webm -acodec copy -vcodec copy -ss 00:19:27 -to 02:18:51 trimmed.webm
102 }
103
104
105 echo STARTING STREAM in 2 seconds. Press CTRL+C to end the session.
106 sleep 2
107 echo Streaming at http://$SERVER/recording.webm
108 echo
109
110 # GSTREAMER 1.0 pipeline for capture, save to file, and stream to icecast2
111 gst-launch-1.0 --eos-on-shutdown ximagesrc use-damage=false $XPARAM show-pointer=$SHOWPOINTER \
112 ! videorate ! video/x-raw,framerate=$FRAMERATE/1 \
113 ! videoscale ! video/x-raw, width=$WIDTH, height=$HEIGHT \
114 ! videoconvert ! queue ! vp8enc target_bitrate=$BITRATE cpu-used=0 deadline=1 threads=4 keyframe-max-dist=15 ! queue ! mux. \
115 pulsesrc ! vorbisenc quality=$SOUNDQUALITY ! queue ! mux. \
116 webmmux streamable=true name=mux ! tee name=stream ! queue ! filesink location=$MOUNT-${DATE}.webm sync=false \
117 stream. ! queue max-size-time=0 max-size-bytes=135528448 max-size-buffers=0 ! shout2send ip=$SERVER port=$PORT mount=$MOUNT.webm password=$PASSWORD sync=false \
118 2>&1 | tee $MOUNT-${DATE}.log
119
120