126 lines
2.7 KiB
Go
126 lines
2.7 KiB
Go
package common
|
|
|
|
import (
|
|
"bufio"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type File struct {
|
|
name string
|
|
size int
|
|
}
|
|
|
|
func (file File) GetSize() int {
|
|
return file.size
|
|
}
|
|
|
|
func (file File) GetName() string {
|
|
return file.name
|
|
}
|
|
|
|
type Folder struct {
|
|
name string
|
|
subfolders []*Folder
|
|
files []File
|
|
parent *Folder
|
|
}
|
|
|
|
func (folder Folder) GetSize() int {
|
|
sum := 0
|
|
for _, folder := range folder.subfolders {
|
|
sum += folder.GetSize()
|
|
}
|
|
for _, file := range folder.files {
|
|
sum += file.GetSize()
|
|
}
|
|
return sum
|
|
}
|
|
|
|
func (folder Folder) GetSubfoldersUnderSize(size int) []*Folder {
|
|
folders := []*Folder{}
|
|
for _, subfolder := range folder.subfolders {
|
|
if subfolder.GetSize() <= size {
|
|
folders = append(folders, subfolder)
|
|
}
|
|
folders = append(folders, subfolder.GetSubfoldersUnderSize(size)...)
|
|
}
|
|
return folders
|
|
}
|
|
|
|
func (folder Folder) GetSubfoldersOverSize(size int) []*Folder {
|
|
folders := []*Folder{}
|
|
for _, subfolder := range folder.subfolders {
|
|
if subfolder.GetSize() >= size {
|
|
folders = append(folders, subfolder)
|
|
}
|
|
folders = append(folders, subfolder.GetSubfoldersOverSize(size)...)
|
|
}
|
|
return folders
|
|
}
|
|
|
|
func (folder Folder) GetName() string {
|
|
return folder.name
|
|
}
|
|
|
|
func (folder Folder) GetSubfolders() []*Folder {
|
|
return folder.subfolders
|
|
}
|
|
|
|
func (folder Folder) GetFilers() []File {
|
|
return folder.files
|
|
}
|
|
|
|
func (folder *Folder) GetOrCreateFolder(name string) *Folder {
|
|
for _, subfolder := range folder.subfolders {
|
|
if subfolder.GetName() == name {
|
|
return subfolder
|
|
}
|
|
}
|
|
newFolder := Folder{name, []*Folder{}, []File{}, folder}
|
|
folder.subfolders = append(folder.subfolders, &newFolder)
|
|
return &newFolder
|
|
}
|
|
|
|
func (folder *Folder) GetOrCreateFile(name string, size int) File {
|
|
for _, file := range folder.files {
|
|
if file.GetName() == name {
|
|
return file
|
|
}
|
|
}
|
|
newFile := File{name, size}
|
|
folder.files = append(folder.files, newFile)
|
|
return newFile
|
|
}
|
|
|
|
func Parse(scanner bufio.Scanner) Folder {
|
|
rootFolder := Folder{"/", []*Folder{}, []File{}, nil}
|
|
var currentFolder *Folder
|
|
|
|
for scanner.Scan() {
|
|
line := scanner.Text()
|
|
switch line[0:4] {
|
|
case "$ cd":
|
|
if line[5:] == "/" {
|
|
currentFolder = &rootFolder
|
|
} else if line[5:] == ".." {
|
|
currentFolder = currentFolder.parent
|
|
} else {
|
|
currentFolder = currentFolder.GetOrCreateFolder(line[5:])
|
|
}
|
|
|
|
case "$ ls":
|
|
|
|
case "dir ":
|
|
currentFolder.GetOrCreateFolder(line[4:])
|
|
|
|
default:
|
|
splittedLine := strings.Split(line, " ")
|
|
size, _ := strconv.Atoi(splittedLine[0])
|
|
currentFolder.GetOrCreateFile(splittedLine[1], size)
|
|
}
|
|
}
|
|
|
|
return rootFolder
|
|
}
|