ente/lib/face_search_manager.dart

64 lines
1.8 KiB
Dart
Raw Normal View History

2020-04-05 12:22:38 +00:00
import 'package:dio/dio.dart';
import 'package:photos/core/configuration.dart';
2020-05-18 15:38:15 +00:00
import 'package:photos/db/photo_db.dart';
2020-05-02 16:28:54 +00:00
import 'package:logging/logging.dart';
2020-04-05 12:22:38 +00:00
import 'package:photos/models/face.dart';
import 'package:photos/models/photo.dart';
2020-04-05 12:22:38 +00:00
class FaceSearchManager {
2020-06-02 14:24:37 +00:00
final _logger = Logger("FaceSearchManager");
2020-04-05 12:22:38 +00:00
final _dio = Dio();
2020-04-05 14:00:44 +00:00
FaceSearchManager._privateConstructor();
static final FaceSearchManager instance =
FaceSearchManager._privateConstructor();
2020-04-05 12:22:38 +00:00
Future<List<Face>> getFaces() {
return _dio
2020-05-17 10:18:09 +00:00
.get(
Configuration.instance.getHttpEndpoint() + "/photos/faces",
options: Options(
headers: {"X-Auth-Token": Configuration.instance.getToken()}),
)
2020-04-05 12:22:38 +00:00
.then((response) => (response.data["faces"] as List)
.map((face) => new Face.fromJson(face))
.toList())
.catchError(_onError);
}
2020-04-17 08:17:37 +00:00
Future<List<Photo>> getFaceSearchResults(Face face) async {
2020-06-02 14:24:37 +00:00
final result = await _dio
2020-05-17 10:18:09 +00:00
.get(
2020-06-02 14:24:37 +00:00
Configuration.instance.getHttpEndpoint() +
"/photos/search/face/" +
face.faceID.toString(),
queryParameters: {
"limit": 100,
},
options:
Options(headers: {"X-Auth-Token": Configuration.instance.getToken()}),
)
.then((response) {
return (response.data["result"] as List)
.map((p) => Photo.fromJson(p))
.toList();
}).catchError(_onError);
final photos = List<Photo>();
for (Photo photo in result) {
try {
photos.add(await PhotoDB.instance.getMatchingPhoto(photo.localId,
photo.title, photo.deviceFolder, photo.createTimestamp));
} catch (e) {
// Not available locally
photos.add(photo);
}
}
return photos;
2020-04-05 14:00:44 +00:00
}
2020-04-05 12:22:38 +00:00
void _onError(error) {
2020-06-02 14:24:37 +00:00
_logger.severe(error);
2020-04-05 12:22:38 +00:00
}
}