mirror of https://github.com/k3s-io/k3s
125 lines
2.9 KiB
Go
125 lines
2.9 KiB
Go
// Copyright 2013 ChaiShushan <chaishushan{AT}gmail.com>. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package mo
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
type moHeader struct {
|
|
MagicNumber uint32
|
|
MajorVersion uint16
|
|
MinorVersion uint16
|
|
MsgIdCount uint32
|
|
MsgIdOffset uint32
|
|
MsgStrOffset uint32
|
|
HashSize uint32
|
|
HashOffset uint32
|
|
}
|
|
|
|
type moStrPos struct {
|
|
Size uint32 // must keep fields order
|
|
Addr uint32
|
|
}
|
|
|
|
func encodeFile(f *File) []byte {
|
|
hdr := &moHeader{
|
|
MagicNumber: MoMagicLittleEndian,
|
|
}
|
|
data := encodeData(hdr, f)
|
|
data = append(encodeHeader(hdr), data...)
|
|
return data
|
|
}
|
|
|
|
// encode data and init moHeader
|
|
func encodeData(hdr *moHeader, f *File) []byte {
|
|
msgList := []Message{f.MimeHeader.toMessage()}
|
|
for _, v := range f.Messages {
|
|
if len(v.MsgId) == 0 {
|
|
continue
|
|
}
|
|
if len(v.MsgStr) == 0 && len(v.MsgStrPlural) == 0 {
|
|
continue
|
|
}
|
|
msgList = append(msgList, v)
|
|
}
|
|
sort.Sort(byMessages(msgList))
|
|
|
|
var buf bytes.Buffer
|
|
var msgIdPosList = make([]moStrPos, len(msgList))
|
|
var msgStrPosList = make([]moStrPos, len(msgList))
|
|
for i, v := range msgList {
|
|
// write msgid
|
|
msgId := encodeMsgId(v)
|
|
msgIdPosList[i].Addr = uint32(buf.Len() + MoHeaderSize)
|
|
msgIdPosList[i].Size = uint32(len(msgId))
|
|
buf.WriteString(msgId)
|
|
// write msgstr
|
|
msgStr := encodeMsgStr(v)
|
|
msgStrPosList[i].Addr = uint32(buf.Len() + MoHeaderSize)
|
|
msgStrPosList[i].Size = uint32(len(msgStr))
|
|
buf.WriteString(msgStr)
|
|
}
|
|
|
|
hdr.MsgIdOffset = uint32(buf.Len() + MoHeaderSize)
|
|
binary.Write(&buf, binary.LittleEndian, msgIdPosList)
|
|
hdr.MsgStrOffset = uint32(buf.Len() + MoHeaderSize)
|
|
binary.Write(&buf, binary.LittleEndian, msgStrPosList)
|
|
|
|
hdr.MsgIdCount = uint32(len(msgList))
|
|
return buf.Bytes()
|
|
}
|
|
|
|
// must called after encodeData
|
|
func encodeHeader(hdr *moHeader) []byte {
|
|
var buf bytes.Buffer
|
|
binary.Write(&buf, binary.LittleEndian, hdr)
|
|
return buf.Bytes()
|
|
}
|
|
|
|
func encodeMsgId(v Message) string {
|
|
if v.MsgContext != "" && v.MsgIdPlural != "" {
|
|
return v.MsgContext + EotSeparator + v.MsgId + NulSeparator + v.MsgIdPlural
|
|
}
|
|
if v.MsgContext != "" && v.MsgIdPlural == "" {
|
|
return v.MsgContext + EotSeparator + v.MsgId
|
|
}
|
|
if v.MsgContext == "" && v.MsgIdPlural != "" {
|
|
return v.MsgId + NulSeparator + v.MsgIdPlural
|
|
}
|
|
return v.MsgId
|
|
}
|
|
|
|
func encodeMsgStr(v Message) string {
|
|
if v.MsgIdPlural != "" {
|
|
return strings.Join(v.MsgStrPlural, NulSeparator)
|
|
}
|
|
return v.MsgStr
|
|
}
|
|
|
|
type byMessages []Message
|
|
|
|
func (d byMessages) Len() int {
|
|
return len(d)
|
|
}
|
|
func (d byMessages) Less(i, j int) bool {
|
|
if a, b := d[i].MsgContext, d[j].MsgContext; a != b {
|
|
return a < b
|
|
}
|
|
if a, b := d[i].MsgId, d[j].MsgId; a != b {
|
|
return a < b
|
|
}
|
|
if a, b := d[i].MsgIdPlural, d[j].MsgIdPlural; a != b {
|
|
return a < b
|
|
}
|
|
return false
|
|
}
|
|
func (d byMessages) Swap(i, j int) {
|
|
d[i], d[j] = d[j], d[i]
|
|
}
|