Compare commits
5 commits
Author | SHA1 | Date | |
---|---|---|---|
4fb4a2d0c3 | |||
c6ea3fd36f | |||
491e9b5a6f | |||
34ba6e782b | |||
ba1bc1732e |
12 changed files with 210 additions and 92 deletions
|
@ -162,15 +162,15 @@ public partial class Gambling : GamblingModule<GamblingService>
|
|||
|
||||
if (password is not null)
|
||||
{
|
||||
var img = GetPasswordImage(password);
|
||||
var img = _captchaService.GetPasswordImage(password);
|
||||
await using var stream = await img.ToStreamAsync();
|
||||
var toSend = Response()
|
||||
.File(stream, "timely.png");
|
||||
|
||||
#if GLOBAL_ELLIE
|
||||
if (_rng.Next(0, 5) == 0)
|
||||
if (_rng.Next(0, 8) == 0)
|
||||
toSend = toSend
|
||||
.Confirm("[Sub on Patreon](https://patreon.com/elliebot) to remove captcha.");
|
||||
.Text("*[Sub on Patreon](https://patreon.com/elliebot) to remove captcha.*");
|
||||
#endif
|
||||
|
||||
var captchaMessage = await toSend.SendAsync();
|
||||
|
@ -194,39 +194,6 @@ public partial class Gambling : GamblingModule<GamblingService>
|
|||
await ClaimTimely();
|
||||
}
|
||||
|
||||
private Image<Rgba32> GetPasswordImage(string password)
|
||||
{
|
||||
var img = new Image<Rgba32>(50, 24);
|
||||
|
||||
var font = _fonts.NotoSans.CreateFont(22);
|
||||
var outlinePen = new SolidPen(Color.Black, 0.5f);
|
||||
var strikeoutRun = new RichTextRun
|
||||
{
|
||||
Start = 0,
|
||||
End = password.GetGraphemeCount(),
|
||||
Font = font,
|
||||
StrikeoutPen = new SolidPen(Color.White, 4),
|
||||
TextDecorations = TextDecorations.Strikeout
|
||||
};
|
||||
// draw password on the image
|
||||
img.Mutate(x =>
|
||||
{
|
||||
x.DrawText(new RichTextOptions(font)
|
||||
{
|
||||
HorizontalAlignment = HorizontalAlignment.Center,
|
||||
VerticalAlignment = VerticalAlignment.Center,
|
||||
FallbackFontFamilies = _fonts.FallBackFonts,
|
||||
Origin = new(25, 12),
|
||||
TextRuns = [strikeoutRun]
|
||||
},
|
||||
password,
|
||||
Brushes.Solid(Color.White),
|
||||
outlinePen);
|
||||
});
|
||||
|
||||
return img;
|
||||
}
|
||||
|
||||
private async Task ClaimTimely()
|
||||
{
|
||||
var period = Config.Timely.Cooldown;
|
||||
|
|
|
@ -16,7 +16,7 @@ public sealed class CaptchaService(FontProvider fonts, IBotCache cache, IPatrona
|
|||
|
||||
public Image<Rgba32> GetPasswordImage(string password)
|
||||
{
|
||||
var img = new Image<Rgba32>(50, 24);
|
||||
var img = new Image<Rgba32>(60, 34);
|
||||
|
||||
var font = fonts.NotoSans.CreateFont(22);
|
||||
var outlinePen = new SolidPen(Color.Black, 0.5f);
|
||||
|
@ -38,7 +38,7 @@ public sealed class CaptchaService(FontProvider fonts, IBotCache cache, IPatrona
|
|||
HorizontalAlignment = HorizontalAlignment.Center,
|
||||
VerticalAlignment = VerticalAlignment.Center,
|
||||
FallbackFontFamilies = fonts.FallBackFonts,
|
||||
Origin = new(25, 12),
|
||||
Origin = new(30, 15),
|
||||
TextRuns = [strikeoutRun]
|
||||
},
|
||||
password,
|
||||
|
|
|
@ -33,9 +33,9 @@ public partial class Games
|
|||
.File(stream, "timely.png");
|
||||
|
||||
#if GLOBAL_ELLIE
|
||||
if (_rng.Next(0, 5) == 0)
|
||||
if (_rng.Next(0, 8) == 0)
|
||||
toSend = toSend
|
||||
.Confirm("[Sub on Patreon](https://patreon.com/elliebot) to remove captcha.");
|
||||
.Text("*[Sub on Patreon](https://patreon.com/elliebot) to remove captcha.*");
|
||||
#endif
|
||||
var captcha = await toSend.SendAsync();
|
||||
|
||||
|
|
|
@ -2,6 +2,8 @@
|
|||
using CsvHelper;
|
||||
using CsvHelper.Configuration;
|
||||
using System.Globalization;
|
||||
using System.Net;
|
||||
using System.Net.Http.Json;
|
||||
using System.Text.Json;
|
||||
|
||||
namespace EllieBot.Modules.Searches;
|
||||
|
@ -9,54 +11,57 @@ namespace EllieBot.Modules.Searches;
|
|||
public sealed class DefaultStockDataService : IStockDataService, IEService
|
||||
{
|
||||
private readonly IHttpClientFactory _httpClientFactory;
|
||||
private readonly IBotCache _cache;
|
||||
|
||||
public DefaultStockDataService(IHttpClientFactory httpClientFactory)
|
||||
=> _httpClientFactory = httpClientFactory;
|
||||
public DefaultStockDataService(IHttpClientFactory httpClientFactory, IBotCache cache)
|
||||
=> (_httpClientFactory, _cache) = (httpClientFactory, cache);
|
||||
|
||||
private static TypedKey<StockData> GetStockDataKey(string query)
|
||||
=> new($"stockdata:{query}");
|
||||
|
||||
public async Task<StockData?> GetStockDataAsync(string query)
|
||||
{
|
||||
ArgumentException.ThrowIfNullOrWhiteSpace(query);
|
||||
|
||||
return await _cache.GetOrAddAsync(GetStockDataKey(query.Trim().ToLowerInvariant()),
|
||||
() => GetStockDataInternalAsync(query),
|
||||
expiry: TimeSpan.FromHours(1));
|
||||
}
|
||||
|
||||
public async Task<StockData?> GetStockDataInternalAsync(string query)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (!query.IsAlphaNumeric())
|
||||
return default;
|
||||
|
||||
using var http = _httpClientFactory.CreateClient();
|
||||
var info = await GetNasdaqDataResponse<NasdaqSummaryResponse>(
|
||||
$"https://api.nasdaq.com/api/quote/{query}/summary?assetclass=stocks");
|
||||
|
||||
var quoteHtmlPage = $"https://finance.yahoo.com/quote/{query.ToUpperInvariant()}";
|
||||
|
||||
var config = Configuration.Default.WithDefaultLoader();
|
||||
using var document = await BrowsingContext.New(config).OpenAsync(quoteHtmlPage);
|
||||
|
||||
var tickerName = document.QuerySelector("div.top > .left > .container > h1")
|
||||
?.TextContent;
|
||||
|
||||
if (tickerName is null)
|
||||
if (info?.Data is not { } d || d.SummaryData is not { } sd)
|
||||
return default;
|
||||
|
||||
var marketcap = document
|
||||
.QuerySelector("li > span > fin-streamer[data-field='marketCap']")
|
||||
?.TextContent;
|
||||
|
||||
var closePrice = double.Parse(sd.PreviousClose.Value?.Substring(1) ?? "0",
|
||||
NumberStyles.Any,
|
||||
CultureInfo.InvariantCulture);
|
||||
|
||||
var volume = document.QuerySelector("li > span > fin-streamer[data-field='regularMarketVolume']")
|
||||
?.TextContent;
|
||||
|
||||
var close = document.QuerySelector("li > span > fin-streamer[data-field='regularMarketPreviousClose']")
|
||||
?.TextContent
|
||||
?? "0";
|
||||
|
||||
var price = document.QuerySelector("fin-streamer.livePrice > span")
|
||||
?.TextContent
|
||||
?? "0";
|
||||
var price = d.BidAsk.Bid.Value.IndexOf('*') is var idx and > 0
|
||||
&& double.TryParse(d.BidAsk.Bid.Value.Substring(1, idx - 1),
|
||||
NumberStyles.Any,
|
||||
CultureInfo.InvariantCulture,
|
||||
out var bid)
|
||||
? bid
|
||||
: double.NaN;
|
||||
|
||||
return new()
|
||||
{
|
||||
Name = tickerName,
|
||||
Symbol = query,
|
||||
Price = double.Parse(price, NumberStyles.Any, CultureInfo.InvariantCulture),
|
||||
Close = double.Parse(close, NumberStyles.Any, CultureInfo.InvariantCulture),
|
||||
MarketCap = marketcap,
|
||||
DailyVolume = (long)double.Parse(volume ?? "0", NumberStyles.Any, CultureInfo.InvariantCulture),
|
||||
Name = query,
|
||||
Symbol = info.Data.Symbol,
|
||||
Price = price,
|
||||
Close = closePrice,
|
||||
MarketCap = sd.MarketCap.Value,
|
||||
DailyVolume =
|
||||
(long)double.Parse(sd.AverageVolume.Value ?? "0", NumberStyles.Any, CultureInfo.InvariantCulture),
|
||||
};
|
||||
}
|
||||
catch (Exception ex)
|
||||
|
@ -66,6 +71,36 @@ public sealed class DefaultStockDataService : IStockDataService, IEService
|
|||
}
|
||||
}
|
||||
|
||||
private async Task<NasdaqDataResponse<T>?> GetNasdaqDataResponse<T>(string url)
|
||||
{
|
||||
using var httpClient = _httpClientFactory.CreateClient("google:search");
|
||||
|
||||
var req = new HttpRequestMessage(HttpMethod.Get,
|
||||
url)
|
||||
{
|
||||
Headers =
|
||||
{
|
||||
{ "Host", "api.nasdaq.com" },
|
||||
{ "User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:132.0) Gecko/20100101 Firefox/132.0" },
|
||||
{ "Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8" },
|
||||
{ "Accept-Language", "en-US,en;q=0.5" },
|
||||
{ "Accept-Encoding", "gzip, deflate, br, zstd" },
|
||||
{ "Connection", "keep-alive" },
|
||||
{ "Upgrade-Insecure-Requests", "1" },
|
||||
{ "Sec-Fetch-Dest", "document" },
|
||||
{ "Sec-Fetch-Mode", "navigate" },
|
||||
{ "Sec-Fetch-Site", "none" },
|
||||
{ "Sec-Fetch-User", "?1" },
|
||||
{ "Priority", "u=0, i" },
|
||||
{ "TE", "trailers" }
|
||||
}
|
||||
};
|
||||
var res = await httpClient.SendAsync(req);
|
||||
|
||||
var info = await res.Content.ReadFromJsonAsync<NasdaqDataResponse<T>>();
|
||||
return info;
|
||||
}
|
||||
|
||||
public async Task<IReadOnlyCollection<SymbolData>> SearchSymbolAsync(string query)
|
||||
{
|
||||
if (string.IsNullOrWhiteSpace(query))
|
||||
|
@ -91,22 +126,37 @@ public sealed class DefaultStockDataService : IStockDataService, IEService
|
|||
.ToList();
|
||||
}
|
||||
|
||||
private static CsvConfiguration _csvConfig = new(CultureInfo.InvariantCulture);
|
||||
private static TypedKey<IReadOnlyCollection<CandleData>> GetCandleDataKey(string query)
|
||||
=> new($"candledata:{query}");
|
||||
|
||||
public async Task<IReadOnlyCollection<CandleData>> GetCandleDataAsync(string query)
|
||||
=> await _cache.GetOrAddAsync(GetCandleDataKey(query),
|
||||
async () => await GetCandleDataInternalAsync(query),
|
||||
expiry: TimeSpan.FromHours(4))
|
||||
?? [];
|
||||
|
||||
public async Task<IReadOnlyCollection<CandleData>> GetCandleDataInternalAsync(string query)
|
||||
{
|
||||
using var http = _httpClientFactory.CreateClient();
|
||||
await using var resStream = await http.GetStreamAsync(
|
||||
$"https://query1.finance.yahoo.com/v7/finance/download/{query}"
|
||||
+ $"?period1={DateTime.UtcNow.Subtract(30.Days()).ToTimestamp()}"
|
||||
+ $"&period2={DateTime.UtcNow.ToTimestamp()}"
|
||||
+ "&interval=1d");
|
||||
|
||||
using var textReader = new StreamReader(resStream);
|
||||
using var csv = new CsvReader(textReader, _csvConfig);
|
||||
var records = csv.GetRecords<YahooFinanceCandleData>().ToArray();
|
||||
var now = DateTime.UtcNow;
|
||||
var fromdate = now.Subtract(30.Days()).ToString("yyyy-MM-dd");
|
||||
var todate = now.ToString("yyyy-MM-dd");
|
||||
|
||||
return records
|
||||
.Map(static x => new CandleData(x.Open, x.Close, x.High, x.Low, x.Volume));
|
||||
var res = await GetNasdaqDataResponse<NasdaqChartResponse>(
|
||||
$"https://api.nasdaq.com/api/quote/{query}/chart?assetclass=stocks"
|
||||
+ $"&fromdate={fromdate}"
|
||||
+ $"&todate={todate}");
|
||||
|
||||
if (res?.Data?.Chart is not { } chart)
|
||||
return Array.Empty<CandleData>();
|
||||
|
||||
|
||||
return chart.Select(d => new CandleData(d.Z.Open,
|
||||
d.Z.Close,
|
||||
d.Z.High,
|
||||
d.Z.Low,
|
||||
(long)double.Parse(d.Z.Volume, NumberStyles.Any, CultureInfo.InvariantCulture)))
|
||||
.ToList();
|
||||
}
|
||||
}
|
|
@ -0,0 +1,20 @@
|
|||
namespace EllieBot.Modules.Searches;
|
||||
|
||||
public sealed class NasdaqChartResponse
|
||||
{
|
||||
public required NasdaqChartResponseData[] Chart { get; init; }
|
||||
|
||||
public sealed class NasdaqChartResponseData
|
||||
{
|
||||
public required CandleData Z { get; init; }
|
||||
|
||||
public sealed class CandleData
|
||||
{
|
||||
public required decimal High { get; init; }
|
||||
public required decimal Low { get; init; }
|
||||
public required decimal Open { get; init; }
|
||||
public required decimal Close { get; init; }
|
||||
public required string Volume { get; init; }
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,6 @@
|
|||
namespace EllieBot.Modules.Searches;
|
||||
|
||||
public sealed class NasdaqDataResponse<T>
|
||||
{
|
||||
public required T? Data { get; init; }
|
||||
}
|
|
@ -0,0 +1,44 @@
|
|||
using System.Text.Json.Serialization;
|
||||
|
||||
namespace EllieBot.Modules.Searches;
|
||||
|
||||
public sealed class NasdaqSummaryResponse
|
||||
{
|
||||
public required string Symbol { get; init; }
|
||||
|
||||
public required NasdaqSummaryResponseData SummaryData { get; init; }
|
||||
public required NasdaqSummaryBidAsk BidAsk { get; init; }
|
||||
|
||||
public sealed class NasdaqSummaryBidAsk
|
||||
{
|
||||
[JsonPropertyName("Bid * Size")]
|
||||
public required NasdaqBid Bid { get; init; }
|
||||
|
||||
public sealed class NasdaqBid
|
||||
{
|
||||
public required string Value { get; init; }
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class NasdaqSummaryResponseData
|
||||
{
|
||||
public required PreviousCloseData PreviousClose { get; init; }
|
||||
public required MarketCapData MarketCap { get; init; }
|
||||
public required AverageVolumeData AverageVolume { get; init; }
|
||||
|
||||
public sealed class PreviousCloseData
|
||||
{
|
||||
public required string Value { get; init; }
|
||||
}
|
||||
|
||||
public sealed class MarketCapData
|
||||
{
|
||||
public required string Value { get; init; }
|
||||
}
|
||||
|
||||
public sealed class AverageVolumeData
|
||||
{
|
||||
public required string Value { get; init; }
|
||||
}
|
||||
}
|
||||
}
|
|
@ -183,7 +183,7 @@ public partial class Utility
|
|||
{
|
||||
var time = DateTime.UtcNow + ts;
|
||||
|
||||
if (ts > TimeSpan.FromDays(60))
|
||||
if (ts > TimeSpan.FromDays(366))
|
||||
return false;
|
||||
|
||||
if (ctx.Guild is not null)
|
||||
|
|
|
@ -150,7 +150,26 @@ public partial class Utility
|
|||
[Cmd]
|
||||
public async Task TodoArchiveAdd([Leftover] string name)
|
||||
{
|
||||
var result = await _service.ArchiveTodosAsync(ctx.User.Id, name);
|
||||
var result = await _service.ArchiveTodosAsync(ctx.User.Id, name, false);
|
||||
if (result == ArchiveTodoResult.NoTodos)
|
||||
{
|
||||
await Response().Error(strs.todo_no_todos).SendAsync();
|
||||
return;
|
||||
}
|
||||
|
||||
if (result == ArchiveTodoResult.MaxLimitReached)
|
||||
{
|
||||
await Response().Error(strs.todo_archive_max_limit).SendAsync();
|
||||
return;
|
||||
}
|
||||
|
||||
await ctx.OkAsync();
|
||||
}
|
||||
|
||||
[Cmd]
|
||||
public async Task TodoArchiveDone([Leftover] string name)
|
||||
{
|
||||
var result = await _service.ArchiveTodosAsync(ctx.User.Id, name, true);
|
||||
if (result == ArchiveTodoResult.NoTodos)
|
||||
{
|
||||
await Response().Error(strs.todo_no_todos).SendAsync();
|
||||
|
@ -193,7 +212,7 @@ public partial class Utility
|
|||
|
||||
foreach (var archivedList in items)
|
||||
{
|
||||
eb.AddField($"id: {archivedList.Id.ToString()}", archivedList.Name, true);
|
||||
eb.AddField($"id: {new kwum(archivedList.Id)}", archivedList.Name, true);
|
||||
}
|
||||
|
||||
return eb;
|
||||
|
@ -202,7 +221,7 @@ public partial class Utility
|
|||
}
|
||||
|
||||
[Cmd]
|
||||
public async Task TodoArchiveShow(int id)
|
||||
public async Task TodoArchiveShow(kwum id)
|
||||
{
|
||||
var list = await _service.GetArchivedTodoListAsync(ctx.User.Id, id);
|
||||
if (list == null || list.Items.Count == 0)
|
||||
|
@ -234,7 +253,7 @@ public partial class Utility
|
|||
}
|
||||
|
||||
[Cmd]
|
||||
public async Task TodoArchiveDelete(int id)
|
||||
public async Task TodoArchiveDelete(kwum id)
|
||||
{
|
||||
if (!await _service.ArchiveDeleteAsync(ctx.User.Id, id))
|
||||
{
|
||||
|
|
|
@ -6,8 +6,8 @@ namespace EllieBot.Modules.Utility;
|
|||
|
||||
public sealed class TodoService : IEService
|
||||
{
|
||||
private const int ARCHIVE_MAX_COUNT = 9;
|
||||
private const int TODO_MAX_COUNT = 27;
|
||||
private const int ARCHIVE_MAX_COUNT = 18;
|
||||
private const int TODO_MAX_COUNT = 36;
|
||||
|
||||
private readonly DbService _db;
|
||||
|
||||
|
@ -111,7 +111,7 @@ public sealed class TodoService : IEService
|
|||
.DeleteAsync();
|
||||
}
|
||||
|
||||
public async Task<ArchiveTodoResult> ArchiveTodosAsync(ulong userId, string name)
|
||||
public async Task<ArchiveTodoResult> ArchiveTodosAsync(ulong userId, string name, bool onlyDone)
|
||||
{
|
||||
// create a new archive
|
||||
|
||||
|
@ -140,7 +140,7 @@ public sealed class TodoService : IEService
|
|||
|
||||
var updated = await ctx
|
||||
.GetTable<TodoModel>()
|
||||
.Where(x => x.UserId == userId && x.ArchiveId == null)
|
||||
.Where(x => x.UserId == userId && (!onlyDone || x.IsDone) && x.ArchiveId == null)
|
||||
.Set(x => x.ArchiveId, inserted.Id)
|
||||
.UpdateAsync();
|
||||
|
||||
|
|
|
@ -1441,6 +1441,11 @@ todoarchivedelete:
|
|||
- del
|
||||
- remove
|
||||
- rm
|
||||
todoarchivedone:
|
||||
- done
|
||||
- compelete
|
||||
- finish
|
||||
- completed
|
||||
todoedit:
|
||||
- edit
|
||||
- change
|
||||
|
|
|
@ -4524,6 +4524,13 @@ todoarchiveadd:
|
|||
params:
|
||||
- name:
|
||||
desc: "The name of the archive to be created."
|
||||
todoarchivedone:
|
||||
desc: Creates a new archive with the specified name using only completed current todos.
|
||||
ex:
|
||||
- Success!
|
||||
params:
|
||||
- name:
|
||||
desc: "The name of the archive to be created."
|
||||
todoarchivelist:
|
||||
desc: Lists all archived todo lists.
|
||||
ex:
|
||||
|
|
Loading…
Add table
Reference in a new issue