import 'dart:io';
import 'dart:typed_data';

import 'package:flutter/material.dart';
import 'package:path_provider/path_provider.dart';
import 'package:share_location/enums.dart';
import 'package:video_player/video_player.dart';

class RawMemoryDisplay extends StatefulWidget {
  final Uint8List data;
  final MemoryType type;
  final bool loopVideo;
  final String? filename;
  final void Function(VideoPlayerController)? onVideoControllerInitialized;
  final BoxFit? fit;

  const RawMemoryDisplay({
    Key? key,
    required this.data,
    required this.type,
    this.loopVideo = false,
    this.fit = BoxFit.cover,
    this.filename,
    this.onVideoControllerInitialized,
  }) : super(key: key);

  @override
  State<RawMemoryDisplay> createState() => _RawMemoryDisplayState();
}

class _RawMemoryDisplayState extends State<RawMemoryDisplay> {
  VideoPlayerController? videoController;

  @override
  void initState() {
    super.initState();

    if (widget.type == MemoryType.video) {
      initializeVideo();
    }
  }

  Future<File> createTempVideo() async {
    final tempDirectory = await getTemporaryDirectory();
    final path = '${tempDirectory.path}/${widget.filename ?? 'video.mp4'}';
    final file = File(path);

    if (await file.exists()) {
      // File already exists, so just return it
      return file;
    }

    // File needs to be created
    await file.create();
    await file.writeAsBytes(widget.data);

    return file;
  }

  Future<void> initializeVideo() async {
    final file = await createTempVideo();

    videoController = VideoPlayerController.file(file);
    videoController!.initialize().then((value) {
      setState(() {});
      videoController!.setLooping(widget.loopVideo);
      videoController!.play();

      if (widget.onVideoControllerInitialized != null) {
        widget.onVideoControllerInitialized!(videoController!);
      }
    });
  }

  @override
  void dispose() {
    videoController?.dispose();

    super.dispose();
  }

  @override
  Widget build(BuildContext context) {
    switch (widget.type) {
      case MemoryType.photo:
        return Image.memory(
          widget.data,
          fit: widget.fit,
        );
      case MemoryType.video:
        if (videoController == null) {
          return const SizedBox();
        }

        switch (widget.fit) {
          case BoxFit.contain:
            return Align(
              child: AspectRatio(
                aspectRatio: videoController!.value.aspectRatio,
                child: VideoPlayer(videoController!),
              ),
            );
          default:
            return AspectRatio(
              aspectRatio: videoController!.value.aspectRatio,
              child: VideoPlayer(videoController!),
            );
        }
      default:
        throw Exception('Unknown memory type: ${widget.type}');
    }
  }
}