subject-two/entry/src/main/ets/common/service/videoService.ts

import mediaLibrary from '@ohos.multimedia.mediaLibrary'
import onvifclient from '@ohos.onvifclient';
import fs from '@ohos.file.fs'

// @ts-ignore
import photoAccessHelper from '@ohos.file.photoAccessHelper'
import dataSharePredicates from '@ohos.data.dataSharePredicates'
import { getCurrentTime } from '../utils/tools'
// import rtsp_server from '@ohos.rtsprecord';
import record from '@ohos.rtsprecord';
import { FileHelper } from './FileHelper';

// const rtsp_server = record.createServer();
//开始录屏
const FILE_ASSET_FETCH_COLUMNS = [photoAccessHelper.PhotoKeys.URI,
photoAccessHelper.PhotoKeys.PHOTO_TYPE,
photoAccessHelper.PhotoKeys.DISPLAY_NAME,
photoAccessHelper.PhotoKeys.SIZE,
photoAccessHelper.PhotoKeys.DATE_ADDED,
photoAccessHelper.PhotoKeys.DATE_MODIFIED,
photoAccessHelper.PhotoKeys.DURATION,
photoAccessHelper.PhotoKeys.WIDTH,
photoAccessHelper.PhotoKeys.HEIGHT,
photoAccessHelper.PhotoKeys.DATE_TAKEN,
photoAccessHelper.PhotoKeys.ORIENTATION,
photoAccessHelper.PhotoKeys.FAVORITE,
photoAccessHelper.PhotoKeys.TITLE,
photoAccessHelper.PhotoKeys.POSITION,
photoAccessHelper.PhotoKeys.DATE_TRASHED,
photoAccessHelper.PhotoKeys.HIDDEN];

export async function startRecordVideo(param, td, context, dir) {
  return new Promise((reslove, reject) => {
    const rtsp_server = record.createServer();
    var video_uri = `rtsp://${param.userName}:${param.pwd}@${param.ip}:${param.port}/h264/ch${td}/main/av_stream`;
    if (rtsp_server != null) {
      console.log(`Rtsprecord startRecordVideo begin,video_uri`, video_uri);
      const num = Math.floor(Math.random() * 10000)
      const fileName = `movie_record${num}.mp4`
      // @ts-ignore
      var recordResult = rtsp_server.startRecordVideo(context, video_uri, fileName, dir);
      const handleId = recordResult.dataInt;
      console.log(`Rtsprecord startRecordVideo begin,handleId`, handleId);

      reslove(handleId)
      console.log(`Rtsprecord startRecordVideo record_handle` + recordResult.dataInt);
    }
    else {
      console.log(`Rtsprecord startRecordVideo rtsp_server isnull`);
      reslove(0)
    }
    console.log(`Rtsprecord startRecordVideo end`);
  })

}

//结束录屏
export async function endRecordVideo(record_handle) {
  const rtsp_server = record.createServer();
  if (rtsp_server != null && record_handle > 0) {
    console.log(`Rtsprecord endRecordVideo begin`);
    // @ts-ignore
    var recordResult = rtsp_server.endRecordVideo(record_handle);
    console.log(`Rtsprecord startRecordVideo record_handle` + recordResult.dataInt + ` filename:` + recordResult.fileName);
  }
  else {
    console.log(`Rtsprecord endRecordVideo handleId isnull`);
  }
  // var result = onvifclient.endRecordVideo(record_handle);
  // fs.closeSync(record_handle);
  // file_asset.close(record_handle);

}
/**
 *拍照
 * @param param
 * @param context
 * @param type=0不获取base64
 */
interface  takePhotoParam {
  name?:string,
  base64?:string,
  fileSize?:number,
}
/**
 *
 * @param param
 * @param context
 * @param dir
 * @param flag=1 //1同步调用 0异步线程调用
 * @param callback
 * @returns
 */
