cleaned up the clone logic
[build.git] / vbuild-init-vserver.sh
1 #!/bin/bash
2 # -*-shell-*-
3
4 COMMAND=$(basename $0)
5 DIRNAME=$(dirname $0)
6
7 # lst parsing utilities
8 PATH=$(dirname $0):$PATH . build.common
9
10 function failure () {
11     echo "$COMMAND : Bailing out"
12     exit 1
13 }
14
15 # overwrite vserver's internal yum config from what is in
16 # .distributions/<distrib>/yum/yum.conf and /yum.repos.d 
17
18 function configure_yum_in_vserver () {
19     set -x 
20     set -e 
21     trap failure ERR INT
22
23     vserver=$1; shift
24     fcdistro=$1; shift
25
26     cd /etc/vservers/.distributions/${fcdistro}
27     if [ -f yum/yum.conf ] ; then
28         echo "Initializing yum.conf in $vserver from $(pwd)/yum"
29         sed -e "s!@YUMETCDIR@!/etc!g;
30                 s!@YUMCACHEDIR@!/var/cache/yum!g;
31                 s!@YUMLOGDIR@!/var/log!g;
32                 s!@YUMLOCKDIR@!/var/lock!g;
33                " yum/yum.conf > /vservers/$vserver/etc/yum.conf
34
35         # post process the various @...@ variables from this yum.conf file.
36     else
37         echo "Using $fcdistro default for yum.conf"
38     fi
39
40     if [ -d yum.repos.d ] ; then
41         echo "Initializing yum.repos.d in $vserver from $(pwd)/yum.repos.d"
42         rm -rf /vservers/$vserver/etc/yum.repos.d
43         tar cf - yum.repos.d | tar -C /vservers/$vserver/etc -xvf -
44     else
45         echo "Cannot initialize yum.repos.d in $vserver"
46     fi
47
48     if [ -n "$MYPLC_MODE" ] ; then
49         if [ ! -d /vservers/$vserver/etc/yum.repos.d ] ; then
50             echo "WARNING : cannot create myplc repo"
51         else
52             # exclude kernel from fedora repos 
53             for i in /vservers/$vserver/etc/yum.repos.d/* ; do
54                 echo "exclude=kernel* ulogd iptables" >> $i
55             done
56             # the build repo is not signed at this stage
57             cat > /vservers/$vserver/etc/yum.repos.d/myplc.repo <<EOF
58 [myplc]
59 name= MyPLC
60 baseurl=$REPO_URL
61 enabled=1
62 gpgcheck=0
63 EOF
64         fi
65     fi
66     cd -
67 }    
68
69 function setup_vserver () {
70
71     set -x
72     set -e
73     trap failure ERR INT
74
75     vserver=$1; shift
76     fcdistro=$1; shift
77     personality=$1; shift
78
79     if [ -d /etc/vservers/$vserver ] ; then
80         echo "$COMMAND : vserver $vserver seems to exist - bailing out"
81         exit 1
82     fi
83
84     # create it
85     $personality vserver $VERBOSE $vserver build -m yum -- -d $fcdistro
86
87     if [ ! -z "$personality" ] ; then
88         l32=$(grep $personality /etc/vservers/$vserver/personality | wc -l)
89         [ $l32 -eq 0 ] && echo $personality >> /etc/vservers/$vserver/personality
90     fi
91
92     if [ -n "$VBUILD_MODE" ] ; then 
93     # set up appropriate vserver capabilities to mount, mknod and IPC_LOCK
94         BCAPFILE=/etc/vservers/$vserver/bcapabilities
95         touch $BCAPFILE
96         cap=$(grep ^CAP_SYS_ADMIN /etc/vservers/$vserver/bcapabilities | wc -l)
97         [ $cap -eq 0 ] && echo 'CAP_SYS_ADMIN' >> /etc/vservers/$vserver/bcapabilities
98         cap=$(grep ^CAP_MKNOD /etc/vservers/$vserver/bcapabilities | wc -l)
99         [ $cap -eq 0 ] && echo 'CAP_MKNOD' >> /etc/vservers/$vserver/bcapabilities
100         cap=$(grep ^CAP_IPC_LOCK /etc/vservers/$vserver/bcapabilities | wc -l)
101         [ $cap -eq 0 ] && echo 'CAP_IPC_LOCK' >> /etc/vservers/$vserver/bcapabilities
102     fi
103
104     # start the vserver so we can do the following operations
105     if [ $CLONED -eq 0 ] ; then
106         $personality vyum $vserver -- -y install yum
107         $personality vserver $VERBOSE $vserver pkgmgmt internalize
108     fi
109     $personality vserver $VERBOSE $vserver start
110     $personality vserver $VERBOSE $vserver exec rm -f /var/lib/rpm/__db*
111     $personality vserver $VERBOSE $vserver exec rpm --rebuilddb
112
113     # minimal config in the vserver for yum to work
114     configure_yum_in_vserver $vserver $fcdistro 
115
116     # set up resolv.conf
117     cp /etc/resolv.conf /vservers/$vserver/etc/resolv.conf
118 }
119
120 function devel_tools () {
121
122     set -x 
123     set -e 
124     trap failure ERR INT
125
126     vserver=$1; shift
127     fcdistro=$1; shift
128     pldistro=$1; shift
129     personality=$1; shift
130
131     # check for .lst file based on pldistro
132     if [ -n "$VBUILD_MODE" ] ; then
133         lst=${pldistro}-devel.lst
134     else
135         lst=${pldistro}-shell.lst
136     fi
137     if [ -f $lst ] ; then
138         echo "$COMMAND: Using $lst"
139     elif [ -f $DIRNAME/$lst ] ; then
140         lst=$DIRNAME/$lst
141         echo "$COMMAND: Using $lst"
142     else
143         echo "$COMMAND : Cannot locate $lst - exiting"
144         usage
145     fi
146
147     # install individual packages, then groups
148     packages=$(pl_getPackages2 ${fcdistro} $lst)
149     groups=$(pl_getGroups2 ${fcdistro} $lst)
150
151     [ -n "$packages" ] && $personality vserver $vserver exec yum -y install $packages
152     [ -n "$groups" ] && $personality vserver $vserver exec yum -y groupinstall $groups
153     return 0
154 }
155
156 function post_install () {
157     if [ -n "$VBUILD_MODE" ] ; then
158         post_install_vbuild "$@" 
159     else
160         post_install_myplc "$@"
161     fi
162 }
163
164 function post_install_vbuild () {
165
166     set -x 
167     set -e 
168     trap failure ERR INT
169
170     vserver=$1; shift
171     personality=$1; shift
172
173 ### From myplc-devel-native.spec
174 # be careful to backslash $ in this, otherwise it's the root context that's going to do the evaluation
175     cat << EOF | $personality vserver $VERBOSE $vserver exec bash -x
176     # set up /dev/loop* in vserver
177     for i in \$(seq 0 255) ; do
178         mknod -m 640 /dev/loop\$i b 7 \$i
179     done
180     
181     # create symlink for /dev/fd
182     ln -fs /proc/self/fd /dev/fd
183
184     # modify /etc/rpm/macros to not use /sbin/new-kernel-pkg
185     sed -i 's,/sbin/new-kernel-pkg:,,' /etc/rpm/macros
186     if [ -h "/sbin/new-kernel-pkg" ] ; then
187         filename=\$(readlink -f /sbin/new-kernel-pkg)
188         if [ "\$filename" == "/sbin/true" ] ; then
189                 echo "WARNING: /sbin/new-kernel-pkg symlinked to /sbin/true"
190                 echo "\tmost likely /etc/rpm/macros has /sbin/new-kernel-pkg declared in _netsharedpath."
191                 echo "\tPlease remove /sbin/new-kernel-pkg from _netsharedpath and reintall mkinitrd."
192                 exit 1
193         fi
194     fi
195     
196     # customize root's prompt
197     cat << PROFILE > /root/.profile
198 export PS1="[$vserver] \\w # "
199 PROFILE
200
201     uid=2000
202     gid=2000
203     
204     # add a "build" user to the system
205     builduser=\$(grep "^build:" /etc/passwd | wc -l)
206     if [ \$builduser -eq 0 ] ; then
207         groupadd -o -g \$gid build;
208         useradd -o -c 'Automated Build' -u \$uid -g \$gid -n -M -s /bin/bash build;
209     fi
210
211 # Allow build user to build certain RPMs as root
212     if [ -f /etc/sudoers ] ; then
213         buildsudo=\$(grep "^build.*ALL=(ALL).*NOPASSWD:.*ALL"  /etc/sudoers | wc -l)
214         if [ \$buildsudo -eq 0 ] ; then
215             echo "build   ALL=(ALL)       NOPASSWD: ALL" >> /etc/sudoers
216         fi
217         sed -i 's,^Defaults.*requiretty,#Defaults requiretty,' /etc/sudoers
218     fi
219 #
220 EOF
221
222 }
223
224 function post_install_myplc  () {
225     set -x 
226     set -e 
227     trap failure ERR INT
228
229     vserver=$1; shift
230     personality=$1; shift
231
232 # be careful to backslash $ in this, otherwise it's the root context that's going to do the evaluation
233     cat << EOF | $personality vserver $VERBOSE $vserver exec bash -x
234
235     # customize root's prompt
236     cat << PROFILE > /root/.profile
237 export PS1="[$vserver] \\w # "
238 PROFILE
239
240 EOF
241 }
242
243 COMMAND_VBUILD="vbuild-init-vserver.sh"
244 COMMAND_MYPLC="myplc-init-vserver.sh"
245 function usage () {
246     set +x 
247     echo "Usage: $COMMAND_VBUILD [-s|s|p] [-v] vserver-name distribution pldistro [personality]"
248     echo "Usage: $COMMAND_MYPLC [-s|s|p] [-v] vserver-name distribution pldistro repo-url [personality]"
249     echo "Requirements: you need to have a vserver-compliant kernel,"
250     echo "  as well as the util-vserver RPM installed"
251     echo "Description:"
252     echo "  This command creates a fresh vserver instance"
253     echo "  . vserver-name : the vserver's name"
254     echo "  . distribution : for creating the root filesystem, e.g. fc6"
255     echo "  . pldistro: e.g. onelab"
256     echo "  . repo-url: for myplc vserver, used to create a yum repository"
257     echo "  . personality: last, optional, argument defaults to linux32"
258     echo "This is done in three steps"
259     echo " (*) setup phase : vserver creation, yum internalization and config (from /etc/vservers)"
260     echo " (*) tools install the tools required for building are installed"
261     echo "     based on the folowing file for the actual set of packages and groups"
262     echo "     vbuild mode :  <pldistro>-devel.lst"
263     echo "     myplc mode :   <pldistro>-shell.lst"
264     echo " (*) post-install : various tunings required, as well as create a build user (vbuild only)"
265     echo "Options:"
266     echo " -s : skips the setup phase"
267     echo " -t : skips the tools phase"
268     echo " -p : skips the post-install"
269     echo " -v : passes -v to calls to vserver"
270     exit 1
271 }
272
273 ### parse args and 
274 function main () {
275
276     set -e
277     trap failure ERR INT
278
279     case "$COMMAND" in
280         $COMMAND_VBUILD)
281             VBUILD_MODE=true ;;
282         $COMMAND_MYPLC)
283             MYPLC_MODE=true;;
284         *)
285             usage ;;
286     esac
287
288     DO_SETUP=true
289     DO_TOOLS=true
290     DO_POST=true
291     VERBOSE=
292     while getopts "stpvh" opt ; do
293         case $opt in
294             s) DO_SETUP="" ;;
295             t) DO_TOOLS="" ;;
296             p) DO_POST="" ;;
297             v) VERBOSE="-v" ;;
298             h|*) usage ;;
299         esac
300     done
301         
302     shift $(($OPTIND - 1))
303     
304     [[ -z "$@" ]] && usage
305     vserver=$1 ; shift
306     [[ -z "$@" ]] && usage
307     fcdistro=$1 ; shift
308     [[ -z "$@" ]] && usage
309     pldistro=$1 ; shift
310     if [ -n "$MYPLC_MODE" ] ; then
311         [[ -z "$@" ]] && usage
312         REPO_URL=$1 ; shift
313     fi
314     if [[ -z "$@" ]] ; then
315         personality=linux32
316     else
317         personality=$1; shift
318     fi
319     [[ -n "$@" ]] && usage
320
321     [ -n "$DO_SETUP" ] && setup_vserver $vserver $fcdistro $personality 
322     [ -n "$DO_TOOLS" ] && devel_tools $vserver $fcdistro $pldistro $personality
323     [ -n "$DO_POST" ] && post_install $vserver $personality
324
325 }
326
327 main "$@"