// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
// Use of this source code is governed by the GPL 3.0

package main

import (
	"fmt"

	"go.wit.com/lib/protobuf/zoopb"
	"go.wit.com/log"
)

func doUpgrade() error {
	if !argv.DryRun {
		checkSuperuser()

		aptUpdate()
	}

	me.machine, _ = zoopb.InitMachine()

	var installed []string

	fmt.Println("Installed Packages:")
	loop := me.machine.Wit.SortByName()
	for loop.Scan() {
		p := loop.Next()
		if p.Name == "" {
			log.Infof("odd /var/lib/apt/ list parse error. p.Name was blank (should be the package name) p=%v\n", p)
			continue
		}
		if !me.machine.IsInstalled(p.Name) {
			continue
		}

		found := me.machine.FindInstalledByName(p.Name)

		if found.Version == p.Version {
			log.Info("name version already installed", p.Name, found.Version)
			if !argv.Force {
				continue
			}
		} else {
			log.Info("name ver diff", p.Name, found.Version, p.Version)
		}

		if argv.DryRun {
			log.Info("should install package", p.Name)
			continue
		}

		installed = append(installed, p.Name)
	}
	if argv.Force {
		// force remove the packages. can be used if binaries change but versions didn't
		for _, pkgname := range installed {
			aptRemove(pkgname)
		}
	}
	for _, pkgname := range installed {
		aptInstall(pkgname)
	}
	okExit("installed")
	return nil
}