summaryrefslogtreecommitdiffstats
path: root/cmd/files.go
blob: 6b608caf08487973994d1b9bb2e2feaba0558ab8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package cmd

import (
	"git.db.org.ai/dborg/internal/formatter"
	"git.db.org.ai/dborg/internal/models"
	"github.com/spf13/cobra"
)

var filesCmd = &cobra.Command{
	Use:     "files [url]",
	Aliases: []string{"f"},
	Short:   "Search open directory files",
	Long:    `Search for files in open directories using various filters (free OSINT endpoint)`,
	Args:    argsOrStdin(1),
	RunE:    runFilesSearch,
}

func init() {
	rootCmd.AddCommand(filesCmd)
	filesCmd.Flags().StringP("filename", "n", "", "Search term to match in filenames")
	filesCmd.Flags().StringP("extension", "e", "", "Filter by file extension(s) - comma-separated (e.g., pdf,doc,txt)")
	filesCmd.Flags().StringP("exclude", "x", "html,HTML", "Exclude file extension(s) - comma-separated")
	filesCmd.Flags().IntP("size", "s", 10, "Number of results to return (max 40)")
	filesCmd.Flags().IntP("from", "f", 0, "Starting offset for pagination")
}

func runFilesSearch(cmd *cobra.Command, args []string) error {
	c, err := newUnauthenticatedClient()
	if err != nil {
		return err
	}

	filename, _ := cmd.Flags().GetString("filename")
	extension, _ := cmd.Flags().GetString("extension")
	exclude, _ := cmd.Flags().GetString("exclude")
	size, _ := cmd.Flags().GetInt("size")
	from, _ := cmd.Flags().GetInt("from")

	return forEachQuery(args, func(url string) error {
		params := &models.OpenDirectorySearchParams{
			URL:       url,
			Filename:  filename,
			Extension: extension,
			Exclude:   exclude,
			Size:      size,
			From:      from,
		}
		response, err := c.SearchOpenDirectoryFiles(params)
		if err != nil {
			return err
		}
		return formatter.FormatFilesResults(*response, IsJSONOutput())
	})
}