mattercontrol/MatterControlLib/Library/Providers/GitHub/GitHubContainer.cs

314 lines
No EOL
9.1 KiB
C#

/*
Copyright (c) 2019, John Lewin
All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
1. Redistributions of source code must retain the above copyright notice, this
list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright notice,
this list of conditions and the following disclaimer in the documentation
and/or other materials provided with the distribution.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
The views and conclusions contained in the software and documentation are those
of the authors and should not be interpreted as representing official policies,
either expressed or implied, of the FreeBSD Project.
*/
using System;
using System.Collections.Generic;
using System.IO;
using System.Net.Http;
using System.Threading.Tasks;
using MatterHackers.Agg;
using MatterHackers.Agg.Image;
using MatterHackers.Agg.Platform;
using Newtonsoft.Json;
namespace MatterHackers.MatterControl.Library
{
public class GitHubContainer : LibraryContainer
{
#pragma warning disable SA1307 // Accessible fields should begin with upper-case letter
#pragma warning disable SA1310 // Field names should not contain underscore
internal struct FileInfo
{
public LinkFields _links;
public string download_url;
public string name;
public string type;
}
// JSON parsing methods
internal struct LinkFields
{
public string self;
}
#pragma warning restore SA1310 // Field names should not contain underscore
#pragma warning restore SA1307 // Accessible fields should begin with upper-case letter
public string Account { get; }
public string Repository { get; }
public string RepoDirectory { get; }
private object locker = new object();
public GitHubContainer(string containerName, string account, string repositor, string repoDirectory)
{
this.ChildContainers = new List<ILibraryContainerLink>();
this.Items = new List<ILibraryItem>();
this.Name = containerName;
this.Account = account;
this.Repository = repositor;
this.RepoDirectory = repoDirectory;
}
public override void Load()
{
var uri = $"https://api.github.com/repos/{Account}/{Repository}/contents/{RepoDirectory}";
// get the directory contents
WebCache.RetrieveText(uri,
(content) =>
{
lock (locker)
{
ParseJson(content);
}
},
false,
AddCromeHeaders);
}
private void ParseJson(string jsonStr)
{
// parse result
FileInfo[] dirContents = JsonConvert.DeserializeObject<FileInfo[]>(jsonStr);
// read in data
foreach (FileInfo file in dirContents)
{
if (file.type == "dir")
{
ChildContainers.Add(new GitHubContainerLink(file.name,
Account,
Repository,
RepoDirectory + "/" + file.name));
}
else if (file.type == "file")
{
if (Path.GetExtension(file.name).ToLower() == ".library")
{
ChildContainers.Add(new GitHubLibraryLink(Path.GetFileNameWithoutExtension(file.name),
Account,
Repository,
file.download_url));
}
else if (file.name.ToLower() == "index.md")
{
}
else
{
this.Items.Add(new GitHubLibraryItem(file.name, file.download_url));
}
}
}
OnContentChanged();
}
private static Dictionary<string, List<(string name, string url)>> imageUrlCaches = new Dictionary<string, List<(string name, string url)>>();
public override Task<ImageBuffer> GetThumbnail(ILibraryItem item, int width, int height)
{
var existingThumbnail = base.GetThumbnail(item, width, height);
if (existingThumbnail.Result == null)
{
var imageUrlCache = LoadImageUrlCache();
foreach (var imageUrl in imageUrlCache)
{
if (imageUrl.name.Contains(item.ID))
{
// download the image and cache it
var image = new ImageBuffer(LibraryConfig.DefaultItemIcon);
image.SetRecieveBlender(new BlenderPreMultBGRA());
WebCache.RetrieveImageAsync(image, imageUrl.url, false);
return Task.FromResult<ImageBuffer>(image);
}
}
}
return existingThumbnail;
}
private List<(string name, string url)> LoadImageUrlCache()
{
lock (locker)
{
var key = $"{Account}:{Repository}";
if (!imageUrlCaches.ContainsKey(key))
{
var imageUrlCache = new List<(string name, string url)>();
// Check if we can find the thumbnail in the GitHub .images directory
var uri = $"https://api.github.com/repos/{Account}/{Repository}/contents/.images";
// get the directory contents
WebCache.RetrieveText(uri,
(content) =>
{
lock (locker)
{
FileInfo[] dirContents = JsonConvert.DeserializeObject<FileInfo[]>(content);
// read in data
foreach (FileInfo file in dirContents)
{
if (file.type == "file")
{
imageUrlCache.Add((file.name, file.download_url));
}
}
imageUrlCaches[key] = imageUrlCache;
}
},
false,
AddCromeHeaders);
imageUrlCaches[key] = imageUrlCache;
}
return imageUrlCaches[key];
}
}
public static void AddCromeHeaders(HttpRequestMessage request)
{
request.Headers.Add("Connection", "keep-alive");
request.Headers.Add("Upgrade-Insecure-Requests", "1");
request.Headers.Add("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.105 Safari/537.36");
request.Headers.Add("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9");
request.Headers.Add("Sec-Fetch-Site", "none");
request.Headers.Add("Sec-Fetch-Mode", "navigate");
request.Headers.Add("Sec-Fetch-User", "?1");
request.Headers.Add("Sec-Fetch-Dest", "document");
request.Headers.Add("Accept-Language", "en-US,en;q=0.9");
}
private class StaticDataItem : ILibraryAssetStream
{
public StaticDataItem()
{
}
public StaticDataItem(string relativePath)
{
this.AssetPath = relativePath;
}
public string AssetPath { get; }
public string Category { get; } = "";
public string ContentType => Path.GetExtension(AssetPath).ToLower().Trim('.');
public DateTime DateCreated { get; } = DateTime.Now;
public DateTime DateModified { get; } = DateTime.Now;
public string FileName => Path.GetFileName(AssetPath);
public long FileSize { get; } = -1;
public string ID => agg_basics.GetLongHashCode(AssetPath).ToString();
public bool IsProtected => true;
public bool IsVisible => true;
public bool LocalContentExists => true;
public string Name => this.FileName;
public Task<StreamAndLength> GetStream(Action<double, string> progress)
{
return Task.FromResult(new StreamAndLength()
{
Stream = AggContext.StaticData.OpenStream(AssetPath),
Length = -1
});
}
}
public class GitHubContainerLink : ILibraryContainerLink
{
private readonly string owner;
private readonly string repository;
protected string Path { get; }
public GitHubContainerLink(string containerName, string owner, string repository, string path)
{
this.Name = containerName;
this.owner = owner;
this.repository = repository;
this.Path = path;
}
public bool IsReadOnly { get; set; } = true;
public bool UseIncrementedNameDuringTypeChange { get; set; }
public string ID => Name
.GetLongHashCode(owner
.GetLongHashCode(repository
.GetLongHashCode(Path
.GetLongHashCode()))).ToString();
public string Name { get; }
public bool IsProtected => false;
public bool IsVisible => true;
public DateTime DateModified => DateTime.Now;
public DateTime DateCreated => DateTime.Now;
public virtual Task<ILibraryContainer> GetContainer(Action<double, string> reportProgress)
{
return Task.FromResult<ILibraryContainer>(new GitHubContainer(Name, owner, repository, Path));
}
}
public class GitHubLibraryLink : GitHubContainerLink
{
public GitHubLibraryLink(string containerName, string owner, string repository, string path)
: base(containerName, owner, repository, path)
{
}
public override async Task<ILibraryContainer> GetContainer(Action<double, string> reportProgress)
{
var content = WebCache.GetCachedText(Path, false, AddCromeHeaders);
return await LibraryJsonFile.ContainerFromJson(Name, content).GetContainer(null);
}
}
}
}