97 lines
3.5 KiB
C#
97 lines
3.5 KiB
C#
using Dapper;
|
|
using Microsoft.Data.Sqlite;
|
|
|
|
var builder = WebApplication.CreateBuilder(args);
|
|
|
|
var dbPath = builder.Configuration["DbPath"]
|
|
?? Path.Combine(AppContext.BaseDirectory, "..", "..", "..", "..", "sessions.db");
|
|
|
|
builder.Services.AddSingleton<Func<SqliteConnection>>(() =>
|
|
new SqliteConnection($"Data Source={Path.GetFullPath(dbPath)}"));
|
|
|
|
var app = builder.Build();
|
|
|
|
app.UseDefaultFiles();
|
|
app.UseStaticFiles();
|
|
|
|
// GET /api/cost-buckets?from=&to=&bucketMinutes=
|
|
// Returns N buckets of aggregated cost
|
|
app.MapGet("/api/cost-buckets", (Func<SqliteConnection> factory, string from, string to, int bucketMinutes) =>
|
|
{
|
|
using var db = factory();
|
|
var rows = db.Query(@"
|
|
SELECT
|
|
CAST((strftime('%s', timestamp) - strftime('%s', @from)) / (@bucketSec) AS INTEGER) AS bucket,
|
|
SUM(CAST(json_extract(data, '$.message.usage.cost.total') AS REAL)) AS cost,
|
|
COUNT(*) AS count,
|
|
MIN(timestamp) AS bucketStart,
|
|
MAX(timestamp) AS bucketEnd
|
|
FROM entries
|
|
WHERE json_extract(data, '$.message.usage.cost.total') IS NOT NULL
|
|
AND timestamp >= @from AND timestamp < @to
|
|
GROUP BY bucket
|
|
ORDER BY bucket ASC",
|
|
new { from, to, bucketSec = bucketMinutes * 60 });
|
|
|
|
return Results.Ok(rows);
|
|
});
|
|
|
|
// GET /api/cost-entries?from=&to=
|
|
// Returns all cost entries in a time range
|
|
app.MapGet("/api/cost-entries", (Func<SqliteConnection> factory, string from, string to) =>
|
|
{
|
|
using var db = factory();
|
|
var rows = db.Query(@"
|
|
SELECT id, session, type, timestamp,
|
|
json_extract(data, '$.message.usage.cost.total') AS cost,
|
|
json_extract(data, '$.message.usage.input') AS inputTokens,
|
|
json_extract(data, '$.message.usage.output') AS outputTokens,
|
|
json_extract(data, '$.message.model') AS model,
|
|
data
|
|
FROM entries
|
|
WHERE json_extract(data, '$.message.usage.cost.total') IS NOT NULL
|
|
AND timestamp >= @from AND timestamp < @to
|
|
ORDER BY timestamp ASC",
|
|
new { from, to });
|
|
|
|
return Results.Ok(rows);
|
|
});
|
|
|
|
// GET /api/cost-summary?from=&to=
|
|
app.MapGet("/api/cost-summary", (Func<SqliteConnection> factory, string? from, string? to) =>
|
|
{
|
|
using var db = factory();
|
|
var conditions = new List<string> { "json_extract(data, '$.message.usage.cost.total') IS NOT NULL" };
|
|
if (from != null) conditions.Add("timestamp >= @from");
|
|
if (to != null) conditions.Add("timestamp < @to");
|
|
var where = string.Join(" AND ", conditions);
|
|
|
|
var row = db.QuerySingle($@"
|
|
SELECT COALESCE(SUM(CAST(json_extract(data, '$.message.usage.cost.total') AS REAL)), 0) AS totalCost,
|
|
COUNT(*) AS totalEntries
|
|
FROM entries
|
|
WHERE {where}",
|
|
new { from, to });
|
|
|
|
return Results.Ok(row);
|
|
});
|
|
|
|
// POST /api/sync - run export.mjs to import new entries
|
|
app.MapPost("/api/sync", async () =>
|
|
{
|
|
var scriptPath = Path.GetFullPath(Path.Combine(AppContext.BaseDirectory, "..", "..", "..", "..", "export.mjs"));
|
|
var psi = new System.Diagnostics.ProcessStartInfo("node", scriptPath)
|
|
{
|
|
RedirectStandardOutput = true,
|
|
RedirectStandardError = true,
|
|
WorkingDirectory = Path.GetDirectoryName(scriptPath)!
|
|
};
|
|
var proc = System.Diagnostics.Process.Start(psi)!;
|
|
var stdout = await proc.StandardOutput.ReadToEndAsync();
|
|
var stderr = await proc.StandardError.ReadToEndAsync();
|
|
await proc.WaitForExitAsync();
|
|
return Results.Ok(new { exitCode = proc.ExitCode, output = (stdout + stderr).Trim() });
|
|
});
|
|
|
|
app.Run();
|