|
| 1 | +package lifecycle |
| 2 | + |
| 3 | +import ( |
| 4 | + "encoding/json" |
| 5 | + "fmt" |
| 6 | + "os" |
| 7 | + "time" |
| 8 | + |
| 9 | + "github.com/sirupsen/logrus" |
| 10 | +) |
| 11 | + |
| 12 | +type logs struct { |
| 13 | + file *os.File |
| 14 | + filePath string |
| 15 | +} |
| 16 | + |
| 17 | +const separator = "\n" |
| 18 | +const ender = "\n__________________________________________________________________\n" |
| 19 | + |
| 20 | +// createLogs creates a new log file and returns a logs struct with the file and file path |
| 21 | +func createLogs(fileName string) (*logs, error) { |
| 22 | + filePath := fmt.Sprintf("logs/%s", fileName) |
| 23 | + |
| 24 | + // Create the logs directory if it doesn't exist |
| 25 | + err := os.MkdirAll("logs", 0755) |
| 26 | + if err != nil { |
| 27 | + logrus.Errorf("Error while creating logs directory: %s", err) |
| 28 | + } |
| 29 | + |
| 30 | + // Open the file in append mode |
| 31 | + file, err := os.OpenFile(filePath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644) |
| 32 | + if err != nil { |
| 33 | + return nil, err |
| 34 | + } |
| 35 | + |
| 36 | + return &logs{file: file, filePath: filePath}, nil |
| 37 | +} |
| 38 | + |
| 39 | +// writeHEAD writes the header of the log file with the version rules, dates and necessary information |
| 40 | +// to help analyze the current situation on the charts versions regarding the release process. |
| 41 | +func (l *logs) writeHEAD(versionRules *VersionRules, title string) { |
| 42 | + l.write(title, "INFO") |
| 43 | + currentTime := time.Now() |
| 44 | + l.write(currentTime.Format("2006-01-02 15:04:05"), "INFO") |
| 45 | + l.write(fmt.Sprintf("Branch Version: %.1f", versionRules.branchVersion), "INFO") |
| 46 | + l.write(fmt.Sprintf("minimal version: %d", versionRules.minVersion), "INFO") |
| 47 | + l.write(fmt.Sprintf("max version: %d", versionRules.maxVersion), "INFO") |
| 48 | + l.write(fmt.Sprintf("development branch: %s", versionRules.devBranch), "INFO") |
| 49 | + l.write(fmt.Sprintf("production branch: %s", versionRules.prodBranch), "INFO") |
| 50 | + |
| 51 | + rules := make(map[string]string, len(versionRules.rules)) |
| 52 | + for k, v := range versionRules.rules { |
| 53 | + rules[fmt.Sprintf("%.1f", k)] = fmt.Sprintf("min: %s, max: %s", v.min, v.max) |
| 54 | + } |
| 55 | + |
| 56 | + rulesJSON, err := json.MarshalIndent(rules, "", " ") |
| 57 | + if err != nil { |
| 58 | + logrus.Errorf("JSON marshaling failed: %s", err) |
| 59 | + l.write(fmt.Sprintf("rules: %v\n", versionRules.rules), "INFO") |
| 60 | + } else { |
| 61 | + l.write(fmt.Sprintf("rules: %s\n", rulesJSON), "INFO") |
| 62 | + } |
| 63 | +} |
| 64 | + |
| 65 | +// write writes the data to the log file and prints it to the console with customizations. |
| 66 | +func (l *logs) write(data string, logType string) { |
| 67 | + switch logType { |
| 68 | + case "INFO": |
| 69 | + logrus.Info(data) |
| 70 | + if _, err := l.file.WriteString("INFO=" + data + "\n"); err != nil { |
| 71 | + logrus.Errorf("Error while writing logs: %s", err) |
| 72 | + } |
| 73 | + case "WARN": |
| 74 | + logrus.Warn(data) |
| 75 | + if _, err := l.file.WriteString("WARN=" + data + "\n"); err != nil { |
| 76 | + logrus.Errorf("Error while writing logs: %s", err) |
| 77 | + } |
| 78 | + case "ERROR": |
| 79 | + logrus.Error(data) |
| 80 | + if _, err := l.file.WriteString("ERROR=" + data + "\n"); err != nil { |
| 81 | + logrus.Errorf("Error while writing logs: %s", err) |
| 82 | + } |
| 83 | + case "SEPARATE": |
| 84 | + fmt.Printf(separator) |
| 85 | + if _, err := l.file.WriteString(separator); err != nil { |
| 86 | + logrus.Errorf("Error while writing logs: %s", err) |
| 87 | + } |
| 88 | + case "END": |
| 89 | + fmt.Printf(ender) |
| 90 | + if _, err := l.file.WriteString("\n" + ender + "\n"); err != nil { |
| 91 | + logrus.Errorf("Error while writing logs: %s", err) |
| 92 | + } |
| 93 | + default: |
| 94 | + fmt.Printf(data) |
| 95 | + if _, err := l.file.WriteString(data + "\n"); err != nil { |
| 96 | + logrus.Errorf("Error while writing logs: %s", err) |
| 97 | + } |
| 98 | + } |
| 99 | +} |
| 100 | + |
| 101 | +// writeVersions receives the loaded assets versions map and writes it to the log file |
| 102 | +// in human-readable format |
| 103 | +func (l *logs) writeVersions(assetsVersions map[string][]Asset, logType string) { |
| 104 | + for asset, versions := range assetsVersions { |
| 105 | + l.write("", "SEPARATE") |
| 106 | + l.write(asset, logType) |
| 107 | + for _, version := range versions { |
| 108 | + l.write(version.version, "") |
| 109 | + } |
| 110 | + } |
| 111 | +} |
0 commit comments