MARKUS

using System;

using System.Collections.Generic;

using System.ComponentModel;

using System.Linq;

using System.Text;

using System.Threading.Tasks;

using System.Windows.Media.Imaging;

namespace KCOM.PageManager

{

    public class PageStorage

    {

        private const int DEFUALT_TALK_PAGE_COUNT = 10;

        bool IsRunWork = false;

        BackgroundWorker backgroundWorker;

        List<PageItem> fileItems = new List<PageItem>();

        string _localStorage;

        string _fileExt;

        string _BaseUri;

        int _TotalPages;

        int _TakeCount;

        BitmapFrame PageImage;

        public PageStorage(string BaseUri, string localStoragePath,string fileExt,int totalPages,int takeCount = 5)

        {

            try

            {

                backgroundWorker = new BackgroundWorker {WorkerSupportsCancellation = true };

                backgroundWorker.DoWork += BackgroundWorker_DoWork;

                _fileExt = fileExt;

                _BaseUri = BaseUri;

                _localStorage = localStoragePath;

                _TotalPages = totalPages;

                _TakeCount = takeCount;

                System.IO.Directory.CreateDirectory(_localStorage);

                //backgroundWorker.RunWorkerAsync(new int[] { 1, 10 });

            }

            catch (Exception ex)

            {

                throw new Exception("PageStorage", ex);

            }

        }

        private void BackgroundWorker_DoWork(object sender, DoWorkEventArgs e)

        {

            int StartPageNo = -1;

            int TalkCount = -1;

            if (e.Argument != null)

            {

                var values = (e.Argument as int[]);

                if (values.Count() == 2)

                {

                    StartPageNo = values[0];

                    TalkCount = values[1];

                    DownloadPagesAsync( StartPageNo, TalkCount);

                }

            }

        }

        public void ResetImage()

        {

            //if (PageImage != null)

            //{

            //    PageImage = null;

            //}

            //PageImage.CreateOptions = BitmapCreateOptions.IgnoreImageCache;

        }

        public async Task<BitmapFrame> GetPageImageAsync(System.Threading.CancellationToken cts,int PageNo,bool IsRestart =false)

        {

            try

            {

                var localUri = await GetPageUriAsync(cts,PageNo);

                if (localUri != null || !cts.IsCancellationRequested)

                {

                    PageImage = BitmapFrame.Create(localUri, BitmapCreateOptions.None, BitmapCacheOption.OnLoad);

                }

            }

            catch (Exception ex)

            {

                throw new Exception("GetPageImageAsync(int PageNo,bool IsRestart =false)", ex);

            }

            finally

            {

            }

            return PageImage;

        }

        public async Task<Uri> GetPageUriAsync( System.Threading.CancellationToken cts, int PageNo, bool IsRestart = false)

        {

            Uri result = null;

            try

            {

                System.Diagnostics.Debug.WriteLine("GetPageAsync");

                var pageItem = await DownloadPageAsync(PageNo,cts);

                if (!IsRunWork && pageItem != null)

                {

                    var takePageNoList = Enumerable.Range(PageNo + 1, _TakeCount);

                    if (takePageNoList.Any(x => fileItems.Count(y => y.PageNo == x) == 0))

                    {

                        while (backgroundWorker.IsBusy)

                        {

                            if (backgroundWorker.IsBusy)

                            {

                                backgroundWorker.CancelAsync();

                            }

                            await Task.Delay(100);

                        }

                        backgroundWorker.RunWorkerAsync(new int[] { PageNo + 1, _TakeCount });

                        if (_TotalPages == _TakeCount)

                        {

                            IsRunWork = true;

                        }

                    }

                }

                result = pageItem.LocalUri;

            }

            catch (Exception ex)

            {

                if (!IsRestart || !cts.IsCancellationRequested)

                {

                    await Task.Delay(100);

                    result = await GetPageUriAsync(cts,PageNo, true);

                }

                //throw new Exception("GetPageAsync(string BasePageUri,int PageNo)", ex);

            }

            finally

            {

            }

            return result;

        }

