using System; using System.Collections.Generic; using System.Diagnostics; using System.IO; using System.Linq; using System.Runtime.InteropServices; using System.Security.Principal; using System.Text; using System.Threading.Tasks; using System.Windows.Media; using System.Windows.Media.Imaging; using System.Xml.Linq; using Windows.ApplicationModel; using Windows.Management.Deployment; using AppxPackaing; using Shell; using Wox.Infrastructure; using Wox.Infrastructure.Logger; using IStream = AppxPackaing.IStream; using Rect = System.Windows.Rect; namespace Wox.Plugin.Program.Programs { public class UWP { public string Name { get; } public string FullName { get; } public string FamilyName { get; } public string Location { get; set; } public Application[] Apps { get; set; } public Package Package { get; } public PackageVersion Version { get; set; } public UWP(Package package) { Package = package; Location = Package.InstalledLocation.Path; Name = Package.Id.Name; FullName = Package.Id.FullName; FamilyName = Package.Id.FamilyName; InitializeAppInfo(); Apps = Apps.Where(a => { var valid = !string.IsNullOrEmpty(a.UserModelId) && !string.IsNullOrEmpty(a.DisplayName); return valid; }).ToArray(); } private void InitializeAppInfo() { var path = Path.Combine(Location, "AppxManifest.xml"); var namespaces = XmlNamespaces(path); InitPackageVersion(namespaces); var appxFactory = new AppxFactory(); IStream stream; const uint noAttribute = 0x80; const Stgm exclusiveRead = Stgm.Read | Stgm.ShareExclusive; var result = SHCreateStreamOnFileEx(path, exclusiveRead, noAttribute, false, null, out stream); if (result == Hresult.Ok) { var reader = appxFactory.CreateManifestReader(stream); var manifestApps = reader.GetApplications(); var apps = new List(); while (manifestApps.GetHasCurrent() != 0) { var manifestApp = manifestApps.GetCurrent(); var appListEntry = manifestApp.GetStringValue("AppListEntry"); if (appListEntry != "none") { var app = new Application(manifestApp, this); apps.Add(app); } manifestApps.MoveNext(); } Apps = apps.Where(a => a.AppListEntry != "none").ToArray(); } } /// http://www.hanselman.com/blog/GetNamespacesFromAnXMLDocumentWithXPathDocumentAndLINQToXML.aspx private string[] XmlNamespaces(string path) { XDocument z = XDocument.Load(path); if (z.Root != null) { var namespaces = z.Root.Attributes(). Where(a => a.IsNamespaceDeclaration). GroupBy( a => a.Name.Namespace == XNamespace.None ? string.Empty : a.Name.LocalName, a => XNamespace.Get(a.Value) ).Select( g => g.First().ToString() ).ToArray(); return namespaces; } else { Log.Error($"can't find namespaces for <{path}>"); return new string[] { }; } } private void InitPackageVersion(string[] namespaces) { var versionFromNamespace = new Dictionary { {"http://schemas.microsoft.com/appx/manifest/foundation/windows10", PackageVersion.Windows10}, {"http://schemas.microsoft.com/appx/2013/manifest", PackageVersion.Windows81}, {"http://schemas.microsoft.com/appx/2010/manifest", PackageVersion.Windows8}, }; foreach (var n in versionFromNamespace.Keys) { if (namespaces.Contains(n)) { Version = versionFromNamespace[n]; return; } } Log.Error($"Unknown Appmanifest version: {FullName}, Package location: <{Location}>."); Version = PackageVersion.Unknown; } public static Application[] All() { var windows10 = new Version(10, 0); var support = Environment.OSVersion.Version.Major >= windows10.Major; if (support) { var watch = new System.Diagnostics.Stopwatch(); watch.Start(); var applications = CurrentUserPackages().AsParallel().SelectMany(p => new UWP(p).Apps).ToArray(); watch.Stop(); Log.Info("UWP ALL" + watch.ElapsedMilliseconds); return applications; } else { return new Application[] { }; } } private static IEnumerable CurrentUserPackages() { var user = WindowsIdentity.GetCurrent().User; if (user != null) { var userSecurityId = user.Value; var packageManager = new PackageManager(); var packages = packageManager.FindPackagesForUser(userSecurityId); packages = packages.Where( p => !p.IsFramework && !p.IsDevelopmentMode && !string.IsNullOrEmpty(p.InstalledLocation.Path)); return packages; } else { return new Package[] { }; } } public override string ToString() { return FamilyName; } public override bool Equals(object obj) { var uwp = obj as UWP; if (uwp != null) { return FamilyName.Equals(uwp.FamilyName); } else { return false; } } public override int GetHashCode() { return FamilyName.GetHashCode(); } public class Application : IProgram { public string AppListEntry { get; set; } public string DisplayName { get; set; } public string Description { get; set; } public string UserModelId { get; set; } public string BackgroundColor { get; set; } public string LogoUri { get; set; } public string LogoPath { get; set; } public UWP Package { get; set; } private int Score(string query) { var score1 = StringMatcher.Score(DisplayName, query); var score2 = StringMatcher.ScoreForPinyin(DisplayName, query); var score3 = StringMatcher.Score(Description, query); var score = new[] { score1, score2, score3 }.Max(); return score; } public Result Result(string query, IPublicAPI api) { var result = new Result { SubTitle = Package.Location, Icon = Logo, Score = Score(query), ContextData = this, Action = e => { Launch(api); return true; } }; if (Description.Length >= DisplayName.Length && Description.Substring(0, DisplayName.Length) == DisplayName) { result.Title = Description; } else if (!string.IsNullOrEmpty(Description)) { result.Title = $"{DisplayName}: {Description}"; } else { result.Title = DisplayName; } return result; } public List ContextMenus(IPublicAPI api) { var contextMenus = new List { new Result { Title = api.GetTranslation("wox_plugin_program_open_containing_folder"), Action = _ => { var hide = Main.StartProcess(new ProcessStartInfo(Package.Location)); return hide; }, IcoPath = "Images/folder.png" } }; return contextMenus; } private async void Launch(IPublicAPI api) { var appManager = new ApplicationActivationManager(); uint unusedPid; const string noArgs = ""; const ACTIVATEOPTIONS noFlags = ACTIVATEOPTIONS.AO_NONE; await Task.Run(() => { try { appManager.ActivateApplication(UserModelId, noArgs, noFlags, out unusedPid); } catch (Exception) { var name = "Plugin: Program"; var message = $"Can't start UWP: {DisplayName}"; api.ShowMsg(name, message, string.Empty); } }); } public Application(IAppxManifestApplication manifestApp, UWP package) { UserModelId = manifestApp.GetAppUserModelId(); DisplayName = manifestApp.GetStringValue("DisplayName"); Description = manifestApp.GetStringValue("Description"); BackgroundColor = manifestApp.GetStringValue("BackgroundColor"); Package = package; DisplayName = ResourceFromPri(package.FullName, DisplayName); Description = ResourceFromPri(package.FullName, Description); LogoUri = LogoUriFromManifest(manifestApp); LogoPath = LogoPathFromUri(LogoUri); } internal string ResourceFromPri(string packageFullName, string resourceReference) { const string prefix = "ms-resource:"; if (!string.IsNullOrWhiteSpace(resourceReference) && resourceReference.StartsWith(prefix)) { // magic comes from @talynone // https://github.com/talynone/Wox.Plugin.WindowsUniversalAppLauncher/blob/master/StoreAppLauncher/Helpers/NativeApiHelper.cs#L139-L153 string key = resourceReference.Substring(prefix.Length); string parsed; if (key.StartsWith("//")) { parsed = prefix + key; } else if (key.StartsWith("/")) { parsed = prefix + "//" + key; } else { parsed = prefix + "///resources/" + key; } var outBuffer = new StringBuilder(128); string source = $"@{{{packageFullName}? {parsed}}}"; var capacity = (uint)outBuffer.Capacity; var hResult = SHLoadIndirectString(source, outBuffer, capacity, IntPtr.Zero); if (hResult == Hresult.Ok) { var loaded = outBuffer.ToString(); if (!string.IsNullOrEmpty(loaded)) { return loaded; } else { Log.Error($"Load {source} failed, null or empty result. Package location: <{Package.Location}>."); return string.Empty; } } else { // known hresult 2147942522: // 'Microsoft Corporation' violates pattern constraint of '\bms-resource:.{1,256}'. // for // Microsoft.MicrosoftOfficeHub_17.7608.23501.0_x64__8wekyb3d8bbwe: ms-resource://Microsoft.MicrosoftOfficeHub/officehubintl/AppManifest_GetOffice_Description // Microsoft.BingFoodAndDrink_3.0.4.336_x64__8wekyb3d8bbwe: ms-resource:AppDescription Log.Error($"Load {source} failed, HResult error code: {hResult}. Package location: <{Package.Location}>."); var exception = Marshal.GetExceptionForHR((int)hResult); Log.Exception(exception); return string.Empty; } } else { return resourceReference; } } internal string LogoUriFromManifest(IAppxManifestApplication app) { var logoKeyFromVersion = new Dictionary { {PackageVersion.Windows10, "Square44x44Logo"}, {PackageVersion.Windows81, "Square30x30Logo"}, {PackageVersion.Windows8, "SmallLogo"}, }; if (logoKeyFromVersion.ContainsKey(Package.Version)) { var key = logoKeyFromVersion[Package.Version]; var logoUri = app.GetStringValue(key); return logoUri; } else { return string.Empty; } } internal string LogoPathFromUri(string uri) { // all https://msdn.microsoft.com/windows/uwp/controls-and-patterns/tiles-and-notifications-app-assets // windows 10 https://msdn.microsoft.com/en-us/library/windows/apps/dn934817.aspx // windows 8.1 https://msdn.microsoft.com/en-us/library/windows/apps/hh965372.aspx#target_size // windows 8 https://msdn.microsoft.com/en-us/library/windows/apps/br211475.aspx string path; if (uri.Contains("\\")) { path = Path.Combine(Package.Location, uri); } else { // for C:\Windows\MiracastView etc path = Path.Combine(Package.Location, "Assets", uri); } var extension = Path.GetExtension(path); if (extension != null) { var end = path.Length - extension.Length; var prefix = path.Substring(0, end); var paths = new List { path }; // todo hidpi icon if (Package.Version == PackageVersion.Windows10) { paths.Add($"{prefix}.scale-100{extension}"); paths.Add($"{prefix}.scale-200{extension}"); } else if (Package.Version == PackageVersion.Windows81) { paths.Add($"{prefix}.scale-100{extension}"); paths.Add($"{prefix}.scale-120{extension}"); paths.Add($"{prefix}.scale-140{extension}"); paths.Add($"{prefix}.scale-160{extension}"); paths.Add($"{prefix}.scale-180{extension}"); } else if (Package.Version == PackageVersion.Windows8) { paths.Add($"{prefix}.scale-100{extension}"); } var selected = paths.FirstOrDefault(File.Exists); if (!string.IsNullOrEmpty(selected)) { return selected; } else { Log.Error($"<{UserModelId}> can't find logo uri: <{uri}>, Package location: <{Package.Location}>."); return string.Empty; } } else { Log.Error($"<{UserModelId}> cantains uri doesn't have extension: <{uri}>, Package location: <{Package.Location}>."); return string.Empty; } } public ImageSource Logo() { var logo = ImageFromPath(LogoPath); var plated = PlatedImage(logo); // todo magic! temp fix for cross thread object plated.Freeze(); return plated; } private BitmapImage ImageFromPath(string path) { if (File.Exists(path)) { var image = new BitmapImage(new Uri(path)); return image; } else { Log.Error($"Can't get logo for <{UserModelId}> with path <{path}>, Package location: <{Package.Location}>.."); return new BitmapImage(new Uri(Constant.ErrorIcon)); } } private ImageSource PlatedImage(BitmapImage image) { if (!string.IsNullOrEmpty(BackgroundColor) && BackgroundColor != "transparent") { var width = image.Width; var height = image.Height; var x = 0; var y = 0; var group = new DrawingGroup(); var converted = ColorConverter.ConvertFromString(BackgroundColor); if (converted != null) { var color = (Color)converted; var brush = new SolidColorBrush(color); var pen = new Pen(brush, 1); var backgroundArea = new Rect(0, 0, width, width); var rectabgle = new RectangleGeometry(backgroundArea); var rectDrawing = new GeometryDrawing(brush, pen, rectabgle); group.Children.Add(rectDrawing); var imageArea = new Rect(x, y, image.Width, image.Height); var imageDrawing = new ImageDrawing(image, imageArea); group.Children.Add(imageDrawing); // http://stackoverflow.com/questions/6676072/get-system-drawing-bitmap-of-a-wpf-area-using-visualbrush var visual = new DrawingVisual(); var context = visual.RenderOpen(); context.DrawDrawing(group); context.Close(); const int dpiScale100 = 96; var bitmap = new RenderTargetBitmap( Convert.ToInt32(width), Convert.ToInt32(height), dpiScale100, dpiScale100, PixelFormats.Pbgra32 ); bitmap.Render(visual); return bitmap; } else { Log.Error($"Can't convert background string <{BackgroundColor}> to color, Package location: <{Package.Location}>."); return new BitmapImage(new Uri(Constant.ErrorIcon)); } } else { // todo use windows theme as background return image; } } public override string ToString() { return $"{DisplayName}: {Description}"; } } public enum PackageVersion { Windows10, Windows81, Windows8, Unknown } [Flags] private enum Stgm : uint { Read = 0x0, ShareExclusive = 0x10, } private enum Hresult : uint { Ok = 0x0000, } [DllImport("shlwapi.dll", CharSet = CharSet.Unicode)] private static extern Hresult SHCreateStreamOnFileEx(string fileName, Stgm grfMode, uint attributes, bool create, IStream reserved, out IStream stream); [DllImport("shlwapi.dll", CharSet = CharSet.Unicode)] private static extern Hresult SHLoadIndirectString(string pszSource, StringBuilder pszOutBuf, uint cchOutBuf, IntPtr ppvReserved); } }