129 lines
3.0 KiB
Go
129 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"encoding/json"
|
|
"fmt"
|
|
"ldiff-converter/pb"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
|
|
reader := bufio.NewReader(os.Stdin)
|
|
fmt.Print("Enter zip ldiff path: ")
|
|
ldiff, _ := reader.ReadString('\n')
|
|
ldiff = strings.TrimSpace(ldiff)
|
|
if ldiff == "" {
|
|
fmt.Fprintln(os.Stderr, "no ldiff file provided")
|
|
os.Exit(1)
|
|
}
|
|
|
|
fmt.Print("Enter zip hdiff output: ")
|
|
hdiff, _ := reader.ReadString('\n')
|
|
hdiff = strings.TrimSpace(hdiff)
|
|
if hdiff == "" {
|
|
fmt.Fprintln(os.Stderr, "no hdiff output provided")
|
|
os.Exit(1)
|
|
}
|
|
|
|
tmpFolderPath := filepath.Join(".", "temp")
|
|
if err := os.MkdirAll(tmpFolderPath, 0755); err != nil {
|
|
fmt.Fprintln(os.Stderr, "error creating temp dir:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err := Unzip(ldiff, tmpFolderPath); err != nil {
|
|
fmt.Fprintln(os.Stderr, "error:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
ldiffPath := filepath.Join(tmpFolderPath, "ldiff")
|
|
manifestPath := filepath.Join(tmpFolderPath, "manifest")
|
|
hdiffFolderPath := filepath.Join(".", "hdiff")
|
|
|
|
manifestProto, err := LoadManifestProto(manifestPath)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "error loading manifest proto:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
ldiffEntries, err := os.ReadDir(ldiffPath)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "error reading ldiff dir:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
for _, ldiffEntry := range ldiffEntries {
|
|
assetName := ldiffEntry.Name()
|
|
var matchingAssets []struct {
|
|
AssetName string
|
|
AssetSize int64
|
|
Asset *pb.AssetManifest
|
|
}
|
|
|
|
for _, assetGroup := range manifestProto.Assets {
|
|
if data := assetGroup.AssetData; data != nil {
|
|
for _, asset := range data.Assets {
|
|
if asset.ChunkFileName == assetName {
|
|
matchingAssets = append(matchingAssets, struct {
|
|
AssetName string
|
|
AssetSize int64
|
|
Asset *pb.AssetManifest
|
|
}{assetGroup.AssetName, assetGroup.AssetSize, asset})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, ma := range matchingAssets {
|
|
err := LDiffFile(ma.Asset, ma.AssetName, ma.AssetSize, ldiffPath, hdiffFolderPath)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
diffMapNames := make([]string, len(ldiffEntries))
|
|
for i, e := range ldiffEntries {
|
|
diffMapNames[i] = e.Name()
|
|
}
|
|
|
|
diffMapList, err := MakeDiffMap(manifestProto, diffMapNames)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "error making diff map:", err)
|
|
os.Exit(1)
|
|
}
|
|
diffMapJson, err := json.Marshal(diffMapList)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "error marshal diff map:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
diffMapJsonPath := filepath.Join(hdiffFolderPath, "hdifffiles.json")
|
|
if err := os.WriteFile(diffMapJsonPath, diffMapJson, 0644); err != nil {
|
|
fmt.Fprintln(os.Stderr, "error writing diff map:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err := os.RemoveAll(tmpFolderPath); err != nil {
|
|
fmt.Fprintln(os.Stderr, "error removing temp dir:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err := Zip(hdiffFolderPath, hdiff); err != nil {
|
|
fmt.Fprintln(os.Stderr, "error zip hdiff:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err := os.RemoveAll(hdiffFolderPath); err != nil {
|
|
fmt.Fprintln(os.Stderr, "error removing temp dir:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
fmt.Println("done!")
|
|
|
|
}
|