package hotline
import (
+ "bytes"
"encoding/binary"
+ "errors"
+ "io"
+ "path"
"strings"
)
Name []byte
}
-func NewFilePathItem(b []byte) FilePathItem {
- return FilePathItem{
- Len: b[2],
- Name: b[3:],
- }
+type FilePath struct {
+ ItemCount [2]byte
+ Items []FilePathItem
}
-type FilePath struct {
- PathItemCount []byte
- PathItems []FilePathItem
+func (fp *FilePath) UnmarshalBinary(b []byte) error {
+ reader := bytes.NewReader(b)
+ err := binary.Read(reader, binary.BigEndian, &fp.ItemCount)
+ if err != nil && !errors.Is(err, io.EOF) {
+ return err
+ }
+ if errors.Is(err, io.EOF) {
+ return nil
+ }
+
+ for i := uint16(0); i < fp.Len(); i++ {
+ // skip two bytes for the file path delimiter
+ _, _ = reader.Seek(2, io.SeekCurrent)
+
+ // read the length of the next pathItem
+ segLen, err := reader.ReadByte()
+ if err != nil {
+ return err
+ }
+
+ pBytes := make([]byte, segLen)
+
+ _, err = reader.Read(pBytes)
+ if err != nil && !errors.Is(err, io.EOF) {
+ return err
+ }
+
+ fp.Items = append(fp.Items, FilePathItem{Len: segLen, Name: pBytes})
+ }
+
+ return nil
}
-func NewFilePath(b []byte) FilePath {
- if b == nil {
- return FilePath{}
+func (fp *FilePath) IsDropbox() bool {
+ if fp.Len() == 0 {
+ return false
}
- fp := FilePath{PathItemCount: b[0:2]}
+ return strings.Contains(strings.ToLower(string(fp.Items[fp.Len()-1].Name)), "drop box")
+}
- // number of items in the path
- pathItemLen := binary.BigEndian.Uint16(b[0:2])
- pathData := b[2:]
- for i := uint16(0); i < pathItemLen; i++ {
- segLen := pathData[2]
- fp.PathItems = append(fp.PathItems, NewFilePathItem(pathData[:segLen+3]))
- pathData = pathData[3+segLen:]
+func (fp *FilePath) IsUploadDir() bool {
+ if fp.Len() == 0 {
+ return false
}
- return fp
+ return strings.Contains(strings.ToLower(string(fp.Items[fp.Len()-1].Name)), "upload")
+}
+
+func (fp *FilePath) Len() uint16 {
+ return binary.BigEndian.Uint16(fp.ItemCount[:])
}
func (fp *FilePath) String() string {
- var out []string
- for _, i := range fp.PathItems {
+ out := []string{"/"}
+ for _, i := range fp.Items {
out = append(out, string(i.Name))
}
- return strings.Join(out, pathSeparator)
+
+ return path.Join(out...)
+}
+
+func ReadFilePath(filePathFieldData []byte) string {
+ var fp FilePath
+ err := fp.UnmarshalBinary(filePathFieldData)
+ if err != nil {
+ // TODO
+ }
+ return fp.String()
+}
+
+func readPath(fileRoot string, filePath, fileName []byte) (fullPath string, err error) {
+ var fp FilePath
+ if filePath != nil {
+ if err = fp.UnmarshalBinary(filePath); err != nil {
+ return "", err
+ }
+ }
+
+ fullPath = path.Join(
+ "/",
+ fileRoot,
+ fp.String(),
+ path.Join("/", string(fileName)),
+ )
+
+ return fullPath, nil
}