#!/usr/bin/env bash

. "test/test-migrate-fixtures.sh"
. "test/testlib.sh"

begin_test "migrate export (default branch)"
(
  set -e

  setup_multiple_local_branches_tracked

  md_oid="$(calc_oid "$(cat a.md)")"
  txt_oid="$(calc_oid "$(cat a.txt)")"

  git checkout my-feature
  md_feature_oid="$(calc_oid "$(cat a.md)")"
  git checkout master

  assert_pointer "refs/heads/master" "a.md" "$md_oid" "140"
  assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "120"
  assert_pointer "refs/heads/my-feature" "a.md" "$md_feature_oid" "30"

  git lfs migrate export --include="*.md, *.txt"

  [ ! $(assert_pointer "refs/heads/master" "a.md" "$md_oid" "140") ]
  [ ! $(assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "120") ]
  assert_pointer "refs/heads/my-feature" "a.md" "$md_feature_oid" "30"

  refute_local_object "$md_oid" "140"
  refute_local_object "$txt_oid" "120"
  assert_local_object "$md_feature_oid" "30"

  master="$(git rev-parse refs/heads/master)"
  feature="$(git rev-parse refs/heads/my-feature)"

  master_attrs="$(git cat-file -p "$master:.gitattributes")"
  feature_attrs="$(git cat-file -p "$feature:.gitattributes")"

  echo "$master_attrs" | grep -q "*.md text !filter !merge !diff"
  echo "$master_attrs" | grep -q "*.txt text !filter !merge !diff"

  [ ! $(echo "$feature_attrs" | grep -q "*.md text !filter !merge !diff") ]
  [ ! $(echo "$feature_attrs" | grep -q "*.txt text !filter !merge !diff") ]
)
end_test

begin_test "migrate export (with remote)"
(
  set -e

  setup_single_remote_branch_tracked

  git push origin master

  md_oid="$(calc_oid "$(cat a.md)")"
  txt_oid="$(calc_oid "$(cat a.txt)")"

  assert_pointer "refs/heads/master" "a.md" "$md_oid" "50"
  assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "30"

  assert_pointer "refs/remotes/origin/master" "a.md" "$md_oid" "50"
  assert_pointer "refs/remotes/origin/master" "a.txt" "$txt_oid" "30"

  # Flush the cache to ensure all objects have to be downloaded
  rm -rf .git/lfs/objects

  git lfs migrate export --everything --include="*.md, *.txt"

  [ ! $(assert_pointer "refs/heads/master" "a.md" "$md_oid" "50") ]
  [ ! $(assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "30") ]

  [ ! $(assert_pointer "refs/remotes/origin/master" "a.md" "$md_oid" "50") ]
  [ ! $(assert_pointer "refs/remotes/origin/master" "a.txt" "$txt_oid" "30") ]

  refute_local_object "$md_oid" "50"
  refute_local_object "$txt_oid" "30"

  master="$(git rev-parse refs/heads/master)"
  master_attrs="$(git cat-file -p "$master:.gitattributes")"

  echo "$master_attrs" | grep -q "*.md text !filter !merge !diff"
  echo "$master_attrs" | grep -q "*.txt text !filter !merge !diff"
)
end_test

begin_test "migrate export (include/exclude args)"
(
  set -e

  setup_multiple_local_branches_tracked

  md_oid="$(calc_oid "$(cat a.md)")"
  txt_oid="$(calc_oid "$(cat a.txt)")"

  assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "120"

  git lfs migrate export --include="*" --exclude="a.md"

  [ ! $(assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "120") ]
  assert_pointer "refs/heads/master" "a.md" "$md_oid" "140"

  refute_local_object "$txt_oid" "120"
  assert_local_object "$md_oid" "140"

  master="$(git rev-parse refs/heads/master)"

  master_attrs="$(git cat-file -p "$master:.gitattributes")"

  echo "$master_attrs" | grep -q "* text !filter !merge !diff"
  echo "$master_attrs" | grep -q "a.md filter=lfs diff=lfs merge=lfs"

)
end_test

begin_test "migrate export (bare repository)"
(
  set -e

  setup_single_remote_branch_tracked
  git push origin master
  make_bare

  git lfs migrate export --everything --include="*"
)
end_test

