path: root/doPatch.go

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

package main

import (
	"path/filepath"

	"go.wit.com/lib/fhelp"
	"go.wit.com/lib/protobuf/forgepb"
	"go.wit.com/log"
)

func doPatchInit() {
	if me.forge.Patchsets != nil {
		if me.forge.Patchsets.Len() == 0 {
			// log.Info("IGNORE: patches are empty")
		} else {
			log.Info("IGNORE: patches already initalized len =", me.forge.Patchsets.Len())
		}
	}
	if err := me.forge.LoadPatchsets(); err != nil {
		log.Info("patches failed to open", err)
		if err := me.forge.SavePatchsets(); err != nil {
			log.Warn("savePatchsets() failed", err)
		}
	}
}

func isPatchingSafe() bool {
	if me.forge.Config.Mode == forgepb.ForgeMode_NORMAL {
		return true
	}
	log.Info("This patch command is not safe to run now")
	log.Info("you must reset the state of your git repositories. Run:")
	log.Info("")
	log.Info("forge normal")
	log.Info("")
	return false
}

// submit's current working patches
func doPatchSubmit() error {
	pset, err := me.forge.MakeDevelPatchSet("testing")
	if err != nil {
		return err
	}
	if pset.Patches == nil {
		log.Info("pset.Patches == nil")
		return err
	}
	if pset.Patches.Len() == 0 {
		log.Info("did not find any patches")
		return nil
	}
	pset.PrintTable()
	_, _, err = pset.HttpPost(myServer(), "new")
	return err
}

func doPatch() error {
	if argv.Patch.Repos != nil {
		showWorkRepos()
		return nil
	}

	if argv.Patch.Submit != nil {
		return doPatchSubmit()
	}

	if !isPatchingSafe() {
		return log.Errorf("not safe to work on patches")
	}

	if argv.Patch.Get != nil {
		psets := forgepb.NewSets()
		newpb, _, _ := psets.HttpPostVerbose(myServer(), "get")
		newpb.PrintTable()
		me.forge.Patchsets = newpb
		me.forge.SavePatchsets()
		return nil
	}

	if argv.Patch.Check != nil {
		/*
			old := findExpired()
			// old.PrintTable()
			for p := range old.IterAll() {
				log.Info("patch", p.Filename, p.Namespace)
			}
			newpb, err := old.HttpPostVerbose(myServer(), "check")
			if err != nil {
				return err
			}
			newpb.PrintTable()
		*/
		log.Info("do something here to find patches merged to devel")
		doMergeReport()
		return nil
	}

	if argv.Patch.List != nil {
		return doPatchList()
	}

	// list patches by default
	return doPatchList()
}

