public async Task <IActionResult> GetDirectNp(
[FromQuery(Name = "s")] int setId,
[FromQuery(Name = "b")] int beatmapId,
[FromQuery(Name = "u")] string userName,
[FromQuery(Name = "h")] string pass,
[FromServices] SoraDbContext ctx,
[FromServices] Pisstaube pisstaube
)
{
Response.ContentType = "text/plain";
var user = await DbUser.GetDbUser(ctx, userName);
if (user == null)
{
return(Ok("err: pass"));
}
if (!user.IsPassword(pass))
{
return(Ok("err: pass"));
}
return(Ok(await(setId != 0
? pisstaube.FetchDirectBeatmapSetAsync(setId)
: pisstaube.FetchDirectBeatmapAsync(beatmapId))));
}
public async Task <IActionResult> GetSearchDirect(
[FromQuery(Name = "m")] int playMode,
[FromQuery(Name = "r")] int rankedStatus,
[FromQuery(Name = "p")] int page,
[FromQuery(Name = "q")] string query,
[FromQuery(Name = "u")] string userName,
[FromQuery(Name = "h")] string pass,
[FromServices] SoraDbContext ctx,
[FromServices] Pisstaube pisstaube
)
{
Response.ContentType = "text/plain";
var user = await DbUser.GetDbUser(ctx, userName);
if (user == null)
{
return(Ok("err: pass"));
}
if (!user.IsPassword(pass))
{
return(Ok("err: pass"));
}
var searchResult = await pisstaube.SearchDirectAsync(query, rankedStatus, playMode, page);
return(Ok(searchResult)); // this no longer needs to be cached...
}
public async Task<IActionResult> GetBeatmap(
string map,
[FromServices] Pisstaube pisstaube)
{
Logger.Debug(map);
if (!Directory.Exists("data/beatmaps"))
Directory.CreateDirectory("data/beatmaps");
var beatmap = await pisstaube.DownloadBeatmapAsync(map, false);
// No config reading for you :3
map = map.Replace("..", string.Empty);
if (!System.IO.File.Exists(beatmap))
return NotFound($"Could not find Beatmap with the Name of {map}");
return File(System.IO.File.OpenRead(beatmap), "text/html");
}
public async Task <IActionResult> GetScoreResult(
[FromQuery(Name = "v")] ScoreboardType type,
[FromQuery(Name = "c")] string fileMd5,
[FromQuery(Name = "f")] string f,
[FromQuery(Name = "m")] PlayMode playMode,
[FromQuery(Name = "i")] int i,
[FromQuery(Name = "mods")] Mod mods,
[FromQuery(Name = "us")] string us,
[FromQuery(Name = "ha")] string pa,
[FromServices] IServiceProvider serviceProvider,
[FromServices] SoraDbContext ctx,
[FromServices] DbContextPool <SoraDbContext> ctxPool,
[FromServices] Pisstaube pisstaube,
[FromServices] Cache cache)
{
try
{
var dbUser = await DbUser.GetDbUser(ctx, us);
if (dbUser?.IsPassword(pa) != true)
{
return(Ok("error: pass"));
}
var cacheHash =
Hex.ToHex(
Crypto.GetMd5(
$"{fileMd5}{playMode}{mods}{type}{dbUser.Id}{dbUser.UserName}"
)
);
if (cache.TryGet($"sora:Scoreboards:{cacheHash}", out string cachedData))
{
return(Ok(cachedData));
}
var scores = DbScore.GetScores(ctx, fileMd5, dbUser, playMode,
type == ScoreboardType.Friends,
type == ScoreboardType.Country,
type == ScoreboardType.Mods,
mods);
BeatmapSet set = null;
DbScore ownScore = null;
var beatmap = DbBeatmap.GetBeatmap(ctx, fileMd5);
if (beatmap == null)
{
var apiSet = await pisstaube.FetchBeatmapSetAsync(fileMd5);
if (apiSet == null)
{
goto JustContinue;
}
var beatmaps = DbBeatmap.FromBeatmapSet(apiSet).ToList();
var beatmapChecksums = beatmaps.Select(s => s.FileMd5);
var dbBeatmaps =
ctx.Beatmaps.Where(rset => beatmapChecksums.Any(lFileMd5 => rset.FileMd5 == lFileMd5))
.ToList();
var pool = serviceProvider.GetRequiredService <DbContextPool <SoraDbContext> >();
Task.WaitAll(beatmaps.Select(rawBeatmap => Task.Run(async() =>
{
var context = pool.Rent();
try
{
var dbBeatmap = dbBeatmaps.FirstOrDefault(s => s.FileMd5 == rawBeatmap.FileMd5);
if (dbBeatmap != null && (dbBeatmap.Flags & DbBeatmapFlags.RankedFreeze) != 0)
{
rawBeatmap.RankedStatus = dbBeatmap.RankedStatus;
rawBeatmap.Flags = dbBeatmap.Flags;
}
context.Beatmaps.AddOrUpdate(rawBeatmap);
await context.SaveChangesAsync();
}
finally
{
pool.Return(context);
}
})).ToArray());
beatmap = beatmaps.FirstOrDefault(s => s.FileMd5 == fileMd5);
}
await foreach (var score in DbScore.GetScores(ctx, fileMd5, dbUser, playMode, false, false, false, mods, true))
{
ownScore = score;
break;
}
if (beatmap != null)
{
set = new BeatmapSet
{
SetID = beatmap.Id,
Artist = beatmap.Artist,
Title = beatmap.Title,
RankedStatus = beatmap.RankedStatus,
ChildrenBeatmaps = new List <Beatmap>
{
new Beatmap
{
FileMD5 = beatmap.FileMd5,
DiffName = beatmap.DiffName,
ParentSetID = beatmap.SetId,
BeatmapID = beatmap.Id,
Mode = beatmap.PlayMode,
},
},
}
}
;
JustContinue:
var sboard = new Scoreboard(set?.ChildrenBeatmaps.FirstOrDefault(bm => bm.FileMD5 == fileMd5),
set, scores, ownScore);
cache.Set($"sora:Scoreboards:{cacheHash}", cachedData = await sboard.ToOsuString(ctxPool), TimeSpan.FromSeconds(30));
return(Ok(cachedData));
}
catch (Exception ex)
{
Logger.Err(ex);
return(Ok("Failed"));
}
}