begin_test "migrate export (given branch)"
(
  set -e

  setup_multiple_local_branches_tracked

  md_oid="$(calc_oid "$(cat a.md)")"
  txt_oid="$(calc_oid "$(cat a.txt)")"

  git checkout my-feature
  md_feature_oid="$(calc_oid "$(cat a.md)")"
  git checkout master

  assert_pointer "refs/heads/my-feature" "a.md" "$md_feature_oid" "30"
  assert_pointer "refs/heads/my-feature" "a.txt" "$txt_oid" "120"
  assert_pointer "refs/heads/master" "a.md" "$md_oid" "140"
  assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "120"

  git lfs migrate export --include="*.md,*.txt" my-feature

  [ ! $(assert_pointer "refs/heads/my-feature" "a.md" "$md_feature_oid" "30") ]
  [ ! $(assert_pointer "refs/heads/my-feature" "a.txt" "$txt_oid" "120") ]
  [ ! $(assert_pointer "refs/heads/master" "a.md" "$md_oid" "140") ]
  [ ! $(assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "120") ]

  refute_local_object "$md_feature_oid" "30"
  refute_local_object "$txt_oid" "120"
  refute_local_object "$md_oid" "140"

  master="$(git rev-parse refs/heads/master)"
  feature="$(git rev-parse refs/heads/my-feature)"

  master_attrs="$(git cat-file -p "$master:.gitattributes")"
  feature_attrs="$(git cat-file -p "$feature:.gitattributes")"

  echo "$master_attrs" | grep -q "*.md text !filter !merge !diff"
  echo "$master_attrs" | grep -q "*.txt text !filter !merge !diff"
  echo "$feature_attrs" | grep -q "*.md text !filter !merge !diff"
  echo "$feature_attrs" | grep -q "*.txt text !filter !merge !diff"
)
end_test

begin_test "migrate export (no filter)"
(
  set -e

  setup_multiple_local_branches_tracked

  git lfs migrate export 2>&1 | tee migrate.log
  if [ ${PIPESTATUS[0]} -eq 0 ]; then
    echo >&2 "fatal: expected git lfs migrate export to fail, didn't"
    exit 1
  fi

  grep "fatal: one or more files must be specified with --include" migrate.log
)
end_test

begin_test "migrate export (exclude remote refs)"
(
  set -e

  setup_single_remote_branch_tracked

  md_oid="$(calc_oid "$(cat a.md)")"
  txt_oid="$(calc_oid "$(cat a.txt)")"

  git checkout refs/remotes/origin/master
  md_remote_oid="$(calc_oid "$(cat a.md)")"
  txt_remote_oid="$(calc_oid "$(cat a.txt)")"
  git checkout master

  assert_pointer "refs/heads/master" "a.md" "$md_oid" "50"
  assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "30"

  assert_pointer "refs/remotes/origin/master" "a.md" "$md_remote_oid" "140"
  assert_pointer "refs/remotes/origin/master" "a.txt" "$txt_remote_oid" "120"

  git lfs migrate export --include="*.md,*.txt"

  [ ! $(assert_pointer "refs/heads/master" "a.md" "$md_oid" "50") ]
  [ ! $(assert_pointer "refs/heads/master" "a.txt" "$txt_oid" "30") ]

  refute_local_object "$md_oid" "50"
  refute_local_object "$txt_oid" "30"

  assert_pointer "refs/remotes/origin/master" "a.md" "$md_remote_oid" "140"
  assert_pointer "refs/remotes/origin/master" "a.txt" "$txt_remote_oid" "120"

  assert_local_object "$md_remote_oid" "140"
  assert_local_object "$txt_remote_oid" "120"

  master="$(git rev-parse refs/heads/master)"
  remote="$(git rev-parse refs/remotes/origin/master)"

  master_attrs="$(git cat-file -p "$master:.gitattributes")"
  remote_attrs="$(git cat-file -p "$remote:.gitattributes")"

  echo "$master_attrs" | grep -q "*.md text !filter !merge !diff"
  echo "$master_attrs" | grep -q "*.txt text !filter !merge !diff"

  [ ! $(echo "$remote_attrs" | grep -q "*.md text !filter !merge !diff") ]
  [ ! $(echo "$remote_attrs" | grep -q "*.txt text !filter !merge !diff") ]
)
end_test

