UserStream取得

TwitterのUserStreamを取得して、どんどんとmesしていくスクリプト。
hspextのpipeexec命令でも使って、別プロセスとして動かすことを想定して作りました。
固まることが多々あります。だからpipeを使えってことなんですけども。
cookies146
2011/9/15 22:14
タグ：
hsp
userstream
twitter
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
#runtime "hsp3cl"
#packopt name "USA"
#module
#defcfunc strCnt str p1, str p2
    string = p1
    count=0 : i=-1
    repeat
        i = instr(string, i+1, p2)
        if i=-1 : break
        count++
    loop
    logmes "21"
return count
#deffunc getBlockStr str p1, int p2, int p3, var block, var other
    string = p1
    count= 0 : i=0
    sdim p2_s
    poke p2_s, 0, p2
    head=instr(string, 0, p2_s)
    if head=-1 : block="" : other=string : return block
    
    repeat ,head
        if peek(string, cnt)=$00 : return "-1"
        if peek(string, cnt)=p2 : count++
        if peek(string, cnt)=p3 : count--
        if count=0 : break
        i++
    loop
    
    block=strmid(string, head, i+1)
    other=strmid(string, head+i+1, strlen(string))
return block
#global
#include "TsubuyakiSoup.as"
#module UserStream
#uselib "crtdll.dll"
#func time "time" var
#uselib "wininet.dll"
#cfunc InternetOpen "InternetOpenA" sptr, int, sptr, sptr, int
#cfunc InternetOpenUrl "InternetOpenUrlA" int, str, sptr, int, int, int
#func InternetReadFile "InternetReadFile" int, var, int, var
#func InternetCloseHandle "InternetCloseHandle" int
#cfunc InternetConnect "InternetConnectA" int, str, int, sptr, sptr, int, int, int
#cfunc HttpOpenRequest "HttpOpenRequestA" int, sptr, str, sptr, sptr, sptr, int, int
#cfunc HttpSendRequest "HttpSendRequestA" int, sptr, int, sptr, int
#cfunc HttpQueryInfo "HttpQueryInfoA" int, int, var, var, int
#func InternetQueryDataAvailable "InternetQueryDataAvailable" int, int, int, int
#func InternetSetOption "InternetSetOptionA" int, int, int, int
#define INTERNET_FLAG_IGNORE_CERT_DATE_INVALID 0x2000
#define INTERNET_FLAG_IGNORE_CERT_CN_INVALID 0x1000
//------------------------------------------------------------
//OAuth認証を含めた、UserStreamの初期化
#deffunc UserStreamInit var p1, var p2, int p3, str p4, array p5
//  引数チェック＆初期化
    sdim p1
    sdim p2
    API = p4
    if vartype(p5) != 2 : return 0
    hConnect = 0        //InternetConnectのハンドル
    hRequest = 0        //HttpOpenRequestのハンドル
    API_statcode = 0    //リクエストの結果コード
    API_p1Length = 0    //データ長
    API_rsize = 1024    //バッファ初期値
    API_hsize = 0       //取得したバイト数が代入される変数
    
    dup TS_AccessToken,TS_AccessToken@TsubuyakiSoup
    dup TS_Consumer_Key, TS_Consumer_Key@TsubuyakiSoup
    dup TS_Consumer_Secret,TS_Consumer_Secret@TsubuyakiSoup
    dup TS_AccessTokenSecret, TS_AccessTokenSecret@TsubuyakiSoup
    dup TS_RequestToken, TS_RequestToken@TsubuyakiSoup
    dup TS_RequestTokenSecret, TS_RequestTokenSecret@TsubuyakiSoup
//  メソッドの設定
    if (p3 = 1) {
        Method = "POST"
    } else {
        Method = "GET"
    }
//  ポート＆フラグの設定
    UsePort = 443 : RequestFlag = -2147483648 | 0x800000 | INTERNET_FLAG_IGNORE_CERT_DATE_INVALID |INTERNET_FLAG_IGNORE_CERT_CN_INVALID : Scheme = "https"
    VersionStr = "2/"
    TokenStr = TS_AccessToken
    SigKey = TS_Consumer_Secret+" "+TS_AccessTokenSecret
    if (strmid(API,0,5) = "oauth") {
        VersionStr = ""
        if (API = "oauth/access_token") {
            repeat length(p5)
                if (p5(cnt) = "x_auth_mode=client_auth") : UsePort = 443 : RequestFlag = -2139082752 : Scheme = "https" : break
                if cnt = length(p5)-1 : TokenStr = TS_RequestToken : SigKey = TS_Consumer_Secret+" "+TS_RequestTokenSecret
            loop
        }
    }
