From b129b7cbc9fd9a9c11a77e5922861ef08893efa1 Mon Sep 17 00:00:00 2001 From: Jeff Halter <868228+jhalter@users.noreply.github.com> Date: Sat, 8 Jun 2024 20:36:04 -0700 Subject: Convert bespoke methods to io.Reader/io.Writer interfaces --- hotline/file_name_with_info.go | 38 ++++++++++++++++++++------------------ 1 file changed, 20 insertions(+), 18 deletions(-) (limited to 'hotline/file_name_with_info.go') diff --git a/hotline/file_name_with_info.go b/hotline/file_name_with_info.go index f230856..ac64c74 100644 --- a/hotline/file_name_with_info.go +++ b/hotline/file_name_with_info.go @@ -3,6 +3,8 @@ package hotline import ( "bytes" "encoding/binary" + "io" + "slices" ) type FileNameWithInfo struct { @@ -24,28 +26,28 @@ func (f *fileNameWithInfoHeader) nameLen() int { return int(binary.BigEndian.Uint16(f.NameSize[:])) } -func (f *FileNameWithInfo) MarshalBinary() (data []byte, err error) { - var buf bytes.Buffer - err = binary.Write(&buf, binary.LittleEndian, f.fileNameWithInfoHeader) - if err != nil { - return data, err - } - - _, err = buf.Write(f.name) - if err != nil { - return data, err - } - - return buf.Bytes(), err +// Read implements io.Reader for FileNameWithInfo +func (f *FileNameWithInfo) Read(b []byte) (int, error) { + return copy(b, + slices.Concat( + f.Type[:], + f.Creator[:], + f.FileSize[:], + f.RSVD[:], + f.NameScript[:], + f.NameSize[:], + f.name, + ), + ), io.EOF } -func (f *FileNameWithInfo) UnmarshalBinary(data []byte) error { - err := binary.Read(bytes.NewReader(data), binary.BigEndian, &f.fileNameWithInfoHeader) +func (f *FileNameWithInfo) Write(p []byte) (int, error) { + err := binary.Read(bytes.NewReader(p), binary.BigEndian, &f.fileNameWithInfoHeader) if err != nil { - return err + return 0, err } headerLen := binary.Size(f.fileNameWithInfoHeader) - f.name = data[headerLen : headerLen+f.nameLen()] + f.name = p[headerLen : headerLen+f.nameLen()] - return err + return len(p), nil } -- cgit