public WhereLeaf(WherePredicate pred, string db_field, string db_table)
{
this.pred = pred;
this.db_field = db_field;
this.db_table = db_table;
}
private static ResponseImage GetDanbooruImage(string[] tags, RequestProtocol protocol, long channelId)
{
List <DanbooruBlockList> blockList = Database.GetMany <DanbooruBlockList>();
string[] blockedTags = new string[blockList.Count];
int i = 0;
foreach (DanbooruBlockList entry in blockList)
{
blockedTags[i] = entry.Tag;
i++;
}
if (tags.Any(tag => blockedTags.Contains(tag)))
{
return(new ResponseImage());
}
string requestTags = string.Join("+", tags);
byte[] authToken = Encoding.ASCII.GetBytes($"{Config.DanbooruUser}:{Config.DanbooruApiKey}");
int page = 1;
const int limit = 50;
DanbooruResults result = null;
WherePredicate whereChannel = new WherePredicate()
{
Source = "channel_id",
Comparitor = "=",
Target = channelId
};
WherePredicate whereProtocol = new WherePredicate()
{
Source = "protocol",
Comparitor = "=",
Target = protocol
};
string[] cache = { };
if (channelId != 0)
{
WherePredicate whereExpires = new WherePredicate()
{
Source = "expires",
Comparitor = ">",
Target = DateTime.UtcNow
};
List <DanbooruCache> danCache = Database.GetMany <DanbooruCache>(new[] { whereExpires, whereProtocol, whereChannel });
i = 0;
cache = new string[danCache.Count];
foreach (DanbooruCache item in danCache)
{
cache[i] = item.FileUrl;
i++;
}
}
using HttpClient client = new HttpClient();
client.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Basic", Convert.ToBase64String(authToken));
while (true)
{
string requestUrl = $"https://danbooru.donmai.us/posts.json?limit={limit}&page={page}&tags={requestTags}";
HttpResponseMessage request = client.GetAsync(requestUrl).Result;
string content = request.Content.ReadAsStringAsync().Result;
List <DanbooruResults> results = JsonConvert.DeserializeObject <List <DanbooruResults> >(content);
if (results.Count == 0)
{
break;
}
result = Rng.PickRandom(results);
results.Remove(result);
bool badImage = false;
while (results.Count > 0)
{
if (badImage)
{
result = Rng.PickRandom(results);
results.Remove(result);
badImage = false;
}
if (cache.Contains(result.FileUrl))
{
badImage = true;
continue;
}
string[] resultTags = result.TagString.Split(" ");
if (blockedTags.Any(tag => resultTags.Contains(tag)))
{
badImage = true;
}
if (!badImage)
{
break;
}
}
if (!badImage)
{
break;
}
result = null;
page++;
}
if (result != null)
{
TextInfo ti = new CultureInfo("en-US", false).TextInfo;
bool isValidUri = Uri.TryCreate(result.FileUrl, UriKind.RelativeOrAbsolute, out Uri uriResult);
bool isValidUriScheme = uriResult.Scheme == Uri.UriSchemeHttp || uriResult.Scheme == Uri.UriSchemeHttps;
string fileUrl = (isValidUri && isValidUriScheme) ? result.FileUrl : $"http://danbooru.donmai.us{result.FileUrl}";
string[] characterTags = result.TagStringCharacter.Split(" ");
string characterString = "";
if (characterTags.Length > 0)
{
characterString = ti.ToTitleCase(characterTags[0].Split("(")[0].Replace("_", " "));
if (characterTags.Length == 2)
{
string secondCharacter = ti.ToTitleCase(characterTags[1].Split("(")[0].Replace("_", " "));
characterString = $"{characterString} and {secondCharacter}";
}
else if (characterTags.Length > 2)
{
characterString = "Multiple";
}
}
string[] copyrightTags = result.TagStringCopyright.Split(" ");
string copyrightString = "";
if (copyrightTags.Length > 0)
{
copyrightString = ti.ToTitleCase(copyrightTags[0].Replace("_", " "));
}
string[] artist = result.TagStringArtist.Split("_");
string artistString = string.Join(" ", artist);
string descriptionString;
if (characterString != "" && copyrightString != "")
{
descriptionString = $"{characterString} - {copyrightString}";
}
else if (copyrightString != "")
{
descriptionString = $"Unknown - {copyrightString}";
}
else
{
descriptionString = $"Original";
}
descriptionString += $"\nDrawn by {artistString}";
if (channelId != 0)
{
WherePredicate whereFileUrl = new WherePredicate
{
Source = "file_url",
Comparitor = "=",
Target = result.FileUrl
};
List <DanbooruCache> cachedItems = Database.GetMany <DanbooruCache>(new[] { whereFileUrl, whereProtocol, whereChannel });
if (cachedItems.Count > 1)
{
throw new Exception("Multiple entries for file in Danbooru cache.");
}
else if (cachedItems.Count == 1)
{
cachedItems[0].Expires = DateTime.UtcNow.AddDays(1);
cachedItems[0].Update();
}
else
{
DanbooruCache cacheItem = new DanbooruCache()
{
FileUrl = result.FileUrl,
Expires = DateTime.UtcNow.AddDays(1),
Protocol = protocol,
ChannelId = channelId
};
cacheItem.Insert();
}
}
return(new ResponseImage()
{
Url = fileUrl,
Source = $"https://danbooru.donmai.us/posts/{result.Id}",
Description = descriptionString,
Referrer = "danbooru.donmai.us",
});
}
return(new ResponseImage());
}
static IEnumerable Where (IEnumerable source, WherePredicate predicate)
{
foreach (var e in source)
if (predicate (e))
yield return e;
}
