Premier commit

twitter/twitter.go

@@ -0,0 +1,71 @@
+package twitter
+
+import (
+	"net/http"
+	"regexp"
+
+	"github.com/foxdeveloper/namecheck"
+	"github.com/foxdeveloper/namecheck/util"
+)
+
+const (
+	EndpointAPI            string = "https://twitter.com/"
+	AvailableHTTPCode      int    = http.StatusNotFound
+	UsernameShort          int    = 1
+	UsernameLong           int    = 15
+	UsernameLegalPattern   string = "^[A-Za-z0-9_]+$"
+	UsernameIllegalPattern string = "twitter"
+)
+
+var (
+	legalRegexp   = regexp.MustCompile(UsernameLegalPattern)
+	illegalRegexp = regexp.MustCompile(UsernameIllegalPattern)
+)
+
+type Twitter struct{}
+
+func init() {
+	namecheck.AddSn(&Twitter{})
+}
+
+func (t *Twitter) String() string {
+	return "Twitter"
+}
+
+// IsValid check the validity of the username
+func (t *Twitter) IsValid(username string) bool {
+	return isLongEnough(username) &&
+		isShortEnough(username) &&
+		containsNoIllegalPattern(username) &&
+		containsOnlyLegalChars(username)
+}
+
+// IsAvailable check username availability
+func (t *Twitter) IsAvailable(username string) (bool, error) {
+	res, err := http.Get(EndpointAPI + username)
+	if err != nil {
+		myErr := &namecheck.ErrUnknownAvailability{Username: username, Cause: err}
+		return false, myErr
+	}
+
+	// Prevent memory leaks
+	defer res.Body.Close()
+
+	return res.StatusCode == AvailableHTTPCode, nil
+}
+
+func isLongEnough(username string) bool {
+	return util.Counter(username) >= UsernameShort
+}
+
+func isShortEnough(username string) bool {
+	return util.Counter(username) <= UsernameLong
+}
+
+func containsOnlyLegalChars(username string) bool {
+	return legalRegexp.MatchString(util.ToLower(username))
+}
+
+func containsNoIllegalPattern(username string) bool {
+	return len(illegalRegexp.FindStringIndex(util.ToLower(username))) == 0
+}

twitter/twitter_test.go

@@ -0,0 +1,69 @@
+package twitter_test
+
+import (
+	"github.com/foxdeveloper/namecheck/twitter"
+	"testing"
+)
+
+var (
+	tw twitter.Twitter = twitter.Twitter{}
+)
+
+func TestName(t *testing.T) {
+	want := "Twitter"
+	got := tw.String()
+
+	if got != want {
+		t.Errorf("twitter.Name() = %s; want %s", got, want)
+	}
+
+}
+func TestUsernameTooShort(t *testing.T) {
+	username := ""
+	want := false
+	got := tw.IsValid(username)
+
+	if got != want {
+		t.Errorf("twitter.IsValid(%s) = %t; want %t", username, got, want)
+	}
+}
+func TestUsernameTooLong(t *testing.T) {
+	username := "azetrgdkalakdzjazdzaidadhazdazidazdazmdazldkazda"
+	want := false
+	got := tw.IsValid(username)
+
+	if got != want {
+		t.Errorf("twitter.IsValid(%s) = %t; want %t", username, got, want)
+	}
+}
+func TestUsernameContainsIllegalPattern(t *testing.T) {
+	username := "psetwiTterdo"
+	want := false
+	got := tw.IsValid(username)
+
+	if got != want {
+		t.Errorf("twitter.IsValid(%s) = %t; want %t", username, got, want)
+	}
+}
+func TestUsernameContainsNotOnlyLegalPattern(t *testing.T) {
+	username := "pseudo-"
+	want := false
+	got := tw.IsValid(username)
+
+	if got != want {
+		t.Errorf("twitter.IsValid(%s) = %t; want %t", username, got, want)
+	}
+}
+func TestUsernameAvailablePattern(t *testing.T) {
+	username := "jubobs"
+	want := false
+	got, err := tw.IsAvailable(username)
+
+	if err != nil {
+		t.Errorf("%v", err)
+	}
+
+	if got != want {
+		t.Errorf("twitter.IsValid(%s) = %t; want %t", username, got, want)
+	}
+}