immich/mobile/lib/utils/image_url_builder.dart

import 'package:immich_mobile/constants/constants.dart';
import 'package:immich_mobile/domain/models/store.model.dart';
import 'package:immich_mobile/entities/album.entity.dart';
import 'package:immich_mobile/entities/asset.entity.dart';
import 'package:immich_mobile/entities/store.entity.dart';
import 'package:openapi/api.dart';

String getThumbnailUrl(final Asset asset, {AssetMediaSize type = AssetMediaSize.thumbnail}) {
  return getThumbnailUrlForRemoteId(asset.remoteId!, type: type);
}

String getThumbnailCacheKey(final Asset asset, {AssetMediaSize type = AssetMediaSize.thumbnail}) {
  return getThumbnailCacheKeyForRemoteId(asset.remoteId!, type: type);
}

String getThumbnailCacheKeyForRemoteId(final String id, {AssetMediaSize type = AssetMediaSize.thumbnail}) {
  if (type == AssetMediaSize.thumbnail) {
    return 'thumbnail-image-$id';
  } else {
    return '${id}_previewStage';
  }
}

String getAlbumThumbnailUrl(final Album album, {AssetMediaSize type = AssetMediaSize.thumbnail}) {
  if (album.thumbnail.value?.remoteId == null) {
    return '';
  }
  return getThumbnailUrlForRemoteId(album.thumbnail.value!.remoteId!, type: type);
}

String getAlbumThumbNailCacheKey(final Album album, {AssetMediaSize type = AssetMediaSize.thumbnail}) {
  if (album.thumbnail.value?.remoteId == null) {
    return '';
  }
  return getThumbnailCacheKeyForRemoteId(album.thumbnail.value!.remoteId!, type: type);
}

String getOriginalUrlForRemoteId(final String id) {
  return ImageUrlBuilder.build('/assets/$id/original');
}

String getImageCacheKey(final Asset asset) {
  // Assets from response DTOs do not have an isar id, querying which would give us the default autoIncrement id
  final isFromDto = asset.id == noDbId;
  return '${isFromDto ? asset.remoteId : asset.id}_fullStage';
}

String getThumbnailUrlForRemoteId(final String id, {AssetMediaSize type = AssetMediaSize.thumbnail}) {
  return ImageUrlBuilder.build('/assets/$id/thumbnail?size=${type.value}');
}

String getPreviewUrlForRemoteId(final String id) {
  return ImageUrlBuilder.build('/assets/$id/thumbnail?size=${AssetMediaSize.preview}');
}

String getPlaybackUrlForRemoteId(final String id) {
  return ImageUrlBuilder.build('/assets/$id/video/playback?');
}

String getFaceThumbnailUrl(final String personId) {
  return ImageUrlBuilder.build('/people/$personId/thumbnail');
}

class ImageUrlBuilder {
  static String? host;
  static Map<String, String>? queryParams;

  static void setHost(String? host) {
    ImageUrlBuilder.host = host;
  }

  static bool isSharedLink() {
    return ImageUrlBuilder.host != null && ImageUrlBuilder.queryParams!.containsKey('key');
  }

  static void setParameter(String key, String value) {
    ImageUrlBuilder.queryParams ??= {};
    ImageUrlBuilder.queryParams![key] = value;
  }

  static String build(String path) {
    final endpoint = host ?? Store.get(StoreKey.serverEndpoint);

    final uri = Uri.parse('$endpoint$path');
    if (queryParams == null || queryParams!.isEmpty) {
      return uri.toString();
    }
    final updatedUri = uri.replace(queryParameters: {...uri.queryParameters, ...queryParams!});
    return updatedUri.toString();
  }

  static void clear() {
    ImageUrlBuilder.host = null;
    ImageUrlBuilder.queryParams = null;
  }
}