~nskaggs/+junk/xenial-test

datasource := simplestreams.NewURLSignedDataSource("bootstrap metadata", baseURL, publicKey, utils.NoVerifySSLHostnames, simplestreams.CUSTOM_CLOUD_DATA, false)

591

592

// Read the image metadata, as we'll want to upload it to the environment.

593

imageConstraint := imagemetadata.NewImageConstraint(simplestreams.LookupParams{})

594

existingMetadata, _, err := imagemetadata.Fetch([]simplestreams.DataSource{datasource}, imageConstraint)

595

if err != nil && !errors.IsNotFound(err) {

596

return nil, errors.Annotate(err, "cannot read image metadata")

597

}

598

599

// Add an image metadata datasource for constraint validation, etc.

600

environs.RegisterUserImageDataSourceFunc("bootstrap metadata", func(environs.Environ) (simplestreams.DataSource, error) {

601

return datasource, nil

602

})

603

logger.Infof("custom image metadata added to search path")

604

return existingMetadata, nil

605

}

606

607

func validateConstraints(env environs.Environ, cons constraints.Value) error {

608

validator, err := env.ConstraintsValidator()

609

if err != nil {

610

return errors.Trace(err)

611

}

612

unsupported, err := validator.Validate(cons)

613

return errors.Annotatef(err, "unsupported constraints: %v", unsupported)

614

}

615

616

// guiArchive returns information on the GUI archive that will be uploaded

617

// to the controller. Possible errors in retrieving the GUI archive information

618

// do not prevent the model to be bootstrapped. If dataSourceBaseURL is

619

// non-empty, remote GUI archive info is retrieved from simplestreams using it

620

// as the base URL. The given logProgress function is used to inform users

621

// about errors or progress in setting up the Juju GUI.

622

func guiArchive(dataSourceBaseURL string, logProgress func(string)) *coretools.GUIArchive {

623

// The environment variable is only used for development purposes.

624

path := os.Getenv("JUJU_GUI")

625

if path != "" {

626

vers, err := guiVersion(path)

627

if err != nil {

628

logProgress(fmt.Sprintf("Cannot use Juju GUI at %q: %s", path, err))

629

return nil

630

}

631

hash, size, err := hashAndSize(path)

632

if err != nil {

633

logProgress(fmt.Sprintf("Cannot use Juju GUI at %q: %s", path, err))

634

return nil

635

}

636

logProgress(fmt.Sprintf("Preparing for Juju GUI %s installation from local archive", vers))

637

return &coretools.GUIArchive{

638

Version: vers,

639

URL: "file://" + filepath.ToSlash(path),

640

SHA256: hash,

641

Size: size,

642

}

643

}

644

// Check if the user requested to bootstrap with no GUI.

645

if dataSourceBaseURL == "" {

646

logProgress("Juju GUI installation has been disabled")

647

return nil

648

}

649

// Fetch GUI archives info from simplestreams.

650

source := gui.NewDataSource(dataSourceBaseURL)

651

allMeta, err := guiFetchMetadata(gui.ReleasedStream, source)

652

if err != nil {

653

logProgress(fmt.Sprintf("Unable to fetch Juju GUI info: %s", err))

654

return nil

655

}

656

if len(allMeta) == 0 {

657

logProgress("No available Juju GUI archives found")

658

return nil

659

}

660

// Metadata info are returned in descending version order.

661

logProgress(fmt.Sprintf("Preparing for Juju GUI %s release installation", allMeta[0].Version))

662

return &coretools.GUIArchive{

663

Version: allMeta[0].Version,

664

URL: allMeta[0].FullPath,

665

SHA256: allMeta[0].SHA256,

666

Size: allMeta[0].Size,

667

}

668

}

669

670

// guiFetchMetadata is defined for testing purposes.

671

var guiFetchMetadata = gui.FetchMetadata

672

673

// guiVersion retrieves the GUI version from the juju-gui-* directory included

674

// in the bz2 archive at the given path.

675

func guiVersion(path string) (version.Number, error) {

676

var number version.Number

677

f, err := os.Open(path)

678

if err != nil {

679

return number, errors.Annotate(err, "cannot open Juju GUI archive")

680

}

681

defer f.Close()

682

prefix := "jujugui-"

683

r := tar.NewReader(bzip2.NewReader(f))

684

for {

685

hdr, err := r.Next()

686

if err == io.EOF {

687

break

688

}

689

if err != nil {

690

return number, errors.New("cannot read Juju GUI archive")

691

}

692

info := hdr.FileInfo()

693

if !info.IsDir() || !strings.HasPrefix(hdr.Name, prefix) {

694

continue

695

}

696

n := info.Name()[len(prefix):]

697

number, err = version.Parse(n)

698

if err != nil {

699

return number, errors.Errorf("cannot parse version %q", n)

700

}

701

return number, nil

702

}

703

return number, errors.New("cannot find Juju GUI version")

704

}

705

706

// hashAndSize calculates and returns the SHA256 hash and the size of the file

707

// located at the given path.

708

func hashAndSize(path string) (hash string, size int64, err error) {

709

f, err := os.Open(path)

710

if err != nil {

711

return "", 0, errors.Mask(err)

712

}

713

defer f.Close()

714

h := sha256.New()

715

size, err = io.Copy(h, f)

716

if err != nil {

717

return "", 0, errors.Mask(err)

718

}

719

return fmt.Sprintf("%x", h.Sum(nil)), size, nil

720

}

Older »