52 lines
		
	
	
		
			1013 B
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			52 lines
		
	
	
		
			1013 B
		
	
	
	
		
			Go
		
	
	
	
package cmd
 | 
						|
 | 
						|
import (
 | 
						|
	"github.com/spf13/cobra"
 | 
						|
 | 
						|
	"github.com/filebrowser/filebrowser/v2/users"
 | 
						|
)
 | 
						|
 | 
						|
func init() {
 | 
						|
	usersCmd.AddCommand(usersFindCmd)
 | 
						|
	usersCmd.AddCommand(usersLsCmd)
 | 
						|
}
 | 
						|
 | 
						|
var usersFindCmd = &cobra.Command{
 | 
						|
	Use:   "find <id|username>",
 | 
						|
	Short: "Find a user by username or id",
 | 
						|
	Long:  `Find a user by username or id. If no flag is set, all users will be printed.`,
 | 
						|
	Args:  cobra.ExactArgs(1),
 | 
						|
	Run:   findUsers,
 | 
						|
}
 | 
						|
 | 
						|
var usersLsCmd = &cobra.Command{
 | 
						|
	Use:   "ls",
 | 
						|
	Short: "List all users.",
 | 
						|
	Args:  cobra.NoArgs,
 | 
						|
	Run:   findUsers,
 | 
						|
}
 | 
						|
 | 
						|
var findUsers = python(func(cmd *cobra.Command, args []string, d pythonData) {
 | 
						|
	var (
 | 
						|
		list []*users.User
 | 
						|
		user *users.User
 | 
						|
		err  error
 | 
						|
	)
 | 
						|
 | 
						|
	if len(args) == 1 {
 | 
						|
		username, id := parseUsernameOrID(args[0])
 | 
						|
		if username != "" {
 | 
						|
			user, err = d.store.Users.Get("", username)
 | 
						|
		} else {
 | 
						|
			user, err = d.store.Users.Get("", id)
 | 
						|
		}
 | 
						|
 | 
						|
		list = []*users.User{user}
 | 
						|
	} else {
 | 
						|
		list, err = d.store.Users.Gets("")
 | 
						|
	}
 | 
						|
 | 
						|
	checkErr(err)
 | 
						|
	printUsers(list)
 | 
						|
}, pythonConfig{})
 |