1d0825bca7fe65beaee391d30da42e937db621564Steve Block# Copyright (C) 2010 Google Inc. All rights reserved. 2d0825bca7fe65beaee391d30da42e937db621564Steve Block# 3231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# Redistribution and use in source and binary forms, with or without 4231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# modification, are permitted provided that the following conditions are 5231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# met: 6231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# 7231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# * Redistributions of source code must retain the above copyright 8231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# notice, this list of conditions and the following disclaimer. 9231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# * Redistributions in binary form must reproduce the above 10231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# copyright notice, this list of conditions and the following disclaimer 11231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# in the documentation and/or other materials provided with the 12231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# distribution. 13231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# * Neither the name of Google Inc. nor the names of its 14231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# contributors may be used to endorse or promote products derived from 15231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# this software without specific prior written permission. 16231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# 17231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block 29d0825bca7fe65beaee391d30da42e937db621564Steve Blockimport os 30d0825bca7fe65beaee391d30da42e937db621564Steve Block 31dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockfrom webkitpy.tool.steps.abstractstep import AbstractStep 32dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockfrom webkitpy.tool.steps.options import Options 33d0825bca7fe65beaee391d30da42e937db621564Steve Block 34d0825bca7fe65beaee391d30da42e937db621564Steve Block 35d0825bca7fe65beaee391d30da42e937db621564Steve Blockclass CleanWorkingDirectory(AbstractStep): 36d0825bca7fe65beaee391d30da42e937db621564Steve Block def __init__(self, tool, options, allow_local_commits=False): 37d0825bca7fe65beaee391d30da42e937db621564Steve Block AbstractStep.__init__(self, tool, options) 38d0825bca7fe65beaee391d30da42e937db621564Steve Block self._allow_local_commits = allow_local_commits 39231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block 40d0825bca7fe65beaee391d30da42e937db621564Steve Block @classmethod 41d0825bca7fe65beaee391d30da42e937db621564Steve Block def options(cls): 4221939df44de1705786c545cd1bf519d47250322dBen Murdoch return AbstractStep.options() + [ 43d0825bca7fe65beaee391d30da42e937db621564Steve Block Options.force_clean, 44d0825bca7fe65beaee391d30da42e937db621564Steve Block Options.clean, 45d0825bca7fe65beaee391d30da42e937db621564Steve Block ] 46231d4e3152a9c27a73b6ac7badbe6be673aa3ddfSteve Block 47d0825bca7fe65beaee391d30da42e937db621564Steve Block def run(self, state): 48f05b935882198ccf7d81675736e3aeb089c5113aBen Murdoch if not self._options.clean: 49f05b935882198ccf7d81675736e3aeb089c5113aBen Murdoch return 50cad810f21b803229eb11403f9209855525a25d57Steve Block # FIXME: This chdir should not be necessary and can be removed as 51cad810f21b803229eb11403f9209855525a25d57Steve Block # soon as ensure_no_local_commits and ensure_clean_working_directory 52cad810f21b803229eb11403f9209855525a25d57Steve Block # are known to set the CWD to checkout_root when calling run_command. 53d0825bca7fe65beaee391d30da42e937db621564Steve Block os.chdir(self._tool.scm().checkout_root) 54d0825bca7fe65beaee391d30da42e937db621564Steve Block if not self._allow_local_commits: 55d0825bca7fe65beaee391d30da42e937db621564Steve Block self._tool.scm().ensure_no_local_commits(self._options.force_clean) 56f05b935882198ccf7d81675736e3aeb089c5113aBen Murdoch self._tool.scm().ensure_clean_working_directory(force_clean=self._options.force_clean) 57