code.grnet.gr Git - snf-image-creator/blob - image_creator/dialog_menu.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # Copyright 2012 GRNET S.A. All rights reserved.
   4 #
   5 # Redistribution and use in source and binary forms, with or
   6 # without modification, are permitted provided that the following
   7 # conditions are met:
   8 #
   9 #   1. Redistributions of source code must retain the above
  10 #      copyright notice, this list of conditions and the following
  11 #      disclaimer.
  12 #
  13 #   2. Redistributions in binary form must reproduce the above
  14 #      copyright notice, this list of conditions and the following
  15 #      disclaimer in the documentation and/or other materials
  16 #      provided with the distribution.
  17 #
  18 # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
  19 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  21 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
  22 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  25 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  26 # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  28 # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29 # POSSIBILITY OF SUCH DAMAGE.
  30 #
  31 # The views and conclusions contained in the software and
  32 # documentation are those of the authors and should not be
  33 # interpreted as representing official policies, either expressed
  34 # or implied, of GRNET S.A.
  35
  36 """This module implements the "expert" mode of the dialog-based version of
  37 snf-image-creator.
  38 """
  39
  40 import os
  41 import textwrap
  42 import StringIO
  43 import json
  44
  45 from image_creator import __version__ as version
  46 from image_creator.util import MD5, FatalError
  47 from image_creator.output.dialog import GaugeOutput, InfoBoxOutput
  48 from image_creator.kamaki_wrapper import Kamaki, ClientError
  49 from image_creator.help import get_help_file
  50 from image_creator.dialog_util import SMALL_WIDTH, WIDTH, \
  51     update_background_title, confirm_reset, confirm_exit, Reset, \
  52     extract_image, extract_metadata_string, add_cloud, edit_cloud
  53
  54 CONFIGURATION_TASKS = [
  55     ("Partition table manipulation", ["FixPartitionTable"],
  56         ["linux", "windows"]),
  57     ("File system resize",
  58         ["FilesystemResizeUnmounted", "FilesystemResizeMounted"],
  59         ["linux", "windows"]),
  60     ("Swap partition configuration", ["AddSwap"], ["linux"]),
  61     ("SSH keys removal", ["DeleteSSHKeys"], ["linux"]),
  62     ("Temporal RDP disabling", ["DisableRemoteDesktopConnections"],
  63         ["windows"]),
  64     ("SELinux relabeling at next boot", ["SELinuxAutorelabel"], ["linux"]),
  65     ("Hostname/Computer Name assignment", ["AssignHostname"],
  66         ["windows", "linux"]),
  67     ("Password change", ["ChangePassword"], ["windows", "linux"]),
  68     ("File injection", ["EnforcePersonality"], ["windows", "linux"])
  69 ]
  70
  71
  72 class MetadataMonitor(object):
  73     """Monitors image metadata chages"""
  74     def __init__(self, session, meta):
  75         self.session = session
  76         self.meta = meta
  77
  78     def __enter__(self):
  79         self.old = {}
  80         for (k, v) in self.meta.items():
  81             self.old[k] = v
  82
  83     def __exit__(self, type, value, traceback):
  84         d = self.session['dialog']
  85
  86         altered = {}
  87         added = {}
  88
  89         for (k, v) in self.meta.items():
  90             if k not in self.old:
  91                 added[k] = v
  92             elif self.old[k] != v:
  93                 altered[k] = v
  94
  95         if not (len(added) or len(altered)):
  96             return
  97
  98         msg = "The last action has changed some image properties:\n\n"
  99         if len(added):
 100             msg += "New image properties:\n"
 101             for (k, v) in added.items():
 102                 msg += '    %s: "%s"\n' % (k, v)
 103             msg += "\n"
 104         if len(altered):
 105             msg += "Updated image properties:\n"
 106             for (k, v) in altered.items():
 107                 msg += '    %s: "%s" -> "%s"\n' % (k, self.old[k], v)
 108             msg += "\n"
 109
 110         self.session['metadata'].update(added)
 111         self.session['metadata'].update(altered)
 112         d.msgbox(msg, title="Image Property Changes", width=SMALL_WIDTH)
 113
 114
 115 def upload_image(session):
 116     """Upload the image to the storage service"""
 117     d = session["dialog"]
 118     image = session['image']
 119     meta = session['metadata']
 120     size = image.size
 121
 122     if "account" not in session:
 123         d.msgbox("You need to select a valid cloud before you can upload "
 124                  "images to it", width=SMALL_WIDTH)
 125         return False
 126
 127     while 1:
 128         if 'upload' in session:
 129             init = session['upload']
 130         elif 'OS' in meta:
 131             init = "%s.diskdump" % meta['OS']
 132         else:
 133             init = ""
 134         (code, answer) = d.inputbox("Please provide a filename:", init=init,
 135                                     width=WIDTH)
 136
 137         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 138             return False
 139
 140         filename = answer.strip()
 141         if len(filename) == 0:
 142             d.msgbox("Filename cannot be empty", width=SMALL_WIDTH)
 143             continue
 144
 145         kamaki = Kamaki(session['account'], None)
 146         overwrite = []
 147         for f in (filename, "%s.md5sum" % filename, "%s.meta" % filename):
 148             if kamaki.object_exists(f):
 149                 overwrite.append(f)
 150
 151         if len(overwrite) > 0:
 152             if d.yesno("The following storage service object(s) already "
 153                        "exist(s):\n%s\nDo you want to overwrite them?" %
 154                        "\n".join(overwrite), width=WIDTH, defaultno=1):
 155                 continue
 156
 157         session['upload'] = filename
 158         break
 159
 160     gauge = GaugeOutput(d, "Image Upload", "Uploading ...")
 161     try:
 162         out = image.out
 163         out.add(gauge)
 164         kamaki.out = out
 165         try:
 166             if 'checksum' not in session:
 167                 md5 = MD5(out)
 168                 session['checksum'] = md5.compute(image.device, size)
 169
 170             try:
 171                 # Upload image file
 172                 with open(image.device, 'rb') as f:
 173                     session["pithos_uri"] = \
 174                         kamaki.upload(f, size, filename,
 175                                       "Calculating block hashes",
 176                                       "Uploading missing blocks")
 177                 # Upload md5sum file
 178                 out.output("Uploading md5sum file ...")
 179                 md5str = "%s %s\n" % (session['checksum'], filename)
 180                 kamaki.upload(StringIO.StringIO(md5str), size=len(md5str),
 181                               remote_path="%s.md5sum" % filename)
 182                 out.success("done")
 183
 184             except ClientError as e:
 185                 d.msgbox(
 186                     "Error in storage service client: %s" % e.message,
 187                     title="Storage Service Client Error", width=SMALL_WIDTH)
 188                 if 'pithos_uri' in session:
 189                     del session['pithos_uri']
 190                 return False
 191         finally:
 192             out.remove(gauge)
 193     finally:
 194         gauge.cleanup()
 195
 196     d.msgbox("Image file `%s' was successfully uploaded" % filename,
 197              width=SMALL_WIDTH)
 198
 199     return True
 200
 201
 202 def register_image(session):
 203     """Register image with the compute service"""
 204     d = session["dialog"]
 205
 206     is_public = False
 207
 208     if "account" not in session:
 209         d.msgbox("You need to select a valid cloud before you "
 210                  "can register an images with it", width=SMALL_WIDTH)
 211         return False
 212
 213     if "pithos_uri" not in session:
 214         d.msgbox("You need to upload the image to the cloud before you can "
 215                  "register it", width=SMALL_WIDTH)
 216         return False
 217
 218     while 1:
 219         (code, answer) = d.inputbox("Please provide a registration name:",
 220                                     width=WIDTH)
 221         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 222             return False
 223
 224         name = answer.strip()
 225         if len(name) == 0:
 226             d.msgbox("Registration name cannot be empty", width=SMALL_WIDTH)
 227             continue
 228
 229         ret = d.yesno("Make the image public?\\nA public image is accessible "
 230                       "by every user of the service.", defaultno=1,
 231                       width=WIDTH)
 232         if ret not in (0, 1):
 233             continue
 234
 235         is_public = True if ret == 0 else False
 236
 237         break
 238
 239     metadata = {}
 240     metadata.update(session['metadata'])
 241     if 'task_metadata' in session:
 242         for key in session['task_metadata']:
 243             metadata[key] = 'yes'
 244
 245     img_type = "public" if is_public else "private"
 246     gauge = GaugeOutput(d, "Image Registration", "Registering image ...")
 247     try:
 248         out = session['image'].out
 249         out.add(gauge)
 250         try:
 251             try:
 252                 out.output("Registering %s image with the cloud ..." %
 253                            img_type)
 254                 kamaki = Kamaki(session['account'], out)
 255                 result = kamaki.register(name, session['pithos_uri'], metadata,
 256                                          is_public)
 257                 out.success('done')
 258                 # Upload metadata file
 259                 out.output("Uploading metadata file ...")
 260                 metastring = unicode(json.dumps(result, ensure_ascii=False))
 261                 kamaki.upload(StringIO.StringIO(metastring),
 262                               size=len(metastring),
 263                               remote_path="%s.meta" % session['upload'])
 264                 out.success("done")
 265                 if is_public:
 266                     out.output("Sharing metadata and md5sum files ...")
 267                     kamaki.share("%s.meta" % session['upload'])
 268                     kamaki.share("%s.md5sum" % session['upload'])
 269                     out.success('done')
 270             except ClientError as e:
 271                 d.msgbox("Error in storage service client: %s" % e.message)
 272                 return False
 273         finally:
 274             out.remove(gauge)
 275     finally:
 276         gauge.cleanup()
 277
 278     d.msgbox("%s image `%s' was successfully registered with the cloud as `%s'"
 279              % (img_type.title(), session['upload'], name), width=SMALL_WIDTH)
 280     return True
 281
 282
 283 def modify_clouds(session):
 284     """Modify existing cloud accounts"""
 285     d = session['dialog']
 286
 287     while 1:
 288         clouds = Kamaki.get_clouds()
 289         if not len(clouds):
 290             if not add_cloud(session):
 291                 break
 292             continue
 293
 294         choices = []
 295         for (name, cloud) in clouds.items():
 296             descr = cloud['description'] if 'description' in cloud else ''
 297             choices.append((name, descr))
 298
 299         (code, choice) = d.menu(
 300             "In this menu you can edit existing cloud accounts or add new "
 301             " ones. Press <Edit> to edit an existing account or <Add> to add "
 302             " a new one. Press <Back> or hit <ESC> when done.", height=18,
 303             width=WIDTH, choices=choices, menu_height=10, ok_label="Edit",
 304             extra_button=1, extra_label="Add", cancel="Back", help_button=1,
 305             title="Clouds")
 306
 307         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 308             return True
 309         elif code == d.DIALOG_OK:  # Edit button
 310             edit_cloud(session, choice)
 311         elif code == d.DIALOG_EXTRA:  # Add button
 312             add_cloud(session)
 313
 314
 315 def delete_clouds(session):
 316     """Delete existing cloud accounts"""
 317     d = session['dialog']
 318
 319     choices = []
 320     for (name, cloud) in Kamaki.get_clouds().items():
 321         descr = cloud['description'] if 'description' in cloud else ''
 322         choices.append((name, descr, 0))
 323
 324     if len(choices) == 0:
 325         d.msgbox("No available clouds to delete!", width=SMALL_WIDTH)
 326         return True
 327
 328     (code, to_delete) = d.checklist("Choose which cloud accounts to delete:",
 329                                     choices=choices, width=WIDTH)
 330     to_delete = map(lambda x: x.strip('"'), to_delete)  # Needed for OpenSUSE
 331
 332     if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 333         return False
 334
 335     if not len(to_delete):
 336         d.msgbox("Nothing selected!", width=SMALL_WIDTH)
 337         return False
 338
 339     if not d.yesno("Are you sure you want to remove the selected cloud "
 340                    "accounts?", width=WIDTH, defaultno=1):
 341         for i in to_delete:
 342             Kamaki.remove_cloud(i)
 343             if 'cloud' in session and session['cloud'] == i:
 344                 del session['cloud']
 345                 if 'account' in session:
 346                     del session['account']
 347     else:
 348         return False
 349
 350     d.msgbox("%d cloud accounts were deleted." % len(to_delete),
 351              width=SMALL_WIDTH)
 352     return True
 353
 354
 355 def kamaki_menu(session):
 356     """Show kamaki related actions"""
 357     d = session['dialog']
 358     default_item = "Cloud"
 359
 360     if 'cloud' not in session:
 361         cloud = Kamaki.get_default_cloud_name()
 362         if cloud:
 363             session['cloud'] = cloud
 364             session['account'] = Kamaki.get_account(cloud)
 365             if not session['account']:
 366                 del session['account']
 367         else:
 368             default_item = "Add/Edit"
 369
 370     while 1:
 371         cloud = session["cloud"] if "cloud" in session else "<none>"
 372         if 'account' not in session and 'cloud' in session:
 373             cloud += " <invalid>"
 374
 375         upload = session["upload"] if "upload" in session else "<none>"
 376
 377         choices = [("Add/Edit", "Add/Edit cloud accounts"),
 378                    ("Delete", "Delete existing cloud accounts"),
 379                    ("Cloud", "Select cloud account to use: %s" % cloud),
 380                    ("Upload", "Upload image to the cloud"),
 381                    ("Register", "Register image with the cloud: %s" % upload)]
 382
 383         (code, choice) = d.menu(
 384             text="Choose one of the following or press <Back> to go back.",
 385             width=WIDTH, choices=choices, cancel="Back", height=13,
 386             menu_height=5, default_item=default_item,
 387             title="Image Registration Menu")
 388
 389         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 390             return False
 391
 392         if choice == "Add/Edit":
 393             if modify_clouds(session):
 394                 default_item = "Cloud"
 395         elif choice == "Delete":
 396             if delete_clouds(session):
 397                 if len(Kamaki.get_clouds()):
 398                     default_item = "Cloud"
 399                 else:
 400                     default_time = "Add/Edit"
 401             else:
 402                 default_time = "Delete"
 403         elif choice == "Cloud":
 404             default_item = "Cloud"
 405             clouds = Kamaki.get_clouds()
 406             if not len(clouds):
 407                 d.msgbox("No clouds available. Please add a new cloud!",
 408                          width=SMALL_WIDTH)
 409                 default_item = "Add/Edit"
 410                 continue
 411
 412             if 'cloud' not in session:
 413                 session['cloud'] = clouds.keys()[0]
 414
 415             choices = []
 416             for name, info in clouds.items():
 417                 default = 1 if session['cloud'] == name else 0
 418                 descr = info['description'] if 'description' in info else ""
 419                 choices.append((name, descr, default))
 420
 421             (code, answer) = d.radiolist("Please select a cloud:",
 422                                          width=WIDTH, choices=choices)
 423             if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 424                 continue
 425             else:
 426                 session['account'] = Kamaki.get_account(answer)
 427
 428                 if session['account'] is None:  # invalid account
 429                     if not d.yesno("The cloud %s' is not valid! Would you "
 430                                    "like to edit it?" % answer, width=WIDTH):
 431                         if edit_cloud(session, answer):
 432                             session['account'] = Kamaki.get_account(answer)
 433                             Kamaki.set_default_cloud(answer)
 434
 435                 if session['account'] is not None:
 436                     session['cloud'] = answer
 437                     Kamaki.set_default_cloud(answer)
 438                     default_item = "Upload"
 439                 else:
 440                     del session['account']
 441                     del session['cloud']
 442         elif choice == "Upload":
 443             if upload_image(session):
 444                 default_item = "Register"
 445             else:
 446                 default_item = "Upload"
 447         elif choice == "Register":
 448             if register_image(session):
 449                 return True
 450             else:
 451                 default_item = "Register"
 452
 453
 454 def add_property(session):
 455     """Add a new property to the image"""
 456     d = session['dialog']
 457
 458     while 1:
 459         (code, answer) = d.inputbox("Please provide a name for a new image"
 460                                     " property:", width=WIDTH)
 461         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 462             return False
 463
 464         name = answer.strip()
 465         if len(name) == 0:
 466             d.msgbox("A property name cannot be empty", width=SMALL_WIDTH)
 467             continue
 468
 469         break
 470
 471     while 1:
 472         (code, answer) = d.inputbox("Please provide a value for image "
 473                                     "property %s" % name, width=WIDTH)
 474         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 475             return False
 476
 477         value = answer.strip()
 478         if len(value) == 0:
 479             d.msgbox("Value cannot be empty", width=SMALL_WIDTH)
 480             continue
 481
 482         break
 483
 484     session['metadata'][name] = value
 485
 486     return True
 487
 488
 489 def modify_properties(session):
 490     """Modify an existing image property"""
 491     d = session['dialog']
 492
 493     while 1:
 494         choices = []
 495         for (key, val) in session['metadata'].items():
 496             choices.append((str(key), str(val)))
 497
 498         (code, choice) = d.menu(
 499             "In this menu you can edit existing image properties or add new "
 500             "ones. Be careful! Most properties have special meaning and "
 501             "alter the image deployment behaviour. Press <HELP> to see more "
 502             "information about image properties. Press <BACK> when done.",
 503             height=18, width=WIDTH, choices=choices, menu_height=10,
 504             ok_label="Edit", extra_button=1, extra_label="Add", cancel="Back",
 505             help_button=1, title="Image Properties")
 506
 507         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 508             return True
 509         # Edit button
 510         elif code == d.DIALOG_OK:
 511             (code, answer) = d.inputbox("Please provide a new value for the "
 512                                         "image property with name `%s':" %
 513                                         choice,
 514                                         init=session['metadata'][choice],
 515                                         width=WIDTH)
 516             if code not in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 517                 value = answer.strip()
 518                 if len(value) == 0:
 519                     d.msgbox("Value cannot be empty!")
 520                     continue
 521                 else:
 522                     session['metadata'][choice] = value
 523         # ADD button
 524         elif code == d.DIALOG_EXTRA:
 525             add_property(session)
 526         elif code == 'help':
 527             help_file = get_help_file("image_properties")
 528             assert os.path.exists(help_file)
 529             d.textbox(help_file, title="Image Properties", width=70, height=40)
 530
 531
 532 def delete_properties(session):
 533     """Delete an image property"""
 534     d = session['dialog']
 535
 536     choices = []
 537     for (key, val) in session['metadata'].items():
 538         choices.append((key, "%s" % val, 0))
 539
 540     (code, to_delete) = d.checklist("Choose which properties to delete:",
 541                                     choices=choices, width=WIDTH)
 542     to_delete = map(lambda x: x.strip('"'), to_delete)  # needed for OpenSUSE
 543
 544     # If the user exits with ESC or CANCEL, the returned tag list is empty.
 545     for i in to_delete:
 546         del session['metadata'][i]
 547
 548     cnt = len(to_delete)
 549     if cnt > 0:
 550         d.msgbox("%d image properties were deleted." % cnt, width=SMALL_WIDTH)
 551         return True
 552     else:
 553         return False
 554
 555
 556 def exclude_tasks(session):
 557     """Exclude specific tasks from running during image deployment"""
 558     d = session['dialog']
 559
 560     index = 0
 561     displayed_index = 1
 562     choices = []
 563     mapping = {}
 564     if 'excluded_tasks' not in session:
 565         session['excluded_tasks'] = []
 566
 567     if -1 in session['excluded_tasks']:
 568         if not d.yesno("Image deployment configuration is disabled. "
 569                        "Do you wish to enable it?", width=SMALL_WIDTH):
 570             session['excluded_tasks'].remove(-1)
 571         else:
 572             return False
 573
 574     for (msg, task, osfamily) in CONFIGURATION_TASKS:
 575         if session['metadata']['OSFAMILY'] in osfamily:
 576             checked = 1 if index in session['excluded_tasks'] else 0
 577             choices.append((str(displayed_index), msg, checked))
 578             mapping[displayed_index] = index
 579             displayed_index += 1
 580         index += 1
 581
 582     while 1:
 583         (code, tags) = d.checklist(
 584             text="Please choose which configuration tasks you would like to "
 585                  "prevent from running during image deployment. "
 586                  "Press <No Config> to supress any configuration. "
 587                  "Press <Help> for more help on the image deployment "
 588                  "configuration tasks.",
 589             choices=choices, height=19, list_height=8, width=WIDTH,
 590             help_button=1, extra_button=1, extra_label="No Config",
 591             title="Exclude Configuration Tasks")
 592         tags = map(lambda x: x.strip('"'), tags)  # Needed for OpenSUSE
 593
 594         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 595             return False
 596         elif code == d.DIALOG_HELP:
 597             help_file = get_help_file("configuration_tasks")
 598             assert os.path.exists(help_file)
 599             d.textbox(help_file, title="Configuration Tasks",
 600                       width=70, height=40)
 601         # No Config button
 602         elif code == d.DIALOG_EXTRA:
 603             session['excluded_tasks'] = [-1]
 604             session['task_metadata'] = ["EXCLUDE_ALL_TASKS"]
 605             break
 606         elif code == d.DIALOG_OK:
 607             session['excluded_tasks'] = []
 608             for tag in tags:
 609                 session['excluded_tasks'].append(mapping[int(tag)])
 610
 611             exclude_metadata = []
 612             for task in session['excluded_tasks']:
 613                 exclude_metadata.extend(CONFIGURATION_TASKS[task][1])
 614
 615             session['task_metadata'] = map(lambda x: "EXCLUDE_TASK_%s" % x,
 616                                            exclude_metadata)
 617             break
 618
 619     return True
 620
 621
 622 def sysprep(session):
 623     """Perform various system preperation tasks on the image"""
 624     d = session['dialog']
 625     image = session['image']
 626
 627     # Is the image already shrinked?
 628     if 'shrinked' in session and session['shrinked']:
 629         msg = "It seems you have shrinked the image. Running system " \
 630               "preparation tasks on a shrinked image is dangerous."
 631
 632         if d.yesno("%s\n\nDo you really want to continue?" % msg,
 633                    width=SMALL_WIDTH, defaultno=1):
 634             return
 635
 636     wrapper = textwrap.TextWrapper(width=WIDTH - 5)
 637
 638     syspreps = image.os.list_syspreps()
 639
 640     if len(syspreps) == 0:
 641         d.msgbox("No system preparation task available to run!",
 642                  title="System Preperation", width=SMALL_WIDTH)
 643         return
 644
 645     while 1:
 646         choices = []
 647         index = 0
 648
 649         help_title = "System Preperation Tasks"
 650         sysprep_help = "%s\n%s\n\n" % (help_title, '=' * len(help_title))
 651
 652         for sysprep in syspreps:
 653             name, descr = image.os.sysprep_info(sysprep)
 654             display_name = name.replace('-', ' ').capitalize()
 655             sysprep_help += "%s\n" % display_name
 656             sysprep_help += "%s\n" % ('-' * len(display_name))
 657             sysprep_help += "%s\n\n" % wrapper.fill(" ".join(descr.split()))
 658             enabled = 1 if sysprep.enabled else 0
 659             choices.append((str(index + 1), display_name, enabled))
 660             index += 1
 661
 662         (code, tags) = d.checklist(
 663             "Please choose which system preparation tasks you would like to "
 664             "run on the image. Press <Help> to see details about the system "
 665             "preparation tasks.", title="Run system preparation tasks",
 666             choices=choices, width=70, ok_label="Run", help_button=1)
 667         tags = map(lambda x: x.strip('"'), tags)  # Needed for OpenSUSE
 668
 669         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 670             return False
 671         elif code == d.DIALOG_HELP:
 672             d.scrollbox(sysprep_help, width=WIDTH)
 673         elif code == d.DIALOG_OK:
 674             # Enable selected syspreps and disable the rest
 675             for i in range(len(syspreps)):
 676                 if str(i + 1) in tags:
 677                     image.os.enable_sysprep(syspreps[i])
 678                 else:
 679                     image.os.disable_sysprep(syspreps[i])
 680
 681             if len([s for s in image.os.list_syspreps() if s.enabled]) == 0:
 682                 d.msgbox("No system preperation task is selected!",
 683                          title="System Preperation", width=SMALL_WIDTH)
 684                 continue
 685
 686             infobox = InfoBoxOutput(d, "Image Configuration")
 687             try:
 688                 image.out.add(infobox)
 689                 try:
 690                     # The checksum is invalid. We have mounted the image rw
 691                     if 'checksum' in session:
 692                         del session['checksum']
 693
 694                     # Monitor the metadata changes during syspreps
 695                     with MetadataMonitor(session, image.os.meta):
 696                         try:
 697                             image.os.do_sysprep()
 698                             infobox.finalize()
 699                         except FatalError as e:
 700                             title = "System Preparation"
 701                             d.msgbox("System Preparation failed: %s" % e,
 702                                      title=title, width=SMALL_WIDTH)
 703                 finally:
 704                     image.out.remove(infobox)
 705             finally:
 706                 infobox.cleanup()
 707             break
 708     return True
 709
 710
 711 def shrink(session):
 712     """Shrink the image"""
 713     d = session['dialog']
 714     image = session['image']
 715
 716     shrinked = 'shrinked' in session and session['shrinked']
 717
 718     if shrinked:
 719         d.msgbox("The image is already shrinked!", title="Image Shrinking",
 720                  width=SMALL_WIDTH)
 721         return True
 722
 723     msg = "This operation will shrink the last partition of the image to " \
 724           "reduce the total image size. If the last partition is a swap " \
 725           "partition, then this partition is removed and the partition " \
 726           "before that is shrinked. The removed swap partition will be " \
 727           "recreated during image deployment."
 728
 729     if not d.yesno("%s\n\nDo you want to continue?" % msg, width=WIDTH,
 730                    height=12, title="Image Shrinking"):
 731         with MetadataMonitor(session, image.meta):
 732             infobox = InfoBoxOutput(d, "Image Shrinking", height=4)
 733             image.out.add(infobox)
 734             try:
 735                 image.shrink()
 736                 infobox.finalize()
 737             finally:
 738                 image.out.remove(infobox)
 739
 740         session['shrinked'] = True
 741         update_background_title(session)
 742     else:
 743         return False
 744
 745     return True
 746
 747
 748 def customization_menu(session):
 749     """Show image customization menu"""
 750     d = session['dialog']
 751
 752     choices = [("Sysprep", "Run various image preparation tasks"),
 753                ("Shrink", "Shrink image"),
 754                ("View/Modify", "View/Modify image properties"),
 755                ("Delete", "Delete image properties"),
 756                ("Exclude", "Exclude various deployment tasks from running")]
 757
 758     default_item = 0
 759
 760     actions = {"Sysprep": sysprep,
 761                "Shrink": shrink,
 762                "View/Modify": modify_properties,
 763                "Delete": delete_properties,
 764                "Exclude": exclude_tasks}
 765     while 1:
 766         (code, choice) = d.menu(
 767             text="Choose one of the following or press <Back> to exit.",
 768             width=WIDTH, choices=choices, cancel="Back", height=13,
 769             menu_height=len(choices), default_item=choices[default_item][0],
 770             title="Image Customization Menu")
 771
 772         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 773             break
 774         elif choice in actions:
 775             default_item = [entry[0] for entry in choices].index(choice)
 776             if actions[choice](session):
 777                 default_item = (default_item + 1) % len(choices)
 778
 779
 780 def main_menu(session):
 781     """Show the main menu of the program"""
 782     d = session['dialog']
 783
 784     update_background_title(session)
 785
 786     choices = [("Customize", "Customize image & cloud deployment options"),
 787                ("Register", "Register image to a cloud"),
 788                ("Extract", "Dump image to local file system"),
 789                ("Reset", "Reset everything and start over again"),
 790                ("Help", "Get help for using snf-image-creator")]
 791
 792     default_item = "Customize"
 793
 794     actions = {"Customize": customization_menu, "Register": kamaki_menu,
 795                "Extract": extract_image}
 796     while 1:
 797         (code, choice) = d.menu(
 798             text="Choose one of the following or press <Exit> to exit.",
 799             width=WIDTH, choices=choices, cancel="Exit", height=13,
 800             default_item=default_item, menu_height=len(choices),
 801             title="Image Creator for synnefo (snf-image-creator version %s)" %
 802                   version)
 803
 804         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
 805             if confirm_exit(d):
 806                 break
 807         elif choice == "Reset":
 808             if confirm_reset(d):
 809                 d.infobox("Resetting snf-image-creator. Please wait ...",
 810                           width=SMALL_WIDTH)
 811                 raise Reset
 812         elif choice == "Help":
 813             d.msgbox("For help, check the online documentation:\n\nhttp://www"
 814                      ".synnefo.org/docs/snf-image-creator/latest/",
 815                      width=WIDTH, title="Help")
 816         elif choice in actions:
 817             actions[choice](session)
 818
 819 # vim: set sta sts=4 shiftwidth=4 sw=4 et ai :