//  シグネチャ生成
    SigArrayMax = 6 + length(p5)
    sdim SigArray, 500, SigArrayMax
    SigNonce = RandomString(8,32)
    time SigTime
    SigArray(0) = "oauth_consumer_key=" + TS_Consumer_Key
    SigArray(1) = "oauth_nonce=" + SigNonce
    SigArray(2) = "oauth_signature_method=HMAC-SHA1"
    SigArray(3) = "oauth_timestamp=" + SigTime
    SigArray(4) = "oauth_token="+ TokenStr
    SigArray(5) = "oauth_version=1.0"
    repeat SigArrayMax - 6
        SigArray(6+cnt) = p5(cnt)
    loop
    //ソート
    SortString SigArray
    //"&"で連結
    TransStr = ""+ Method +" "+ Scheme +"://userstream.twitter.com/"+ VersionStr + API +" "
    repeat SigArrayMax
        if SigArray(cnt) = "" : continue
        TransStr += SigArray(cnt) +"&"
    loop
    TransStr = strmid(TransStr, 0, strlen(TransStr)-1)
    Signature = SignatureEncode(TransStr, SigKey)
//  データ整形
    if (p3 = 1) {
        //POST
        PostStr = ""
        repeat SigArrayMax
            PostStr += SigArray(cnt) +"&"
        loop
        PostStr += "oauth_signature="+ Signature
        PostStrLen = strlen(PostStr)
        AddUrl = ""
    } else {
        //GET
        PostStr = 0
        PostStrLen = 0
        AddUrl = "?"
        repeat SigArrayMax
            AddUrl += SigArray(cnt) +"&"
        loop
        AddUrl += "oauth_signature="+ Signature
    }
    logmes TransStr
    //サーバへ接続
    hConnect = InternetConnect(TwiH_hInet@TsubuyakiSoup, "userstream.twitter.com", UsePort, 0, 0, 3, 0, 0)
    if (hConnect) {
        //リクエストの初期化
        hRequest = HttpOpenRequest(hConnect, Method, VersionStr+API+AddUrl, "HTTP/1.1", 0, 0, RequestFlag, 0)
        if (hRequest) {
            logmes VersionStr+API+AddUrl
            //サーバへリクエスト送信
            if ( HttpSendRequest(hRequest, 0, 0, PostStr, PostStrLen)) {
                API_statcode = 0
            } else {
                //サーバへリクエスト送信できなかった場合
                API_statcode = -2
            }
        } else {
            //Requestハンドルを取得できなかった場合
            API_statcode = -3
        }
        } else {
        //Connectハンドルを取得できなかった場合
        API_statcode = -4
    }
return API_statcode
//適当な部分もある、UserStreamのヘッダ取得
#deffunc getHead_US var p2
                //ヘッダを取得する変数の初期化
                p2Size = 3000
                sdim p2, p2Size
                //ヘッダの取得
                if ( HttpQueryInfo(hRequest, 22, p2, p2Size, 0) ) {
                    //ヘッダの解析
                    notesel p2
                    repeat notemax
                        noteget API_BufStr, cnt
                        API_buf = instr(API_BufStr, 0, "Status: ")              //ステータスコード
                        if (API_Buf != -1) : API_statcode = int(strmid(API_BufStr, API_buf+8, 3))
                        API_buf = instr(API_BufStr, 0, "Content-Length: ")      //長さ
                        if (API_Buf != -1) : API_p1Length = int(strmid(API_BufStr, -1, strlen(API_BufStr)-API_buf+16))
                        API_buf = instr(API_BufStr, 0, "X-RateLimit-Limit: ")       //60分間にAPIを実行できる回数
                        if (API_Buf != -1) : TS_RateLimit(0) = int(strmid(API_BufStr, -1, strlen(API_BufStr)-(API_buf+19)))
                        API_buf = instr(API_BufStr, 0, "X-RateLimit-Remaining: ")   //APIを実行できる残り回数
                        if (API_Buf != -1) : TS_RateLimit(1) = int(strmid(API_BufStr, -1, strlen(API_BufStr)-(API_buf+23)))
                        API_buf = instr(API_BufStr, 0, "X-RateLimit-Reset: ")       //リセットする時間
                        if (API_Buf != -1) : TS_RateLimit(2) = int(strmid(API_BufStr, -1, strlen(API_BufStr)-(API_buf+19)))
                    loop
                    noteunsel
                } else {
                    //ヘッダの取得ができなかった場合
                    API_statcode = -1
                }
