54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
|
package resource
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"git.defalsify.org/festive/state"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Sizer struct {
|
||
|
|
outputSize uint32
|
||
|
|
menuSize uint16
|
||
|
|
memberSizes map[string]uint16
|
||
|
|
totalMemberSize uint32
|
||
|
|
sink string
|
||
|
|
}
|
||
|
|
|
||
|
|
func SizerFromState(st *state.State) (Sizer, error){
|
||
|
|
sz := Sizer{
|
||
|
|
outputSize: st.GetOutputSize(),
|
||
|
|
menuSize: st.GetMenuSize(),
|
||
|
|
memberSizes: make(map[string]uint16),
|
||
|
|
}
|
||
|
|
sizes, err := st.Sizes()
|
||
|
|
if err != nil {
|
||
|
|
return sz, err
|
||
|
|
}
|
||
|
|
for k, v := range sizes {
|
||
|
|
if v == 0 {
|
||
|
|
sz.sink = k
|
||
|
|
}
|
||
|
|
sz.memberSizes[k] = v
|
||
|
|
sz.totalMemberSize += uint32(v)
|
||
|
|
}
|
||
|
|
return sz, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func(szr *Sizer) Check(s string) (uint32, bool) {
|
||
|
|
l := uint32(len(s))
|
||
|
|
if szr.outputSize > 0 && l > szr.outputSize {
|
||
|
|
log.Printf("sizer check fails with length %v: %s", l, szr)
|
||
|
|
return l, false
|
||
|
|
}
|
||
|
|
return l, true
|
||
|
|
}
|
||
|
|
|
||
|
|
func(szr *Sizer) String() string {
|
||
|
|
var diff uint32
|
||
|
|
if szr.outputSize > 0 {
|
||
|
|
diff = szr.outputSize - szr.totalMemberSize - uint32(szr.menuSize)
|
||
|
|
}
|
||
|
|
return fmt.Sprintf("output: %v, member: %v, menu: %v, diff: %v", szr.outputSize, szr.totalMemberSize, szr.menuSize, diff)
|
||
|
|
}
|