1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2020 The ChromiumOS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""The script to generate a cleanup script after setup.sh.
This script takes a set of flags, telling it what setup.sh changed
during the set up process. Based on the values of the input flags, it
generates a cleanup script, named ${BOARD}_cleanup.sh, which will
undo the changes made by setup.sh, returning everything to its
original state.
"""
from __future__ import print_function
import argparse
import sys
def Usage(parser, msg):
print("ERROR: " + msg)
parser.print_help()
sys.exit(1)
def Main(argv):
"""Generate a script to undo changes done by setup.sh
The script setup.sh makes a change that needs to be
undone, namely it creates a soft link making /build/${board} point
to /build/${board}.work. To do this, it had to see if
/build/${board} already existed, and if so, whether it was a real
tree or a soft link. If it was soft link, it saved the old value
of the link, then deleted it and created the new link. If it was
a real tree, it renamed the tree to /build/${board}.save, and then
created the new soft link. If the /build/${board} did not
previously exist, then it just created the new soft link.
This function takes arguments that tell it exactly what setup.sh
actually did, then generates a script to undo those exact changes.
"""
parser = argparse.ArgumentParser()
parser.add_argument(
"--board",
dest="board",
required=True,
help="Chromeos board for packages/image.",
)
parser.add_argument(
"--old_tree_missing",
dest="tree_existed",
action="store_false",
help="Did /build/${BOARD} exist.",
default=True,
)
parser.add_argument(
"--renamed_tree",
dest="renamed_tree",
action="store_true",
help="Was /build/${BOARD} saved & renamed.",
default=False,
)
parser.add_argument(
"--old_link",
dest="old_link",
help=("The original build tree soft link."),
)
options = parser.parse_args(argv[1:])
if options.old_link or options.renamed_tree:
if not options.tree_existed:
Usage(
parser,
"If --tree_existed is False, cannot have "
"--renamed_tree or --old_link",
)
if options.old_link and options.renamed_tree:
Usage(parser, "--old_link and --renamed_tree are incompatible options.")
if options.tree_existed:
if not options.old_link and not options.renamed_tree:
Usage(
parser,
"If --tree_existed is True, then must have either "
"--old_link or --renamed_tree",
)
out_filename = "cros_pkg/" + options.board + "_cleanup.sh"
with open(out_filename, "w", encoding="utf-8") as out_file:
out_file.write("#!/bin/bash\n\n")
# First, remove the 'new' soft link.
out_file.write("sudo rm /build/%s\n" % options.board)
if options.tree_existed:
if options.renamed_tree:
# Old build tree existed and was a real tree, so it got
# renamed. Move the renamed tree back to the original tree.
out_file.write(
"sudo mv /build/%s.save /build/%s\n"
% (options.board, options.board)
)
else:
# Old tree existed and was already a soft link. Re-create the
# original soft link.
original_link = options.old_link
if original_link[0] == "'":
original_link = original_link[1:]
if original_link[-1] == "'":
original_link = original_link[:-1]
out_file.write(
"sudo ln -s %s /build/%s\n" % (original_link, options.board)
)
out_file.write("\n")
# Remove common.sh file
out_file.write("rm common/common.sh\n")
return 0
if __name__ == "__main__":
retval = Main(sys.argv)
sys.exit(retval)
|