        public async void DownloadPagesAsync(int StartPageNo, int TalkCount = 5)

        {

            try

            {

                if (StartPageNo + TalkCount > _TotalPages)

                {

                    TalkCount = _TotalPages - StartPageNo;

                }

                if (TalkCount > 0)

                {

                    for (int i = StartPageNo; i < StartPageNo + TalkCount; i++)

                    {

                        try

                        {

                            await DownloadPageAsync(i,null);

                            System.Threading.Thread.Sleep(200);

                        }

                        catch (Exception ex)

                        {

                            System.Diagnostics.Debug.WriteLine("DownloadPagesAsync err", ex);

                        }

                        finally

                        {

                        }

                    }

                }

            }

            catch (Exception ex)

            {

                throw new Exception("DownloadPagesAsync : ", ex);

            }

            finally

            {

                GC.Collect(2);

                GC.Collect(2);

            }

        }

        public async Task<PageItem> DownloadPageAsync(int PageNo,System.Threading.CancellationToken? cts)

        {

            PageItem result = new PageItem { PageNo = PageNo };

            try

            {

                var page = fileItems.Where(x => x.PageNo == PageNo);

                if (page.Count() > 0)

                {

                    System.Diagnostics.Debug.WriteLine("DownloadPageAsync fileItems");

                    PageItem item = page.First();

                    /// 파일 체크 후 없으면 다시 다운로드

                    if (System.IO.File.Exists(item.LocalFilePath))

                    {

                        result = page.First();

                    }

                    else

                    {

                        fileItems.Remove(item);

                        result = await DownloadPageAsync(PageNo,cts);

                    }

                }

                else

                {

                    System.Diagnostics.Debug.WriteLine("DownloadPageAsync down");

                    string downloadFilePath = System.IO.Path.Combine(_localStorage, PageNo.ToString() + "." + _fileExt);

                    Uri originalUri = new Uri(_BaseUri.Replace("{PageNo}", PageNo.ToString()));

                    using (System.Net.WebClient client = new System.Net.WebClient())

                    {

                        client.UseDefaultCredentials = true;

                        System.Net.IWebProxy webProxy = client.Proxy;

                        if (webProxy != null)

                        {

                            // Use the default credentials of the logged on user.

                            webProxy.Credentials = System.Net.CredentialCache.DefaultCredentials;

                        }

                        //client.DownloadFileCompleted += (snd, evt) =>

                        //{

                        //    //(snd as System.Net.WebClient).Dispose();

                        //};

                        if(cts != null)

                            client.DownloadProgressChanged += (snd, ect) =>

                            {

                                if(cts.Value.IsCancellationRequested)

                                {

                                    client.CancelAsync();

                                }

                            };

                        System.Diagnostics.Debug.WriteLine("Download : " + downloadFilePath);

                        await client.DownloadFileTaskAsync(originalUri, downloadFilePath);

                    }

                    result = new PageItem

                    {

                        PageNo = PageNo,

                        OriginalUri = originalUri,

                        LocalUri = new Uri(downloadFilePath, UriKind.Absolute),

                        LocalFilePath = downloadFilePath

                    };

                    lock (fileItems)

                    {

                        fileItems.Add(result);

                    }

                    //System.Diagnostics.Debug.WriteLine("Download : " + result.LocalFilePath);

                }

            }

            catch (Exception ex)

            {

                throw new Exception("DownloadPageAsync : ", ex);

            }

            finally

            {

            }

            return result;

        }

        public void Clear()

        {

            try

            {

                ResetImage();

            }

            catch (Exception ex)

            {

                System.Diagnostics.Debug.WriteLine(ex.ToString());

            }

            try

            {

                backgroundWorker.CancelAsync();

                backgroundWorker.Dispose();

                fileItems.ForEach(x =>

                {

                    System.IO.File.Delete(x.LocalFilePath);

                });

                System.IO.Directory.Delete(_localStorage, true);

            }

            catch (Exception ex)

            {

                System.Diagnostics.Debug.WriteLine(ex.ToString());

            }

        }

    }

}