BililiveRecorder/BililiveRecorder.Core/RecordedRoom.cs

using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.ComponentModel;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Net;
using System.Text;
using BililiveRecorder.FlvProcessor;

namespace BililiveRecorder.Core
{
    public class RecordedRoom : INotifyPropertyChanged
    {
        public int roomID { get; private set; }
        public RoomInfo roomInfo { get; private set; }

        public RecordStatus Status;

        public StreamMonitor streamMonitor;
        public FlvStreamProcessor Processor; // FlvProcessor
        public readonly ObservableCollection<FlvClipProcessor> Clips = new ObservableCollection<FlvClipProcessor>();
        private HttpWebRequest webRequest;

        public RecordedRoom()
        {
            Processor.BlockProcessed += Processor_BlockProcessed;
            streamMonitor.StreamStatusChanged += StreamMonitor_StreamStatusChanged;

            UpdateRoomInfo();
        }

        private void StreamMonitor_StreamStatusChanged(object sender, StreamStatusChangedArgs e)
        {
            if (e.status.isStreaming)
            {
                // TODO: 失败重试逻辑 & 掉线重连逻辑
                _StartRecord();
            }
        }

        public void StartRecord()
        {
            throw new NotImplementedException();
        }

        private void _StartRecord()
        {
            // throw new NotImplementedException();
            if (webRequest != null)
            {
                //TODO: cleanup
                webRequest = null;
            }
            if (Processor != null)
            {
                //TODO: cleanup
                Processor = null;
            }


            string flv_path = BililiveAPI.GetPlayUrl(roomInfo.RealRoomid);

            webRequest = WebRequest.CreateHttp(flv_path);
            _SetupFlvRequest(webRequest);
            HttpWebResponse response = webRequest.GetResponse() as HttpWebResponse;
            if (response.StatusCode != HttpStatusCode.OK)
            {
                //TODO: Log
                response.Close();
                webRequest = null;
                return;
            }
            Stream flvStream = response.GetResponseStream();
            const int BUF_SIZE = 1024 * 8;// 8 KiB
            byte[] buffer = new byte[BUF_SIZE];

            AsyncCallback callback = null;

            callback = ar =>
            {
                try
                {
                    int bytesRead = flvStream.EndRead(ar);

                    if (bytesRead == 0)
                    {
                        // TODO: connection closed
                    }
                    else
                    {
                        if (bytesRead != buffer.Length)
                        {
                            Processor.AddBytes(buffer.Take(bytesRead).ToArray());
                        }
                        else
                        {
                            Processor.AddBytes(buffer);
                        }


                        Console.Write('#');

                        flvStream.BeginRead(buffer, 0, BUF_SIZE, callback, null);
                    }
                }
                catch (Exception e)
                {
                    throw; //TODO
                }
            };

            flvStream.BeginRead(buffer, 0, BUF_SIZE, callback, null);
        }

        private static void _SetupFlvRequest(HttpWebRequest r)
        {
            r.Accept = "*/*";
            r.AllowAutoRedirect = true;
            r.Connection = "keep-alive";
            r.Referer = "https://live.bilibili.com";
            r.Headers["Origin"] = "https://live.bilibili.com";
            r.UserAgent = "Mozilla/5.0 BililiveRecorder/0.0.0.0 (+https://github.com/Bililive/BililiveRecorder;bliverec@genteure.com)";
        }

        void StartWebRequest()
        {

            webRequest.BeginGetResponse(new AsyncCallback(FinishWebRequest), webRequest);
        }

        void FinishWebRequest(IAsyncResult result)
        {
            HttpWebResponse response = (result.AsyncState as HttpWebRequest).EndGetResponse(result) as HttpWebResponse;
        }

        public bool UpdateRoomInfo()
        {
            try
            {
                roomInfo = BililiveAPI.GetRoomInfo(roomID);
                return true;
            }
            catch (Exception ex)
            {
                Debug.WriteLine(ex.ToString());
                return false;
            }
        }

        // Called by API or GUI
        public void Clip()
        {
            var clip = Processor.Clip();
            // TODO: 多个线程同时运行，这个位置有可能会导致 Clip 丢数据
            // 考虑在此处加锁， Clip 操作时停止向主 Processor 添加数据
            clip.ClipFinalized += CallBack_ClipFinalized;
            Clips.Add(clip);
        }

        private void CallBack_ClipFinalized(object sender, ClipFinalizedArgs e)
        {
            if (Clips.Remove(e.ClipProcessor))
            {
                Debug.WriteLine("Clip Finalized");
            }
            else
            {
                Debug.WriteLine("Warning! Clip Finalized but was not in Collection.");
            }
        }

        private void Processor_BlockProcessed(object sender, BlockProcessedArgs e)
        {
            Clips.ToList().ForEach((fcp) => fcp.AddBlock(e.DataBlock));
        }


        public event PropertyChangedEventHandler PropertyChanged;
    }
}