clr-installer/tui/install.go

// Copyright © 2020 Intel Corporation
//
// SPDX-License-Identifier: GPL-3.0-only

package tui

import (
	"strings"
	"time"

	"github.com/VladimirMarkelov/clui"

	"github.com/clearlinux/clr-installer/controller"
	"github.com/clearlinux/clr-installer/network"
	"github.com/clearlinux/clr-installer/progress"
	"github.com/clearlinux/clr-installer/swupd"
)

// InstallPage is the Page implementation for installation progress page, it also implements
// the progress.Client interface
type InstallPage struct {
	BasePage
	rebootBtn *SimpleButton
	exitBtn   *SimpleButton
	prgBar    *clui.ProgressBar
	prgLabel  *clui.Label
	prgMax    int
}

var (
	loopWaitDuration = 2 * time.Second
)

// Success is part of the progress.Client implementation and represents the
// successful progress completion of a task by setting
// the progress bar to "full"
func (page *InstallPage) Success() {
	page.prgBar.SetValue(page.prgMax)
	clui.RefreshScreen()
}

// Failure is part of the progress.Client implementation and represents the
// unsuccessful progress completion of a task by setting
// the progress bar to "fail"
func (page *InstallPage) Failure() {
	page.prgBar.SetValue(0)
	for i := 1; i <= 5; i++ {
		page.prgBar.SetStyle("AltProgress")
		clui.RefreshScreen()
		time.Sleep(100 * time.Millisecond)
		page.prgBar.SetStyle("")
		clui.RefreshScreen()
		time.Sleep(100 * time.Millisecond)
	}
}

// Step is part of the progress.Client implementation and moves the progress bar one step
// case it becomes full it starts again
func (page *InstallPage) Step() {
	if page.prgBar.Value() == page.prgMax {
		page.prgBar.SetValue(0)
	} else {
		page.prgBar.Step()
	}
	clui.RefreshScreen()
}

// Desc is part of the progress.Client implementation and sets the progress bar label
func (page *InstallPage) Desc(desc string) {
	page.prgBar.SetValue(0)

	// The target prefix is used by the massinstaller to separate target, offline, and ISO
	// content installs. It is unnecessary for the TUI.
	desc = strings.TrimPrefix(desc, swupd.TargetPrefix)

	page.prgLabel.SetTitle(desc)
	clui.RefreshScreen()
}

// Partial is part of the progress.Client implementation and adjusts the progress bar to the
// current completion percentage
func (page *InstallPage) Partial(total int, step int) {
	perc := float32(step) / float32(total)
	value := int(float32(page.prgMax) * perc)
	page.prgBar.SetValue(int(value))
	clui.RefreshScreen()
}

// LoopWaitDuration is part of the progress.Client implementation and returns the time duration
// each step should wait until calling Step again
func (page *InstallPage) LoopWaitDuration() time.Duration {
	return loopWaitDuration
}

// Activate is called when the page is "shown"
func (page *InstallPage) Activate() {
	go func() {
		progress.Set(page)

		err := controller.Install(page.tui.rootDir, page.getModel(), page.tui.options)
		if err != nil {
			page.Panic(err)
			return // In a panic state, do not continue
		}

		go func() {
			_ = network.DownloadInstallerMessage("Post-Installation",
				network.PostInstallConf)
		}()
		page.rebootBtn.SetEnabled(true)
		page.exitBtn.SetEnabled(true)
		clui.ActivateControl(page.GetWindow(), page.rebootBtn)
		clui.RefreshScreen()

		page.tui.installReboot = true
	}()
}

func newInstallPage(tui *Tui) (Page, error) {
	page := &InstallPage{}
	page.setup(tui, TuiPageInstall, NoButtons, TuiPageMenu)

	lbl := clui.CreateLabel(page.content, 2, 2, "Installing Clear Linux* OS", Fixed)
	lbl.SetPaddings(0, 2)

	progressFrame := clui.CreateFrame(page.content, AutoSize, 3, BorderNone, clui.Fixed)
	progressFrame.SetPack(clui.Vertical)

	page.prgBar = clui.CreateProgressBar(progressFrame, AutoSize, AutoSize, clui.Fixed)

	page.prgMax, _ = page.prgBar.Size()
	page.prgBar.SetLimits(0, page.prgMax)

	page.prgLabel = clui.CreateLabel(progressFrame, 1, 1, "Installing", Fixed)
	page.prgLabel.SetPaddings(0, 3)

	page.rebootBtn = CreateSimpleButton(page.cFrame, AutoSize, AutoSize, "Reboot", Fixed)
	page.rebootBtn.OnClick(func(ev clui.Event) {
		go clui.Stop()
	})
	page.rebootBtn.SetEnabled(false)

	page.exitBtn = CreateSimpleButton(page.cFrame, AutoSize, AutoSize, "Exit", Fixed)
	page.exitBtn.OnClick(func(ev clui.Event) {
		page.tui.installReboot = false
		go clui.Stop()
	})
	page.exitBtn.SetEnabled(false)

	return page, nil
}