Windowsで文字化けしない zip ファイルを golang で作る

はじめに

MacやLinuxでZIPファイルを作るとWindowsで文字化けする場合があります。
Shift-JISに対応したソフトを使用すれば良いのですが今回は golang でささっと圧縮プログラムを作ってみます。

やってみよう

必用なもの

• golang

仕様

• CLIコマンド
• 入力文字コードはUTF8固定（Mac、Linux共にUTF8がデフォルト文字コード）
• パラメータで圧縮対処フォルダを指定する
• Windowsで文字化けしないようにファイル／フォルダ名をCP932に変換して保存する

実装

まずはUTF8文字列 -> Shift-JIS文字列の変換関数を用意します。

main.go

func Utf82Sjis(str string) (string, error) {
    iostr := strings.NewReader(str)
    rio := transform.NewReader(iostr, japanese.ShiftJIS.NewEncoder())
    ret, err := ioutil.ReadAll(rio)
    if err != nil {
        return "", err
    }
    return string(ret), err
}

次にZIPファイルのヘッダを作成する関数を用意します。
ヘッダ作成時にファイルのタイムスタンプも修正します。

main.go

func FileInfoHeader(fi os.FileInfo) (*zip.FileHeader, error) {
    size := fi.Size()
    fh := &zip.FileHeader{
        Name:               fi.Name(),
        UncompressedSize64: uint64(size),
    }

    local := time.Now().Local()
    _, offset := local.Zone()
    fh.SetModTime(fi.ModTime().Add(time.Duration(offset) * time.Second))
    fh.SetMode(fi.Mode())
    var uintmax = uint32((1 << 32) - 1)
    if fh.UncompressedSize64 > uint64(uintmax) {
        fh.UncompressedSize = uintmax
    } else {
        fh.UncompressedSize = uint32(fh.UncompressedSize64)
    }
    return fh, nil
}

最後に指定フォルダ配下をZIPする関数を用意します。

main.go

func Zip(source, output string) error {
    var f *os.File
    if "-" == output {
        f = os.Stdout
    } else {
        var err error
        f, err = os.Create(output)
        if err != nil {
            return err
        }
    }
    defer f.Close()

    archive := zip.NewWriter(f)
    defer archive.Close()

    info, err := os.Stat(source)
    if err != nil {
        return err
    }

    var baseDir string
    if info.IsDir() {
        baseDir = filepath.Base(source)
    }

    filepath.Walk(source, func(path string, info os.FileInfo, err error) error {
        if err != nil {
            return err
        }

        header, err := FileInfoHeader(info)
        if err != nil {
            return err
        }

        if baseDir != "" {
            //header.Name = filepath.Join(baseDir, strings.TrimPrefix(path, source))
            name, err := Utf82Sjis(filepath.Join(baseDir, strings.TrimPrefix(path, source)))
            if err != nil {
                return err
            }
            header.Name = name
        }

        if info.IsDir() {
            header.Name += "/"
        } else {
            //header.Method = zip.Deflate
            header.Method = zip.Store
        }

        writer, err := archive.CreateHeader(header)
        if err != nil {
            return err
        }

        if info.IsDir() {
            return nil
        }

        file, err := os.Open(path)
        if err != nil {
            return err
        }
        _, err = io.Copy(writer, file)
        file.Close()
        return err
    })

    return err
}

あとは作ったZip関数を実行する main() を作れば完了。

main.go

func main() {
    if 3 > len(os.Args) {
        fmt.Printf("usage:\n\t%s [zip file] [archive directory]\n", os.Args[0])
        os.Exit(1)
    }

    output := os.Args[1]
    source := os.Args[2]

    if err := Zip(source, output); err != nil {
        panic(err)
    }
}

今回のコード一式は以下にあります。
やっぱツール作るなら golang がお気に入りです。
https://github.com/noridas80/zi18np