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 configAddress from '../../common/utils/FileConfig'
import { getCurrentTime } from '../utils/tools'
// import rtsp_server from '@ohos.rtsprecord';
import record from '@ohos.rtsprecord';
// 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
 */
export async function takePhoto(param, context, type = 0, 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${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 (type == 1) {
      reslove(snapResult.dataString)
    } else {
      reslove('')
    }
  })

}

export async function deleteAllFileByPiC(dir, context) {
  const userFileMgr = photoAccessHelper.getPhotoAccessHelper(context);

  console.log('baoyihu deleteAllPictures');
  let photoFetchResult = null;
  try {
    let predicates = new dataSharePredicates.DataSharePredicates();
    let fetchOptions = {
      fetchColumns: FILE_ASSET_FETCH_COLUMNS,
      predicates: predicates
    };
    photoFetchResult = await userFileMgr.getAssets(fetchOptions);

    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 + ', photoType : ' + onfile.photoType + ', displayName : ' + onfile.displayName);
      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();
    }
  }
}

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)
//   })
//
// }