begin_test "migrate export (--skip-fetch)"
(
  set -e

  setup_single_remote_branch_tracked

  md_master_oid="$(calc_oid "$(cat a.md)")"
  txt_master_oid="$(calc_oid "$(cat a.txt)")"

  git checkout refs/remotes/origin/master
  md_remote_oid="$(calc_oid "$(cat a.md)")"
  txt_remote_oid="$(calc_oid "$(cat a.txt)")"
  git checkout master

  git tag pseudo-remote "$(git rev-parse refs/remotes/origin/master)"
  # Remove the refs/remotes/origin/master ref, and instruct 'git lfs migrate' to
  # not fetch it.
  git update-ref -d refs/remotes/origin/master

  assert_pointer "refs/heads/master" "a.md" "$md_master_oid" "50"
  assert_pointer "pseudo-remote" "a.md" "$md_remote_oid" "140"
  assert_pointer "refs/heads/master" "a.txt" "$txt_master_oid" "30"
  assert_pointer "pseudo-remote" "a.txt" "$txt_remote_oid" "120"

  git lfs migrate export --skip-fetch --include="*.md,*.txt"

  [ ! $(assert_pointer "refs/heads/master" "a.md" "$md_master_oid" "50") ]
  [ ! $(assert_pointer "pseudo-remote" "a.md" "$md_remote_oid" "140") ]
  [ ! $(assert_pointer "refs/heads/master" "a.txt" "$txt_master_oid" "30") ]
  [ ! $(assert_pointer "pseudo-remote" "a.txt" "$txt_remote_oid" "120") ]

  refute_local_object "$md_master_oid" "50"
  refute_local_object "$md_remote_oid" "140"
  refute_local_object "$txt_master_oid" "30"
  refute_local_object "$txt_remote_oid" "120"

  master="$(git rev-parse refs/heads/master)"
  remote="$(git rev-parse pseudo-remote)"

  master_attrs="$(git cat-file -p "$master:.gitattributes")"
  remote_attrs="$(git cat-file -p "$remote:.gitattributes")"

  echo "$master_attrs" | grep -q "*.md text !filter !merge !diff"
  echo "$master_attrs" | grep -q "*.txt text !filter !merge !diff"
  echo "$remote_attrs" | grep -q "*.md text !filter !merge !diff"
  echo "$remote_attrs" | grep -q "*.txt text !filter !merge !diff"
)
end_test

begin_test "migrate export (include/exclude ref)"
(
  set -e

  setup_multiple_remote_branches_gitattrs

  md_master_oid="$(calc_oid "$(cat a.md)")"
  txt_master_oid="$(calc_oid "$(cat a.txt)")"

  git checkout refs/remotes/origin/master
  md_remote_oid="$(calc_oid "$(cat a.md)")"
  txt_remote_oid="$(calc_oid "$(cat a.txt)")"

  git checkout my-feature
  md_feature_oid="$(calc_oid "$(cat a.md)")"
  txt_feature_oid="$(calc_oid "$(cat a.txt)")"

  git checkout master

  git lfs migrate export \
    --include="*.txt" \
    --include-ref=refs/heads/my-feature \
    --exclude-ref=refs/heads/master
  
  assert_pointer "refs/heads/master" "a.md" "$md_master_oid" "21"
  assert_pointer "refs/heads/master" "a.txt" "$txt_master_oid" "20"

  assert_pointer "refs/remotes/origin/master" "a.md" "$md_remote_oid" "11"
  assert_pointer "refs/remotes/origin/master" "a.txt" "$txt_remote_oid" "10"

  assert_pointer "refs/heads/my-feature" "a.md" "$md_feature_oid" "31"
  [ ! $(assert_pointer "refs/heads/my-feature" "a.txt" "$txt_feature_oid" "30") ]

  assert_local_object "$md_master_oid" "21"
  assert_local_object "$txt_master_oid" "20"

  assert_local_object "$md_remote_oid" "11"
  assert_local_object "$txt_remote_oid" "10"

  assert_local_object "$md_feature_oid" "31"
  refute_local_object "$txt_feature_oid" "30"

  master="$(git rev-parse refs/heads/master)"
  feature="$(git rev-parse refs/heads/my-feature)"
  remote="$(git rev-parse refs/remotes/origin/master)"

  master_attrs="$(git cat-file -p "$master:.gitattributes")"
  remote_attrs="$(git cat-file -p "$remote:.gitattributes")"
  feature_attrs="$(git cat-file -p "$feature:.gitattributes")"

  [ ! $(echo "$master_attrs" | grep -q "*.txt text !filter !merge !diff") ]
  [ ! $(echo "$remote_attrs" | grep -q "*.txt text !filter !merge !diff") ]
  echo "$feature_attrs" | grep -q "*.txt text !filter !merge !diff"
)
end_test

begin_test "migrate export (--object-map)"
(
  set -e

  setup_multiple_local_branches_tracked

  output_dir=$(mktemp -d)

  git log --all --pretty='format:%H' > "${output_dir}/old_sha.txt"
  git lfs migrate export --everything --include="*" --object-map "${output_dir}/object-map.txt"
  git log --all --pretty='format:%H' > "${output_dir}/new_sha.txt"
  paste -d',' "${output_dir}/old_sha.txt" "${output_dir}/new_sha.txt" > "${output_dir}/expected-map.txt"

  diff -u <(sort "${output_dir}/expected-map.txt") <(sort "${output_dir}/object-map.txt")
)
end_test

begin_test "migrate export (--verbose)"
(
  set -e

  setup_multiple_local_branches_tracked

  git lfs migrate export --everything --include="*" --verbose 2>&1 | grep -q "migrate: commit "
)
end_test