AssetBundleCacher

AssetBundleCacher.cs

@@ -0,0 +1,265 @@
+// Copyright(c) 2019-2020 pypy. All rights reserved.
+//
+// This work is licensed under the terms of the MIT license.
+// For a copy, see <https://opensource.org/licenses/MIT>.
+
+using Microsoft.Win32;
+using System;
+using System.Diagnostics;
+using System.Text.RegularExpressions;
+using System.IO;
+using System.Security.Cryptography;
+using System.Net;
+using System.Text;
+using System.ComponentModel;
+
+namespace VRCX
+{
+    public class AssetBundleCacher
+    {
+        public static readonly AssetBundleCacher Instance;
+
+        static AssetBundleCacher()
+        {
+            Instance = new AssetBundleCacher();
+        }
+
+        public static string AssetBundleCacherTemp;
+        public static string VRChatCacheLocation;
+        public static string CacheDestinationLocation;
+        public static string AssetBundleCacherArgs;
+        public static int DownloadProgress;
+        public static bool DownloadCanceled;
+        public static string AssetId;
+        public static string AssetVersion;
+        public static WebClient client;
+        public static Process process;
+
+        public string GetAssetId(string id)
+        {
+            byte[] hash = SHA256.Create().ComputeHash(Encoding.UTF8.GetBytes(id));
+            StringBuilder idHex = new StringBuilder(hash.Length * 2);
+            foreach (byte b in hash)
+            {
+                idHex.AppendFormat("{0:x2}", b);
+            }
+            return idHex.ToString().ToUpper().Substring(0, 16);
+        }
+
+        public string GetAssetVersion(int version)
+        {
+            byte[] bytes = BitConverter.GetBytes(version);
+            string versionHex = String.Empty;
+            foreach (byte b in bytes)
+            {
+                versionHex += b.ToString("X2");
+            }
+            return versionHex.PadLeft(32, '0');
+        }
+
+        public string GetVRChatCacheLocation(string id, int version, string cacheDir)
+        {
+            var cachePath = Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData) + @"Low\VRChat\VRChat\Cache-WindowsPlayer";
+            if (cacheDir != String.Empty && Directory.Exists(cacheDir))
+            {
+                cachePath = Path.Combine(cacheDir, @"Cache-WindowsPlayer");
+            }
+            var idHash = GetAssetId(id);
+            var versionLocation = GetAssetVersion(version);
+            return Path.Combine(cachePath, idHash, versionLocation);
+        }
+
+        public long CheckVRChatCache(string id, int version, string cacheDir)
+        {
+            var FullLocation = GetVRChatCacheLocation(id, version, cacheDir);
+            var FileLocation = Path.Combine(FullLocation, "__data");
+            if (File.Exists(FileLocation))
+            {
+                FileInfo data = new FileInfo(FileLocation);
+                return data.Length;
+            }
+            return -1;
+        }
+
+        public void DownloadCacheFile(string url, string id, int version, string AppVersion, string cacheDir)
+        {
+            if (!File.Exists(Path.Combine(Program.BaseDirectory, "AssetBundleCacher\\AssetBundleCacher.exe")))
+            {
+                DownloadProgress = -10;
+                return;
+            }
+            if (!File.Exists(Path.Combine(Program.BaseDirectory, "AssetBundleCacher\\UnityPlayer.dll")))
+            {
+                using (var key = Registry.ClassesRoot.OpenSubKey(@"VRChat\shell\open\command"))
+                {
+                    // "C:\Program Files (x86)\Steam\steamapps\common\VRChat\launch.exe" "%1" %*
+                    var match = Regex.Match(key.GetValue(string.Empty) as string, "(?!\")(.+?\\\\VRChat.*)(!?\\\\launch.exe\")");
+                    if (match.Success == true)
+                    {
+                        var fileLocation = Path.Combine(match.Groups[1].Value, "UnityPlayer.dll");
+                        if (File.Exists(fileLocation))
+                        {
+                            File.Copy(fileLocation, Path.Combine(Program.BaseDirectory, "AssetBundleCacher\\UnityPlayer.dll"));
+                        }
+                        else
+                        {
+                            DownloadProgress = -11;
+                            return;
+                        }
+                    }
+                }
+            }
+            DownloadProgress = 0;
+            VRChatCacheLocation = Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData) + @"Low\VRChat\VRChat";
+            if (cacheDir != String.Empty && Directory.Exists(cacheDir))
+            {
+                VRChatCacheLocation = cacheDir;
+            }
+            CacheDestinationLocation = GetVRChatCacheLocation(id, version, cacheDir);
+            if (File.Exists(Path.Combine(CacheDestinationLocation, "__data")))
+            {
+                DownloadProgress = -12;
+                return;
+            }
+            AssetId = GetAssetId(id);
+            AssetVersion = GetAssetVersion(version);
+            var DownloadTempLocation = Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId);
+            AssetBundleCacherTemp = Path.Combine(VRChatCacheLocation, "AssetBundleCacher\\Cache");
+            Directory.CreateDirectory(AssetBundleCacherTemp);
+            AssetBundleCacherArgs = $@" -url ""file:\\{DownloadTempLocation}"" -id ""{id}"" -ver {version} -batchmode -path ""{AssetBundleCacherTemp}""";
+            DownloadCanceled = false;
+            client = new WebClient();
+            client.Headers.Add("user-agent", AppVersion);
+            client.DownloadProgressChanged += new DownloadProgressChangedEventHandler(DownloadProgressCallback);
+            client.DownloadFileCompleted += new AsyncCompletedEventHandler(DownloadCompletedCallback);
+            client.DownloadFileAsync(new System.Uri(url), DownloadTempLocation);
+        }
+
+        public void CancelDownload()
+        {
+            DownloadCanceled = true;
+            if (client != null)
+            {
+                client.CancelAsync();
+            }
+            if (process != null && !process.HasExited)
+            {
+                process.Kill();
+                if (File.Exists(Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId)))
+                    File.Delete(Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId));
+                if (Directory.Exists(Path.Combine(AssetBundleCacherTemp, AssetId)))
+                    Directory.Delete(Path.Combine(AssetBundleCacherTemp, AssetId), true);
+            }
+            DownloadProgress = -4;
+        }
+
+        public int CheckDownloadProgress()
+        {
+            return DownloadProgress;
+        }
+
+        private static void DownloadProgressCallback(object sender, DownloadProgressChangedEventArgs e)
+        {
+            DownloadProgress = e.ProgressPercentage;
+        }
+
+        private static void DownloadCompletedCallback(object sender, AsyncCompletedEventArgs e)
+        {
+            if (DownloadCanceled)
+            {
+                if (File.Exists(Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId)))
+                    File.Delete(Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId));
+                return;
+            }
+            DownloadProgress = -1;
+            process = new Process();
+            process.StartInfo.FileName = Path.Combine(Program.BaseDirectory, "AssetBundleCacher\\AssetBundleCacher.exe");
+            process.StartInfo.Arguments = AssetBundleCacherArgs;
+            process.Start();
+            process.WaitForExit(1000 * 60 * 2); //2mins
+            if (DownloadCanceled)
+            {
+                if (File.Exists(Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId)))
+                    File.Delete(Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId));
+                return;
+            }
+            try
+            {
+                if (File.Exists(Path.Combine(CacheDestinationLocation, "__data")))
+                {
+                    if (File.Exists(Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId)))
+                        File.Delete(Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId));
+                    if (Directory.Exists(Path.Combine(AssetBundleCacherTemp, AssetId)))
+                        Directory.Delete(Path.Combine(AssetBundleCacherTemp, AssetId), true);
+                    DownloadProgress = -12;
+                    return;
+                }
+                if (Directory.Exists(CacheDestinationLocation))
+                    Directory.Delete(CacheDestinationLocation, true);
+                var CacheSource = Path.Combine(AssetBundleCacherTemp, AssetId, AssetVersion);
+                if (!File.Exists(Path.Combine(CacheSource, "__data")))
+                {
+                    DownloadProgress = -13;
+                    return;
+                }
+                if (!Directory.Exists(Path.Combine(VRChatCacheLocation, "Cache-WindowsPlayer", AssetId)))
+                    Directory.CreateDirectory(Path.Combine(VRChatCacheLocation, "Cache-WindowsPlayer", AssetId));
+                Directory.Move(CacheSource, CacheDestinationLocation);
+                if (File.Exists(Path.Combine(VRChatCacheLocation, "Cache-WindowsPlayer", "__info")))
+                    File.Delete(Path.Combine(VRChatCacheLocation, "Cache-WindowsPlayer", "__info"));
+                File.Move(Path.Combine(AssetBundleCacherTemp, "__info"), Path.Combine(VRChatCacheLocation, "Cache-WindowsPlayer", "__info"));
+                Directory.Delete(Path.Combine(VRChatCacheLocation, "AssetBundleCacher\\Cache", AssetId), true);
+                File.Delete(Path.Combine(VRChatCacheLocation, "AssetBundleCacher", AssetId));
+            }
+            catch
+            {
+                DownloadProgress = -14;
+                return;
+            }
+            DownloadProgress = -3;
+        }
+
+        public void DeleteCache(string cacheDir)
+        {
+            var cachePath = Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData) + @"Low\VRChat\VRChat\Cache-WindowsPlayer";
+            if (cacheDir != String.Empty && Directory.Exists(cacheDir))
+                cachePath = Path.Combine(cacheDir, @"Cache-WindowsPlayer");
+            if (Directory.Exists(cachePath))
+            {
+                Directory.Delete(cachePath, true);
+                Directory.CreateDirectory(cachePath);
+            }
+        }
+
+        public long GetCacheSize(string cacheDir)
+        {
+            var cachePath = Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData) + @"Low\VRChat\VRChat\Cache-WindowsPlayer";
+            if (cacheDir != String.Empty && Directory.Exists(cacheDir))
+                cachePath = Path.Combine(cacheDir, @"Cache-WindowsPlayer");
+            if (Directory.Exists(cachePath))
+            {
+                return DirSize(new DirectoryInfo(cachePath));
+            }
+            else
+            {
+                return 0;
+            }
+        }
+
+        public long DirSize(DirectoryInfo d)
+        {
+            long size = 0;
+            FileInfo[] files = d.GetFiles();
+            foreach (FileInfo file in files)
+            {
+                size += file.Length;
+            }
+            DirectoryInfo[] directories = d.GetDirectories();
+            foreach (DirectoryInfo directory in directories)
+            {
+                size += DirSize(directory);
+            }
+            return size;
+        }
+    }
+}