diff --git a/extension-vsc/src/recorder/bin/windows/ffmpeg.exe b/extension-vsc/src/recorder/bin/windows/ffmpeg.exe new file mode 100755 index 0000000..f9208f2 Binary files /dev/null and b/extension-vsc/src/recorder/bin/windows/ffmpeg.exe differ diff --git a/extension-vsc/src/recorder/index.ts b/extension-vsc/src/recorder/index.ts index 4ef23ef..3bc0914 100644 --- a/extension-vsc/src/recorder/index.ts +++ b/extension-vsc/src/recorder/index.ts @@ -1,41 +1,47 @@ import * as os from 'os'; import MacOSRecorder from './macos'; import LinuxRecorder from './linux' +import WindowsRecorder from './windows' import {IRecorder} from '../types'; enum OS {OSX, macOS, Linux, Windows} function getCurrentOS() : OS | null { const platform = os.platform(); const release = os.release(); if (platform === 'darwin') { return release ? (Number(release.split('.')[0]) > 15 ? OS.macOS : OS.OSX) : OS.macOS; } else if (platform === 'linux') { return OS.Linux; + } else if (platform === 'win32') { + return OS.Windows; } return null; } function createRecorder() : IRecorder{ const currentOS = getCurrentOS(); switch (currentOS) { case (OS.macOS || OS.OSX): { return new MacOSRecorder(); } case (OS.Linux): { return new LinuxRecorder(); } + case (OS.Windows): { + return new WindowsRecorder(); + } } throw new Error(`Unsupported OS: ${os.platform()} ${os.release()}`); } const recorder = createRecorder(); export default recorder; diff --git a/extension-vsc/src/recorder/linux.ts b/extension-vsc/src/recorder/linux.ts index 796907a..0f10cb5 100644 --- a/extension-vsc/src/recorder/linux.ts +++ b/extension-vsc/src/recorder/linux.ts @@ -1,180 +1,180 @@ -import {InputDevices, InputDevice, InputScreenDeviceLinux, InputDeviceType, IRecorder, RecorderOptions} from '../types'; +import {InputDevices, InputDevice, InputScreenDevice, InputDeviceType, IRecorder, RecorderOptions} from '../types'; import * as tempy from 'tempy'; import {spawn, exec as callbackExec} from 'child_process' import config from '../config' import * as vscode from 'vscode'; import * as path from 'path' import {promisify} from 'util'; const exec = promisify(callbackExec); const ffmpegPath = path.join(__dirname, './bin/linux/ffmpeg'); const execFfmpegCommand = (args) => { return exec(`${ffmpegPath} ${args}`) .catch(({stderr}) => { return stderr }) }; class LinuxRecorder implements IRecorder { options: RecorderOptions; ffmpegChild: any; private recording: boolean; - private selectedScreenDevice: InputScreenDeviceLinux; + private selectedScreenDevice: InputScreenDevice; private selectedAudioDeviceIndex = 0; constructor(options?: RecorderOptions) { if (options) { this.options = options; } } - private static async getScreens(): Promise { + private static async getScreens(): Promise { const {stdout} = await exec('xrandr --listmonitors'); const regExp = /(\d+):.* (\d+)\/\d+x(\d+).*\+(\d+)\+(\d+)/g; if (!regExp.flags.includes('g')) { console.error('Invalid RegExp, will cause infinite loop.'); return [] } const screens = []; let match; while ((match = regExp.exec(stdout))) { const [, index, resX, resY, offsetX, offsetY] = match; screens.push({ index: Number(index), type: InputDeviceType.Screen, resolution: { x: Number(resX), y: Number(resY), }, offset: { x: Number(offsetX), y: Number(offsetY), } }) } return screens.map((screen, i) => ({ ...screen, name: (i === 0 ? 'Main Screen' : `Screen ${i+1}`) + ` (${screen.resolution.x}x${screen.resolution.y})` })) } private static async getMicrophones(): Promise { return []; } async getDevices(): Promise { return { screen: await LinuxRecorder.getScreens(), microphone: await LinuxRecorder.getMicrophones(), } } setInputDevice(device) { if (device.type === InputDeviceType.Screen) { this.selectedScreenDevice = device; } else if (device.type === InputDeviceType.Microphone) { this.selectedAudioDeviceIndex = device.index; } } async start(options: RecorderOptions) { const destinationPath = tempy.file({extension: 'mkv'}); const isLowQuality = vscode.workspace.getConfiguration().get('gitduck.video-quality') === 'low'; const {resolution, offset} = this.selectedScreenDevice; let args = [ '-hide_banner -loglevel info -f x11grab -video_size', `${resolution.x}x${resolution.y} -i ":0.0+${offset.x},${offset.y}" -f alsa`, `-ac 2 -i "default" -c:a aac -c:v libx264 -preset ultrafast -threads 0`, '-vf "scale=-2:1080,format=yuv420p" -tune zerolatency', '-framerate 20 -g 40 -f flv -pix_fmt yuv420p -b:a 128k -r 20', `rtmp://${config.liveHost}/app/${options.streamKey}` ].join(' '); if (isLowQuality) { // -f alsa -ac 2 -i "default" args = [ `-hide_banner -loglevel info -f x11grab -video_size ${resolution.x}x${resolution.y} -i ":0.0+${offset.x},${offset.y}"`, `-c:a aac -c:v libx264 -threads 0`, '-x264-params "nal-hrd=cbr" -preset ultrafast', '-vf "scale=-2:720,format=yuv420p" -tune zerolatency -framerate 30', '-g 60 -b:a 128k -pix_fmt yuv420p -crf 30 -b:v 600k', `-r 15 -f flv`, `rtmp://${config.liveHost}/app/${options.streamKey}` ].join(' ') } console.log(`Running ${ffmpegPath} ${args}`); this.ffmpegChild = spawn( ffmpegPath, args.split(' '), {shell: true, detached: true} ); this.recording = true; this.ffmpegChild.unref(); console.log('Deataching and unreffing ffmpeg process....'); this.ffmpegChild.on('close', (code, signal) => { this.recording = false; console.debug('FFMPEG Close event. GOT A CODE', code, '+SIGNAL:', signal); }); this.ffmpegChild.on('disconnect', () => { console.debug('FFMPEG Disconnect event'); }); this.ffmpegChild.on('error', err => { console.debug('FFMPEG Error event. GOT AN ERROR', err); if (options.onError) { options.onError() } throw new Error('Error in ffmpeg') }); this.ffmpegChild.on('exit', (code, signal) => { this.recording = false; console.debug('FFMPEG Exit event. GOT A CODE', code, '\n+SIGNAL:', signal); if ((code > 0 && code !== 255) || signal === 'SIGABRT') { if (options.onError) { options.onError() } throw new Error('Error in ffmpeg') } }); this.ffmpegChild.stdout.on('data', data => { console.debug('[FFMPEG STDOUT]:', data.toString()); }); this.ffmpegChild.stderr.on('data', data => { console.debug('[FFMPEG STDERR]:', data.toString()); }); return null; } async pause() { } isRecording() { return !!this.recording; } async stop() { this.recording = false; return this.ffmpegChild.kill(); } } export default LinuxRecorder; diff --git a/extension-vsc/src/recorder/windows.ts b/extension-vsc/src/recorder/windows.ts new file mode 100644 index 0000000..1d655e5 --- /dev/null +++ b/extension-vsc/src/recorder/windows.ts @@ -0,0 +1,193 @@ +import {InputDevices, InputDeviceType, InputScreenDevice, InputDevice, IRecorder, RecorderOptions} from '../types'; +import * as tempy from 'tempy'; +import {spawn, exec as callbackExec} from 'child_process' +import config from '../config' +import * as vscode from 'vscode'; +import * as path from 'path' +import {promisify} from 'util'; +import * as systemInfo from 'systeminformation'; + +const exec = promisify(callbackExec); + +const ffmpegPath = path.join(__dirname, './bin/windows/ffmpeg'); + +let execFfmpegCommand = (args) => { + return exec(`${ffmpegPath} ${args}`) + .catch(({stderr}) => { + return stderr + }) +}; + +class WindowsRecorder implements IRecorder { + options?: RecorderOptions; + ffmpegChild: any; + private recording: boolean; + + private selectedVideoDevice: InputScreenDevice; + private selectedAudioDevice: InputDevice; + + constructor(options?: RecorderOptions) { + if (options) { + this.options = options; + } + } + + static async getScreens(): Promise { + const {displays} = await systemInfo.graphics(); + + return displays.map((display, i) => { + return { + index: i, + name: display.main ? 'Main screen' : 'Screen ' + i, + type: InputDeviceType.Screen, + resolution: { + x: Number(display.resolutionx), + y: Number(display.resolutiony), + }, + offset: { + x: Number(display.positionX), + y: Number(display.positionY), + } + } + }); + } + + static async getMicrophones(): Promise { + const ffmpegOutput = await execFfmpegCommand('-hide_banner -list_devices true -f dshow -i dummy'); + const audioDevicesIndex = ffmpegOutput.indexOf('DirectShow audio devices'); + const audioDevicesOutput = ffmpegOutput.substring(audioDevicesIndex); + + const regExp = /"(.*)"\r\n.*Alternative name "(.*)"/g; + if (!regExp.flags.includes('g')) { + console.error('Invalid RegExp, will cause infinite loop.'); + return [] + } + + const microphones = []; + let match; + let i = 0; + while ((match = regExp.exec(audioDevicesOutput))) { + const [, name, id] = match; + microphones.push({ + index: i, + type: InputDeviceType.Microphone, + id: id, + name: name, + }); + ++i; + } + + return microphones; + } + + async getDevices(): Promise { + const screen = WindowsRecorder.getScreens(); + const microphone = WindowsRecorder.getMicrophones(); + return { + screen: await screen, + microphone: await microphone, + } + } + + setInputDevice(device) { + if (device.type === InputDeviceType.Screen) { + this.selectedVideoDevice = device; + } else if (device.type === InputDeviceType.Microphone) { + this.selectedAudioDevice = device; + } + } + + async start(options: RecorderOptions) { + const lowQuality = vscode.workspace.getConfiguration().get('gitduck.video-quality') === 'low'; + + const {resolution, offset} = this.selectedVideoDevice; + const resolutionStr = resolution.x + 'x' + resolution.y; + let args = [ + '-hide_banner -loglevel info -f gdigrab', + `-video_size ${resolutionStr} -offset_x ${offset.x} -offset_y ${offset.y} -i desktop`, + `-f dshow -ac 2 -i audio="${this.selectedAudioDevice.id}" -c:a aac -c:v libx264 -preset ultrafast -threads 0`, + '-vf "scale=-2:1080,format=yuv420p" -tune zerolatency', + '-framerate 20 -g 40 -f flv -pix_fmt yuv420p -b:a 128k -r 20', + `rtmp://${config.liveHost}/app/${options.streamKey}` + ].join(' '); + + if (lowQuality) { + args = [ + `-hide_banner -loglevel info -f gdigrab`, + `-video_size ${resolutionStr} -offset_x ${offset.x} -offset_y ${offset.y} -i desktop`, + `-f dshow -ac 2 -i audio="${this.selectedAudioDevice.id}" -c:a aac -c:v libx264 -threads 0`, + '-x264-params "nal-hrd=cbr" -preset ultrafast', + '-vf "scale=-2:720,format=yuv420p" -tune zerolatency -framerate 30', + '-g 60 -b:a 128k -pix_fmt yuv420p -crf 30 -b:v 600k', + `-r 15 -f flv`, + `rtmp://${config.liveHost}/app/${options.streamKey}` + ].join(' ') + } + + console.log(`Running ${ffmpegPath} ${args}`); + + this.ffmpegChild = spawn( + ffmpegPath, + args.split(' '), + {shell: true, detached: false, windowsHide: true} + ); + this.recording = true; + this.ffmpegChild.unref(); + + console.log('Deataching and unreffing ffmpeg process....'); + + this.ffmpegChild.on('close', (code, signal) => { + this.recording = false; + console.debug('FFMPEG Close event. GOT A CODE', code, '+SIGNAL:', signal); + }); + + this.ffmpegChild.on('disconnect', () => { + console.debug('FFMPEG Disconnect event'); + }); + + this.ffmpegChild.on('error', err => { + console.debug('FFMPEG Error event. GOT AN ERROR', err); + if (options.onError) { + options.onError() + } + throw new Error('Error in ffmpeg') + }); + + this.ffmpegChild.on('exit', (code, signal) => { + this.recording = false; + console.debug('FFMPEG Exit event. GOT A CODE', code, '\n+SIGNAL:', signal); + if ((code > 1 && code !== 255) || signal === 'SIGABRT') { + if (options.onError) { + options.onError() + } + throw new Error('Error in ffmpeg') + } + }); + + this.ffmpegChild.stdout.on('data', data => { + console.debug('[FFMPEG STDOUT]:', data.toString()); + }); + + this.ffmpegChild.stderr.on('data', data => { + console.debug('[FFMPEG STDERR]:', data.toString()); + }); + + + return null; + } + + async pause() { + + } + + isRecording() { + return !!this.recording; + } + + async stop() { + this.recording = false; + await exec('taskkill ' + ['/pid', this.ffmpegChild.pid, '/f', '/t'].join(' ')); + } +} + +export default WindowsRecorder; diff --git a/extension-vsc/src/types.ts b/extension-vsc/src/types.ts index 109c98b..5ebd4dd 100644 --- a/extension-vsc/src/types.ts +++ b/extension-vsc/src/types.ts @@ -1,57 +1,58 @@ export type Commit = { authorEmail: string, authorName: string, body: string, date: Date, videoTimestamp: number, hash: string, message: string, refs: string, rawCommit: string, rawCommitFull: string, } export interface RecorderOptions { fps?: number; streamKey: string; onError?: Function; } export enum InputDeviceType { Microphone = 'microphone', Screen = 'screen' } export type InputDevice = { index: number, + id?: string, name: string, type: InputDeviceType, } -export type InputScreenDeviceLinux = { +export type InputScreenDevice = { index: number, name: string, type: InputDeviceType, resolution: { x: number, y: number, }, offset: { x: number, y: number, }, } export type InputDevices = { - screen: (InputDevice | InputScreenDeviceLinux)[], + screen: (InputDevice | InputScreenDevice)[], microphone: InputDevice[], } export interface IRecorder { options?: RecorderOptions; start(options: RecorderOptions): Promise; pause(): Promise; stop(): Promise; setInputDevice(device: InputDevice): void; getDevices(): Promise isRecording(): boolean; }

GitDuck for Windows (3)

More coding videos

Fixing a small bug from GetStream.io React library

Uploading code-snippets to GitDuck #2

Adding front-camera and possibility to turn on/off cam and/or mic

Small change on indexing

Making it clearer when the extension is recording

Coding session on October 9th