package installer import ( "os" "path/filepath" "sync" "github.com/hashicorp/go-multierror" "github.com/mudler/luet/pkg/api/core/template" "github.com/mudler/luet/pkg/api/core/types" fileHelper "github.com/mudler/luet/pkg/helpers/file" "github.com/mudler/luet/pkg/tree" ) type System struct { Database types.PackageDatabase Target string fileIndex map[string]*types.Package fileIndexPackages map[string]*types.Package sync.Mutex } func (s *System) World() (types.Packages, error) { return s.Database.World(), nil } func (s *System) OSCheck(ctx types.Context) (notFound types.Packages) { s.buildFileIndex() s.Lock() defer s.Unlock() for f, p := range s.fileIndex { targetFile := filepath.Join(s.Target, f) if _, err := os.Lstat(targetFile); err != nil { if _, err := s.Database.FindPackage(p); err == nil { ctx.Debugf("Missing file '%s' from '%s'", targetFile, p.HumanReadableString()) notFound = append(notFound, p) } } } notFound = notFound.Unique() return } func (s *System) ExecuteFinalizers(ctx types.Context, packs []*types.Package) error { var errs error executedFinalizer := map[string]bool{} for _, p := range packs { if fileHelper.Exists(p.Rel(tree.FinalizerFile)) { out, err := template.RenderWithValues([]string{p.Rel(tree.FinalizerFile)}, p.Rel(types.PackageDefinitionFile)) if err != nil { ctx.Warning("Failed rendering finalizer for ", p.HumanReadableString(), err.Error()) errs = multierror.Append(errs, err) continue } if _, exists := executedFinalizer[p.GetFingerPrint()]; !exists { executedFinalizer[p.GetFingerPrint()] = true ctx.Info("Executing finalizer for " + p.HumanReadableString()) finalizer, err := NewLuetFinalizerFromYaml([]byte(out)) if err != nil { ctx.Warning("Failed reading finalizer for ", p.HumanReadableString(), err.Error()) errs = multierror.Append(errs, err) continue } err = finalizer.RunInstall(ctx, s) if err != nil { ctx.Warning("Failed running finalizer for ", p.HumanReadableString(), err.Error()) errs = multierror.Append(errs, err) continue } } } } return errs } func (s *System) buildFileIndex() { // XXX: Replace with cache s.Lock() defer s.Unlock() if s.fileIndex == nil { s.fileIndex = make(map[string]*types.Package) } if s.fileIndexPackages == nil { s.fileIndexPackages = make(map[string]*types.Package) } // Check if cache is empty or if it got modified if len(s.Database.GetPackages()) != len(s.fileIndexPackages) { s.fileIndexPackages = make(map[string]*types.Package) for _, p := range s.Database.World() { files, _ := s.Database.GetPackageFiles(p) for _, f := range files { s.fileIndex[f] = p } s.fileIndexPackages[p.GetPackageName()] = p } } } func (s *System) Clean() { s.Lock() defer s.Unlock() s.fileIndex = nil } func (s *System) FileIndex() map[string]*types.Package { s.buildFileIndex() s.Lock() defer s.Unlock() return s.fileIndex } func (s *System) ExistsPackageFile(file string) (bool, *types.Package, error) { s.buildFileIndex() s.Lock() defer s.Unlock() if p, exists := s.fileIndex[file]; exists { return exists, p, nil } return false, nil, nil }