func doPatchList() error {
	// if nothing selected, list the patches
	var changed bool
	newpatches := new(forgepb.Set)
	newpatches.Patches = forgepb.NewPatches()
	for pset := range me.forge.Patchsets.IterAll() {
		pset.PrintTable()
		if pset.Patches.Len() == 0 {
			log.Info("pset is empty")
			me.forge.Patchsets.Delete(pset)
			changed = true
			continue
		}
		for patch := range pset.Patches.IterAll() {
			changed = true
			if err := isPatchIdApplied(patch); err == nil {
				log.Infof("%s patchId already applied %s\n", patch.PatchId, patch.Filename)
				pset.Patches.Delete(patch)
				changed = true
				continue
			}

			if patch.NewHash == "" || patch.NewHash == "na" {
				if newpatches.Patches.AppendByPatchId(patch) {
					log.Infof("%s patchId added here\n", patch.PatchId)
				} else {
					log.Infof("%s patchId already here\n", patch.PatchId)
					pset.Patches.Delete(patch)
					changed = true
				}
			} else {
				if err := setNewCommitHash(patch); err != nil {
					log.Infof("%s bad check on patch failure %v\n", patch.Filename, err)
					return err
				}
				log.Infof("%s %s newhash set already here\n", patch.PatchId, patch.NewHash)
			}
		}
	}
	log.Info("NEW PATCHES TABLE")
	newpatches.PrintTable()
	for patch := range newpatches.Patches.IterAll() {
		if err := setNewCommitHash(patch); err == nil {
			log.Info("newhash set already here", patch.PatchId, patch.NewHash)
			changed = true
			continue
		}
		log.Infof("%s is new\n", patch.Filename)
		repo := me.forge.FindByGoPath(patch.Namespace)
		if repo == nil {
			log.Info("\tCould not find namespace:", patch.Namespace)
			continue
		}
		if fhelp.QuestionUser("apply this patch?") {
			newhash, err := applyAndTrackPatch(repo, patch)
			log.Info("apply results:", newhash, err)
		}
	}
	if changed {
		if err := me.forge.SavePatchsets(); err != nil {
			log.Warn("savePatchsets() failed", err)
		}
	}
	return nil
	/*
		if newpatches.Len() != 0 {
			for patch := range newpatches.IterAll() {
				log.Info("new patch:", patch.CommitHash, patch.NewHash, patch.Filename)
				repo := me.forge.FindByGoPath(patch.Namespace)
				if repo == nil {
					log.Info("\tCould not find namespace:", patch.Namespace)
					continue
				}
			}
			return log.Errorf("patches need to be applied")
		}

		// return doPatchList()
		applied := findApplied()
		if applied == nil || applied.Len() == 0 {
			log.Info("no patches have been appled to the devel branch yet")
			return nil
		}
		// for patch := range applied.IterAll() {
		// 	log.Info("SEND APPLIED: newhash:", patch.NewHash, "commithash:", patch.CommitHash, "patch", patch.Namespace)
		// }
		newpb, _, err := applied.HttpPostVerbose(myServer(), "applied")
		if err != nil {
			return err
		}
		newpb.PrintTable()
		return nil
	*/
}

// Shows repos that are:
//   - git dirty repos
//   - repos with 'user' branch patches not in 'devel' branch
//   - repos with awaiting master branch verions
//
// return true if any are found
func showWorkRepos() bool {
	// always run dirty first
	me.forge.CheckDirtyQuiet()

	// if no option is given to patch, list out the
	// repos that have patches ready in them
	found := findReposWithPatches()
	if found.Len() == 0 {
		log.Info("you currently have no repos with patches")
		return false
	} else {
		me.forge.PrintHumanTable(found)
	}
	return true
}

// returns bad if patches can not be applied
// logic is not great here but it was a first pass
func showPatchset(pset *forgepb.Set) bool {
	// don't even bother to continue if we already know it's broken
	if pset.State == "BROKEN" {
		log.Printf("Patchset Name: %-24s Author: %s <%s> IS BAD\n", pset.Name, pset.GetGitAuthorName(), pset.GetGitAuthorEmail())
		return false
	} else {
		log.Printf("Patchset Name: %-24s Author: %s <%s> IS GOOD\n", pset.Name, pset.GetGitAuthorName(), pset.GetGitAuthorEmail())
	}

	var count int
	var bad int
	all := pset.Patches.SortByFilename()
	for all.Scan() {
		p := all.Next()
		if IsValidPatch(p) {
			// ok
		} else {
			pset.State = "BROKEN"
			bad += 1
		}
		count += 1
	}
	log.Info("pset has", count, "total patches, ", bad, "bad patches")
	if bad == 0 {
		return true
	}
	return false
}

func IsValidPatch(p *forgepb.Patch) bool {
	basepath, filename := filepath.Split(p.Filename)
	repo := me.forge.FindByGoPath(basepath)
	if argv.Verbose {
		log.Info("start:", p.StartHash, "end:", p.CommitHash, "file:", basepath, filename, "devel version", repo.GetDevelVersion())
	}
	if repo == nil {
		log.Info("can not apply patch! repo not found", basepath, filename)
		return false
	}
	if repo.ActualDevelHash() != p.StartHash {
		log.Info("can not apply patch! devel hash mismatch", basepath, filename)
		return false
	}
	if repo.ActualDevelHash() == p.StartHash {
		log.Info("local devel hash:", repo.ActualDevelHash(), "matches patch hash", p.StartHash, "and can be applied")
	}
	log.Info("start:", p.StartHash, "end:", p.CommitHash, "file:", basepath, filename, "devel version", repo.GetDevelVersion())
	for _, line := range p.Files {
		log.Info("\t", line)
	}
	return true
}