First commit
This commit is contained in:
commit
87d4ab0490
|
@ -0,0 +1,60 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
"forge.cadoles.com/pcaseiro/namecheck"
|
||||
"forge.cadoles.com/pcaseiro/namecheck/twitter"
|
||||
)
|
||||
|
||||
func checkUsername(ch chan<- string, wg *sync.WaitGroup, sn namecheck.SocialNetwork, username string) (bool, error) {
|
||||
defer wg.Done()
|
||||
// s1 := rand.NewSource(time.Now().UnixNano())
|
||||
// time.Sleep(rand.New(s1).Intn(100))
|
||||
res, err := sn.IsAvailable(username)
|
||||
if err != nil {
|
||||
ch <- fmt.Sprintf("Error : {%v}\n", err)
|
||||
return false, err
|
||||
}
|
||||
|
||||
if !res {
|
||||
ch <- fmt.Sprintf("User %s is not available\n", username)
|
||||
return false, nil
|
||||
}
|
||||
ch <- fmt.Sprintf("User %s is available\n", username)
|
||||
return true, nil
|
||||
}
|
||||
|
||||
func main() {
|
||||
var wg sync.WaitGroup
|
||||
var socialNetworks []namecheck.SocialNetwork
|
||||
|
||||
// Create a channel for result
|
||||
ch := make(chan string, 20)
|
||||
|
||||
// Create the "social networks"
|
||||
for i := 0; i < 20; i++ {
|
||||
socialNetworks = append(socialNetworks, &twitter.Twitter{})
|
||||
}
|
||||
|
||||
// Start the routines for check
|
||||
username := os.Args[1]
|
||||
for _, sn := range socialNetworks {
|
||||
wg.Add(1)
|
||||
go checkUsername(ch, &wg, sn, username)
|
||||
}
|
||||
|
||||
// Wait in background
|
||||
go func() {
|
||||
wg.Wait()
|
||||
close(ch)
|
||||
}()
|
||||
|
||||
// Print the result
|
||||
for li := range ch {
|
||||
fmt.Printf(li)
|
||||
}
|
||||
|
||||
}
|
|
@ -0,0 +1,70 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"sync"
|
||||
|
||||
"forge.cadoles.com/pcaseiro/namecheck"
|
||||
"forge.cadoles.com/pcaseiro/namecheck/twitter"
|
||||
)
|
||||
|
||||
func checkUsername(ch chan<- string, wg *sync.WaitGroup, sn namecheck.SocialNetwork, username string) (bool, error) {
|
||||
defer wg.Done()
|
||||
// s1 := rand.NewSource(time.Now().UnixNano())
|
||||
// time.Sleep(rand.New(s1).Intn(100))
|
||||
res, err := sn.IsAvailable(username)
|
||||
if err != nil {
|
||||
ch <- fmt.Sprintf("Error : {%v} for %v\n", err, sn.String())
|
||||
return false, err
|
||||
}
|
||||
|
||||
if !res {
|
||||
ch <- fmt.Sprintf("User %s is not available on %v\n", username, sn.String())
|
||||
return false, nil
|
||||
}
|
||||
ch <- fmt.Sprintf("User %s is available on %v\n", username, sn.String())
|
||||
return true, nil
|
||||
}
|
||||
|
||||
func msgHandler(w http.ResponseWriter, req *http.Request) {
|
||||
var wg sync.WaitGroup
|
||||
var socialNetworks []namecheck.SocialNetwork
|
||||
|
||||
// Create a channel for result
|
||||
ch := make(chan string, 20)
|
||||
|
||||
// Create the "social networks"
|
||||
for i := 0; i < 20; i++ {
|
||||
socialNetworks = append(socialNetworks, &twitter.Twitter{})
|
||||
}
|
||||
|
||||
// Start the routines for check
|
||||
urlQuery := req.URL.Query() //os.Args[1]
|
||||
usernames := urlQuery["u"]
|
||||
for _, user := range usernames {
|
||||
for _, sn := range socialNetworks {
|
||||
wg.Add(1)
|
||||
go checkUsername(ch, &wg, sn, user)
|
||||
}
|
||||
}
|
||||
|
||||
// Wait in background
|
||||
go func() {
|
||||
wg.Wait()
|
||||
close(ch)
|
||||
}()
|
||||
|
||||
// Print the result
|
||||
for li := range ch {
|
||||
io.WriteString(w, li)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func main() {
|
||||
http.HandleFunc("/", msgHandler)
|
||||
log.Fatal(http.ListenAndServe(":8080", nil))
|
||||
}
|
|
@ -0,0 +1,21 @@
|
|||
package namecheck
|
||||
|
||||
import "fmt"
|
||||
|
||||
type ErrSocialNetwork interface {
|
||||
Error() error
|
||||
Unwrap() error
|
||||
}
|
||||
|
||||
type ErrUnknownAvailability struct {
|
||||
Username string
|
||||
Cause error
|
||||
}
|
||||
|
||||
func (e *ErrUnknownAvailability) Error() string {
|
||||
return fmt.Sprintf("Error during check for username '%s'\n[%s]\n", e.Username, e.Cause)
|
||||
}
|
||||
|
||||
func (e *ErrUnknownAvailability) Unwrap() error {
|
||||
return e.Cause
|
||||
}
|
|
@ -0,0 +1,19 @@
|
|||
package namecheck
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
)
|
||||
|
||||
type Validator interface {
|
||||
IsValid(string) bool
|
||||
}
|
||||
|
||||
type Provider interface {
|
||||
IsAvailable(string) (bool, error)
|
||||
}
|
||||
|
||||
type SocialNetwork interface {
|
||||
fmt.Stringer
|
||||
Validator
|
||||
Provider
|
||||
}
|
|
@ -0,0 +1,22 @@
|
|||
mode: set
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:29.51,31.2 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:33.53,34.65 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:37.2,37.13 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:34.65,36.3 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:40.41,42.2 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:44.42,46.2 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:48.49,49.41 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:52.2,52.39 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:55.2,55.29 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:59.2,59.30 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:62.2,62.13 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:49.41,51.3 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:52.39,54.3 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:55.29,57.3 1 0
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:59.30,61.3 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:66.33,68.2 1 0
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:71.62,74.16 3 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:78.2,81.44 3 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:85.2,85.18 1 1
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:74.16,77.3 2 0
|
||||
forge.cadoles.com/pcaseiro/namecheck/twitter/twitter.go:81.44,84.3 2 1
|
|
@ -0,0 +1,3 @@
|
|||
module forge.cadoles.com/pcaseiro/namecheck
|
||||
|
||||
go 1.13
|
|
@ -0,0 +1,92 @@
|
|||
package twitter
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"regexp"
|
||||
"strings"
|
||||
"unicode/utf8"
|
||||
|
||||
"forge.cadoles.com/pcaseiro/namecheck"
|
||||
)
|
||||
|
||||
const (
|
||||
sizeMin = 1
|
||||
sizeMax = 15
|
||||
legalChars = "^[a-zA-Z0-9_]+$"
|
||||
illegalPattern = "twitter"
|
||||
twitterUrl = "https://twitter.com"
|
||||
)
|
||||
|
||||
type Twitter struct {
|
||||
HttpClient http.Client
|
||||
Responds bool
|
||||
}
|
||||
|
||||
var (
|
||||
legal = regexp.MustCompile(legalChars)
|
||||
illegal = regexp.MustCompile(illegalPattern)
|
||||
)
|
||||
|
||||
func containsOnlyLegalChars(username string) bool {
|
||||
return legal.MatchString(username)
|
||||
}
|
||||
|
||||
func containsNoIllegalPattern(username string) bool {
|
||||
if strings.Contains(strings.ToLower(username), illegalPattern) {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func isLongEnough(username string) bool {
|
||||
return utf8.RuneCountInString(username) > sizeMin
|
||||
}
|
||||
|
||||
func isShortEnough(username string) bool {
|
||||
return utf8.RuneCountInString(username) <= sizeMax
|
||||
}
|
||||
|
||||
func (t *Twitter) IsValid(username string) bool {
|
||||
if !containsNoIllegalPattern(username) {
|
||||
return false
|
||||
}
|
||||
if !containsOnlyLegalChars(username) {
|
||||
return false
|
||||
}
|
||||
if !isLongEnough(username) {
|
||||
return false
|
||||
}
|
||||
|
||||
if !isShortEnough(username) {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
// Name return a string with the name of the "Social Media"
|
||||
func (t *Twitter) String() string {
|
||||
return "Twitter"
|
||||
}
|
||||
|
||||
// IsAvailable check if a user name is valid and available on twitter
|
||||
func (t *Twitter) IsAvailable(username string) (bool, error) {
|
||||
resp := t.IsValid(username)
|
||||
if resp {
|
||||
url := fmt.Sprintf("%s/%s", twitterUrl, username)
|
||||
resp, err := t.HttpClient.Get(url)
|
||||
if err != nil {
|
||||
t.Responds = false
|
||||
nerr := &namecheck.ErrUnknownAvailability{Username: username, Cause: err}
|
||||
return false, nerr
|
||||
}
|
||||
t.Responds = true
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusNotFound {
|
||||
return false, nil
|
||||
}
|
||||
return true, nil
|
||||
}
|
||||
return false, fmt.Errorf("Username is not valid")
|
||||
}
|
|
@ -0,0 +1,115 @@
|
|||
package twitter
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestContainsOnlyLegalChars(t *testing.T) {
|
||||
goodUsername := "cadoles"
|
||||
badUsername := "cadoles!"
|
||||
|
||||
if !containsOnlyLegalChars(goodUsername) {
|
||||
t.Errorf("The user is valid but containsOnlyLegalChars says it's not")
|
||||
}
|
||||
|
||||
if containsOnlyLegalChars(badUsername) {
|
||||
t.Errorf("The user is bad but containsOnlyLegalChars says it's OK")
|
||||
}
|
||||
}
|
||||
|
||||
func TestContainsNoIllegalPattern(t *testing.T) {
|
||||
goodUsername := "puppetmaster"
|
||||
if !containsNoIllegalPattern(goodUsername) {
|
||||
t.Errorf("The username is valid but containsNoIllegalPattern says it's not")
|
||||
}
|
||||
|
||||
badUsername := "twitterPuppetMaster"
|
||||
if containsNoIllegalPattern(badUsername) {
|
||||
t.Errorf("The username is bad but containsNoIllegalPattern says it's OK")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestIsLongEnough(t *testing.T) {
|
||||
goodUsername := "cadoles_test"
|
||||
if !isLongEnough(goodUsername) {
|
||||
t.Errorf("The username is long enough but isLongEnough says it's not")
|
||||
}
|
||||
|
||||
badUsername := ""
|
||||
if isLongEnough(badUsername) {
|
||||
t.Errorf("The username is not long enought but isLongEnough says it's OK")
|
||||
}
|
||||
}
|
||||
|
||||
func TestIsShortEnough(t *testing.T) {
|
||||
goodUsername := "cadoles_test"
|
||||
if !isShortEnough(goodUsername) {
|
||||
t.Errorf("The username is short enough but isShortEnough says it's not !")
|
||||
}
|
||||
|
||||
badUsername := "cadoles_cadoles_cadoles_cadoles_cadoles"
|
||||
if isShortEnough(badUsername) {
|
||||
t.Errorf("The username is not short enough but isShortEnough says it's OK !")
|
||||
}
|
||||
}
|
||||
|
||||
func TestIsValid(t *testing.T) {
|
||||
var tw Twitter
|
||||
goodUsername := "cadoles"
|
||||
if !tw.IsValid(goodUsername) {
|
||||
t.Errorf("The username is valid and isValid says it's not !")
|
||||
}
|
||||
}
|
||||
|
||||
func TestIsToShort(t *testing.T) {
|
||||
var tw Twitter
|
||||
toShort := ""
|
||||
if tw.IsValid(toShort) {
|
||||
t.Errorf("The is to short but isValid says it's OK !")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestIsToLong(t *testing.T) {
|
||||
var tw Twitter
|
||||
toLong := "cadoles_cadoles_cadoles_cadoles_cadoles_cadoles"
|
||||
if tw.IsValid(toLong) {
|
||||
t.Errorf("The username is to long but isValid says it's OK !")
|
||||
}
|
||||
}
|
||||
|
||||
func TestContainsBadPattern(t *testing.T) {
|
||||
var tw Twitter
|
||||
twitterBad := "twittercadoles"
|
||||
if tw.IsValid(twitterBad) {
|
||||
t.Errorf("The username contains twitter but isValid says it's OK !")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestTakenUsername(t *testing.T) {
|
||||
var tw Twitter
|
||||
username := "puppetmaster"
|
||||
res, err := tw.IsAvailable(username)
|
||||
if err == nil {
|
||||
t.Errorf("The user is supposed to be taken ! [%s]", err)
|
||||
}
|
||||
|
||||
if res {
|
||||
t.Errorf("The user is not supposed to be available !")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestFreeUsername(t *testing.T) {
|
||||
var tw Twitter
|
||||
username := "caseiro23"
|
||||
res, err := tw.IsAvailable(username)
|
||||
if err != nil {
|
||||
t.Errorf("The user is supposed to be free ! [%s]", err)
|
||||
}
|
||||
|
||||
if res == false {
|
||||
t.Errorf("The user is supposed to be available !")
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in New Issue