geekbot/Geekbot.net/Lib/AudioClientCache.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Security.Cryptography;
using Discord.Audio;
using Discord.Net;

namespace Geekbot.net.Lib
{
    public class AudioUtils : IAudioUtils
    {
        private string _tempFolderPath;
        private Dictionary<ulong, IAudioClient> _audioClients;

        public AudioUtils()
        {
            _audioClients = new Dictionary<ulong, IAudioClient>();
            _tempFolderPath = Path.GetFullPath("./temp/");
            if (!Directory.Exists(_tempFolderPath))
            {
                Directory.CreateDirectory(_tempFolderPath);
            }
        }
        
        public IAudioClient GetAudioClient(ulong guildId)
        {
            return _audioClients[guildId];
        }

        public void StoreAudioClient(ulong guildId, IAudioClient client)
        {
            _audioClients[guildId] = client;
        }
        
        public Process CreateStreamFromFile(string path)
        {
            var ffmpeg = new ProcessStartInfo
            {
                FileName = "ffmpeg",
                Arguments = $"-i {path} -ac 2 -f s16le -ar 48000 pipe:1",
                UseShellExecute = false,
                RedirectStandardOutput = true,
            };
            return Process.Start(ffmpeg);
        }

        public Process CreateStreamFromYoutube(string url, ulong guildId)
        {
            var ytdlMediaUrl = GetYoutubeMediaUrl(url);
            DownloadMediaUrl(ytdlMediaUrl, guildId);
            return CreateStreamFromFile($"{_tempFolderPath}\\{guildId}.mp3");
        }

        private string GetYoutubeMediaUrl(string url)
        {
            var ytdl = new ProcessStartInfo()
            {
                FileName = "youtube-dl",
                Arguments = $"-f bestaudio -g {url}",
                UseShellExecute = false,
                RedirectStandardOutput = true
            };
            var output = Process.Start(ytdl).StandardOutput.ReadToEnd();
            if (string.IsNullOrWhiteSpace(output))
            {
                throw new System.Exception("Could not get Youtube Media URL");
            }
            return output;
        }

        private void DownloadMediaUrl(string url, ulong guildId)
        {
            var ffmpeg = new ProcessStartInfo
            {
                FileName = "ffmpeg",
                Arguments = $"-re -i \"${url}\" -c:a mp3 -b:a 256k {_tempFolderPath}\\{guildId}.mp3",
                UseShellExecute = false,
                RedirectStandardOutput = true,
            };
            Process.Start(ffmpeg).WaitForExit();
            return;
        }
    }

    public interface IAudioUtils
    {
        IAudioClient GetAudioClient(ulong guildId);
        void StoreAudioClient(ulong guildId, IAudioClient client);
        Process CreateStreamFromFile(string path);
        Process CreateStreamFromYoutube(string url, ulong guildId);
    }
}