OpenHarmony-v6.0-Release/s

// @ts-nocheck
/*
 * Copyright (c) 2023 Shenzhen Kaihong Digital Industry Development Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import fileIO from '@ohos.fileio';
import { stashOrGetObject } from '../utils/SingleInstanceUtils';
import { Log } from '../utils/Log';
import { hiSysEventDataQueryTimedOut } from '../utils/hisysEventUtil';
import photoAccessHelper from '@ohos.file.photoAccessHelper';
import dataSharePredicates from '@ohos.data.dataSharePredicates';
import { MediaConstants } from '../constants/MediaConstants'
import { getSystemAlbumDisplayName } from './UserFileDataHelper';
import { SimpleAlbumDataItem } from '../common/SimpleAlbumDataItem';
import bundleManager from '@ohos.bundle.bundleManager';

const TAG = 'UserFileModel';

export class UserFileModelItem {
  fileAsset: photoAccessHelper.PhotoAsset;
  counts: number;
}

class UserFileModel {
  private userFileMgr: photoAccessHelper.PhotoAccessHelper = undefined;

  constructor() {
  }

  onCreate(context): void {
    if (this.userFileMgr == undefined) {
      this.userFileMgr = photoAccessHelper.getPhotoAccessHelper(context);
    }
  }

  getUserFileMgr(): photoAccessHelper.PhotoAccessHelper {
    return this.userFileMgr;
  }

  async createOne(displayName: string, albumUri: string): Promise<photoAccessHelper.PhotoAsset> {
    Log.info(TAG, 'createOne displayName:' + displayName + ' albumUri: ' + albumUri);
    let fileAsset = await this.userFileMgr.createAsset(displayName, albumUri);
    let album = await this.getUserAlbumItemByUri(albumUri);
    await album.addAssets([fileAsset]);
    return fileAsset;
  }

  async getUserAlbumItemByUri(uri: string): Promise<photoAccessHelper.Album> {
    let fetchResult: photoAccessHelper.FetchResult<photoAccessHelper.Album> = null;
    let album: photoAccessHelper.Album = null;
    try {
      Log.info(TAG, 'getUserAlbumItemByUri');
      let predicates = new dataSharePredicates.DataSharePredicates();
      predicates.equalTo(photoAccessHelper.AlbumKeys.URI, uri);
      let fetchOptions = {
        fetchColumns: MediaConstants.EMPTY_FETCH_COLUMNS,
        predicates: predicates
      };
      fetchResult = await this.userFileMgr.getAlbums(photoAccessHelper.AlbumType.USER, photoAccessHelper.AlbumSubtype.USER_GENERIC, fetchOptions);
      Log.info(TAG, 'get Album fetchResult, count: ' + fetchResult.getCount());
      if (fetchResult.getCount() > 0) {
        album = fetchResult.getFirstObject();
      }
    } catch (err) {
      Log.error(TAG, 'get Album fetchResult failed with err: ' + err);
    } finally {
      if (fetchResult != null) {
        fetchResult.close();
      }
    }
    return album;
  }

  async copyOne(source: photoAccessHelper.PhotoAsset, target: photoAccessHelper.PhotoAsset): Promise<void> {
    Log.info(TAG, 'copy start: src:' + source.uri + ' target: ' + target.uri);
    let fd: Number = await this.openAsset('R', source);
    if (fd <= 0) {
      throw new Error('fd is invalid');
    }

    let targetFd: Number = await this.openAsset('RW', target);
    if (targetFd <= 0) {
      throw new Error('targetFd is invalid');
    }

    await fileIO.copyFile(fd, targetFd);

    await this.closeAsset(fd, source);
    await this.closeAsset(targetFd, target);

    Log.debug(TAG, 'copy end');
  }

  async deleteOne(uri: string): Promise<void> {
    Log.debug(TAG, 'deleteAsset uri: ' + uri);
    try {
      await this.userFileMgr.deleteAssets([uri]);
    } catch (err) {
      Log.error(TAG, 'deleteOne with error: ' + err);
    }
  }

  async recover(fileAsset: photoAccessHelper.PhotoAsset): Promise<void> {
    let albumFetchResult = null;
    try {
      Log.debug(TAG, 'recoverPhotoAssetsDemoPromise:' + fileAsset.displayName);
      albumFetchResult = await this.userFileMgr.getAlbums(photoAccessHelper.AlbumType.SYSTEM, photoAccessHelper.AlbumSubtype.TRASH);
      let album = await albumFetchResult.getFirstObject();
      Log.debug(TAG, 'album  get');
      album.recoverAssets([fileAsset]).then(() => {
        Log.debug(TAG, 'album recoverPhotoAssets successfully');
      }).catch((err) => {
        Log.info(TAG, 'album recoverPhotoAssets failed with error: ' + err);
      });
    } catch (err) {
      Log.error(TAG, 'recoverPhotoAssetsDemoPromise failed with error: ' + err);
    } finally {
      if (albumFetchResult != null) {
        albumFetchResult.close();
      }
    }
  }

  async permanentDelete(fileAsset: photoAccessHelper.PhotoAsset): Promise<void> {
    let albumFetchResult = null;
    try {
      Log.debug(TAG, 'permanentDelete');
      albumFetchResult = await this.userFileMgr.getAlbums(photoAccessHelper.AlbumType.SYSTEM, photoAccessHelper.AlbumSubtype.TRASH);
      let album = await albumFetchResult.getFirstObject();
      album.deleteAssets([fileAsset]).then(() => {
        Log.debug('album deletePhotoAssets successfully');
      }).catch((err) => {
        Log.error(TAG, 'album deletePhotoAssets failed with error: ' + err);
      });
    } catch (err) {
      Log.error(TAG, 'permanentDelete failed with error: ' + err);
    } finally {
      if (albumFetchResult != null) {
        albumFetchResult.close();
      }
    }
  }

  async createAlbum(albumName: string): Promise<SimpleAlbumDataItem> {
    let newAlbum: SimpleAlbumDataItem = undefined;
    try {
      let album = await this.userFileMgr.createAlbum(albumName);
      newAlbum = new SimpleAlbumDataItem(MediaConstants.ALBUM_ID_USER, albumName, album.albumUri,
        '', '', MediaConstants.ALBUM_TYPE_USER, MediaConstants.ALBUM_SUBTYPE_USER_GENERIC);
    } catch (err) {
      Log.error(TAG, 'createAlbum failed with error: ' + err);
    }
    return newAlbum;
  }

  async deleteAlbum(albumName: string): Promise<void> {
    let fetchResult: photoAccessHelper.FetchResult<photoAccessHelper.Album> = null;
    try {
      let predicates = new dataSharePredicates.DataSharePredicates();
      predicates.equalTo(photoAccessHelper.AlbumKeys.ALBUM_NAME, albumName);
      let fetchOptions = {
        fetchColumns: MediaConstants.EMPTY_FETCH_COLUMNS,
        predicates: predicates
      };
      fetchResult = await this.userFileMgr.getAlbums(photoAccessHelper.AlbumType.USER, photoAccessHelper.AlbumSubtype.USER_GENERIC, fetchOptions);
      for (let i = 0; i < fetchResult.getCount(); i++) {
        let albumAsset = await fetchResult.getObjectByPosition(i);
        Log.info(TAG, 'deleteAlbum albumUri' + albumAsset.albumUri);
        Log.info(TAG, 'deleteAlbum albumName' + albumAsset.albumName);
        this.userFileMgr.deleteAlbums([albumAsset]).then(() => {
          Log.info(TAG, 'deletePhotoAlbumsPromise successfully');
        }).catch((err) => {
          Log.error(TAG, 'deletePhotoAlbumsPromise failed with err: ' + err);
        });
      }
    } catch (err) {
      Log.error(TAG, 'deletePhotoAlbumsPromise failed with error: ' + err);
    } finally {
      if (fetchResult != null) {
        fetchResult.close();
      }
    }
  }

  async deleteAll(fetchOption: photoAccessHelper.FetchOptions): Promise<void> {
    Log.info(TAG, 'deleteAll');
    let fetchFileResult: photoAccessHelper.FetchResult = null;
    try {
      fetchFileResult = await this.userFileMgr.getAssets(fetchOption);
      Log.debug(TAG, 'deleteAll getPhotoAssets');
      let deleteAllGetAllObject = hiSysEventDataQueryTimedOut('deleteAllGetAllObject');
      let fileAssets: photoAccessHelper.PhotoAsset[] = await fetchFileResult.getAllObject();
      clearTimeout(deleteAllGetAllObject);
      for (let i = 0;i < fileAssets.length; i++) {
        await this.deleteOne(fileAssets[i].uri);
      }
      Log.debug(TAG, 'deleteAll getFirstObject');
    } catch (err) {
      Log.error(TAG, 'deleteAll error:' + JSON.stringify(err));
    } finally {
      if (fetchFileResult != null) {
        fetchFileResult.close();
      }
    }
    Log.debug(TAG, 'deleteAll finish');
  }

  async getAllMediaItems(): Promise<photoAccessHelper.PhotoAsset[]> {
    let fileAssets: photoAccessHelper.PhotoAsset[] = [];
    let photoFetchResult: photoAccessHelper.FetchResult<photoAccessHelper.PhotoAsset> = null;
    try {
      let predicates = new dataSharePredicates.DataSharePredicates();
      predicates.orderByDesc(photoAccessHelper.PhotoKeys.DATE_ADDED);
      let emptyFetchOption = {
        fetchColumns: MediaConstants.FILE_ASSET_FETCH_COLUMNS,
        predicates: predicates
      };
      photoFetchResult = await this.userFileMgr.getAssets(emptyFetchOption);
      Log.info(TAG, 'getAllMediaItems count: ' + photoFetchResult.getCount());
      for (let i = 0;i < photoFetchResult.getCount(); i++) {
        fileAssets.push(await photoFetchResult.getObjectByPosition(i));
      }
    } catch (err) {
      Log.error(TAG, 'getAllMediaItems failed with err: ' + err);
    } finally {
      if (photoFetchResult != null) {
        photoFetchResult.close();
      }
    }
    return fileAssets;
  }

  async getAllMovingPhotoItems(): Promise<photoAccessHelper.PhotoAsset[]> {
    let fileAssets: photoAccessHelper.PhotoAsset[] = [];
    let photoFetchResult: photoAccessHelper.FetchResult<photoAccessHelper.PhotoAsset> = null;
    try {
      let predicates = new dataSharePredicates.DataSharePredicates();
      predicates.equalTo(MediaConstants.PHOTO_SUBTYPE, MediaConstants.MOVING_PHOTO);
      predicates.orderByDesc(photoAccessHelper.PhotoKeys.DATE_ADDED);
      let fetchOptions: photoAccessHelper.FetchOptions = {
        fetchColumns: MediaConstants.FILE_ASSET_FETCH_COLUMNS,
        predicates: predicates
      };
      photoFetchResult = await this.userFileMgr.getAssets(fetchOptions);
      Log.info(TAG, 'getAllMovingPhotoItems count: ' + photoFetchResult.getCount());
      for (let i = 0; i < photoFetchResult.getCount(); i++) {
        fileAssets.push(await photoFetchResult.getObjectByPosition(i));
      }
    } catch (err) {
      Log.error(TAG, 'getAllMovingPhotoItems failed with err: ' + err);
    } finally {
      if (photoFetchResult != null) {
        photoFetchResult.close();
      }
    }
    return fileAssets;
  }

  async getAllMediaItemsByType(type: number, subType: number, albumFetchOption, fileFetchOption): Promise<photoAccessHelper.PhotoAsset[]> {
    let fileAssets: photoAccessHelper.PhotoAsset[] = [];
    let fetchResult: photoAccessHelper.FetchResult<photoAccessHelper.Album> = null;
    try {
      Log.info(TAG, 'getAllMediaItemsByUserFile');
      Log.info(TAG, 'type:' + type);
      Log.info(TAG, 'subType:' + subType);
      if (type === photoAccessHelper.AlbumType.USER && albumFetchOption != null) {
        Log.info(TAG, 'albumFetchOption ！= null');
        fetchResult = await this.userFileMgr.getAlbums(type, subType, albumFetchOption);
      } else {
        fetchResult = await this.userFileMgr.getAlbums(type, subType);
      }
      Log.info(TAG, 'get Album fetchResult, count: ' + fetchResult.getCount());
      for (let i = 0; i < fetchResult.getCount(); i++) {
        let albumAsset: photoAccessHelper.Album = await fetchResult.getObjectByPosition(i);
        let photoFetchResult: photoAccessHelper.FetchResult<photoAccessHelper.PhotoAsset> = null;
        try {
          photoFetchResult = await albumAsset.getAssets(fileFetchOption);
          for (let i = 0; i < photoFetchResult.getCount(); i++) {
            let photoAsset = await photoFetchResult.getObjectByPosition(i);
            fileAssets.push(photoAsset);
            Log.info(TAG, 'getPhotoAssets successfully, file displayName: ' + photoAsset.displayName);
          }
        } catch (err) {
          Log.info(TAG, 'get Album getPhotoAssets failed with err: ' + err);
        } finally {
          if (photoFetchResult != null) {
            photoFetchResult.close();
          }
        }
      }
    } catch (err) {
      Log.error(TAG, 'get Album fetchResult failed with err: ' + err);
    } finally {
      if (fetchResult != null) {
        fetchResult.close();
      }
    }
    Log.info(TAG, 'fileAssets: ' + fileAssets.length);
    return fileAssets;
  }

  async getMediaItemByUriFromTrash(uri: string): Promise<photoAccessHelper.PhotoAsset> {
    Log.info(TAG, 'getMediaItemByUriFromTrash');
    Log.info(TAG, 'uri:' + uri);
    let albumFetchResult: photoAccessHelper.FetchResult<photoAccessHelper.Album> = null;
    let fileAsset: photoAccessHelper.PhotoAsset = null;
    try {
      albumFetchResult = await this.userFileMgr.getAlbums(photoAccessHelper.AlbumType.SYSTEM, photoAccessHelper.AlbumSubtype.TRASH);
      let albumAsset = await albumFetchResult.getFirstObject();
      let photoFetchResult: photoAccessHelper.FetchResult<photoAccessHelper.PhotoAsset> = null;
      try {
        let predicates = new dataSharePredicates.DataSharePredicates();
        let emptyFetchOption = {
          fetchColumns: MediaConstants.FILE_ASSET_FETCH_COLUMNS,
          predicates: predicates
        };
        photoFetchResult = await albumAsset.getAssets(emptyFetchOption);
        for (let i = 0; i < photoFetchResult.getCount(); i++) {
          let photoAsset = await photoFetchResult.getObjectByPosition(i);
          if (photoAsset.uri === uri) {
            fileAsset = photoAsset;
            Log.info(TAG, 'getMediaItemByUriFromTrash success: ' + photoAsset.displayName);
            break;
          }
        }
      } catch (err) {
        Log.info(TAG, 'getMediaItemByUriFromTrash getPhotoAssets failed with err: ' + err);
      } finally {
        if (photoFetchResult != null) {
          photoFetchResult.close();
        }
      }
    } catch (err) {
      Log.error(TAG, 'getMediaItemByUriFromTrash failed with error: ' + err);
    } finally {
      if (albumFetchResult != null) {
        albumFetchResult.close();
      }
    }
    return fileAsset;
  }

  async getMediaItemByUri(uri: string): Promise<photoAccessHelper.PhotoAsset> {
    Log.info(TAG, 'getMediaItemByUri');
    Log.info(TAG, 'uri:' + uri);
    let predicates = new dataSharePredicates.DataSharePredicates();
    predicates.equalTo(photoAccessHelper.PhotoKeys.URI, uri);
    let fetchOptions = {
      fetchColumns: MediaConstants.FILE_ASSET_FETCH_COLUMNS,
      predicates: predicates
    };
    let fileAsset: photoAccessHelper.PhotoAsset = null;
    let fetchResult: photoAccessHelper.FetchResult<photoAccessHelper.PhotoAsset> = null
    try {
      Log.info(TAG, 'fetchResult start');
      fetchResult = await this.userFileMgr.getAssets(fetchOptions);
      Log.info(TAG, 'fetchResult count ' + fetchResult.getCount());
      fileAsset = await fetchResult.getFirstObject();
      Log.info(TAG, 'fileAsset ' + fileAsset.displayName);
    } catch (err) {
      Log.error(TAG, 'getMediaItemByUri failed, message = ' + err);
    } finally {
      if (fetchResult != null) {
        fetchResult.close();
      }
    }
    return fileAsset;
  }

  async getMediaItemCountsByDisplayName(displayName: string): Promise<number> {
    Log.info(TAG, 'getMediaItemCountsByDisplayName');
    let count = 0;
    let fetchFileResult: photoAccessHelper.FetchResult = null;
    let predicates = new dataSharePredicates.DataSharePredicates();
    predicates.equalTo(photoAccessHelper.PhotoKeys.DISPLAY_NAME, displayName);
    let fetchOptions = {
      fetchColumns: MediaConstants.FILE_ASSET_FETCH_COLUMNS,
      predicates: predicates
    };
    try {
      fetchFileResult = await this.userFileMgr.getAssets(fetchOptions);
      Log.debug(TAG, 'getMediaItemCountsByDisplayName getPhotoAssets');
      count = await fetchFileResult.getCount();
    } catch (err) {
      Log.error(TAG, 'getMediaItemCountsByDisplayName error:' + JSON.stringify(err));
    } finally {
      if (fetchFileResult != null) {
        fetchFileResult.close();
      }
    }
    Log.debug(TAG, 'getMediaItemCountsByDisplayName finish');
    return count;
  }

  async getUserAlbumCountByName(albumName: string): Promise<number> {
    Log.info(TAG, 'getUserAlbumCountByName');
    Log.info(TAG, 'album_name：' + albumName);
    let count = 0;
    let systemAlbums = await getSystemAlbumDisplayName();
    if (systemAlbums.indexOf(albumName) >= 0) {
      return 1;
    }
    let fetchResult: photoAccessHelper.FetchResult = null;
    try {
      let predicates = new dataSharePredicates.DataSharePredicates();
      predicates.equalTo(photoAccessHelper.AlbumKeys.ALBUM_NAME, albumName);
      let fetchOptions = {
        fetchColumns: MediaConstants.EMPTY_FETCH_COLUMNS,
        predicates: predicates
      };
      fetchResult = await this.userFileMgr.getAlbums(photoAccessHelper.AlbumType.USER, photoAccessHelper.AlbumSubtype.USER_GENERIC, fetchOptions);
      Log.info(TAG, 'fetchResult' + fetchResult.getCount());
      count = fetchResult.getCount();
    } catch (err) {
      Log.error(TAG, 'getUserAlbumCountByName error:' + JSON.stringify(err));
    } finally {
      if (fetchResult != null) {
        fetchResult.close();
      }
    }
    Log.debug(TAG, 'getUserAlbumCountByName finish');
    return count;
  }

  async getActivePeers(): Promise<photoAccessHelper.PeerInfo[]> {
    Log.info(TAG, 'getActivePeers');
    let peers: photoAccessHelper.PeerInfo[] = [];
    try {
      let getActivePeers = hiSysEventDataQueryTimedOut('getActivePeers')
      peers = await this.userFileMgr.getActivePeers();
      clearTimeout(getActivePeers);
    } catch (err) {
      Log.error(TAG, 'getActivePeers error:' + JSON.stringify(err));
    }
    Log.debug(TAG, 'getActivePeers finish');
    return peers;
  }

  async getUserAlbumByName(albumName: string): Promise<photoAccessHelper.Album> {
    Log.info(TAG, 'getUserAlbumByName');
    Log.info(TAG, 'album_name' + albumName);
    let fetchResult: photoAccessHelper.FetchResult = null;
    let album = null;
    try {
      let predicates = new dataSharePredicates.DataSharePredicates();
      predicates.equalTo(photoAccessHelper.AlbumKeys.ALBUM_NAME, albumName);
      let fetchOptions = {
        fetchColumns: MediaConstants.EMPTY_FETCH_COLUMNS,
        predicates: predicates
      };
      fetchResult = await this.userFileMgr.getAlbums(photoAccessHelper.AlbumType.USER, photoAccessHelper.AlbumSubtype.USER_GENERIC, fetchOptions);
      Log.info(TAG, 'fetchResult' + fetchResult.getCount());
      if (fetchResult.getCount() > 0) {
        album = fetchResult.getFirstObject();
      }
    } catch (err) {
      Log.error(TAG, 'getUserAlbumByName error:' + JSON.stringify(err));
    } finally {
      if (fetchResult != null) {
        fetchResult.close();
      }
    }
    Log.debug(TAG, 'getUserAlbumByName finish');
    return album;
  }

  async getAlbums(fetchOption: photoAccessHelper.FetchOptions): Promise<photoAccessHelper.Album[]> {
    Log.info(TAG, 'getAlbums');
    let albums: photoAccessHelper.Album[] = [];
    try {
      albums = await this.userFileMgr.getAlbums(fetchOption);
      Log.info(TAG, 'getAlbums albums ' + albums.getCounts());
    } catch (err) {
      Log.error(TAG, 'getAlbums error:' + JSON.stringify(err));
    }
    Log.debug(TAG, 'getAlbums finish');
    return albums;
  }

  async openAsset(mode: string, fileAsset: photoAccessHelper.PhotoAsset): Promise<number> {
    Log.debug(TAG, 'openAsset start');
    let fd: number = await fileAsset.open(mode);
    Log.info(TAG, 'openAsset end. fd: ' + fd);
    if (fd <= 0) {
      Log.info(TAG, 'openAsset Fail');
    }
    return fd;
  }

  async closeAsset(fd: number, fileAsset: photoAccessHelper.PhotoAsset): Promise<void> {
    Log.debug(TAG, 'closeAsset start');
    await fileAsset.close(fd);
  }

  async addPhotoToAlbumByUserFileMgr(albumUri: string, uri: string): Promise<void> {
    Log.info(TAG, 'addPhotoAssetsDemoPromise');
    Log.info(TAG, 'albumUri' + albumUri);
    Log.info(TAG, 'mediaItem.uri' + uri);
    try {
      let album = await this.getUserAlbumItemByUri(albumUri);
      let predicates = new dataSharePredicates.DataSharePredicates();
      predicates.equalTo(photoAccessHelper.PhotoKeys.URI, uri);
      let fetchOptions = {
        fetchColumns: MediaConstants.FILE_ASSET_FETCH_COLUMNS,
        predicates: predicates
      };
      this.userFileMgr.getAssets(fetchOptions, async (err, fetchResult) => {
        if (fetchResult != undefined) {
          Log.info(TAG, 'fetchResult success');
          let fileAsset = await fetchResult.getFirstObject();
          if (fileAsset != undefined) {
            Log.info(TAG, 'addPhotoToAlbumByUserFileMgr photo displayName : ' + fileAsset.displayName);
            album.addAssets([fileAsset]).then(() => {
              Log.info(TAG, 'album addPhotoAssets successfully');
            }).catch((err) => {
              Log.info(TAG, 'album addPhotoAssets failed with error: ' + err);
            });
          }
          fetchResult.close();
        } else {
          Log.info(TAG, 'fetchResult fail' + err);
        }
      });
    } catch (err) {
      Log.error(TAG, 'addPhotoAssetsDemoPromise failed with error: ' + err);
    }
  }

  async hideSensitives(type: number, uris: string[]): Promise<void> {
    let flags = bundleManager.BundleFlag.GET_BUNDLE_INFO_WITH_APPLICATION;
    let bundleInfo = bundleManager.getBundleInfoForSelfSync(flags);
    let tokenId = bundleInfo.appInfo.accessTokenId;
    for (let uri of uris) {
      try {
        let grantResult = await this.userFileMgr.grantPhotoUriPermission(tokenId, uri, 1, type);
        Log.info(TAG, `grantPhotoUriPermission success, result: ${grantResult}`);
      }catch (err) {
        Log.error(TAG, 'grantPhotoUriPermission failed with error: ' + JSON.stringify(err));
      }
    }
  }
}

export let userFileModel: UserFileModel = stashOrGetObject<UserFileModel>(new UserFileModel(), TAG);