fbmobile/lib/core/viewmodels/upload_model.dart

import 'dart:async';
import 'dart:io';

import 'package:file_picker/file_picker.dart';
import 'package:flutter/material.dart';
import 'package:flutter/services.dart';
import 'package:flutter_translate/flutter_translate.dart';
import 'package:logger/logger.dart';
import 'package:path/path.dart';
import 'package:receive_sharing_intent/receive_sharing_intent.dart';

import '../../locator.dart';
import '../enums/error_code.dart';
import '../enums/refresh_event.dart';
import '../enums/viewstate.dart';
import '../error/rest_service_exception.dart';
import '../error/service_exception.dart';
import '../models/rest/rest_error.dart';
import '../models/rest/uploaded_multi_response.dart';
import '../models/rest/uploaded_response.dart';
import '../services/file_service.dart';
import '../services/link_service.dart';
import '../services/refresh_service.dart';
import '../util/logger.dart';
import 'base_model.dart';

class UploadModel extends BaseModel {
  final Logger _logger = getLogger();
  final FileService _fileService = locator<FileService>();
  final LinkService _linkService = locator<LinkService>();
  final RefreshService _refreshService = locator<RefreshService>();

  TextEditingController _pasteTextController = TextEditingController();
  StreamSubscription _intentDataStreamSubscription;

  bool createMulti = false;
  String fileName;
  List<PlatformFile> paths;
  String _extension;
  bool loadingPath = false;
  String errorMessage;

  TextEditingController get pasteTextController => _pasteTextController;

  void init() {
    // For sharing images coming from outside the app while the app is in the memory
    _intentDataStreamSubscription = ReceiveSharingIntent.getMediaStream().listen((List<SharedMediaFile> value) {
      if (value != null && value.length > 0) {
        setState(ViewState.Busy);
        paths = value.map((sharedFile) {
          return PlatformFile.fromMap({
            'path': sharedFile.path,
            'name': basename(sharedFile.path),
            'size': File(sharedFile.path).lengthSync(),
            'bytes': null
          });
        }).toList();
        setState(ViewState.Idle);
      }
    }, onError: (err) {
      setState(ViewState.Busy);
      errorMessage = translate('upload.retrieval_intent');
      _logger.e('Error while retrieving shared data: $err');
      setState(ViewState.Idle);
    });

    // For sharing images coming from outside the app while the app is closed
    ReceiveSharingIntent.getInitialMedia().then((List<SharedMediaFile> value) {
      if (value != null && value.length > 0) {
        setState(ViewState.Busy);
        paths = value.map((sharedFile) {
          return PlatformFile.fromMap({
            'path': sharedFile.path,
            'name': basename(sharedFile.path),
            'size': File(sharedFile.path).lengthSync(),
            'bytes': null
          });
        }).toList();
        setState(ViewState.Idle);
      }
    });

    // For sharing or opening urls/text coming from outside the app while the app is in the memory
    _intentDataStreamSubscription = ReceiveSharingIntent.getTextStream().listen((String value) {
      if (value != null && value.isNotEmpty) {
        setState(ViewState.Busy);
        pasteTextController.text = value;
        setState(ViewState.Idle);
      }
    }, onError: (err) {
      setState(ViewState.Busy);
      errorMessage = translate('upload.retrieval_intent');
      _logger.e('Error while retrieving shared data: $err');
      setState(ViewState.Idle);
    });

    // For sharing or opening urls/text coming from outside the app while the app is closed
    ReceiveSharingIntent.getInitialText().then((String value) {
      if (value != null && value.isNotEmpty) {
        setState(ViewState.Busy);
        pasteTextController.text = value;
        setState(ViewState.Idle);
      }
    });
  }

  void toggleCreateMulti() {
    setState(ViewState.Busy);
    createMulti = !createMulti;
    setState(ViewState.Idle);
  }

  void openFileExplorer() async {
    setState(ViewState.Busy);
    setStateMessage(translate('upload.file_explorer_open'));
    loadingPath = true;

    try {
      paths = (await FilePicker.platform.pickFiles(
        type: FileType.any,
        allowMultiple: true,
        withData: false,
        withReadStream: true,
        allowedExtensions: (_extension?.isNotEmpty ?? false) ? _extension?.replaceAll(' ', '')?.split(',') : null,
      ))
          ?.files;
    } on PlatformException catch (e) {
      _logger.e('Unsupported operation', e);
    } catch (ex) {
      _logger.e('An unknown error occurred', ex);
    }

    loadingPath = false;
    fileName = paths != null ? paths.map((e) => e.name).toString() : '...';

    setStateMessage(null);
    setState(ViewState.Idle);
  }

  void clearCachedFiles() async {
    setState(ViewState.Busy);
    await FilePicker.platform.clearTemporaryFiles();
    paths = null;
    fileName = null;
    errorMessage = null;
    setState(ViewState.Idle);
  }

  Future<Map<String, bool>> upload() async {
    setState(ViewState.Busy);
    setStateMessage(translate('upload.uploading_now'));

    Map<String, bool> uploadedPasteIds = new Map();
    try {
      List<File> files;
      Map<String, String> additionalFiles;

      if (pasteTextController.text != null && pasteTextController.text.isNotEmpty) {
        additionalFiles = Map.from(
            {'paste-${(new DateTime.now().millisecondsSinceEpoch / 1000).round()}.txt': pasteTextController.text});
      }

      if (paths != null && paths.length > 0) {
        files = paths.map((e) => new File(e.path)).toList();
      }

      UploadedResponse response = await _fileService.upload(files, additionalFiles);
      response.data.ids.forEach((element) {
        uploadedPasteIds.putIfAbsent(element, () => false);
      });

      if (createMulti && response.data.ids.length > 1) {
        UploadedMultiResponse multiResponse = await _fileService.createMulti(response.data.ids);
        uploadedPasteIds.putIfAbsent(multiResponse.data.urlId, () => true);
      }

      clearCachedFiles();
      _pasteTextController.clear();
      _refreshService.addEvent(RefreshEvent.RefreshHistory);
      errorMessage = null;
      return uploadedPasteIds;
    } catch (e) {
      if (e is RestServiceException) {
        if (e.statusCode == HttpStatus.notFound) {
          errorMessage = translate('upload.errors.not_found');
        } else if (e.statusCode == HttpStatus.forbidden) {
          errorMessage = translate('api.forbidden');
        } else if (e.statusCode != HttpStatus.notFound &&
            e.statusCode != HttpStatus.forbidden &&
            e.responseBody is RestError &&
            e.responseBody.message != null) {
          if (e.statusCode == HttpStatus.badRequest) {
            errorMessage = translate('api.bad_request', args: {'reason': e.responseBody.message});
          } else {
            errorMessage = translate('api.general_rest_error_payload', args: {'message': e.responseBody.message});
          }
        } else {
          errorMessage = translate('api.general_rest_error');
        }
      } else if (e is ServiceException && e.code == ErrorCode.SOCKET_ERROR) {
        errorMessage = translate('api.socket_error');
      } else if (e is ServiceException && e.code == ErrorCode.SOCKET_TIMEOUT) {
        errorMessage = translate('api.socket_timeout');
      } else {
        errorMessage = translate('app.unknown_error');
        setStateMessage(null);
        setState(ViewState.Idle);
        _logger.e('An unknown error occurred', e);
        throw e;
      }
    }

    setStateMessage(null);
    setState(ViewState.Idle);
    return null;
  }

  void openLink(String link) {
    _linkService.open(link);
  }

  @override
  void dispose() {
    _pasteTextController.dispose();
    _intentDataStreamSubscription.cancel();
    super.dispose();
  }
}