Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Flag for User Defined Dump folder #58

Merged
merged 2 commits into from
Jan 18, 2024
Merged

Add Flag for User Defined Dump folder #58

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
e325a67
Add flag for user defined dump location, create dump folder if not ex…
gregandcin Jan 18, 2024
8216df3
Add short and log flags
gregandcin Jan 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
69 changes: 52 additions & 17 deletions pinecone.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import (
"flag"
"fmt"
"io"
"log"
"net/http"
"os"
"path/filepath"
Expand All @@ -18,11 +19,12 @@ import (

var (
titles TitleList
updateFlag = flag.Bool("update", false, "Update the JSON data from the source URL")
summarizeFlag = flag.Bool("summarize", false, "Print summary statistics for all titles")
titleIDFlag = flag.String("titleid", "", "Filter statistics by Title ID")
fatxplorer = flag.Bool("fatxplorer", false, "Use FatXplorer's X: drive")
helpFlag = flag.Bool("help", false, "Display help information")
updateFlag = false
summarizeFlag = false
titleIDFlag = ""
fatxplorer = false
dumpLocation = "dump"
helpFlag = false
)

type TitleData struct {
Expand Down Expand Up @@ -73,6 +75,9 @@ func loadJSONData(jsonFilePath, owner, repo, path string, v interface{}, updateF

// Download JSON data
jsonData, err := downloadJSONData(fmt.Sprintf("https://api.github.com/repos/%s/%s/contents/%s", owner, repo, path))
if err != nil {
return err
}

// Check if downloaded JSON is different from existing JSON
if _, err := os.Stat(jsonFilePath); err == nil {
Expand Down Expand Up @@ -399,16 +404,31 @@ func promptForDownload(url string) bool {
return strings.ToLower(response) == "yes"
}
func main() {

flag.BoolVar(&updateFlag, "update", false, "Update the JSON data from the source URL")
flag.BoolVar(&updateFlag, "u", false, "Update the JSON data from the source URL")
flag.BoolVar(&summarizeFlag, "summarize", false, "Print summary statistics for all titles")
flag.BoolVar(&summarizeFlag, "s", false, "Print summary statistics for all titles")
flag.StringVar(&titleIDFlag, "titleid", "", "Filter statistics by Title ID")
flag.StringVar(&titleIDFlag, "tID", "", "Filter statistics by Title ID")
flag.BoolVar(&fatxplorer, "fatxplorer", false, "Use FatXplorer's X: drive")
flag.BoolVar(&fatxplorer, "f", false, "Use FatXplorer's X: drive")
flag.StringVar(&dumpLocation, "location", "dump", "Directory to search for TDATA/UDATA directories")
flag.StringVar(&dumpLocation, "l", "dump", "Directory to search for TDATA/UDATA directories")
flag.BoolVar(&helpFlag, "help", false, "Display help information")
flag.BoolVar(&helpFlag, "h", false, "Display help information")

flag.Parse() // Parse command line flags

// Check for help flag
if *helpFlag {
if helpFlag {
fmt.Println("Usage of Pinecone:")
fmt.Println(" -update: Update the JSON data from the source URL. If not set, uses local copies of data.")
fmt.Println(" -summarize: Print summary statistics for all titles. If not set, checks for content in the TDATA folder.")
fmt.Println(" -titleid: Filter statistics by Title ID (-titleID=ABCD1234). If not set, statistics are computed for all titles.")
fmt.Println(" -fatxplorer: Use FATXPlorer's X drive as the root directory. If not set, runs as normal. (Windows Only)")
fmt.Println(" -help: Display this help information.")
fmt.Println(" -u, --update: Update the JSON data from the source URL. If not set, uses local copies of data.")
fmt.Println(" -s, --summarize: Print summary statistics for all titles. If not set, checks for content in the TDATA folder.")
fmt.Println(" -tID, --titleid: Filter statistics by Title ID (-titleID=ABCD1234). If not set, statistics are computed for all titles.")
fmt.Println(" -f, --fatxplorer: Use FATXPlorer's X drive as the root directory. If not set, runs as normal. (Windows Only)")
fmt.Println(" -l --location: Directory where TDATA/UDATA folders are stored. If not set, checks in \"dump\"")
fmt.Println(" -h, --help: Display this help information.")
return
}
jsonFilePath := "data/id_database.json"
Expand Down Expand Up @@ -436,7 +456,7 @@ func main() {
fmt.Println("Download aborted by user.")
return
}
} else if *updateFlag {
} else if updateFlag {
// Handle manual update
err := loadJSONData(jsonFilePath, "Xbox-Preservation-Project", "Pinecone", "data/id_database.json", &titles, true)
if err != nil {
Expand All @@ -451,17 +471,32 @@ func main() {
return
}
}

if dumpLocation != "dump" {
if _, err := os.Stat(dumpLocation); os.IsNotExist(err) {
log.Fatalln("Directory does not exist, exiting...")
}
} else {
if _, err := os.Stat(dumpLocation); os.IsNotExist(err) {
fmt.Println("Default dump folder not found. Creating...")
if mkDirErr := os.Mkdir(dumpLocation, 0755); mkDirErr != nil {
log.Fatalln("Error creating dump folder:", mkDirErr)
}
log.Fatalln("Please place TDATA folder in the \"dump\" folder")
}
}

fmt.Println("Pinecone v0.4.2b")
fmt.Println("Please share output of this program with the Pinecone team if you find anything interesting!")
flag.Parse()

if *titleIDFlag != "" {
if titleIDFlag != "" {
// if the titleID flag is set, print stats for that title
printStats(*titleIDFlag, false)
} else if *summarizeFlag {
printStats(titleIDFlag, false)
} else if summarizeFlag {
// if the summarize flag is set, print stats for all titles
printStats("", true)
} else if *fatxplorer {
} else if fatxplorer {
if runtime.GOOS == "windows" {
if _, err := os.Stat(`X:\`); os.IsNotExist(err) {
fmt.Println(`FatXplorer's X: drive not found`)
Expand All @@ -476,7 +511,7 @@ func main() {
} else {
// If no flag is set, proceed normally
// Check if TDATA folder exists
if _, err := os.Stat("dump/TDATA"); os.IsNotExist(err) {
if _, err := os.Stat(dumpLocation + "/TDATA"); os.IsNotExist(err) {
fmt.Println("TDATA folder not found. Please place TDATA folder in the dump folder.")
return
}
Expand Down