# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

. "${SRC_ROOT}/platform/dev/toolchain_utils.sh" || exit 1

create_base_image() {
  local image_name=$1
  local disk_layout=$2

  local disk_img="${BUILD_DIR}/${image_name}"
  local mbr_img="/usr/share/syslinux/gptmbr.bin"
  local root_fs_dir="${BUILD_DIR}/rootfs"

  info "Using image type ${disk_layout}"
  "${BUILD_LIBRARY_DIR}/disk_util" --disk_layout="${disk_layout}" \
      format --mbr_boot_code="${mbr_img}" "${disk_img}"

  "${BUILD_LIBRARY_DIR}/disk_util" --disk_layout="${disk_layout}" \
      mount "${disk_img}" "${root_fs_dir}"
  trap "cleanup_mounts '${root_fs_dir}' && delete_prompt" EXIT

  # First thing first, install baselayout with USE=build to create a
  # working directory tree. Don't use binpkgs due to the use flag change.
  sudo -E USE=build "emerge-${BOARD}" --root="${root_fs_dir}" \
      --usepkg=n --buildpkg=n --oneshot --quiet --nodeps sys-apps/baselayout

  # FIXME(marineam): Work around glibc setting EROOT=$ROOT
  # https://bugs.gentoo.org/show_bug.cgi?id=473728#c12
  sudo mkdir -p "${root_fs_dir}/etc/ld.so.conf.d"

  # We "emerge --root=${root_fs_dir} --root-deps=rdeps --usepkgonly" all of the
  # runtime packages for chrome os. This builds up a chrome os image from
  # binary packages with runtime dependencies only.  We use INSTALL_MASK to
  # trim the image size as much as possible.
  emerge_prod_gcc --root="${root_fs_dir}"
  emerge_to_image --root="${root_fs_dir}" ${BASE_PACKAGE}

  # Make sure profile.env and ld.so.cache has been generated
  sudo ROOT="${root_fs_dir}" env-update

  # Record directories installed to the state partition.
  # Ignore /var/tmp, systemd covers this entry.
  sudo "${BUILD_LIBRARY_DIR}/gen_tmpfiles.py" --root="${root_fs_dir}" \
      --output="${root_fs_dir}/usr/lib/tmpfiles.d/base_image_var.conf" \
      --ignore=/var/tmp "${root_fs_dir}/var"
  if [[ "${disk_layout}" == *-usr ]]; then
      sudo "${BUILD_LIBRARY_DIR}/gen_tmpfiles.py" --root="${root_fs_dir}" \
          --output="${root_fs_dir}/usr/lib/tmpfiles.d/base_image_etc.conf" \
          "${root_fs_dir}/etc"
  fi

  # Set /etc/lsb-release on the image.
  "${BUILD_LIBRARY_DIR}/set_lsb_release" \
  --root="${root_fs_dir}" \
  --board="${BOARD}"

  ${BUILD_LIBRARY_DIR}/create_legacy_bootloader_templates.sh \
    --arch=${ARCH} \
    --disk_layout="${disk_layout}" \
    --boot_dir="${root_fs_dir}"/boot \
    --esp_dir="${root_fs_dir}"/boot/efi \
    --boot_args="${FLAGS_boot_args}"

  # Zero all fs free space to make it more compressible so auto-update
  # payloads become smaller, not fatal since it won't work on linux < 3.2
  sudo fstrim "${root_fs_dir}" || true
  if [[ "${disk_layout}" == *-usr ]]; then
    sudo fstrim "${root_fs_dir}/usr" || true
  else
    sudo fstrim "${root_fs_dir}/media/state" || true
  fi

  cleanup_mounts "${root_fs_dir}"
  trap - EXIT
}