initial commit

utils.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"os"
+	"time"
+
+	"git.tordarus.net/tordarus/channel"
+	"git.tordarus.net/tordarus/gmath"
+)
+
+func InitializeRunner() channel.Runner {
+	if ThreadCount <= 0 {
+		return channel.NewUnlimitedRunner()
+	}
+	return channel.NewLimitedRunner(ThreadCount)
+}
+
+func FormatBytes[T gmath.Integer](bytes T) string {
+	value := float64(bytes)
+
+	if value >= 1000000000000 {
+		return fmt.Sprintf("%.02fT", value/1000000000000)
+	} else if value >= 1000000000 {
+		return fmt.Sprintf("%.02fG", value/1000000000)
+	} else if value >= 1000000 {
+		return fmt.Sprintf("%.02fM", value/1000000)
+	} else if value >= 1000 {
+		return fmt.Sprintf("%.02fK", value/1000)
+	} else {
+		return fmt.Sprintf("%.02fB", value)
+	}
+}
+
+func PrintByteChan(w io.Writer, ch <-chan byte) error {
+	for b := range ch {
+		if _, err := w.Write([]byte{b}); err != nil {
+			return err
+		}
+	}
+
+	fmt.Fprintln(w)
+	return nil
+}
+
+func PrintFileHandle(fh *FileHandle) {
+	if err := PrintByteChan(os.Stdout, fh.Chan); err != nil {
+		panic(err) // TODO error handling
+	}
+}
+
+func WaitForNextTelegramSendCycle() {
+	now := time.Now()
+	_, offset := now.Zone()
+	offsetDuration := time.Duration(offset) * time.Second
+	lastCycle := now.Truncate(TelegramOrganizeMessageSendInterval).Add(-offsetDuration).Add(TelegramOrganizeMessageSendOffset)
+
+	if durationUntilLastCycle := time.Until(lastCycle); durationUntilLastCycle > 0 {
+		time.Sleep(durationUntilLastCycle)
+	} else {
+		nextCycle := lastCycle.Add(TelegramOrganizeMessageSendInterval)
+		time.Sleep(time.Until(nextCycle))
+	}
+}