First commit

This commit is contained in:
Philippe Caseiro 2019-12-04 16:58:38 +01:00
commit 87d4ab0490
8 changed files with 402 additions and 0 deletions

60
cmd/cli/main.go Normal file
View File

@ -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)
}
}

70
cmd/server/main.go Normal file
View File

@ -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))
}

21
errors.go Normal file
View File

@ -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
}

19
namecheck.go Normal file
View File

@ -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
}

22
twitter/coverprofile.tmp Normal file
View File

@ -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

3
twitter/go.mod Normal file
View File

@ -0,0 +1,3 @@
module forge.cadoles.com/pcaseiro/namecheck
go 1.13

92
twitter/twitter.go Normal file
View File

@ -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")
}

115
twitter/twitter_test.go Normal file
View File

@ -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 !")
}
}