// Copyright (c) 2019, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
//
// Sample illustrating resource management with an explicit arena.

import 'dart:async';
import 'dart:ffi';

import 'package:expect/expect.dart';
import 'package:ffi/ffi.dart';

import 'utf8_helpers.dart';
import '../dylib_utils.dart';

main() async {
  final ffiTestDynamicLibrary = dlopenPlatformSpecific(
    "ffi_test_dynamic_library",
  );

  final MemMove = ffiTestDynamicLibrary
      .lookupFunction<
        Void Function(Pointer<Void>, Pointer<Void>, IntPtr),
        void Function(Pointer<Void>, Pointer<Void>, int)
      >("MemMove");

  // To ensure resources are freed, wrap them in a [using] call.
  using((Arena arena) {
    final p = arena<Int64>(2);
    p[0] = 24;
    MemMove((p + 1).cast<Void>(), p.cast<Void>(), sizeOf<Int64>());
    print(p[1]);
    Expect.equals(24, p[1]);
  });

  // Resources are freed also when abnormal control flow occurs.
  try {
    using((Arena arena) {
      final p = arena<Int64>(2);
      p[0] = 25;
      MemMove((p + 1).cast<Void>(), p.cast<Void>(), 8);
      print(p[1]);
      Expect.equals(25, p[1]);
      throw Exception("Some random exception");
    });
    // `calloc.free(p)` has been called.
  } on Exception catch (e) {
    print("Caught exception: $e");
  }

  // In a arena multiple resources can be allocated, which will all be freed
  // at the end of the scope.
  using((Arena arena) {
    final p = arena<Int64>(2);
    final p2 = arena<Int64>(2);
    p[0] = 1;
    p[1] = 2;
    MemMove(p2.cast<Void>(), p.cast<Void>(), 2 * sizeOf<Int64>());
    Expect.equals(1, p2[0]);
    Expect.equals(2, p2[1]);
  });

  // If the resource allocation happens in a different scope, then one either
  // needs to pass the arena to that scope.
  f1(Arena arena) {
    return arena<Int64>(2);
  }

  using((Arena arena) {
    final p = f1(arena);
    final p2 = f1(arena);
    p[0] = 1;
    p[1] = 2;
    MemMove(p2.cast<Void>(), p.cast<Void>(), 2 * sizeOf<Int64>());
    Expect.equals(1, p2[0]);
    Expect.equals(2, p2[1]);
  });

  // Using Strings.
  using((Arena arena) {
    final p = "Hello world!".toUtf8(arena);
    print(p.contents());
  });

  final allocateResource = ffiTestDynamicLibrary
      .lookupFunction<
        Pointer<SomeResource> Function(),
        Pointer<SomeResource> Function()
      >("AllocateResource");

  final useResource = ffiTestDynamicLibrary
      .lookupFunction<
        Void Function(Pointer<SomeResource>),
        void Function(Pointer<SomeResource>)
      >("UseResource");

  final releaseResource = ffiTestDynamicLibrary
      .lookupFunction<
        Void Function(Pointer<SomeResource>),
        void Function(Pointer<SomeResource>)
      >("ReleaseResource");

  // Using an FFI call to release a resource.
  using((Arena arena) {
    final r = arena.using(allocateResource(), releaseResource);
    useResource(r);
  });

  // Using an FFI call to release a resource with abnormal control flow.
  try {
    using((Arena arena) {
      final r = arena.using(allocateResource(), releaseResource);
      useResource(r);

      throw Exception("Some random exception");
    });
    // Resource has been freed.
  } on Exception catch (e) {
    print("Caught exception: $e");
  }

  /// [using] waits with releasing its resources until after [Future]s
  /// complete.
  List<int> freed = [];
  freeInt(int i) {
    freed.add(i);
  }

  Future<int> myFutureInt = using((Arena arena) {
    return Future.microtask(() {
      arena.using(1, freeInt);
      return 1;
    });
  });

  Expect.isTrue(freed.isEmpty);
  await myFutureInt;
  Expect.equals(1, freed.single);
}

/// Represents some opaque resource being managed by a library.
final class SomeResource extends Opaque {}