feat: Log usernames with the IDs

This commit is contained in:
BuyMyMojo 2022-07-21 00:23:43 +10:00
parent 5547316ff3
commit f627e97bb1

View file

@ -15,6 +15,7 @@ var (
CsvInput string CsvInput string
CsvOutput string CsvOutput string
unique_old []string unique_old []string
unique_name []string
unique_ids []string unique_ids []string
) )
@ -28,6 +29,7 @@ func init() {
// struct for getting just user ID out of csv file // struct for getting just user ID out of csv file
type DiscordCSV struct { type DiscordCSV struct {
UserID string UserID string
Username string
} }
func main() { func main() {
@ -79,14 +81,19 @@ func process(CsvPath string) {
if err != nil { if err != nil {
panic(err) panic(err)
} }
// Loop through lines & turn into object
for _, line := range lines {
data := DiscordCSV{
UserID: line[0],
}
unique_old = append(unique_old, data.UserID)
}
// Loop through lines & add to DiscordIDs list
for i, _ := range lines {
data := DiscordCSV{
UserID: lines[i][0],
Username: lines[i][1],
}
if !contains(unique_old, data.UserID) {
unique_old = append(unique_old, data.UserID)
unique_name = append(unique_name, data.Username)
}
}
WriteCSV(CsvPath) WriteCSV(CsvPath)
} else if errors.Is(err, os.ErrNotExist) { } else if errors.Is(err, os.ErrNotExist) {
@ -112,6 +119,7 @@ func process(CsvPath string) {
func WriteCSV(InFile string) { func WriteCSV(InFile string) {
// Create DiscordIDs list // Create DiscordIDs list
var DiscordIDs []string var DiscordIDs []string
var DiscordNames []string
// Read CSV file // Read CSV file
lines, err := ReadCsv(InFile) lines, err := ReadCsv(InFile)
@ -122,20 +130,23 @@ func WriteCSV(InFile string) {
OutFile, err := os.OpenFile(CsvOutput, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0600) OutFile, err := os.OpenFile(CsvOutput, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0600)
// Loop through lines & add to DiscordIDs list // Loop through lines & add to DiscordIDs list
for _, line := range lines { for i, _ := range lines {
data := DiscordCSV{ data := DiscordCSV{
UserID: line[0], UserID: lines[i][0],
Username: lines[i][1],
} }
if !contains(DiscordIDs, data.UserID) { if !contains(DiscordIDs, data.UserID) {
DiscordIDs = append(DiscordIDs, data.UserID) DiscordIDs = append(DiscordIDs, data.UserID)
DiscordNames = append(DiscordNames, data.Username)
} }
} }
// Checks if ID is already in list // Checks if ID is already in list
// if it isnt then it gets writen to list // if it isnt then it gets writen to list
for _, ID := range DiscordIDs { for i, ID := range DiscordIDs {
if !contains(unique_old, ID) { if !contains(unique_old, ID) {
OutFile.WriteString(ID + "\n") OutFile.WriteString(ID + "," + DiscordNames[i] + "\n")
} }
} }
} }
@ -149,6 +160,9 @@ func ReadCsv(filename string) ([][]string, error) {
} }
defer f.Close() defer f.Close()
var x string
f.WriteString(x)
// Read File into a Variable // Read File into a Variable
lines, err := csv.NewReader(f).ReadAll() lines, err := csv.NewReader(f).ReadAll()
if err != nil { if err != nil {