How to modify apk in golang

type DexFileHeader struct {
    magic       [8]byte
    checksum    uint32
    signature   [20]byte
    fileSize    uint32
    headerSize  uint32
    endianTag   uint32
    ...
}

apkFile, err := zip.OpenReader(apkPath)
if err != nil {
    panic(err)
}
defer apkFile.Close()

var dexFile *zip.File
for _, f := range apkFile.File {
    if f.Name == "classes.dex" {
        dexFile = f
        break
    }
}
if dexFile == nil {
    panic("no classes.dex found")
}

dexReader, err := dexFile.Open()
if err != nil {
    panic(err)
}
defer dexReader.Close()

tmpDexPath := filepath.Join(tmpDir, "classes.dex")
tmpJarPath := filepath.Join(tmpDir, "classes.jar")

tmpDexFile, err := os.Create(tmpDexPath)
if err != nil {
    panic(err)
}
defer tmpDexFile.Close()

io.Copy(tmpDexFile, dexReader)

cmd := exec.Command("d2j-dex2jar", tmpDexPath, "-f", "-o", tmpJarPath)
if err := cmd.Run(); err != nil {
    panic(err)
}

jarFile, err := jar.Open(tmpJarPath)
if err != nil {
    panic(err)
}
defer jarFile.Close()

for _, classFile := range jarFile.Files() {
    if !strings.HasSuffix(classFile.Name, ".class") {
        continue
    }

    className := strings.TrimSuffix(classFile.Name, ".class")
    className = strings.ReplaceAll(className, "/", ".")

    classReader, err := classFile.Open()
    if err != nil {
        panic(err)
    }
    defer classReader.Close()

    classBytes, err := ioutil.ReadAll(classReader)
    if err != nil {
        panic(err)
    }

    // 修改源代码
    modifiedClassBytes := modifyClassBytes(classBytes)

    tmpClassPath := filepath.Join(tmpDir, className+".class")
    tmpClassFile, err := os.Create(tmpClassPath)
    if err != nil {
        panic(err)
    }
    defer tmpClassFile.Close()

    _, err = tmpClassFile.Write(modifiedClassBytes)
    if err != nil {
        panic(err)
    }
}

unit, err := parser.ParseFile(token.NewFileSet(), "", modifiedSource, parser.ParseComments)
if err != nil {
    panic(err)
}

// 修改AST

var buf bytes.Buffer
printer.Fprint(&buf, token.NewFileSet(), unit)

return buf.Bytes()

cmd = exec.Command("d2j-jar2dex", tmpJarPath, "-o", tmpDexPath)
if err := cmd.Run(); err != nil {
    panic(err)
}

outDex, err := os.Open(tmpDexPath)
if err != nil {
    panic(err)
}
defer outDex.Close()

outDexInfo, err := os.Stat(tmpDexPath)
if err != nil {
    panic(err)
}

outDexHeader := &zip.FileHeader{
    Name:   "classes.dex",
    Method: zip.Store,
}

outDexHeader.SetModTime(outDexInfo.ModTime())

outDexWriter, err := apkWriter.CreateHeader(outDexHeader)
if err != nil {
    panic(err)
}

if _, err := io.Copy(outDexWriter, outDex); err != nil {
    panic(err)
}