Goit

// Copyright (C) 2023, Jakob Wakeling

// All rights reserved.

package util

import (

	"cmp"

	"errors"

	"io/fs"

	"net/http"

	"os"

	"path/filepath"

)

const ModeNotRegular = os.ModeSymlink | os.ModeDevice | os.ModeNamedPipe | os.ModeSocket | os.ModeCharDevice |

	os.ModeIrregular

func If[T any](cond bool, a, b T) T {

	if cond {

		return a

	}

	return b

}

func Min[T cmp.Ordered](a, b T) T {

	if a < b {

		return a

	}

	return b

}

func Max[T cmp.Ordered](a, b T) T {

	if a > b {

		return a

	}

	return b

}

/* Return the named cookie or nil if not found or invalid. */

func Cookie(r *http.Request, name string) *http.Cookie {

	c, err := r.Cookie(name)

	if err == nil && c.Valid() == nil {

		return c

	}

	return nil

}

/* Return a slice of form values with the specified key. */

func FormValues(r *http.Request, key string) []string {

	if r.Form == nil {

		r.ParseMultipartForm(32 << 20 /* 32 MB */)

	}

	return r.Form[key]

}

func ModeString(mode uint32) string {

	s := If((mode&0o40000) != 0, "d", "-")

	s += If((mode&0o400) != 0, "r", "-")

	s += If((mode&0o200) != 0, "w", "-")

	s += If((mode&0o100) != 0, "x", "-")

	s += If((mode&0o040) != 0, "r", "-")

	s += If((mode&0o020) != 0, "w", "-")

	s += If((mode&0o010) != 0, "x", "-")

	s += If((mode&0o004) != 0, "r", "-")

	s += If((mode&0o002) != 0, "w", "-")

	s += If((mode&0o001) != 0, "x", "-")

	return s

}

func DirSize(path string) (uint64, error) {

	var size int64

	err := filepath.WalkDir(path, func(_ string, d fs.DirEntry, err error) error {

		if err != nil {

			if errors.Is(err, os.ErrNotExist) {

				return nil

			}

			return err

		}

		if d.IsDir() {

			return nil

		}

		f, err := d.Info()

		if err != nil {

			return err

		}

		if (f.Mode() & ModeNotRegular) == 0 {

			size += f.Size()

		}

		return nil

	})

	return uint64(size), err

}