| 1 |  |  | /* | 
            
                                                                                                            
                            
            
                                    
            
            
                | 2 |  |  |  * Copyright (c) 2023 Clive Walkden <[email protected]> | 
            
                                                                                                            
                            
            
                                    
            
            
                | 3 |  |  |  * | 
            
                                                                                                            
                            
            
                                    
            
            
                | 4 |  |  |  * Permission is hereby granted, free of charge, to any person obtaining a copy | 
            
                                                                                                            
                            
            
                                    
            
            
                | 5 |  |  |  * of this software and associated documentation files (the "Software"), to deal | 
            
                                                                                                            
                            
            
                                    
            
            
                | 6 |  |  |  * in the Software without restriction, including without limitation the rights | 
            
                                                                                                            
                            
            
                                    
            
            
                | 7 |  |  |  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | 
            
                                                                                                            
                            
            
                                    
            
            
                | 8 |  |  |  * copies of the Software, and to permit persons to whom the Software is | 
            
                                                                                                            
                            
            
                                    
            
            
                | 9 |  |  |  * furnished to do so, subject to the following conditions: | 
            
                                                                                                            
                            
            
                                    
            
            
                | 10 |  |  |  * | 
            
                                                                                                            
                            
            
                                    
            
            
                | 11 |  |  |  * The above copyright notice and this permission notice shall be included in all | 
            
                                                                                                            
                            
            
                                    
            
            
                | 12 |  |  |  * copies or substantial portions of the Software. | 
            
                                                                                                            
                            
            
                                    
            
            
                | 13 |  |  |  * | 
            
                                                                                                            
                            
            
                                    
            
            
                | 14 |  |  |  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | 
            
                                                                                                            
                            
            
                                    
            
            
                | 15 |  |  |  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES | 
            
                                                                                                            
                            
            
                                    
            
            
                | 16 |  |  |  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND | 
            
                                                                                                            
                            
            
                                    
            
            
                | 17 |  |  |  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT | 
            
                                                                                                            
                            
            
                                    
            
            
                | 18 |  |  |  * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, | 
            
                                                                                                            
                            
            
                                    
            
            
                | 19 |  |  |  * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING | 
            
                                                                                                            
                            
            
                                    
            
            
                | 20 |  |  |  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR | 
            
                                                                                                            
                            
            
                                    
            
            
                | 21 |  |  |  * OTHER DEALINGS IN THE SOFTWARE. | 
            
                                                                                                            
                            
            
                                    
            
            
                | 22 |  |  |  */ | 
            
                                                                                                            
                            
            
                                    
            
            
                | 23 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 24 |  |  | package reporting | 
            
                                                                                                            
                            
            
                                    
            
            
                | 25 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 26 |  |  | import ( | 
            
                                                                                                            
                            
            
                                    
            
            
                | 27 |  |  | 	"fmt" | 
            
                                                                                                            
                            
            
                                    
            
            
                | 28 |  |  | 	"github.com/fatih/color" | 
            
                                                                                                            
                            
            
                                    
            
            
                | 29 |  |  | 	"github.com/rodaine/table" | 
            
                                                                                                            
                            
            
                                    
            
            
                | 30 |  |  | 	"github.com/spf13/viper" | 
            
                                                                                                            
                            
            
                                    
            
            
                | 31 |  |  | 	"math" | 
            
                                                                                                            
                            
            
                                    
            
            
                | 32 |  |  | ) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 33 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 34 |  |  | type Result struct { | 
            
                                                                                                            
                            
            
                                    
            
            
                | 35 |  |  | 	Name        string | 
            
                                                                                                            
                            
            
                                    
            
            
                | 36 |  |  | 	Kept        int | 
            
                                                                                                            
                            
            
                                    
            
            
                | 37 |  |  | 	KeptSize    string | 
            
                                                                                                            
                            
            
                                    
            
            
                | 38 |  |  | 	Deleted     int | 
            
                                                                                                            
                            
            
                                    
            
            
                | 39 |  |  | 	DeletedSize string | 
            
                                                                                                            
                            
            
                                    
            
            
                | 40 |  |  | } | 
            
                                                                                                            
                            
            
                                    
            
            
                | 41 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 42 |  |  | type Report struct { | 
            
                                                                                                            
                            
            
                                    
            
            
                | 43 |  |  | 	Result []Result | 
            
                                                                                                            
                            
            
                                    
            
            
                | 44 |  |  | 	DryRun bool | 
            
                                                                                                            
                            
            
                                    
            
            
                | 45 |  |  | } | 
            
                                                                                                            
                            
            
                                    
            
            
                | 46 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 47 |  |  | func Output(report Report) { | 
            
                                                                                                            
                            
            
                                    
            
            
                | 48 |  |  | 	if !viper.GetBool("verbose") { | 
            
                                                                                                            
                            
            
                                    
            
            
                | 49 |  |  | 		fmt.Print("\033[H\033[2J") //clear screen | 
            
                                                                                                            
                            
            
                                    
            
            
                | 50 |  |  | 	} | 
            
                                                                                                            
                            
            
                                    
            
            
                | 51 |  |  | 	headerFmt := color.New(color.FgGreen, color.Underline).SprintfFunc() | 
            
                                                                                                            
                            
            
                                    
            
            
                | 52 |  |  | 	columnFmt := color.New(color.FgYellow).SprintfFunc() | 
            
                                                                                                            
                            
            
                                    
            
            
                | 53 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 54 |  |  | 	heading := color.New(color.FgBlue, color.Bold).PrintlnFunc() | 
            
                                                                                                            
                            
            
                                    
            
            
                | 55 |  |  | 	if report.DryRun { | 
            
                                                                                                            
                            
            
                                    
            
            
                | 56 |  |  | 		heading("Potential Results: (DryRun Mode)") | 
            
                                                                                                            
                            
            
                                    
            
            
                | 57 |  |  | 	} else { | 
            
                                                                                                            
                            
            
                                    
            
            
                | 58 |  |  | 		heading("Results:") | 
            
                                                                                                            
                            
            
                                    
            
            
                | 59 |  |  | 	} | 
            
                                                                                                            
                            
            
                                    
            
            
                | 60 |  |  | 	fmt.Println("") | 
            
                                                                                                            
                            
            
                                    
            
            
                | 61 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 62 |  |  | 	tbl := table.New("Name", "Deleted Files", "Deleted Size", "Deleted (%)", "Remaining Files", "Remaining Size") | 
            
                                                                                                            
                            
            
                                    
            
            
                | 63 |  |  | 	tbl.WithHeaderFormatter(headerFmt).WithFirstColumnFormatter(columnFmt).WithPadding(5) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 64 |  |  |  | 
            
                                                                                                            
                            
            
                                    
            
            
                | 65 |  |  | 	for _, element := range report.Result { | 
            
                                                                                                            
                            
            
                                    
            
            
                | 66 |  |  | 		// Skip empty Structs | 
            
                                                                                                            
                            
            
                                    
            
            
                | 67 |  |  | 		if element.Name == "" { | 
            
                                                                                                            
                            
            
                                    
            
            
                | 68 |  |  | 			continue | 
            
                                                                                                            
                            
            
                                    
            
            
                | 69 |  |  | 		} | 
            
                                                                                                            
                            
            
                                    
            
            
                | 70 |  |  | 		tbl.AddRow(element.Name, element.Deleted, element.DeletedSize, deletedPerc(element), element.Kept, element.KeptSize) | 
            
                                                                                                            
                            
            
                                    
            
            
                | 71 |  |  | 	} | 
            
                                                                                                            
                            
            
                                    
            
            
                | 72 |  |  | 	tbl.Print() | 
            
                                                                                                            
                            
            
                                    
            
            
                | 73 |  |  | 	fmt.Println("") | 
            
                                                                                                            
                            
            
                                    
            
            
                | 74 |  |  | 	fmt.Println("") | 
            
                                                                                                            
                            
            
                                    
            
            
                | 75 |  |  | } | 
            
                                                                                                            
                                                                
            
                                    
            
            
                | 76 |  |  |  | 
            
                                                        
            
                                    
            
            
                | 77 |  |  | func deletedPerc(result Result) (delta float64) { | 
            
                                                        
            
                                    
            
            
                | 78 |  |  | 	originalNumber := float64(result.Kept + result.Deleted) | 
            
                                                        
            
                                    
            
            
                | 79 |  |  | 	decrease := originalNumber - float64(result.Kept) | 
            
                                                        
            
                                    
            
            
                | 80 |  |  | 	delta = (decrease / originalNumber) * 100 | 
            
                                                        
            
                                    
            
            
                | 81 |  |  | 	return math.Round(delta*100) / 100 | 
            
                                                        
            
                                    
            
            
                | 82 |  |  | } | 
            
                                                        
            
                                    
            
            
                | 83 |  |  |  |