OpenHarmony-v6.0-Release/s

/*
 * Copyright (c) 2022-2025 Huawei Device 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 prompt from '@ohos.promptAction';
import router from '@ohos.router';
import mediaQuery from '@ohos.mediaquery';
import photoAccessHelper from '@ohos.file.photoAccessHelper';
import dataSharePredicates from '@ohos.data.dataSharePredicates';
import Logger from '../data/Logger';

const TAG: string = 'ChoicePhoto';

interface ChiceBuilderParam {
  backGround: string,
  showNumber: number
};

@Entry
@Component
export struct ChoicePhotos {
  @State whichShow: Array<boolean> = [true, false];
  @State medias: Array<photoAccessHelper.PhotoAsset> = [];
  @State choiceShow: Array<Resource> = [$r('app.string.choice_photo'), $r('app.string.choice_video')];
  @State taskShow: Array<Resource> = [$r('app.media.photo'), $r('app.media.video')];
  @State textShow: Array<Resource> = [$r('app.string.photograph'), $r('app.string.take_video')];
  @State isShowChoices: Array<boolean> = new Array(this.medias.length).fill(false);
  @State @Watch('choiceChange') choiceMedias: Array<photoAccessHelper.PhotoAsset> = [];
  @State mediaUris: Array<string> = [];
  @State isChoice: boolean = false;
  @State nextText: string = '';
  @State isLand: boolean = false;
  private listener = mediaQuery.matchMediaSync('screen and (min-aspect-ratio: 1.5) or (orientation: landscape)');
  onLand = (mediaQueryResult: mediaQuery.MediaQueryResult) => {
    if (mediaQueryResult.matches) {
      this.isLand = true
    } else {
      this.isLand = false
    }
  };

  choiceChange() {
    if (this.choiceMedias.length > 0) {
      this.isChoice = true
    } else {
      this.isChoice = false
    }
  }

  @Builder showChoiceBuild($$: ChiceBuilderParam) {
    Column() {
      Text($$.showNumber === 0 ? '' : `${$$.showNumber}`)
        .id(`select${$$.showNumber}`)
        .fontSize(14)
        .fontColor(Color.White)
        .height(20)
        .margin({ top: 2 })
    }
    .width(24)
    .height(24)
    .margin({ top: 6, right: 6 })
    .borderRadius(15)
    .border({ width: 1, style: BorderStyle.Solid, color: Color.White })
    .backgroundColor($$.backGround)
  }

  async getFileAssetsFromType(mediaType: photoAccessHelper.PhotoType) {
    Logger.info(TAG, `getFileAssetsFromType`);
    let context = getContext(this);
    let phAccessHelper = photoAccessHelper.getPhotoAccessHelper(context);
    let predicates: dataSharePredicates.DataSharePredicates = new dataSharePredicates.DataSharePredicates();
    predicates.equalTo('media_type', mediaType);
    let fetchOptions: photoAccessHelper.FetchOptions = {
      fetchColumns: ['date_added', 'duration'],
      predicates: predicates
    };

    try {
      let fetchResult: photoAccessHelper.FetchResult<photoAccessHelper.PhotoAsset> =
        await phAccessHelper.getAssets(fetchOptions);
      Logger.info(TAG, 'fetchResult success');
      if (fetchResult !== undefined) {
        Logger.info(TAG, `fetchFileResult = ${JSON.stringify(fetchResult)}, ${fetchResult.getCount()}`);
        if (fetchResult.getCount() > 0) {
          this.medias = await fetchResult.getAllObjects();
          Logger.info(TAG, `this.medias = ${JSON.stringify(this.medias)}`);
        }
      }
    } catch (err) {
      Logger.error(TAG, 'getAssets failed, message = ' + JSON.stringify(err));
    }
  }

  async convertResourceToString(resource: Resource) {
    Logger.info(TAG, `convertResourceToString result = ${getContext(this).resourceManager.getStringSync(resource)}`);
    return getContext(this).resourceManager.getStringSync(resource);
  }

  getMaxHeight(): ConstraintSizeOptions {
    if (!this.isLand && this.isChoice) {
      return { maxHeight: '64%' };
    } else if (!this.isLand && !this.isChoice) {
      return { maxHeight: '75.5%' };
    } else if (this.isLand && this.isChoice) {
      return { maxHeight: '60%' };
    } else if (this.isLand && !this.isChoice) {
      return { maxHeight: '71%' };
    }else {
      return {};
    }
  }

  async aboutToAppear() {
    Logger.info(TAG, `aboutToAppear`);
    this.listener.on('change', this.onLand);
    this.nextText = await this.convertResourceToString($r('app.string.next'));
    this.getFileAssetsFromType(photoAccessHelper.PhotoType.IMAGE);
  }

  build() {
    Column() {
      Row() {
        Image($r('app.media.back'))
          .width(44)
          .height(24)
          .objectFit(ImageFit.Contain)
          .onClick(() => {
            router.back()
          })

        Row() {
          Text($r('app.string.recently_added'))
            .fontColor(Color.Black)
            .fontSize(22)
            .textAlign(TextAlign.Center)
          Image($r('app.media.drop_down'))
            .width(30)
            .height(32)
            .objectFit(ImageFit.Contain)
        }

        Button(`${this.nextText} ${this.isChoice ? `(${this.choiceMedias.length})` : ''}`)
          .id('nextStep')
          .fontSize(20)
          .height(32)
          .backgroundColor(this.isChoice === true ? '#E92F4F' : '#fffa8e8e')
          .margin({ right: 10 })
          .borderRadius(20)
          .onClick(() => {
            if (this.isChoice === false) {
              return
            }
            this.mediaUris = this.choiceMedias.map((item) => {
              return item.uri
            })
            router.pushUrl({
              url: 'pages/Index',
              params: { mediaUris: this.mediaUris, isShowCamera: true }

            })
          })
      }
      .width('100%')
      .height(35)
      .padding({ left: 14 })
      .margin({ top: 20 })
      .justifyContent(FlexAlign.SpaceBetween)

      Column() {
        Row() {
          ForEach(this.choiceShow, (item: Resource, index) => {
            Column() {
              Text(item)
                .fontSize(20)
                .fontWeight(500)
                .fontColor(this.whichShow[index] === true ? '#0000000' : '#99182431')
                .onClick(() => {
                  this.whichShow.fill(false)
                  this.whichShow[index] = true
                  this.medias = []
                  if (index == 0) {
                    this.getFileAssetsFromType(photoAccessHelper.PhotoType.IMAGE);
                  } else {
                    prompt.showDialog({ message: $r('app.string.user_tip') });
                    this.choiceMedias = [];
                    this.isShowChoices = new Array(this.medias.length).fill(false);
                    this.getFileAssetsFromType(photoAccessHelper.PhotoType.VIDEO);
                  }
                })
              if (this.whichShow[index]) {
                Divider()
                  .vertical(false)
                  .strokeWidth(3)
                  .color('#ffff0000')
                  .lineCap(LineCapStyle.Round)
                  .width('40%')
                  .margin({ top: 4 })
              }
            }
            .width('30%')
            .id(`type${index}`)
          })
        }
        .margin({ top: 20 })
      }

      Scroll() {
        Column() {
          Grid() {
            ForEach(this.medias, (item: photoAccessHelper.PhotoAsset, index) => {
              GridItem() {
                Stack({ alignContent: Alignment.TopEnd }) {
                  Image(item.uri)
                    .id(`image${index + 1}`)
                    .width('100%')
                    .height('100%')
                    .borderRadius(10)
                    .objectFit(ImageFit.Fill)
                  if (this.isShowChoices[index]) {
                    this.showChoiceBuild({ backGround: '#fffc0303', showNumber: this.choiceMedias.indexOf(item) + 1 });
                  } else {
                    this.showChoiceBuild({ backGround:'#ffb7b4b4', showNumber: 0 });
                  }
                }
                .width('100%')
                .height('100%')
                .onClick(() => {
                  this.isShowChoices[index] = !this.isShowChoices[index]
                  if (this.isShowChoices[index]) {
                    if (this.choiceMedias.length > 5) {
                      prompt.showDialog({ message: $r('app.string.choice_number') })
                      this.isShowChoices[index] = !this.isShowChoices[index]
                      return
                    }
                    this.choiceMedias.push(item)
                  } else {
                    if (this.choiceMedias.indexOf(item) != -1) {
                      this.choiceMedias.splice(this.choiceMedias.indexOf(item), 1)
                    }
                  }
                })
              }
              .aspectRatio(1)
            })
          }
          .columnsTemplate('1fr 1fr 1fr 1fr')
          .columnsGap(8)
          .rowsGap(8)
        }
        .height('100%')
        .width('95%')
        .margin({ top: 8 })
      }
      .margin({ top: 20 })
      .width('100%')
      .constraintSize(this.getMaxHeight())
      .backgroundColor('#fff5f3f3')

      if (this.isChoice) {
        Grid() {
          ForEach(this.choiceMedias, (item: photoAccessHelper.PhotoAsset, index) => {
            GridItem() {
              Stack({ alignContent: Alignment.TopEnd }) {
                Image(item.uri)
                  .id(`selected${index + 1}`)
                  .width('100%')
                  .height(70)
                  .borderRadius(10)
                Image($r('app.media.delete'))
                  .id(`deleteImage${index + 1}`)
                  .width(20)
                  .height(20)
                  .margin({ top: 5, right: 5 })
                  .onClick(() => {
                    for (let i = 0;i < this.medias.length; i++) {
                      if (this.medias[i] === this.choiceMedias[index]) {
                        this.isShowChoices[i] = false
                      }
                    }
                    this.choiceMedias.splice(index, 1)
                  })
              }
              .width('100%')
            }
          }, (item: photoAccessHelper.PhotoAsset) => item.uri )
        }
        .columnsTemplate('1fr 1fr 1fr 1fr 1fr 1fr')
        .columnsGap(8)
        .rowsGap(8)
        .margin({ top: 8 })
        .width('95%')
        .height(70)

        Divider()
          .vertical(false)
          .strokeWidth(1)
          .color('#ffd9d5d6')
          .lineCap(LineCapStyle.Round)
          .width('100%')
          .margin({ top: 8 })
      }

      Row() {
        ForEach(this.taskShow, (item: Resource, index) => {
          Column() {
            Image(item)
              .width(30)
              .height(30)

            Text(this.textShow[index])
              .fontSize(14)
              .fontColor('#99182431')
              .margin({ top: 2 })
          }
          .width('50%')
        })
      }
      .margin({ top: 8 })
    }
  }
}