pikapython/package/requests/_requests.c

#include "_requests.h"
#include <ctype.h>
#include "_requests_Response.h"
#include "webclient.h"

#define GET_HEADER_BUFSZ 1024
#define GET_RESP_BUFSZ 1024

PikaObj* _requests_request(PikaObj* self, char* method, char* url) {
    PikaObj* response = newNormalObj(New__requests_Response);
    obj_setStr(response, "url", url);

    struct webclient_session* session = RT_NULL;
    unsigned char* buffer = RT_NULL;
    int ret = 0;
    int bytes_read, resp_status;
    int content_length = -1;

    buffer = (unsigned char*)web_calloc(1, GET_RESP_BUFSZ);
    if (buffer == RT_NULL) {
        rt_kprintf("no memory for receive buffer.\n");
        ret = -RT_ENOMEM;
        goto __exit;
    }

    /* create webclient session and set header response size */
    session = webclient_session_create(GET_HEADER_BUFSZ);
    if (session == RT_NULL) {
        ret = -RT_ENOMEM;
        goto __exit;
    }

    if (strEqu("GET", method)) {
        resp_status = webclient_get(session, url);
        obj_setInt(response, "status_code", resp_status);
        /* send GET request by default header */
        if (resp_status != 200) {
            goto __exit;
        }

        content_length = webclient_content_length_get(session);
        obj_setInt(response, "content_length", content_length);

        if (content_length < 0) {
            // rt_kprintf("webclient GET request type is chunked.\n");
            do {
                bytes_read =
                    webclient_read(session, (void*)buffer, GET_RESP_BUFSZ);
                if (bytes_read <= 0) {
                    break;
                }
            } while (1);
        } else {
            int content_pos = 0;
            do {
                bytes_read =
                    webclient_read(session, (void*)buffer,
                                   content_length - content_pos > GET_RESP_BUFSZ
                                       ? GET_RESP_BUFSZ
                                       : content_length - content_pos);
                if (bytes_read <= 0) {
                    break;
                }
                content_pos += bytes_read;
            } while (content_pos < content_length);
        }
        obj_setStr(response, "text", (char*)buffer);
        goto __exit;
    }

__exit:
    if (session) {
        webclient_close(session);
    }

    if (buffer) {
        web_free(buffer);
    }

    if (ret != 0) {
        if (response) {
            obj_deinit(response);
        }
        response = NULL;
    }
    return response;
}

char to_hex(char code) {
    static char hex[] = "0123456789abcdef";
    return hex[code & 15];
}

char* _requests_urlencode(PikaObj* self, char* s) {
    obj_setBytes(self, "encodebuff", NULL, strlen(s) * 3 + 1);
    char* result = (char*)obj_getBytes(self, "encodebuff");
    char* p = result;
    while (*s) {
        if (isalnum(*s) || *s == '-' || *s == '_' || *s == '.' || *s == '~') {
            *p++ = *s;
        } else if (*s == ' ') {
            *p++ = '+';
        } else {
            *p++ = '%';
            *p++ = to_hex(*s >> 4);
            *p++ = to_hex(*s & 0xf);
        }
        s++;
    }
    *p = '\0';
    return result;
}

void _requests___del__(PikaObj* self) {}

void _requests___init__(PikaObj* self) {}