TweetDuck/Core/Handling/RequestHandlerBase.cs

using System;
using System.Collections.Generic;
using System.Collections.Specialized;
using System.Linq;
using System.Text.RegularExpressions;
using CefSharp;
using CefSharp.Handler;
using TweetDuck.Core.Handling.General;
using TweetDuck.Core.Utils;

namespace TweetDuck.Core.Handling{
    class RequestHandlerBase : DefaultRequestHandler{
        private static readonly Regex TweetDeckResourceUrl = new Regex(@"/dist/(.*?)\.(.*?)\.(css|js)$", RegexOptions.Compiled);
        private static readonly SortedList<string, string> TweetDeckHashes = new SortedList<string, string>(4);

        public static void LoadResourceRewriteRules(string rules){
            if (string.IsNullOrEmpty(rules)){
                return;
            }

            TweetDeckHashes.Clear();

            foreach(string rule in rules.Replace(" ", "").ToLower().Split(',')){
                string[] split = rule.Split('=');

                if (split.Length == 2){
                    string key = split[0];
                    string hash = split[1];

                    if (hash.All(chr => char.IsDigit(chr) || (chr >= 'a' && chr <= 'f'))){
                        TweetDeckHashes.Add(key, hash);
                    }
                    else{
                        throw new ArgumentException("Invalid hash characters: "+rule);
                    }
                }
                else{
                    throw new ArgumentException("A rule must have exactly one '=' character: "+rule);
                }
            }
        }

        private readonly bool autoReload;

        public RequestHandlerBase(bool autoReload){
            this.autoReload = autoReload;
        }

        public override bool OnOpenUrlFromTab(IWebBrowser browserControl, IBrowser browser, IFrame frame, string targetUrl, WindowOpenDisposition targetDisposition, bool userGesture){
            return LifeSpanHandler.HandleLinkClick(browserControl, targetDisposition, targetUrl);
        }

        public override CefReturnValue OnBeforeResourceLoad(IWebBrowser browserControl, IBrowser browser, IFrame frame, IRequest request, IRequestCallback callback){
            if (BrowserUtils.HasDevTools){
                NameValueCollection headers = request.Headers;
                headers.Remove("x-devtools-emulate-network-conditions-client-id");
                request.Headers = headers;
            }

            return base.OnBeforeResourceLoad(browserControl, browser, frame, request, callback);
        }

        public override bool OnResourceResponse(IWebBrowser browserControl, IBrowser browser, IFrame frame, IRequest request, IResponse response){
            if ((request.ResourceType == ResourceType.Script || request.ResourceType == ResourceType.Stylesheet) && TweetDeckHashes.Count > 0){
                string url = request.Url;
                Match match = TweetDeckResourceUrl.Match(url);

                if (match.Success && TweetDeckHashes.TryGetValue($"{match.Groups[1]}.{match.Groups[3]}", out string hash)){
                    if (match.Groups[2].Value == hash){
                        Program.Reporter.LogVerbose("[RequestHandlerBase] Accepting " + url);
                    }
                    else{
                        Program.Reporter.LogVerbose("[RequestHandlerBase] Replacing " + url + " hash with " + hash);
                        request.Url = TweetDeckResourceUrl.Replace(url, $"/dist/$1.{hash}.$3");
                        return true;
                    }
                }
            }

            return base.OnResourceResponse(browserControl, browser, frame, request, response);
        }
        
        public override void OnRenderProcessTerminated(IWebBrowser browserControl, IBrowser browser, CefTerminationStatus status){
            if (autoReload){
                browser.Reload();
            }
        }
    }
}