diff options
author | sjlleo <[email protected]> | 2022-05-20 17:15:06 +0800 |
---|---|---|
committer | sjlleo <[email protected]> | 2022-05-20 17:15:06 +0800 |
commit | 5002df0f2bea39784a8ed128291bab424afc681a (patch) | |
tree | 29e6f15353f580565074ea0176a9384767bf994b | |
parent | bdf72ae2308d40ca55666082f6e54891dad0c876 (diff) |
Refactor: Verify Module
-rw-r--r-- | main.go | 391 | ||||
-rw-r--r-- | verify/ipv4.go | 72 | ||||
-rw-r--r-- | verify/ipv6.go | 73 | ||||
-rw-r--r-- | verify/verify.go | 73 | ||||
-rw-r--r-- | verify/verify_test.go | 13 |
5 files changed, 242 insertions, 380 deletions
@@ -1,393 +1,24 @@ package main import ( - "crypto/tls" "flag" - "fmt" - "net" - "net/http" - "net/url" - "os" - "strconv" - "strings" - "time" -) - -const Netflix = "https://www.netflix.com/title/" -var method = flag.String("method","","模式选择(full/lite)") -var custom = flag.String("custom","","自定义测试NF影片ID\n绝命毒师的ID是70143836") -var address = flag.String("address", "", "本机公网IP") - -func RequestIP(requrl string, ip string) string { - if ip == "" { - return "Error" - } - urlValue, err := url.Parse(requrl) - if err != nil { - return "Error" - } - host := urlValue.Host - if ip == "" { - ip = host - } - newrequrl := strings.Replace(requrl, host, ip, 1) - client := &http.Client{ - Transport: &http.Transport{ - TLSClientConfig: &tls.Config{ServerName: host}, - // goodryb pull - Proxy: http.ProxyFromEnvironment, - DialContext: (&net.Dialer{ - LocalAddr: &net.TCPAddr{ - IP: net.ParseIP(*address), - }, - }).DialContext, - }, - CheckRedirect: func(req *http.Request, via []*http.Request) error { return http.ErrUseLastResponse }, - Timeout: 5 * time.Second, - } - req, err := http.NewRequest("GET", newrequrl, nil) - if err != nil { - //return errors.New(strings.ReplaceAll(err.Error(), newrequrl, requrl)) - return "Error" - } - req.Host = host - req.Header.Set("USER-AGENT", "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.108 Safari/537.36") - resp, err := client.Do(req) - if err != nil { - //return errors.New(strings.ReplaceAll(err.Error(), newrequrl, requrl)) - return "Error" - } - defer resp.Body.Close() - - Header := resp.Header - if Header["X-Robots-Tag"] != nil { - if Header["X-Robots-Tag"][0] == "index" { - return "us" - } - } - - if Header["Location"] == nil { - return "Ban" - } else { - return strings.Split(Header["Location"][0], "/")[3] - } -} - -func ParseIP(s string) int { - ip := net.ParseIP(s) - if ip == nil { - return 0 - } - for i := 0; i < len(s); i++ { - switch s[i] { - case '.': - return 4 - case ':': - return 6 - } - } - return 0 -} - -func UnblockTest(MoiveID int, ip string) bool { - testURL := Netflix + strconv.Itoa(MoiveID) - reCode := RequestIP(testURL, ip) - if strings.Contains(reCode, "Ban") { - return false - } else { - return true - } -} - -func ShellPrinter(Num int) { - switch Num { - case 0: - fmt.Println("** NetFlix 解锁检测小工具 v2.61 By \033[1;36m@sjlleo\033[0m **") - case 1: - fmt.Println("\033[0;33mNetFlix不为您测试的出口IP提供服务\033[0m") - case 2: - fmt.Println("\033[0;32mNetFlix在您测试的出口IP所在的地区提供服务,宽松版权的自制剧可以解锁\033[0m") - case 3: - fmt.Println("\033[0;36m[IPv4测试]\033[0m") - case 4: - fmt.Println("\033[0;36m[IPv6测试]\033[0m") - case 5: - fmt.Println("\033[0;32m支持解锁全部的自制剧\033[0m") - case 6: - fmt.Println("\033[0;31m不支持解锁带有强版权的自制剧\033[0m") - case 7: - fmt.Println("->> 正在检查是否完整支持自制剧 <<-") - case 8: - fmt.Println("->> 正在检查支持的NetFlix地区 <<-") - case 9: - fmt.Println("\033[0;32m支持解锁非自制剧\033[0m") - case 10: - fmt.Println("\033[0;31m不支持解锁非自制剧\033[0m") - } - -} - -func FindCountry(Code string) string { - countryName := []string{"美国", "阿富汗", "奥兰群岛", "阿尔巴尼亚", "阿尔及利亚", "美属萨摩亚", "安道尔", "安哥拉", "安圭拉", "南极洲", "安提瓜和巴布达", "阿根廷", "亚美尼亚", "阿鲁巴", "澳大利亚", "奥地利", "阿塞拜疆", "巴哈马", "巴林", "孟加拉国", "巴巴多斯", "白俄罗斯", "比利时", "伯利兹", "贝宁", "百慕大", "不丹", "玻利维亚", "波黑", "博茨瓦纳", "布维岛", "巴西", "英属印度洋领地", "文莱", "保加利亚", "布基纳法索", "布隆迪", "柬埔寨", "喀麦隆", "加拿大", "佛得角", "开曼群岛", "中非", "乍得", "智利", "中国", "圣诞岛", "科科斯(基林)群岛", "哥伦比亚", "科摩罗", "刚果(布)", "刚果(金)", "库克群岛", "哥斯达黎加", "科特迪瓦", "克罗地亚", "古巴", "塞浦路斯", "捷克", "丹麦", "吉布提", "多米尼克", "多米尼加", "厄瓜多尔", "埃及", "萨尔瓦多", "赤道几内亚", "厄立特里亚", "爱沙尼亚", "埃塞俄比亚", "福克兰群岛(马尔维纳斯)", "法罗群岛", "斐济", "芬兰", "法国", "法属圭亚那", "法属波利尼西亚", "法属南部领地", "加蓬", "冈比亚", "格鲁吉亚", "德国", "加纳", "直布罗陀", "希腊", "格陵兰", "格林纳达", "瓜德罗普", "关岛", "危地马拉", "格恩西岛", "几内亚", "几内亚比绍", "圭亚那", "海地", "赫德岛和麦克唐纳岛", "梵蒂冈", "洪都拉斯", "香港", "匈牙利", "冰岛", "印度", "印度尼西亚", "伊朗", "伊拉克", "爱尔兰", "英国属地曼岛", "以色列", "意大利", "牙买加", "日本", "泽西岛", "约旦", "哈萨克斯坦", "肯尼亚", "基里巴斯", "朝鲜", "韩国", "科威特", "吉尔吉斯斯坦", "老挝", "拉脱维亚", "黎巴嫩", "莱索托", "利比里亚", "利比亚", "列支敦士登", "立陶宛", "卢森堡", "澳门", "前南马其顿", "马达加斯加", "马拉维", "马来西亚", "马尔代夫", "马里", "马耳他", "马绍尔群岛", "马提尼克", "毛利塔尼亚", "毛里求斯", "马约特", "墨西哥", "密克罗尼西亚联邦", "摩尔多瓦", "摩纳哥", "蒙古", "黑山", "蒙特塞拉特", "摩洛哥", "莫桑比克", "缅甸", "纳米比亚", "瑙鲁", "尼泊尔", "荷兰", "荷属安的列斯", "新喀里多尼亚", "新西兰", "尼加拉瓜", "尼日尔", "尼日利亚", "纽埃", "诺福克岛", "北马里亚纳", "挪威", "阿曼", "巴基斯坦", "帕劳", "巴勒斯坦", "巴拿马", "巴布亚新几内亚", "巴拉圭", "秘鲁", "菲律宾", "皮特凯恩", "波兰", "葡萄牙", "波多黎各", "卡塔尔", "留尼汪", "罗马尼亚", "俄罗斯联邦", "卢旺达", "圣赫勒拿", "圣基茨和尼维斯", "圣卢西亚", "圣皮埃尔和密克隆", "圣文森特和格林纳丁斯", "萨摩亚", "圣马力诺", "圣多美和普林西比", "沙特阿拉伯", "塞内加尔", "塞尔维亚", "塞舌尔", "塞拉利昂", "新加坡", "斯洛伐克", "斯洛文尼亚", "所罗门群岛", "索马里", "南非", "南乔治亚岛和南桑德韦奇岛", "西班牙", "斯里兰卡", "苏丹", "苏里南", "斯瓦尔巴岛和扬马延岛", "斯威士兰", "瑞典", "瑞士", "叙利亚", "台湾", "塔吉克斯坦", "坦桑尼亚", "泰国", "东帝汶", "多哥", "托克劳", "汤加", "特立尼达和多巴哥", "突尼斯", "土耳其", "土库曼斯坦", "特克斯和凯科斯群岛", "图瓦卢", "乌干达", "乌克兰", "阿联酋", "英国", "美国本土外小岛屿", "乌拉圭", "乌兹别克斯坦", "瓦努阿图", "委内瑞拉", "越南", "英属维尔京群岛", "美属维尔京群岛", "瓦利斯和富图纳", "西撒哈拉", "也门", "赞比亚", "津巴布韦"} - countryCode := []string{"us", "af", "ax", "al", "dz", "as", "ad", "ao", "ai", "aq", "ag", "ar", "am", "aw", "au", "at", "az", "bs", "bh", "bd", "bb", "by", "be", "bz", "bj", "bm", "bt", "bo", "ba", "bw", "bv", "br", "io", "bn", "bg", "bf", "bi", "kh", "cm", "ca", "cv", "ky", "cf", "td", "cl", "cn", "cx", "cc", "co", "km", "cg", "cd", "ck", "cr", "ci", "hr", "cu", "cy", "cz", "dk", "dj", "dm", "do", "ec", "eg", "sv", "gq", "er", "ee", "et", "fk", "fo", "fj", "fi", "fr", "gf", "pf", "tf", "ga", "gm", "ge", "de", "gh", "gi", "gr", "gl", "gd", "gp", "gu", "gt", "gg", "gn", "gw", "gy", "ht", "hm", "va", "hn", "hk", "hu", "is", "in", "id", "ir", "iq", "ie", "im", "il", "it", "jm", "jp", "je", "jo", "kz", "ke", "ki", "kp", "kr", "kw", "kg", "la", "lv", "lb", "ls", "lr", "ly", "li", "lt", "lu", "mo", "mk", "mg", "mw", "my", "mv", "ml", "mt", "mh", "mq", "mr", "mu", "yt", "mx", "fm", "md", "mc", "mn", "me", "ms", "ma", "mz", "mm", "na", "nr", "np", "nl", "an", "nc", "nz", "ni", "ne", "ng", "nu", "nf", "mp", "no", "om", "pk", "pw", "ps", "pa", "pg", "py", "pe", "ph", "pn", "pl", "pt", "pr", "qa", "re", "ro", "ru", "rw", "sh", "kn", "lc", "pm", "vc", "ws", "sm", "st", "sa", "sn", "rs", "sc", "sl", "sg", "sk", "si", "sb", "so", "za", "gs", "es", "lk", "sd", "sr", "sj", "sz", "se", "ch", "sy", "tw", "tj", "tz", "th", "tl", "tg", "tk", "to", "tt", "tn", "tr", "tm", "tc", "tv", "ug", "ua", "ae", "gb", "um", "uy", "uz", "vu", "ve", "vn", "vg", "vi", "wf", "eh", "ye", "zm", "zw"} - for i, v := range countryCode { - if strings.Contains(Code, v) { - return countryName[i] - } - } - return Code -} - -func IsNumeric(val interface{}) bool { - switch val.(type) { - case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64: - case float32, float64, complex64, complex128: - return true - case string: - str := val.(string) - if str == "" { - return false - } - // Trim any whitespace - str = strings.Trim(str, " \\t\\n\\r\\v\\f") - if str[0] == '-' || str[0] == '+' { - if len(str) == 1 { - return false - } - str = str[1:] - } - // hex - if len(str) > 2 && str[0] == '0' && (str[1] == 'x' || str[1] == 'X') { - for _, h := range str[2:] { - if !((h >= '0' && h <= '9') || (h >= 'a' && h <= 'f') || (h >= 'A' && h <= 'F')) { - return false - } - } - return true - } - // 0-9,Point,Scientific - p, s, l := 0, 0, len(str) - for i, v := range str { - if v == '.' { // Point - if p > 0 || s > 0 || i+1 == l { - return false - } - p = i - } else if v == 'e' || v == 'E' { // Scientific - if i == 0 || s > 0 || i+1 == l { - return false - } - s = i - } else if v < '0' || v > '9' { - return false - } - } - return true - } + "github.com/sjlleo/netflix-verify/printer" + "github.com/sjlleo/netflix-verify/verify" +) - return false -} +// var method = flag.String("method", "", "模式选择(full/lite)") +var custom = flag.String("custom", "", "自定义测试NF影片ID\n绝命毒师的ID是70143836") +var address = flag.String("address", "", "本机网卡的IP") func main() { - var ipv4, ipv6 string - var NextLineSignal bool = false - - var areaAvailableID = 80018499 - var SelfMadeAvailableID = 80197526 - var NonSelfMadeAvailableID = 70143836 - dns := "www.netflix.com" - flag.Parse() - - - // 解析ip地址 - ns, err := net.LookupHost(dns) - if err != nil { - fmt.Fprintf(os.Stderr, "Err: %s", err.Error()) - return - } - - switch { - case len(ns) != 0: - for _, n := range ns { - - if ParseIP(n) == 4 { - ipv4 = n - } - if ParseIP(n) == 6 { - ipv6 = "[" + n + "]" - } - - } - - } - - ShellPrinter(0) - - if (*method == "full") { - fmt.Println("\033[0;35m模式:详细信息模式\033[0m") - } else if (*custom == "") { - fmt.Println("\033[0;35m模式:简洁信息模式\033[0m") - } else { - fmt.Println("\033[0;35m模式:自定义影片测试模式\033[0m") - } - - // 拼接非自制剧的URL - testURL := Netflix + strconv.Itoa(NonSelfMadeAvailableID) - ipv4CountryCode := RequestIP(testURL, ipv4) - ipv6CountryCode := RequestIP(testURL, ipv6) - - /*** - * 检查CountryCode返回值: - * Error 代表该网络访问失败 - * Ban 代表无法解锁这个ID种类的影片 - * 此处如果显示值不为Error则都应该继续检测 - ***/ - if !strings.Contains(ipv4CountryCode, "Error") { - //开启换行信号,在IPv4检测完毕后换行 - NextLineSignal = true - ShellPrinter(3) - //检测是否为自定义测试模式 - if (*custom != "") { - if (IsNumeric(*custom) == false) { - fmt.Println("\033[0;34m您输入的不是数字!\033[0m") - return - } else { - MovieID,_ := strconv.Atoi(*custom) - if(UnblockTest(MovieID, ipv4)) { - fmt.Println("\033[0;32m可以解锁此影片\033[0m") - } else { - fmt.Println("\033[0;31m不能解锁此影片\033[0m") - } - } - - } else { - //如果反馈为Ban,那么进一步检测是否支持Netflix地区解锁 - if strings.Contains(ipv4CountryCode, "Ban") { - //检测该IP所在的地区是否支持NF - if UnblockTest(areaAvailableID, ipv4) { - //所在地区支持NF - if (*method == "full") { - ShellPrinter(2) - ShellPrinter(7) - } - //检测是否支持自制 - if UnblockTest(SelfMadeAvailableID, ipv4) { - //支持自制剧 - if (*method == "full"){ - ShellPrinter(5) - ShellPrinter(8) - ShellPrinter(10) - fmt.Println("\n\033[1;34m判断结果:不支持Netflix解锁") - testURL2 := Netflix + strconv.Itoa(SelfMadeAvailableID) - ipv4CountryCode2 := RequestIP(testURL2, ipv4) - fmt.Println("\033[0;36mNF库识别的IP地域信息:\033[1;36m" + FindCountry(ipv4CountryCode2) + "区("+ strings.ToUpper(strings.Split(ipv4CountryCode2,"-")[0]) +") NetFlix 非原生IP\033[0m") - } else { - fmt.Println("\033[0;33m您的出口IP不能解锁Netflix,仅支持自制剧的观看\033[0m") - } - } else { - //不支持自制剧 - ShellPrinter(6) - } - } else { - //所在地区不支持NF - ShellPrinter(1) - } - - } else { - //如果支持非自制剧的解锁,则直接跳过自制剧的解锁 - if (*method == "full") { - ShellPrinter(2) - ShellPrinter(7) - ShellPrinter(5) - ShellPrinter(8) - ShellPrinter(9) - fmt.Println("\n\033[1;34m判断结果:完整支持Netflix解锁") - } else { - fmt.Println("\033[0;32m您的出口IP完整解锁Netflix,支持非自制剧的观看\033[0m") - } - fmt.Println("\033[0;36m原生IP地域解锁信息:\033[1;36m" + FindCountry(ipv4CountryCode) + "区("+ strings.ToUpper(strings.Split(ipv4CountryCode,"-")[0]) +") NetFlix 原生IP\033[0m") - } - } - } - if !strings.Contains(ipv6CountryCode, "Error") { - //如果存在在IPv4检测,那在其完毕后换行 - if NextLineSignal { - fmt.Print("\n") - } - ShellPrinter(4) - //判断是否为自定义检测状态 - if (*custom != "") { - if (IsNumeric(*custom) == false) { - fmt.Println("\033[0;34m您输入的不是数字!\033[0m") - return - } else { - MovieID,_ := strconv.Atoi(*custom) - if(UnblockTest(MovieID, ipv6)) { - fmt.Println("\033[0;32m可以解锁此影片\033[0m") - } else { - fmt.Println("\033[0;31m不能解锁此影片\033[0m") - } - } - return - } - //如果反馈为Ban,那么进一步检测是否支持Netflix地区解锁 - if strings.Contains(ipv6CountryCode, "Ban") { - //检测该IP所在的地区是否支持NF - if UnblockTest(areaAvailableID, ipv6) { - //所在地区支持NF - if (*method == "full") { - ShellPrinter(2) - ShellPrinter(7) - } - //检测是否支持自制 - if UnblockTest(SelfMadeAvailableID, ipv6) { - //支持自制剧 - if (*method == "full"){ - ShellPrinter(5) - ShellPrinter(8) - ShellPrinter(10) - fmt.Println("\n\033[1;34m判断结果:不支持Netflix解锁") - testURL62 := Netflix + strconv.Itoa(SelfMadeAvailableID) - ipv6CountryCode2 := RequestIP(testURL62, ipv6) - fmt.Println("\033[0;36mNF库识别的IP地域信息:\033[1;36m" + FindCountry(ipv6CountryCode2) + "区("+ strings.ToUpper(strings.Split(ipv6CountryCode2,"-")[0]) +") NetFlix 非原生IP\033[0m") - } else { - fmt.Println("\033[0;33m您的出口IP不能解锁Netflix,仅支持自制剧的观看\033[0m") - } - } else { - //不支持自制剧 - ShellPrinter(6) - } - } else { - //所在地区不支持NF - ShellPrinter(1) - } + r := verify.NewVerify(verify.Config{ + LocalAddr: *address, + Custom: *custom, + }) - } else { - //如果支持非自制剧的解锁,则直接跳过自制剧的解锁 - if (*method == "full") { - ShellPrinter(2) - ShellPrinter(7) - ShellPrinter(5) - ShellPrinter(8) - ShellPrinter(9) - fmt.Println("\n\033[0;34m判断结果:完整支持Netflix解锁") - } else { - fmt.Println("\033[0;32m您的出口IP完整解锁Netflix,支持非自制剧的观看\033[0m") - } - fmt.Println("\033[0;36m原生IP地域解锁信息:\033[1;36m" + FindCountry(ipv6CountryCode) + "区("+ strings.ToUpper(strings.Split(ipv6CountryCode,"-")[0]) +") NetFlix 原生IP\033[0m") - } - } else { - if (*method == "full"){ - if NextLineSignal { - fmt.Print("\n") - } - ShellPrinter(4) - fmt.Println("\033[0;31m本机不支持IPv6的访问\033[0m") - } - } - fmt.Println("\n\033[0;36m感谢每一个正在使用本脚本的你,祝解锁NetFlix!\033[0m") + printer.Print(*r) } diff --git a/verify/ipv4.go b/verify/ipv4.go new file mode 100644 index 0000000..89e0cc9 --- /dev/null +++ b/verify/ipv4.go @@ -0,0 +1,72 @@ +package verify + +import ( + "strconv" + + "github.com/sjlleo/netflix-verify/util" +) + +type IPv4Verifier struct { + Config + IP string + unblockStatus int + unblockTestChan chan UnblockTestResult +} + +func (v *IPv4Verifier) Execute() *VerifyResponse { + var err error + var response VerifyResponse + v.unblockStatus = AreaUnavailable + response.Type = 1 + + if v.IP, err = util.DnsResolver(4); err != nil { + return &VerifyResponse{} + } + + v.unblockTestChan = make(chan UnblockTestResult) + defer close(v.unblockTestChan) + + go v.UnblockTest(AreaAvailableID) + go v.UnblockTest(SelfMadeAvailableID) + go v.UnblockTest(NonSelfMadeAvailableID) + + for i := 0; i < 3; i++ { + switch res := <-v.unblockTestChan; { + + case res.CountryCode != "": + switch res.movieID { + case AreaAvailableID: + v.upgradeStatus(AreaAvailable) + case SelfMadeAvailableID: + v.upgradeStatus(UnblockSelfMadeMovie) + case NonSelfMadeAvailableID: + v.upgradeStatus(UnblockNonSelfMadeMovie) + } + response.CountryCode = res.CountryCode + response.CountryName = util.CountryCodeToCountryName(res.CountryCode) + default: + } + } + response.StatusCode = v.unblockStatus + return &response +} + +func (v *IPv4Verifier) upgradeStatus(status int) { + if v.unblockStatus < status { + v.unblockStatus = status + } +} + +func (v *IPv4Verifier) UnblockTest(MoiveID int) { + + testURL := util.Netflix + strconv.Itoa(MoiveID) + if reCode, err := util.RequestIP(testURL, v.IP, v.LocalAddr); err != nil { + if err.Error() == "Banned" { + v.unblockTestChan <- UnblockTestResult{MoiveID, "", nil} + } else { + v.unblockTestChan <- UnblockTestResult{MoiveID, "", err} + } + } else { + v.unblockTestChan <- UnblockTestResult{MoiveID, reCode, nil} + } +} diff --git a/verify/ipv6.go b/verify/ipv6.go new file mode 100644 index 0000000..07ff625 --- /dev/null +++ b/verify/ipv6.go @@ -0,0 +1,73 @@ +package verify + +import ( + "strconv" + + "github.com/sjlleo/netflix-verify/util" +) + +type IPv6Verifier struct { + Config + IP string + unblockStatus int + unblockTestChan chan UnblockTestResult +} + +func (v *IPv6Verifier) Execute() *VerifyResponse { + var err error + var response VerifyResponse + v.unblockStatus = AreaUnavailable + response.Type = 2 + + if v.IP, err = util.DnsResolver(6); err != nil { + response.StatusCode = -2 + return &response + } + + v.unblockTestChan = make(chan UnblockTestResult) + defer close(v.unblockTestChan) + + go v.UnblockTest(AreaAvailableID) + go v.UnblockTest(SelfMadeAvailableID) + go v.UnblockTest(NonSelfMadeAvailableID) + + for i := 0; i < 3; i++ { + switch res := <-v.unblockTestChan; { + + case res.CountryCode != "": + switch res.movieID { + case AreaAvailableID: + v.upgradeStatus(AreaAvailable) + case SelfMadeAvailableID: + v.upgradeStatus(UnblockSelfMadeMovie) + case NonSelfMadeAvailableID: + v.upgradeStatus(UnblockNonSelfMadeMovie) + } + response.CountryCode = res.CountryCode + response.CountryName = util.CountryCodeToCountryName(res.CountryCode) + default: + } + } + response.StatusCode = v.unblockStatus + return &response +} + +func (v *IPv6Verifier) upgradeStatus(status int) { + if v.unblockStatus < status { + v.unblockStatus = status + } +} + +func (v *IPv6Verifier) UnblockTest(MoiveID int) { + + testURL := util.Netflix + strconv.Itoa(MoiveID) + if reCode, err := util.RequestIP(testURL, v.IP, v.LocalAddr); err != nil { + if err.Error() == "Banned" { + v.unblockTestChan <- UnblockTestResult{MoiveID, "", nil} + } else { + v.unblockTestChan <- UnblockTestResult{MoiveID, "", err} + } + } else { + v.unblockTestChan <- UnblockTestResult{MoiveID, reCode, nil} + } +} diff --git a/verify/verify.go b/verify/verify.go new file mode 100644 index 0000000..a657f3d --- /dev/null +++ b/verify/verify.go @@ -0,0 +1,73 @@ +package verify + +var AreaAvailableID = 80018499 +var SelfMadeAvailableID = 80197526 +var NonSelfMadeAvailableID = 70143836 + +const ( + AreaUnavailable = -1 + AreaAvailable = 0 + UnblockSelfMadeMovie = 1 + UnblockNonSelfMadeMovie = 2 +) + +type Verify interface { + Execute() *VerifyResponse +} + +type FinalResult struct { + Res map[int]VerifyResponse +} + +type VerifyResult struct { + Config + response chan *VerifyResponse +} + +type Config struct { + LocalAddr string + Custom string +} + +type VerifyResponse struct { + Type int + StatusCode int + CountryCode string + CountryName string +} + +type UnblockTestResult struct { + movieID int + CountryCode string + err error +} + +func NewVerify(c Config) *FinalResult { + var finalResult FinalResult + finalResult.Res = make(map[int]VerifyResponse) + + vr := VerifyResult{Config: c} + ch := make(chan *VerifyResponse) + vr.response = ch + defer close(ch) + + go vr.IPv4Verifier() + go vr.IPv6Verifier() + + for i := 0; i < 2; i++ { + if res, err := <-ch; err { + finalResult.Res[res.Type] = *res + } + } + return &finalResult +} + +func (vr *VerifyResult) IPv4Verifier() { + verify := &IPv4Verifier{Config: vr.Config} + vr.response <- verify.Execute() +} + +func (vr *VerifyResult) IPv6Verifier() { + verify := &IPv6Verifier{Config: vr.Config} + vr.response <- verify.Execute() +} diff --git a/verify/verify_test.go b/verify/verify_test.go new file mode 100644 index 0000000..6f334d4 --- /dev/null +++ b/verify/verify_test.go @@ -0,0 +1,13 @@ +package verify + +import "testing" + +func TestIPv4Execute(t *testing.T) { + verify := &IPv4Verifier{Config: Config{}} + verify.Execute() +} + +func TestIPv6Execute(t *testing.T) { + verify := &IPv6Verifier{Config: Config{}} + verify.Execute() +} |