summary refs log tree commit diff
path: root/downloadAvatar.c
blob: 45a39acbe6263c541d7b8bde9c3581a34cce0549 (plain)
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
#define _POSIX_C_SOURCE 200112L

#include "cptc.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#include <curl/curl.h>
#include <curl/easy.h>


static size_t write_function(void *contents, size_t size, size_t nmemb, void *userp)
{
    size_t realsize = size * nmemb;
    struct {char *buf; size_t size;} *mem = userp;

    mem->buf = realloc(mem->buf, mem->size + realsize + 1);
    if (!mem->buf)
    {
        perror("realloc()");
        return 0;
    }

    memcpy(&(mem->buf[mem->size]), contents, realsize);
    mem->size += realsize;
    mem->buf[mem->size] = 0;

    return realsize;
}

char *CPTC_downloadAvatar(long long uid, const char *filenamebuf)
{
    char *avatar, link[256], auth[128], *ret;
    CURL *curl;
    CURLcode res;
    FILE *fp;
    struct {char *buf; size_t size;} buffer;
    buffer.buf = malloc(1);
    buffer.size = 0;

    curl = curl_easy_init();
    if (curl)
    {
        snprintf(link, sizeof(link), "https://discord.com/api/v9/users/%lld", uid);
        snprintf(auth, sizeof(auth), "Authorization: %s", CPTC_token);

        curl_easy_setopt(curl, CURLOPT_URL, link);
        curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
        curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_function);
        curl_easy_setopt(curl, CURLOPT_WRITEDATA, &buffer);
        curl_easy_setopt(curl, CURLOPT_HTTPHEADER, curl_slist_append(NULL, auth));

        res = curl_easy_perform(curl);
        if (res != CURLE_OK)
        {
            fprintf(stderr, "curl_easy_perform(): %s\n", curl_easy_strerror(res));
            curl_easy_cleanup(curl);
            free(buffer.buf);
            return NULL;
        }

        curl_easy_cleanup(curl);
    } else
    {
        free(buffer.buf);
        return NULL;
    }
    avatar = strstr(buffer.buf, "avatar\":\"") + 9;
    if (!(avatar - 9))
    {
        fprintf(stderr, "User %lld has no avatar\n", uid);
        free(buffer.buf);
        return NULL;
    }
    *strchr(avatar, '"') = '\0';

    fp = fopen(filenamebuf, "w");
    if (!fp)
    {
        perror("fopen()");
        free(buffer.buf);
        return NULL;
    }

    curl = curl_easy_init();
    if (curl)
    {
        snprintf(link, sizeof(link), "https://cdn.discordapp.com/avatars/%lld/%s.png", uid, avatar);

        curl_easy_setopt(curl, CURLOPT_URL, link);
        curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
        curl_easy_setopt(curl, CURLOPT_WRITEDATA, fp);
        curl_easy_setopt(curl, CURLOPT_HTTPHEADER, curl_slist_append(NULL, auth));

        res = curl_easy_perform(curl);
        if (res != CURLE_OK)
        {
            fprintf(stderr, "curl_easy_perform(): %s\n", curl_easy_strerror(res));
            curl_easy_cleanup(curl);
            fclose(fp);
            free(buffer.buf);
            return NULL;
        }

        curl_easy_cleanup(curl);
    } else
    {
        fclose(fp);
        free(buffer.buf);
        return NULL;
    }

    fclose(fp);
    ret = malloc(strlen(filenamebuf) + 1);
    memcpy(ret, filenamebuf, strlen(filenamebuf) + 1);
    free(buffer.buf);
    return ret;
}