-
Notifications
You must be signed in to change notification settings - Fork 18
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add add subcommand and other tweaks for cookiecloud cmd
- Loading branch information
Showing
15 changed files
with
352 additions
and
105 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,136 @@ | ||
| package get | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "net/url" | ||
|
|
||
| log "github.com/sirupsen/logrus" | ||
| "github.com/spf13/cobra" | ||
|
|
||
| "github.com/sagan/ptool/cmd" | ||
| "github.com/sagan/ptool/cmd/cookiecloud" | ||
| "github.com/sagan/ptool/config" | ||
| "github.com/sagan/ptool/site" | ||
| "github.com/sagan/ptool/util" | ||
| ) | ||
|
|
||
| var ( | ||
| showAll = false | ||
| format = "" | ||
| profile = "" | ||
| ) | ||
|
|
||
| var command = &cobra.Command{ | ||
| Use: "get {site | group | domain | url}...", | ||
| Annotations: map[string]string{"cobra-prompt-dynamic-suggestions": "cookiecloud.get"}, | ||
| Short: "Get cookie for sites or domains from data of cookiecloud servers.", | ||
| Long: `Get cookie for sites or domains from data of cookiecloud servers. | ||
| Each arg can be a site or group name, a domain (or IP), or a full url. It will query cookiecloud servers | ||
| and display all found cookies of the corresponding arg in list. | ||
| If --all flag is NOT set (the default case), only cookies which path attribute constitutes the prefix of | ||
| the url's path will be included (in the case of arg being a domain or IP, it's path is assumed to be "/"). | ||
| If --all flag is set, all cookies associated with the domain of arg will be included, the result is only | ||
| useful / meanful in the case of "js" format. | ||
| By default it will show site cookies in http request "Cookie" header format. | ||
| If --format flag is set to "js", it will show cookies in a JavaScript "document.cookie=''" code snippet format, | ||
| which is suitable for pasting to the browser developer console to set the cookies of corresponding site.`, | ||
| Args: cobra.MatchAll(cobra.MinimumNArgs(1), cobra.OnlyValidArgs), | ||
| RunE: get, | ||
| } | ||
|
|
||
| func init() { | ||
| cmd.AddEnumFlagP(command, &format, "format", "", &cmd.EnumFlag{ | ||
| Description: "Cookies output format", | ||
| Options: [][2]string{ | ||
| {"http", `http request "Cookie" header`}, | ||
| {"js", `JavaScript 'document.cookie=""' code snippet`}, | ||
| }, | ||
| }) | ||
| command.Flags().BoolVarP(&showAll, "all", "a", false, "Show all cookies associated with the domain (no path checking)") | ||
| command.Flags().StringVarP(&profile, "profile", "", "", "Comma-separated string, Set the used cookiecloud profile name(s). If not set, All cookiecloud profiles in config will be used") | ||
| cookiecloud.Command.AddCommand(command) | ||
| } | ||
|
|
||
| func get(cmd *cobra.Command, args []string) error { | ||
| cntError := int64(0) | ||
| cookiecloudProfiles := cookiecloud.ParseProfile(profile) | ||
| if len(cookiecloudProfiles) == 0 { | ||
| return fmt.Errorf("no cookiecloud profile specified or found") | ||
| } | ||
| cookiecloudDatas := []cookiecloud.Ccdata_struct{} | ||
| for _, profile := range cookiecloudProfiles { | ||
| data, err := cookiecloud.GetCookiecloudData(profile.Server, profile.Uuid, profile.Password, profile.Proxy) | ||
| if err != nil { | ||
| log.Errorf("Cookiecloud server %s (uuid %s) connection failed: %v\n", profile.Server, profile.Uuid, err) | ||
| cntError++ | ||
| } else { | ||
| log.Infof("Cookiecloud server %s (uuid %s) connection ok: cookies of %d domains found\n", | ||
| profile.Server, profile.Uuid, len(data.Cookie_data)) | ||
| cookiecloudDatas = append(cookiecloudDatas, cookiecloud.Ccdata_struct{ | ||
| Label: fmt.Sprintf("%s-%s", util.GetUrlDomain(profile.Server), profile.Uuid), | ||
| Sites: profile.Sites, | ||
| Data: data, | ||
| }) | ||
| } | ||
| } | ||
| if len(cookiecloudDatas) == 0 { | ||
| return fmt.Errorf("no cookiecloud server can be connected") | ||
| } | ||
| siteOrDomainOrUrls := config.ParseGroupAndOtherNames(args...) | ||
|
|
||
| fmt.Printf("%-20s %-20s %s\n", "Site/Url/Hostname", "CookieCloud", "Cookie") | ||
| for _, siteOrDomainOrUrl := range siteOrDomainOrUrls { | ||
| domainOrUrl := "" | ||
| if siteConfig := config.GetSiteConfig(siteOrDomainOrUrl); siteConfig != nil { | ||
| siteInstance, err := site.CreateSiteInternal(siteOrDomainOrUrl, siteConfig, config.Get()) | ||
| if err != nil { | ||
| log.Debugf("Failed to create site %s", siteOrDomainOrUrl) | ||
| } else { | ||
| domainOrUrl = util.ParseUrlHostname(siteInstance.GetSiteConfig().Url) | ||
| } | ||
| } else { | ||
| domainOrUrl = siteOrDomainOrUrl | ||
| } | ||
| var cookieScope = domainOrUrl // the valid scope of cookies | ||
| if domainOrUrl == "" { | ||
| fmt.Printf("%-20s %-20s %s\n", | ||
| util.First(util.StringPrefixInWidth(siteOrDomainOrUrl, 20)), "", "// Error: empty hostname") | ||
| cntError++ | ||
| continue | ||
| } else if !util.IsUrl(domainOrUrl) && !util.IsHostname(domainOrUrl) { | ||
| fmt.Printf("%-20s %-20s %s\n", | ||
| util.First(util.StringPrefixInWidth(siteOrDomainOrUrl, 20)), | ||
| "", "// Error: invalid site, url or hostname") | ||
| cntError++ | ||
| continue | ||
| } else if util.IsUrl(domainOrUrl) { | ||
| urlObj, err := url.Parse(domainOrUrl) | ||
| if err == nil { | ||
| cookieScope = urlObj.Hostname() | ||
| if !showAll && urlObj.Path != "/" { | ||
| cookieScope += urlObj.Path | ||
| } | ||
| } | ||
| } | ||
| if showAll { | ||
| cookieScope += " <all>" | ||
| } | ||
| for _, cookiecloudData := range cookiecloudDatas { | ||
| cookie, _ := cookiecloudData.Data.GetEffectiveCookie(domainOrUrl, showAll, format) | ||
| if cookie == "" { | ||
| log.Debugf("No cookie found for %s in cookiecloud %s", siteOrDomainOrUrl, cookiecloudData.Label) | ||
| continue | ||
| } | ||
| cookieStr := cookie + " // " + cookieScope | ||
| fmt.Printf("%-20s %-20s %s\n", util.First(util.StringPrefixInWidth(siteOrDomainOrUrl, 20)), | ||
| util.First(util.StringPrefixInWidth(cookiecloudData.Label, 20)), cookieStr) | ||
| } | ||
| } | ||
|
|
||
| if cntError > 0 { | ||
| return fmt.Errorf("%d errors", cntError) | ||
| } | ||
| return nil | ||
| } |
Oops, something went wrong.