60 lines
1.7 KiB
C#
60 lines
1.7 KiB
C#
using Dapper;
|
|
using Npgsql;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace OpenArchival.Database;
|
|
|
|
public class FilePathProvider : IFilePathProvider
|
|
{
|
|
NpgsqlDataSource _dataSource;
|
|
|
|
public FilePathProvider(NpgsqlDataSource dataSource)
|
|
{
|
|
_dataSource = dataSource;
|
|
}
|
|
|
|
public async Task<bool> AddFile(string path)
|
|
{
|
|
await using var connection = await _dataSource.OpenConnectionAsync();
|
|
|
|
var sql = @"INSERT INTO archivefiles (filename, path) VALUES (@Filename, @Path)";
|
|
|
|
var rowsAffected = await connection.ExecuteAsync(sql, new { Filename = Path.GetFileName(path), Path = path});
|
|
|
|
return rowsAffected == 1;
|
|
}
|
|
|
|
public async Task<bool> RemoveFile(string path)
|
|
{
|
|
await using var connection = await _dataSource.OpenConnectionAsync();
|
|
|
|
var sql = @"DELETE FROM archivefiles WHERE path = @Path";
|
|
|
|
var rowsAffected = await connection.ExecuteAsync(sql, new { Path = path });
|
|
|
|
return rowsAffected == 1;
|
|
}
|
|
|
|
public async Task<IEnumerable<FileInfo>> SearchFiles(string filename)
|
|
{
|
|
await using var connection = await _dataSource.OpenConnectionAsync();
|
|
|
|
var sql = @"SELECT * FROM archivefiles WHERE POSITION(LOWER(@Query) in LOWER(filename)) > 0";
|
|
|
|
return await connection.QueryAsync<FileInfo>(sql, new { Query = filename });
|
|
}
|
|
|
|
public async Task<IEnumerable<FileInfo>> TopFiles(int resultsCount)
|
|
{
|
|
await using var connection = await _dataSource.OpenConnectionAsync();
|
|
|
|
var sql = $@"SELECT * FROM archivefiles ORDER BY filename ASC LIMIT {resultsCount}";
|
|
|
|
return await connection.QueryAsync<FileInfo>(sql);
|
|
}
|
|
}
|