export async function takePhoto(param, context,dir,flag=1,callback?) {

    const rtsp_server = record.createServer();
    var video_uri = `rtsp://${param.userName}:${param.pwd}@${param.ip}:${param.port}/h264/ch${param.pztd}/main/av_stream`;

    const num = Math.floor(Math.random() * 10000)
    const fileName = `picture_record${num}.jpg`
    console.log(`baoyihu Rtsprecord baohaowen getVideoSnapshot fileName：` + fileName);
    // @ts-ignore
    // var snapResult = rtsp_server.getVideoSnapshot(context, video_uri, '', dir);
  if(flag==0){
    rtsp_server.getVideoSnapshot(context, video_uri, fileName,dir,false,(err,snapResult)=>{
      console.log(`baoyihu getVideoSnapshot  return `);
      console.log(`baoyihu getVideoSnapshot  file_name：` + snapResult.fileName);
      console.log(`baoyihu getVideoSnapshot  file_path：` + snapResult.filePath);
      console.log(`baoyihu getVideoSnapshot  fileSize：` + snapResult.fileSize);
      callback({base64: snapResult.dataString,name:snapResult.fileName,fileSize:snapResult.fileSize })
    });
  }else{
    return new Promise<takePhotoParam>((resolve,reject)=>{
      rtsp_server.getVideoSnapshot(context, video_uri, fileName,dir,true,(err,snapResult)=>{
        resolve({base64: snapResult.dataString,name:snapResult.fileName,fileSize:snapResult.fileSize })
      });
    })
  }


}

export async function deleteAllFileByPiC(dirName) {
  const fileHelper = new FileHelper();
  fileHelper.deleteFileOfAlbum(dirName,1);
}

export async function deleteAllVideos(context,type: photoAccessHelper.AlbumType, subType: photoAccessHelper.AlbumSubtype): Promise<void> {
  let fetchResult: photoAccessHelper.FetchResult<photoAccessHelper.Album> = null;
  try {
    const userFileMgr = photoAccessHelper.getPhotoAccessHelper(context);
    fetchResult = await userFileMgr.getAlbums(type, subType);
    console.log('baoyihu 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;
      let count = 0;
      try {
        let predicates = new dataSharePredicates.DataSharePredicates();
        let fetchOptions = {
          fetchColumns: FILE_ASSET_FETCH_COLUMNS,
          predicates: predicates
        };
        photoFetchResult = await albumAsset.getAssets(fetchOptions);
        count = photoFetchResult.getCount();
        console.log('baoyihu photoFetchResult count: ' + count);
        var all_fileAsset = await photoFetchResult.getAllObjects();
        var uri_array = []
        console.log('baoyihu batch delete begin');
        for (let onfile of all_fileAsset) {
          console.log('baoyihu push one uri : ' + onfile.uri);
          uri_array.push(onfile.uri);
        }
        await userFileMgr.deleteAssets(uri_array);

        console.log('baoyihu batch delete end');
      } catch (err) {
        console.log('baoyihu get Album getPhotoAssets failed with err: ' + err);
      } finally {
        if (photoFetchResult != null) {
          photoFetchResult.close();
        }
      }
    }
  } catch (err) {
    console.log('baoyihu get Album fetchResult failed with err: ' + err);
  } finally {
    if (fetchResult != null) {
      fetchResult.close();
    }
  }
}
// export async function takePhoto(param, context, type = 0, dir) {
//   return new Promise((reslove, reject) => {
//     console.log('mmmmmmmttt,04', type)
//
//     // const rtsp_server = record.createServer();
//     var video_uri = `rtsp://${param.userName}:${param.pwd}@${param.ip}:${param.port}/h264/ch${param.pztd}/main/av_stream`;
//     // var snapResult = rtsp_server.getVideoSnapshot(context, video_uri, dir);
//     // console.log(`baoyihu Rtsprecord getVideoSnapshot file：` + snapResult.dataString);
//
//     console.log(`baoyihu Rtsprecord getVideoSnapshot begin：`);
//     // const num = Math.floor(Math.random() * 10000)
//     console.log('mmmmmmmttt,05', video_uri)
//
//     const num = Math.floor(Math.random() * 10000)
//     const fileName = `picture_record${num}.jpg`
//     console.log(`baoyihu Rtsprecord baohaowen getVideoSnapshot fileName：` + fileName);
//     // @ts-ignore
//     var snapResult = rtsp_server.getVideoSnapshot(context, video_uri, fileName, dir, (ret) => {
//       const str = ret.dataString
//       if (type == 1) {
//         console.log('mmmmmmmttt',str.length)
//         reslove(str)
//       } else {
//         reslove('')
//       }
//     })
//     // console.log('mmmmmmmttt,06',snapResult.dataString)
//
//     // console.log(`baoyihu Rtsprecord getVideoSnapshot file：` + snapResult.fileName);
//     // console.log(`baoyihu Rtsprecord getVideoSnapshot dataLen：` + snapResult.dataString.length);
//     // console.log('datadata1',snapResult.dataString.length)
//   })
//
// }