Files
trufflehog/pkg/analyzer/analyzers/huggingface/permissions.go
Abdul BasitandAbdul Basit 3e46b3f221 [Analyzer] Test and generated permissions for HuggingFace, Square & Stripe (#3294)
* stripe analyzer unit test

* add huggingface analyzer unit test

* add permissions.yaml for huggingface and fix in analyzer

* square permissions generated

* permissions generated for stripe

* change permissions to lowercase

* skip unknown permissions for square and stripe

---------

Co-authored-by: Abdul Basit <[email protected]>
2024-09-23 12:17:20 -07:00

67 lines
1.5 KiB
Go

// Code generated by go generate; DO NOT EDIT.
package huggingface
import "errors"
type Permission int
const (
Invalid Permission = iota
Read Permission = iota
Write Permission = iota
)
var (
PermissionStrings = map[Permission]string{
Read: "read",
Write: "write",
}
StringToPermission = map[string]Permission{
"read": Read,
"write": Write,
}
PermissionIDs = map[Permission]int{
Read: 1,
Write: 2,
}
IdToPermission = map[int]Permission{
1: Read,
2: Write,
}
)
// ToString converts a Permission enum to its string representation
func (p Permission) ToString() (string, error) {
if str, ok := PermissionStrings[p]; ok {
return str, nil
}
return "", errors.New("invalid permission")
}
// ToID converts a Permission enum to its ID
func (p Permission) ToID() (int, error) {
if id, ok := PermissionIDs[p]; ok {
return id, nil
}
return 0, errors.New("invalid permission")
}
// PermissionFromString converts a string representation to its Permission enum
func PermissionFromString(s string) (Permission, error) {
if p, ok := StringToPermission[s]; ok {
return p, nil
}
return 0, errors.New("invalid permission string")
}
// PermissionFromID converts an ID to its Permission enum
func PermissionFromID(id int) (Permission, error) {
if p, ok := IdToPermission[id]; ok {
return p, nil
}
return 0, errors.New("invalid permission ID")
}