blob: 40bd3d734ceff2ee9fe7402683e2cef8c58f0cfc [file] [log] [blame]
#!/bin/bash
# Copyright (c) 2011 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.
# This script moves ebuilds between 'stable' and 'live' states.
# By default 'stable' ebuilds point at and build from source at the
# last known good commit. Moving an ebuild to 'live' (via cros_workon start)
# is intended to support development. The current source tip is fetched,
# source modified and built using the unstable 'live' (9999) ebuild.
. "$(dirname "$0")/common.sh" || exit 1
assert_not_root_user
# Script must be run inside the chroot
DEFINE_string board "${DEFAULT_BOARD}" \
"The board to set package keywords for."
DEFINE_boolean host "${FLAGS_FALSE}" \
"Uses the host instead of board"
DEFINE_string remote "" \
"For non-workon projects, the git remote to use."
DEFINE_string revision "" \
"Use to override the manifest defined default revision used for a project"
DEFINE_string command "git status" \
"The command to be run by forall."
DEFINE_boolean all "${FLAGS_FALSE}" \
"Apply to all possible packages for the given command"
FLAGS_HELP="usage: $0 <command> [flags] [<list of packages>|.|--all]
commands:
start: Moves an ebuild to live (intended to support development)
stop: Moves an ebuild to stable (use last known good)
info: Print package name, repo name, and source directory.
list: List of live ebuilds (workon ebuilds if --all)
list-all: List all of the live ebuilds for all setup boards
iterate: For each ebuild, cd to the source dir and run a command"
FLAGS "$@" || { [ "${FLAGS_help}" = "${FLAGS_TRUE}" ] && exit 0; } || exit 1
eval set -- "${FLAGS_ARGV}"
# eat the workon command keywords: start, stop or list.
WORKON_CMD=$1
shift
# Board dir config
# If both are specified, just use host, because board does not
# have to be specified and may come from default, in which case
# there's no way to override.
[ -n "${FLAGS_board}" ] && [ "${FLAGS_host}" = ${FLAGS_TRUE} ] && \
FLAGS_board="" # kill board
if [ -z "${FLAGS_board}" ] && \
[ "${FLAGS_host}" = ${FLAGS_FALSE} ] && \
[ "${WORKON_CMD}" != "list-all" ]; then
flags_help
die "You must specify either --host or --board="
fi
if [ -n "${FLAGS_board}" ]; then
BOARD_DIR=/build/"${FLAGS_board}" # --board specified
EQUERYCMD=equery-"${FLAGS_board}"
EBUILDCMD=ebuild-"${FLAGS_board}"
PORTAGEQCMD=portageq-"${FLAGS_board}"
BOARD_STR="${FLAGS_board}"
else
BOARD_DIR="" # --host specified
EQUERYCMD=equery
EBUILDCMD=ebuild
PORTAGEQCMD=portageq
BOARD_STR="host"
fi
WORKON_DIR=${CHROOT_TRUNK_DIR}/.config/cros_workon
CONFIG_DIR=${BOARD_DIR}/etc/portage
KEYWORDS_DIR=${CONFIG_DIR}/package.keywords
MASK_DIR=${CONFIG_DIR}/package.mask
UNMASK_DIR=${CONFIG_DIR}/package.unmask
WORKON_FILE=${WORKON_DIR}/${FLAGS_board:-host}
MASK_WORKON_FILE=${WORKON_FILE}.mask
KEYWORDS_FILE=${KEYWORDS_DIR}/cros-workon
MASK_FILE=${MASK_DIR}/cros-workon
UNMASK_FILE=${UNMASK_DIR}/cros-workon
CHROME_ATOM=chromeos-base/chromeos-chrome
mkdir -p "${WORKON_DIR}" || die "mkdir -p ${WORKON_DIR}"
touch "${WORKON_FILE}" "${MASK_WORKON_FILE}" || \
die "touch ${WORKON_FILE} ${MASK_WORKON_FILE}"
cmds=(
"mkdir -p '${KEYWORDS_DIR}' '${MASK_DIR}' '${UNMASK_DIR}'"
# Clobber and re-create the WORKON_FILE symlinks every time. This
# is a trivial operation and eliminates all kinds of corner cases
# as well as any possible future renames of WORKON_FILE.
# In particular, chroot is usually built as "amd64-host" but becomes
# just "host" after installation. crosbug.com/23096
"ln -sf '${WORKON_FILE}' '${KEYWORDS_FILE}'"
"ln -sf '${MASK_WORKON_FILE}' '${MASK_FILE}'"
"ln -sf '${WORKON_FILE}' '${UNMASK_FILE}'"
)
# If the board dir doesn't exist yet, we don't want to create it as
# that'll screw up ./setup_board later on.
if [[ -d ${BOARD_DIR:-/} ]] ; then
sudo_multi "${cmds[@]}"
else
die "${BOARD_STR} has not been setup yet"
fi
find_keyword_workon_ebuilds() {
local keyword="${1}"
local overlay
# NOTE: overlay may be a symlink, and we have to use ${overlay}/
for overlay in ${PORTDIR_OVERLAY}; do
# only look up ebuilds named 9999 to eliminate duplicates
find "${overlay}"/*-* -maxdepth 2 -type f -name '*9999.ebuild' \
-exec grep -l 'inherit.*cros-workon' {} + 2>/dev/null | \
xargs egrep -l "KEYWORDS=\".*~([*]|${keyword})[ \"]"
done
}
ebuild_to_package() {
# This changes the absolute path to ebuilds into category/package.
sed -e 's/.*\/\([^/]*\)\/\([^/]*\)\/.*\.ebuild/\1\/\2/'
}
show_project_ebuild_map() {
local keyword="$1"
# Column 1: Repo name
# Column 2: Package name
for EBUILD in $(find_keyword_workon_ebuilds ${keyword}); do
(
eval $(grep -E '^CROS_WORKON' "${EBUILD}")
CP=$(echo "$EBUILD" | ebuild_to_package)
echo "${CROS_WORKON_PROJECT}" "${CP}"
)
done
}
show_project_path_map() {
# Column 1: Repo name
# Column 2: Source directory
repo list | awk -F ' : ' '{ print $2, $1 }'
}
show_workon_ebuilds() {
local keyword="$1"
find_keyword_workon_ebuilds "${keyword}" | ebuild_to_package | sort -u
}
show_workon_info() {
local atoms="$1"
local keyword="$2"
local sort="sort -k1b,1"
# Column 1: Package name
# Column 2: Repo name
# Column 3: Source directory (if present locally)
join \
<(echo "${atoms}" | sed -e 's/ /\n/g' | ${sort}) \
<(join -a 1 -e - -o 1.2,1.1,2.2 \
<(show_project_ebuild_map "${keyword}" | ${sort}) \
<(show_project_path_map | ${sort}) \
| ${sort})
}
# Canonicalize package name to category/package.
canonicalize_name() {
local pkgfile
local pkgname
if grep -qx "=$1-9999" "${WORKON_FILE}" ; then
echo $1
return 0
fi
if ! pkgfile=$(ACCEPT_KEYWORDS="~${ARCH}" ${EQUERYCMD} \
which --include-masked $1); then
# Try to auto-complete it.
local autopkgs=()
for pkgname in $(show_workon_ebuilds); do
if [[ ${pkgname} == *"$1"* ]]; then
autopkgs+=( ${pkgname} )
fi
done
case ${#autopkgs[@]} in
0)
warn "error looking up package $1" 1>&2
return 1
;;
1)
# Do nothing -- fall through.
;;
*)
warn "Multiple autocompletes found: ${autopkgs[*]}"
;;
esac
pkgname=${autopkgs[0]}
info "Autocompleted '$1' to: ${pkgname}"
canonicalize_name "${pkgname}"
return
fi
pkgname=$(echo "${pkgfile}" | awk -F '/' '{ print $(NF-2) "/" $(NF-1) }')
# TODO(rcui): remove special casing of chromeos-chrome here when we make it
# inherit from cros-workon class. Tracked in chromium-os:19259.
if [ "${pkgname}" != "${CHROME_ATOM}" ] && \
! grep -q "cros-workon" ${pkgfile}; then
warn "${pkgname} is not a cros-workon package" 1>&2
return 1
fi
echo "${pkgname}"
return 0
}
# Canonicalize a list of names.
canonicalize_names() {
local atoms=$1
local names=()
local atom
for atom in ${atoms}; do
names+=( $(canonicalize_name "${atom}") ) || return 1
done
echo "${names[*]}"
}
# Locate the package name based on the current directory
locate_package() {
local projectname=$(git config --get remote.cros.projectname ||
git config --get remote.cros-internal.projectname)
if [ -z "${projectname}" ]; then
die "No project in git config: Can not cros_workon . here"
fi
local reponame=$(show_project_ebuild_map |
grep "^${projectname} " | cut -d" " -f2)
if [ -z "${reponame}" ]; then
die "No matching package for ${projectname}: Can not cros_workon . here"
else
echo "${reponame}"
fi
}
# Display ebuilds currently part of the live branch and open for development.
show_live_ebuilds() {
sed -n 's/^=\(.*\)-9999$/\1/p' "${WORKON_FILE}"
}
# Display ebuilds currently part of the live branch and open for development
# for any board that currently has live ebuilds.
show_all_live_ebuilds() {
local workon_file
for workon_file in ${WORKON_DIR}/*; do
if [[ -s ${workon_file} && ${workon_file} != *.mask ]] ; then
echo -e "${V_BOLD_GREEN}$(basename ${workon_file}):${V_VIDOFF}"
sed -n 's/^=\(.*\)-9999$/ \1/p' "${workon_file}"
echo ""
fi
done
}
# This is called only for "cros-workon start". We dont handle the
# "stop" case since the local changes are ignored anyway since the
# 9999.ebuild is masked and we dont want to deal with what to do with
# the user's local changes.
regen_manifest_and_sync() {
# Nothing to do unless you are working on the minilayout
local manifest=${CHROOT_TRUNK_DIR}/.repo/manifest.xml
if [ $(basename $(readlink -f ${manifest})) != "minilayout.xml" ]; then
if [ -z "$(git config -f "${CHROOT_TRUNK_DIR}/.repo/manifests.git/config" \
--get manifest.groups)" ]; then
# Reaching here means that it's a full manifest w/out any groups set-
# literal full manifest.
return
fi
fi
local need_repo_sync=
local pkgname
for pkgname in $(show_live_ebuilds); do
local pkgpath="$(${EQUERYCMD} which "${pkgname}")"
local pkginfo="$(${EBUILDCMD} "${pkgpath}" info |
grep -v 'pkg_info() is not defined')"
if [ -z "${pkginfo}" ]; then
continue # No package information available
fi
eval "${pkginfo}"
local trunkdir=$(readlink -m "${CHROOT_TRUNK_DIR}")
local i=0
need_repo_sync='yes'
for S in "${CROS_WORKON_SRCDIR[@]}"; do
local srcdir=$(readlink -m "${S}")
local revision="${FLAGS_revision:+--revision=${FLAGS_revision}}"
if [ -z "${FLAGS_remote}" ]; then
loman add --workon "${CROS_WORKON_PROJECT[i]}" ${revision}
else
loman add --remote "${FLAGS_remote}" ${revision} \
"${CROS_WORKON_PROJECT[i]}" "${srcdir#${trunkdir}/}"
fi
: $(( ++i ))
done
done
if [ -n "${need_repo_sync}" ]; then
echo "Please run \"repo sync\" now."
fi
}
# Move a stable ebuild to the live development catgeory. The ebuild
# src_unpack step fetches the package source for local development.
ebuild_to_live() {
local atoms=$1
local atoms_success=()
local atom
for atom in ${atoms}; do
if ! grep -qx "=${atom}-9999" "${WORKON_FILE}" ; then
echo "=${atom}-9999" >> "${WORKON_FILE}" || \
die "Could not update ${WORKON_FILE} with ${atom}"
echo "<${atom}-9999" >> "${MASK_WORKON_FILE}" || \
die "Could not update ${MASK_WORKON_FILE} with ${atom}"
atoms_success+=( ${atom} )
else
warn "Already working on ${atom}"
fi
done
[ ${#atoms_success[@]} -gt 0 ] && regen_manifest_and_sync && \
info "Started working on '${atoms_success[*]}' for '${BOARD_STR}'"
}
# Move a live development ebuild back to stable.
ebuild_to_stable() {
local atoms=$1
local atoms_success=()
local atom
for atom in ${atoms}; do
if grep -qx "=${atom}-9999" "${WORKON_FILE}" "${MASK_WORKON_FILE}" ; then
sed -i -e "/^=${atom/\//\\/}-9999\$/d" "${WORKON_FILE}" || \
die "Could not update ${WORKON_FILE} with ${atom}"
sed -i -e "/^<${atom/\//\\/}-9999\$/d" "${MASK_WORKON_FILE}" || \
die "Could not update ${WORKON_FILE} with ${atom}"
atoms_success+=( ${atom} )
else
warn "Not working on ${atom}"
fi
done
[ ${#atoms_success[@]} -gt 0 ] && \
info "Stopped working on '${atoms_success[*]}' for '${BOARD_STR}'"
}
# Run a command on all or a set of repos.
ebuild_iterate() {
local atoms=$1
local atom
for atom in ${atoms}; do
info "Running \"${FLAGS_command}\" on ${atom}"
eval $(${EBUILDCMD} $(${EQUERYCMD} which ${atom}) info)
for S in "${CROS_WORKON_SRCDIR[@]}"; do
(cd "${S}" && bash -c "${FLAGS_command}")
done
done
}
# Only call portageq when absolutely required, and when we do, only run it
# once -- it's a slow beast and can easily take hundreds of milliseconds :(.
if [[ ${WORKON_CMD} != "list" || ${FLAGS_all} != ${FLAGS_FALSE} ]] ; then
portageq_vars="ARCH PORTDIR_OVERLAY"
unset ${portageq_vars}
eval $(${PORTAGEQCMD} envvar -v ${portageq_vars})
fi
# --all makes commands operate on different lists
if [ ${FLAGS_all} = "${FLAGS_TRUE}" ]; then
case ${WORKON_CMD} in
start|info) ATOM_LIST=$(show_workon_ebuilds ${ARCH});;
stop|iterate) ATOM_LIST=$(show_live_ebuilds);;
list) ;;
*) die "--all is invalid for the given command";;
esac
else # not selected --all
case ${WORKON_CMD} in
start|stop|info|iterate)
ATOM_LIST=$@
if [ -z "${ATOM_LIST}" ]; then
die "${WORKON_CMD}: No packages specified"
fi
if [ "${ATOM_LIST}" = "." ]; then
ATOM_LIST=$(locate_package)
fi
if ! ATOM_LIST=$(canonicalize_names "${ATOM_LIST}"); then
die "Error parsing package list"
fi;;
*) ;;
esac
fi
case ${WORKON_CMD} in
start) ebuild_to_live "${ATOM_LIST}" ;;
stop) ebuild_to_stable "${ATOM_LIST}" ;;
info) show_workon_info "${ATOM_LIST}" "${ARCH}" ;;
list) [ ${FLAGS_all} = "${FLAGS_FALSE}" ] && show_live_ebuilds || \
show_workon_ebuilds ${ARCH} ;;
list-all) show_all_live_ebuilds ;;
iterate) ebuild_iterate "${ATOM_LIST}" ;;
*)
flags_help
die "$(basename $0): command '${WORKON_CMD}' not recognized"
;;
esac