return API_statcode
//適当な部分もある、UserStreamのデータ取得
#deffunc getBody_US var p1
                wait 10
                p2Size = 3000
                sdim p2, p2Size
                    //入手可能なデータ量を取得
                    InternetQueryDataAvailable hRequest, varptr(API_rsize), 0, 0
                    //バッファの初期化
                    sdim API_bufStr, API_rsize+1
                    sdim p1, API_rsize+1
                    repeat
                        InternetReadFile hRequest, API_bufStr, API_rsize, API_hsize
                        p1 += strmid(API_bufStr, 0, API_hsize)
                        wait 10
                        if (API_hsize = 1)||(API_hsize!API_rsize)||(API_hsize=0) : break 
                    loop
return API_statcode
#global
#uselib "KERNEL32.DLL"
#cfunc OpenProcess "OpenProcess" int,int,int
#func GetExitCodeProcess "GetExitCodeProcess" int,int
#define PROCESS_QUERY_INFORMATION $400  //OpenProcess時に使う。
//コマンドライン解析用。http://lldev.jp/hsp/tips/miscellaneous.html#cmdlineprmsからＤＬしてください。
#include "cmdline_parser/gm_cmdline_parser.hsp"
;サンプル用設定ファイル
#include "SampleSetting.hsp"
if CONSUMER_KEY = "" : mes "「SampleSetting.hsp」でConsumerKeyとConsumerSecretを設定してください。" : end
    if getarg(0,1)="" : mes "USA: 引数が不正です。" : end       //引数が一つもない状態
    
    // 初期化
    TS_Init "TsubuyakiSoup Sample", CONSUMER_KEY, CONSUMER_SECRET, 20
    // 設定ファイルの有無確認
    xToken = ""
    xSecret = ""
    exist "Setting.txt"
    if strsize != -1 {
        notesel SFA
        noteload "Setting.txt"
        if (Decryption(SFA, "SampleKey")=0) : dialog "ファイルの復号に失敗したため、読み込むことができませんでした。終了します。" : end
        noteget xToken,0
        noteget xSecret,1
        noteunsel
        SetAccessToken xToken, xSecret
    }
    if ( (xToken ="") or (xSecret = "") ) : dialog "「Sample01_OAuth.hsp」か「Sample02_xAuth.hsp」を実行して「Setting.txt」を作成してください。" : end
    
    
    
    repeat                      //コマンドラインの2～の要素を、UserStreamAPIに対しての引数とする。
        if getarg(cnt+2,1)="" : break
        Arg.cnt = getarg(cnt+2,1)
    loop
    parentpid = int(getarg(1,1))
    hParent=OpenProcess(PROCESS_QUERY_INFORMATION, 0, parentpid)    //親プロセス実行状況確認時に使う。
    if hParent=0 : mes "USA: error "+__LINE__ : wait 500 : end      //OpenProcessが実行できなかった場合。
    
    api = getarg(0,1)
    UserStreamInit body, head, 0, api, Arg  ; HTTPS
    
    buf=""
*main
    ;   結果待ちのためのループ
    wait 100
    
    getBody_US body
    buf+=body
    gosub *comp
    
    exitcode = 0
    GetExitCodeProcess hParent, varptr(exitcode)    //親プロセスの実行状況確認
    if exitcode!$103 : goto *exit                   //親プロセスが閉じられていたら、*exitへ。
    wait 30
    goto *main
*exit
    //親プロセスが閉じられたっぽい。
    mes "USA: exit"
    wait 500
    end
    
*comp
    ;   完了
    getBlockStr body, '{', '}', block, buf  //bodyの{と}の間をブロックとしてblockに格納、余ったごみをbufへ。
    mes block
    return
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
このコードをツイートする