コード例 #1
0
ファイル: main.go プロジェクト: andresvia/mtproto
func main() {
	var err error

	if len(os.Args) < 2 {
		usage()
		os.Exit(1)
	}

	commands := map[string]int{"auth": 1, "msg": 2, "list": 0}
	valid := false
	for k, v := range commands {
		if os.Args[1] == k {
			if len(os.Args) < v+2 {
				usage()
				os.Exit(1)
			}
			valid = true
			break
		}
	}

	if !valid {
		usage()
		os.Exit(1)
	}

	m, err := mtproto.NewMTProto(os.Getenv("HOME") + "/.telegram_go")
	if err != nil {
		fmt.Printf("Create failed: %s\n", err)
		os.Exit(2)
	}

	err = m.Connect()
	if err != nil {
		fmt.Printf("Connect failed: %s\n", err)
		os.Exit(2)
	}

	switch os.Args[1] {
	case "auth":
		err = m.Auth(os.Args[2])
	case "msg":
		user_id, _ := strconv.Atoi(os.Args[2])
		err = m.SendMsg(int32(user_id), os.Args[3])

	case "list":
		err = m.GetContacts()
	}

	if err != nil {
		fmt.Println(err)
		os.Exit(2)
	}
}
コード例 #2
0
ファイル: main.go プロジェクト: hugozhu/mtproto
func main() {
	var err error

	if len(os.Args) < 2 {
		usage()
		os.Exit(1)
	}

	valid := false
	for k, v := range commands {
		if os.Args[1] == k {
			if len(os.Args) < v+2 {
				usage()
				os.Exit(1)
			}
			valid = true
			break
		}
	}

	if !valid {
		usage()
		os.Exit(1)
	}

	m, err := mtproto.NewMTProto(os.Getenv("HOME") + "/.telegram_go")
	if err != nil {
		fmt.Printf("Create failed: %s\n", err)
		os.Exit(2)
	}

	err = m.Connect()
	if err != nil {
		fmt.Printf("Connect failed: %s\n", err)
		os.Exit(2)
	}
	switch os.Args[1] {
	case "auth":
		err = m.Auth(os.Args[2])
	case "msg":
		err = m.SendMsg(os.Args[2], os.Args[3])
	case "list":
		err = m.GetContacts()
	case "dialogs":
		err = m.GetChats()
	case "sendmedia":
		err = m.SendMedia(os.Args[2], os.Args[3])
	}

	if err != nil {
		fmt.Println(err)
		os.Exit(2)
	}
}