// 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.

@TestOn('vm')

import 'dart:convert';
import 'dart:io';

import 'package:archive/archive.dart';
import 'package:async/async.dart';
import 'package:path/path.dart' as p;
import 'package:test/test.dart';
import 'package:test_descriptor/test_descriptor.dart' as d;

import 'utils.dart';

void main() {
  group("create()", () {
    test("creates an empty archive", () async {
      await d.archive("test.tar").create();

      var archive =
          TarDecoder().decodeBytes(File(d.path("test.tar")).readAsBytesSync());
      expect(archive.files, isEmpty);
    });

    test("creates an archive with files", () async {
      await d.archive("test.tar", [
        d.file("file1.txt", "contents 1"),
        d.file("file2.txt", "contents 2")
      ]).create();

      var files = TarDecoder()
          .decodeBytes(File(d.path("test.tar")).readAsBytesSync())
          .files;
      expect(files.length, equals(2));
      _expectFile(files[0], "file1.txt", "contents 1");
      _expectFile(files[1], "file2.txt", "contents 2");
    });

    test("creates an archive with files in a directory", () async {
      await d.archive("test.tar", [
        d.dir("dir", [
          d.file("file1.txt", "contents 1"),
          d.file("file2.txt", "contents 2")
        ])
      ]).create();

      var files = TarDecoder()
          .decodeBytes(File(d.path("test.tar")).readAsBytesSync())
          .files;
      expect(files.length, equals(2));
      _expectFile(files[0], "dir/file1.txt", "contents 1");
      _expectFile(files[1], "dir/file2.txt", "contents 2");
    });

    test("creates an archive with files in a nested directory", () async {
      await d.archive("test.tar", [
        d.dir("dir", [
          d.dir("subdir", [
            d.file("file1.txt", "contents 1"),
            d.file("file2.txt", "contents 2")
          ])
        ])
      ]).create();

      var files = TarDecoder()
          .decodeBytes(File(d.path("test.tar")).readAsBytesSync())
          .files;
      expect(files.length, equals(2));
      _expectFile(files[0], "dir/subdir/file1.txt", "contents 1");
      _expectFile(files[1], "dir/subdir/file2.txt", "contents 2");
    });

    group("creates a file in", () {
      test("zip format", () async {
        await d.archive("test.zip", [d.file("file.txt", "contents")]).create();

        var archive = ZipDecoder()
            .decodeBytes(File(d.path("test.zip")).readAsBytesSync());
        _expectFile(archive.files.single, "file.txt", "contents");
      });

      group("gzip tar format", () {
        for (var extension in [".tar.gz", ".tar.gzip", ".tgz"]) {
          test("with $extension", () async {
            await d.archive(
                "test$extension", [d.file("file.txt", "contents")]).create();

            var archive = TarDecoder().decodeBytes(GZipDecoder()
                .decodeBytes(File(d.path("test$extension")).readAsBytesSync()));
            _expectFile(archive.files.single, "file.txt", "contents");
          });
        }
      });

      group("bzip2 tar format", () {
        for (var extension in [".tar.bz2", ".tar.bzip2"]) {
          test("with $extension", () async {
            await d.archive(
                "test$extension", [d.file("file.txt", "contents")]).create();

            var archive = TarDecoder().decodeBytes(BZip2Decoder()
                .decodeBytes(File(d.path("test$extension")).readAsBytesSync()));
            _expectFile(archive.files.single, "file.txt", "contents");
          });
        }
      });
    });

    group("gracefully rejects", () {
      test("an uncreatable descriptor", () async {
        await expectLater(
            d.archive("test.tar", [d.filePattern(RegExp(r"^foo-"))]).create(),
            throwsUnsupportedError);
        await d.nothing("test.tar").validate();
      });

      test("a non-file non-directory descriptor", () async {
        await expectLater(
            d.archive("test.tar", [d.nothing("file.txt")]).create(),
            throwsUnsupportedError);
        await d.nothing("test.tar").validate();
      });

      test("an unknown file extension", () async {
        await expectLater(
            d.archive("test.asdf", [d.nothing("file.txt")]).create(),
            throwsUnsupportedError);
      });
    });
  });

  group("validate()", () {
    group("with an empty archive", () {
      test("succeeds if an empty archive exists", () async {
        File(d.path("test.tar"))
            .writeAsBytesSync(TarEncoder().encode(Archive()));
        await d.archive("test.tar").validate();
      });

      test("succeeds if a non-empty archive exists", () async {
        File(d.path("test.tar")).writeAsBytesSync(
            TarEncoder().encode(Archive()..addFile(_file("file.txt"))));
        await d.archive("test.tar").validate();
      });

      test("fails if no archive exists", () {
        expect(d.archive("test.tar").validate(),
            throwsA(toString(startsWith('File not found: "test.tar".'))));
      });

      test("fails if an invalid archive exists", () {
        d.file("test.tar", "not a valid tar file").create();
        expect(
            d.archive("test.tar").validate(),
            throwsA(toString(
                startsWith('File "test.tar" is not a valid archive.'))));
      });
    });

    test("succeeds if an archive contains a matching file", () async {
      File(d.path("test.tar")).writeAsBytesSync(TarEncoder()
          .encode(Archive()..addFile(_file("file.txt", "contents"))));
      await d.archive("test.tar", [d.file("file.txt", "contents")]).validate();
    });

    test("fails if an archive doesn't contain a file", () async {
      File(d.path("test.tar")).writeAsBytesSync(TarEncoder().encode(Archive()));
      expect(
          d.archive("test.tar", [d.file("file.txt", "contents")]).validate(),
          throwsA(
              toString(startsWith('File not found: "test.tar/file.txt".'))));
    });

    test("fails if an archive contains a non-matching file", () async {
      File(d.path("test.tar")).writeAsBytesSync(TarEncoder()
          .encode(Archive()..addFile(_file("file.txt", "wrong contents"))));
      expect(
          d.archive("test.tar", [d.file("file.txt", "contents")]).validate(),
          throwsA(toString(
              startsWith('File "test.tar/file.txt" should contain:'))));
    });

    test("succeeds if an archive contains a file matching a pattern", () async {
      File(d.path("test.tar")).writeAsBytesSync(TarEncoder()
          .encode(Archive()..addFile(_file("file.txt", "contents"))));
      await d.archive("test.tar",
          [d.filePattern(RegExp(r"f..e\.txt"), "contents")]).validate();
    });

    group("validates a file in", () {
      test("zip format", () async {
        File(d.path("test.zip")).writeAsBytesSync(ZipEncoder()
            .encode(Archive()..addFile(_file("file.txt", "contents"))));

        await d
            .archive("test.zip", [d.file("file.txt", "contents")]).validate();
      });

      group("gzip tar format", () {
        for (var extension in [".tar.gz", ".tar.gzip", ".tgz"]) {
          test("with $extension", () async {
            File(d.path("test$extension")).writeAsBytesSync(GZipEncoder()
                .encode(TarEncoder().encode(
                    Archive()..addFile(_file("file.txt", "contents")))));

            await d.archive(
                "test$extension", [d.file("file.txt", "contents")]).validate();
          });
        }
      });

      group("bzip2 tar format", () {
        for (var extension in [".tar.bz2", ".tar.bzip2"]) {
          test("with $extension", () async {
            File(d.path("test$extension")).writeAsBytesSync(BZip2Encoder()
                .encode(TarEncoder().encode(
                    Archive()..addFile(_file("file.txt", "contents")))));

            await d.archive(
                "test$extension", [d.file("file.txt", "contents")]).validate();
          });
        }
      });
    });

    test("gracefully rejects an unknown file format", () {
      expect(d.archive("test.asdf").validate(), throwsUnsupportedError);
    });
  });

  test("read() is unsupported", () {
    expect(d.archive("test.tar").read(), throwsUnsupportedError);
  });

  test("readAsBytes() returns the contents of the archive", () async {
    var descriptor = d.archive("test.tar",
        [d.file("file1.txt", "contents 1"), d.file("file2.txt", "contents 2")]);

    var files = TarDecoder()
        .decodeBytes(await collectBytes(descriptor.readAsBytes()))
        .files;
    expect(files.length, equals(2));
    _expectFile(files[0], "file1.txt", "contents 1");
    _expectFile(files[1], "file2.txt", "contents 2");
  });

  test("archive returns the in-memory contents", () async {
    var archive = await d.archive("test.tar", [
      d.file("file1.txt", "contents 1"),
      d.file("file2.txt", "contents 2")
    ]).archive;

    var files = archive.files;
    expect(files.length, equals(2));
    _expectFile(files[0], "file1.txt", "contents 1");
    _expectFile(files[1], "file2.txt", "contents 2");
  });

  test("io refers to the file within the sandbox", () {
    expect(d.file('test.tar').io.path, equals(p.join(d.sandbox, 'test.tar')));
  });
}

/// Asserts that [file] has the given [name] and [contents].
void _expectFile(ArchiveFile file, String name, String contents) {
  expect(file.name, equals(name));
  expect(utf8.decode(file.content as List<int>), equals(contents));
}

/// Creates an [ArchiveFile] with the given [name] and [contents].
ArchiveFile _file(String name, [String contents]) {
  var bytes = utf8.encode(contents ?? "");
  return ArchiveFile(name, bytes.length, bytes)
    // Setting the mode and mod time are necessary to work around
    // brendan-duncan/archive#76.
    ..mode = 428
    ..lastModTime = DateTime.now().millisecondsSinceEpoch ~/ 1000;
}