mirror of
https://github.com/packwiz/packwiz.git
synced 2025-04-19 13:06:30 +02:00
319 lines
7.2 KiB
Go
319 lines
7.2 KiB
Go
package curseforge
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/comp500/packwiz/core"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
type importPackFile interface {
|
|
Name() string
|
|
Open() (io.ReadCloser, error)
|
|
}
|
|
|
|
type importPackMetadata interface {
|
|
Name() string
|
|
Versions() map[string]string
|
|
Mods() []struct {
|
|
ModID int
|
|
FileID int
|
|
}
|
|
GetFiles() ([]importPackFile, error)
|
|
}
|
|
|
|
// importCmd represents the import command
|
|
var importCmd = &cobra.Command{
|
|
Use: "import [modpack]",
|
|
Short: "Import an installed curseforge modpack, from a download URL or a downloaded pack zip, or an installed metadata json file",
|
|
Args: cobra.ExactArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
var packImport importPackMetadata
|
|
if strings.HasPrefix(args[0], "http") {
|
|
fmt.Println("it do be a http doe")
|
|
os.Exit(0)
|
|
} else {
|
|
// Attempt to read from file
|
|
f, err := os.Open(args[0])
|
|
if err != nil {
|
|
fmt.Printf("Error opening file: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
defer f.Close()
|
|
|
|
buf := bufio.NewReader(f)
|
|
header, err := buf.Peek(2)
|
|
if err != nil {
|
|
fmt.Printf("Error reading file: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Check if file is a zip
|
|
if string(header) == "PK" {
|
|
fmt.Println("it do be a zip doe")
|
|
os.Exit(0)
|
|
} else {
|
|
// Read the whole file (as we are going to parse it multiple times)
|
|
fileData, err := ioutil.ReadAll(buf)
|
|
if err != nil {
|
|
fmt.Printf("Error reading file: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Determine what format the file is
|
|
var jsonFile map[string]interface{}
|
|
err = json.Unmarshal(fileData, &jsonFile)
|
|
if err != nil {
|
|
fmt.Printf("Error parsing JSON: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
isManifest := false
|
|
if v, ok := jsonFile["manifestType"]; ok {
|
|
isManifest = v.(string) == "minecraftModpack"
|
|
}
|
|
if isManifest {
|
|
fmt.Println("it do be a manifest doe")
|
|
os.Exit(0)
|
|
} else {
|
|
// Replace FileNameOnDisk with fileNameOnDisk
|
|
fileData = bytes.ReplaceAll(fileData, []byte("FileNameOnDisk"), []byte("fileNameOnDisk"))
|
|
packMeta := twitchInstalledPackMeta{}
|
|
err = json.Unmarshal(fileData, &packMeta)
|
|
if err != nil {
|
|
fmt.Printf("Error parsing JSON: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
packImport = packMeta
|
|
}
|
|
}
|
|
}
|
|
|
|
pack, err := core.LoadPack()
|
|
if err != nil {
|
|
fmt.Println("Failed to load existing pack, creating a new one...")
|
|
|
|
// Create a new modpack
|
|
indexFilePath := viper.GetString("init.index-file")
|
|
_, err = os.Stat(indexFilePath)
|
|
if os.IsNotExist(err) {
|
|
// Create file
|
|
err = ioutil.WriteFile(indexFilePath, []byte{}, 0644)
|
|
if err != nil {
|
|
fmt.Printf("Error creating index file: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
fmt.Println(indexFilePath + " created!")
|
|
} else if err != nil {
|
|
fmt.Printf("Error checking index file: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
pack = core.Pack{
|
|
Name: packImport.Name(),
|
|
Index: struct {
|
|
File string `toml:"file"`
|
|
HashFormat string `toml:"hash-format"`
|
|
Hash string `toml:"hash"`
|
|
}{
|
|
File: indexFilePath,
|
|
},
|
|
Versions: packImport.Versions(),
|
|
}
|
|
}
|
|
index, err := pack.LoadIndex()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
modsList := packImport.Mods()
|
|
modIDs := make([]int, len(modsList))
|
|
for i, v := range modsList {
|
|
modIDs[i] = v.ModID
|
|
}
|
|
|
|
fmt.Println("Querying Curse API for mod info...")
|
|
|
|
modInfos, err := getModInfoMultiple(modIDs)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
modInfosMap := make(map[int]modInfo)
|
|
for _, v := range modInfos {
|
|
modInfosMap[v.ID] = v
|
|
}
|
|
|
|
// TODO: multithreading????
|
|
successes := 0
|
|
for _, v := range modsList {
|
|
modInfoValue, ok := modInfosMap[v.ModID]
|
|
if !ok {
|
|
fmt.Printf("Failed to obtain mod information for ID %d\n", v.ModID)
|
|
continue
|
|
}
|
|
|
|
found := false
|
|
var fileInfo modFileInfo
|
|
for _, fileInfo = range modInfoValue.LatestFiles {
|
|
if fileInfo.ID == v.FileID {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
fileInfo, err = getFileInfo(v.ModID, v.FileID)
|
|
if !ok {
|
|
fmt.Printf("Failed to obtain file information for Mod / File %d / %d: %s\n", v.ModID, v.FileID, err)
|
|
continue
|
|
}
|
|
}
|
|
|
|
err = createModFile(modInfoValue, fileInfo, &index)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
fmt.Printf("Imported mod \"%s\" successfully!\n", modInfoValue.Name)
|
|
successes++
|
|
}
|
|
|
|
fmt.Printf("Successfully imported %d/%d mods!\n", successes, len(modsList))
|
|
|
|
// TODO: import existing files (config etc.)
|
|
|
|
err = index.Write()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
err = pack.UpdateIndexHash()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
err = pack.Write()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
curseforgeCmd.AddCommand(importCmd)
|
|
}
|
|
|
|
type diskFile struct {
|
|
NameInternal string
|
|
Base string
|
|
}
|
|
|
|
func (f diskFile) Name() string {
|
|
return f.NameInternal
|
|
}
|
|
|
|
func (f diskFile) Open() (io.ReadCloser, error) {
|
|
return os.Open(filepath.Join(f.Base, f.NameInternal))
|
|
}
|
|
|
|
func diskFilesFromPath(base string) ([]importPackFile, error) {
|
|
list := make([]importPackFile, 0)
|
|
err := filepath.Walk(base, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
list = append(list, diskFile{base, path})
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return list, nil
|
|
}
|
|
|
|
type twitchInstalledPackMeta struct {
|
|
NameInternal string `json:"name"`
|
|
Path string `json:"installPath"`
|
|
// TODO: javaArgsOverride?
|
|
// TODO: allocatedMemory?
|
|
MCVersion string `json:"gameVersion"`
|
|
Modloader struct {
|
|
Name string `json:"name"`
|
|
MavenVersionString string `json:"mavenVersionString"`
|
|
} `json:"baseModLoader"`
|
|
ModpackOverrides []string `json:"modpackOverrides"`
|
|
ModsInternal []struct {
|
|
ID int `json:"addonID"`
|
|
File struct {
|
|
// I've given up on using this cached data, just going to re-request it
|
|
ID int `json:"id"`
|
|
} `json:"installedFile"`
|
|
} `json:"installedAddons"`
|
|
// Used to determine if modpackOverrides should be used or not
|
|
IsUnlocked bool `json:"isUnlocked"`
|
|
}
|
|
|
|
func (m twitchInstalledPackMeta) Name() string {
|
|
return m.NameInternal
|
|
}
|
|
|
|
func (m twitchInstalledPackMeta) Versions() map[string]string {
|
|
vers := make(map[string]string)
|
|
vers["minecraft"] = m.MCVersion
|
|
if strings.HasPrefix(m.Modloader.Name, "forge") {
|
|
if len(m.Modloader.MavenVersionString) > 0 {
|
|
vers["forge"] = strings.TrimPrefix(m.Modloader.MavenVersionString, "net.minecraftforge:forge:")
|
|
} else {
|
|
vers["forge"] = m.MCVersion + "-" + strings.TrimPrefix(m.Modloader.Name, "forge-")
|
|
}
|
|
}
|
|
return vers
|
|
}
|
|
|
|
func (m twitchInstalledPackMeta) Mods() []struct {
|
|
ModID int
|
|
FileID int
|
|
} {
|
|
list := make([]struct {
|
|
ModID int
|
|
FileID int
|
|
}, len(m.ModsInternal))
|
|
for i, v := range m.ModsInternal {
|
|
list[i] = struct {
|
|
ModID int
|
|
FileID int
|
|
}{
|
|
ModID: v.ID,
|
|
FileID: v.File.ID,
|
|
}
|
|
}
|
|
return list
|
|
}
|
|
|
|
func (m twitchInstalledPackMeta) GetFiles() ([]importPackFile, error) {
|
|
if m.IsUnlocked {
|
|
return diskFilesFromPath(m.Path)
|
|
}
|
|
list := make([]importPackFile, len(m.ModpackOverrides))
|
|
for i, v := range m.ModpackOverrides {
|
|
list[i] = diskFile{
|
|
Base: m.Path,
|
|
NameInternal: v,
|
|
}
|
|
}
|
|
return list, nil
|
|
}
|