#include <curl/curl.h>

#include <json-c/json.h>

#include <netinet/in.h>

#include <sys/socket.h>

#include <sys/stat.h>

#include <sys/types.h>

#include <pwd.h>

#include <fcntl.h>

#include <netdb.h>

#include <unistd.h>

#include <stdio.h>

#include <string.h>

#include <errno.h>

#include "downloader.h"

typedef struct {

int id;

char *name;

} fm_channel_t;

#define LOCAL_CHANNEL "999"

#define JING_TOP_CHANNEL "#top"

#define JING_TOP_CHANNEL_NAME "Jing+ tops"

#define JING_PSN_CHANNEL "#psn"

#define JING_PSN_CHANNEL_NAME "Jing+ personal"

#define JING_RAND_CHANNEL "#rand"

#define JING_RAND_CHANNEL_NAME "Jing+ rand"

// utilities definitions

char *escapech(char *buf, char c, char *str)

{

int l = strlen(str) + 1;

// just needs to loop through and add backslash to the front

int i = 0, pi = 0;

while (i < l) {

char ch = str[i++];

if (ch == c)

buf[pi++] = '\\';

buf[pi++] = ch;

}

return buf;

}

char *escapesh(char *buf, char *str)

{

return escapech(buf, '\'', str);

}

void free_channels(fm_channel_t *channels, int number) {

int i;

for (i=0; i<number; i++) {

free(channels[i].name);

}

free(channels);

}

int read_channels_json(json_object *obj, fm_channel_t **channels, int *number) {

int ret = -1;

int i;

if (obj) {

array_list *channel_objs = json_object_get_array(json_object_object_get(obj, "channels"));

if (channel_objs) {

*number = array_list_length(channel_objs);

*channels = (fm_channel_t *) malloc(*number * sizeof(fm_channel_t));

for (i = 0; i < *number; i++) {

json_object *o = (json_object*) array_list_get_idx(channel_objs, i);

int id = json_object_get_int(json_object_object_get(o, "channel_id"));

(*channels)[i].id = id;

(*channels)[i].name = strdup(json_object_get_string(json_object_object_get(o, "name")));

}

ret = 0;

}

json_object_put(obj);

}

return ret;

}

// return 0 on success, -1 otherwise

int read_channels(fm_channel_t **channels, int *number)

{

struct passwd *pwd = getpwuid(getuid());

const char file[128];

strcpy(file, pwd->pw_dir);

strcat(file, "/.rpd/channels");

char buf[4096];

size_t size;

int fd;

int ret = -1;

fd = open(file, O_RDONLY);

if (fd >= 0) {

memset(buf, 0, sizeof(buf));

size = read(fd, buf, sizeof(buf));

if (size > 0) {

ret = read_channels_json(json_tokener_parse(buf), channels, number);

} else {

*number = 0;

ret = 0;

}

close(fd);

}

else {

// download the content of the channel list

downloader_stack_t *s = stack_init();

downloader_t *d = stack_get_idle_downloader(s, dMem);

curl_easy_setopt(d->curl, CURLOPT_URL, "https://www.douban.com/j/app/radio/channels");

curl_easy_setopt(d->curl, CURLOPT_CONNECTTIMEOUT, 10);

stack_perform_until_done(s, d);

ret = read_channels_json(json_tokener_parse(d->content.mbuf->data), channels, number);

int len = ret == 0 ? d->content.mbuf->length : 0;

ret = 0;

FILE *f = fopen(file, "w");

// save to a file

fwrite(d->content.mbuf->data, 1, len, f);

fclose(f);

}

return ret;

}

char *get_local_channel_name() {

/* char *login = getlogin(); */

/* if (!login) */

/* login = "Red-Heart"; */

return "Red-Heart";

}

void print_channels(fm_channel_t *channels, int len)

{

int i;

printf("%5s %s\n", "id", "name");

printf("%5s %s\n", LOCAL_CHANNEL, get_local_channel_name());

printf("%5s %s\n", JING_TOP_CHANNEL, JING_TOP_CHANNEL_NAME);

printf("%5s %s\n", JING_PSN_CHANNEL, JING_PSN_CHANNEL_NAME);

printf("%5s %s\n", JING_RAND_CHANNEL, JING_RAND_CHANNEL_NAME);

for (i = 0; i < len; i++) {

printf("%5d %s\n", channels[i].id, channels[i].name);

}

}

void print_usage()

{

printf("Usage: rpc [-a address] [-p port] [cmd] [argument]\n"

" rpc help - show this help infomation\n"

" rpc info [format] - show current RPD information\n"

" if the format argument is given, the following specifier will be replaced accordingly\n"

" %%a -- artist \n"

" %%t -- song title \n"

" %%b -- album \n"

" %%y -- release year \n"

" %%i -- cover image \n"

" %%d -- douban url \n"

" %%c -- channel \n"

" %%p -- currtime \n"

" %%l -- totaltime \n"

" %%u -- status \n"

" %%k -- kbps \n"

" %%r -- rate (0 or 1) \n"

" %%%% -- a literal %% \n"

" rpc play - start playback\n"

" rpc pause - pause playback\n"

" rpc toggle - toggle between play and pause\n"

" rpc stop - stop playback\n"

" rpc skip/next - skip current song\n"

" rpc ban - don't ever play current song again\n"

" rpc rate - mark current song as \"liked\"\n"

" rpc unrate - unmark current song\n"

" rpc channels - list all FM channels\n"

" rpc webpage - open the douban page using the browser defined in $BROWSER\n"

" rpc setch <id> - set channel through channel's id\n"

" rpc kbps <kbps> - set music quality to the specified kbps\n"

" rpc launch - tell RPD to restart\n"

" rpc end - tell RPD to quit\n"

);

}

void time_str(int time, char *buf)

{

int sec = time % 60;

int min = time / 60;

sprintf(buf, "%d:%02d", min, sec);

}

int main(int argc, char *argv[])

{

int channels_len;

fm_channel_t* channels;

if (read_channels(&channels, &channels_len) != 0) {

printf("Failed to retrieve the channel list. Try again later.\n");

return -1;

}

char *addr = "localhost";

char *port = "10098";

int c;

while ((c = getopt(argc, argv, "a:p:")) != -1) {

switch (c) {

case 'a':

addr = optarg;

break;

case 'p':

port = optarg;

break;

default:

break;

}

}

char input_buf[64] = "";

char output_format[512] = "";

char output_buf[1024] = "";

int buf_size;

if (optind < argc) {

strcpy(input_buf, argv[optind]);

int i;

char *buf = strcmp(input_buf, "info") == 0 ? output_format : input_buf;

for (i = optind + 1; i < argc; i++) {

strcat(buf, " ");

strcat(buf, argv[i]);

}

} else {

strcpy(input_buf, "info");

}

int open_webpage = 0;

if (strcmp(input_buf, "channels") == 0) {

print_channels(channels, channels_len);

return 0;

}

else if (strcmp(input_buf, "help") == 0) {

print_usage();

return 0;

}

else if (strcmp(input_buf, "launch") == 0) {

// forcefully restart RPD

system("pgrep rpd && /usr/local/bin/rpc end && sleep 30; /usr/local/bin/rpd");

return 0;

}

else if (strncmp(input_buf, "webpage", 7) == 0) {

// client told to trigger a webpage

strcpy(input_buf, "info");

open_webpage = 1;

}

struct addrinfo hints, *results, *p;

int sock_fd;

memset(&hints, 0, sizeof(hints));

hints.ai_family = AF_INET;

hints.ai_socktype = SOCK_STREAM;

hints.ai_flags = AI_PASSIVE;

if (getaddrinfo(addr, port, &hints, &results) != 0) {

return -1;

}

for (p = results; p != NULL; p = p->ai_next) {

sock_fd = socket(p->ai_family, p->ai_socktype, p->ai_protocol);

if (sock_fd < 0) {

continue;

}

if (connect(sock_fd, p->ai_addr, p->ai_addrlen) == 0) {

break;

}

close(sock_fd);

}

if (p == NULL) {

perror("connect");

return -1;

}

freeaddrinfo(results);

send(sock_fd, input_buf, strlen(input_buf), 0);

buf_size = recv(sock_fd, output_buf, sizeof(output_buf), 0);

if (buf_size == 0) {

close(sock_fd);

return 0;

}

output_buf[buf_size] = '\0';

close(sock_fd);

json_object *obj = json_tokener_parse(output_buf);

char *status = obj ? strdup(json_object_get_string(json_object_object_get(obj, "status"))) : "error",

*channel = "", *artist = "", *title = "", pos[16] = "", len[16] = "", kbps[8] = "", *album = "", *cover = "", year[8] = "", *douban_url = "", *like = "";

if (strcmp(status, "error") != 0) {

char *chl = strdup(json_object_get_string(json_object_object_get(obj, "channel")));

long int cid;

if (strcmp(chl, LOCAL_CHANNEL) == 0) {

channel = get_local_channel_name();

} else if (strcmp(chl, JING_TOP_CHANNEL) == 0) {

channel = JING_TOP_CHANNEL_NAME;

} else if (strcmp(chl, JING_PSN_CHANNEL) == 0) {

channel = JING_PSN_CHANNEL_NAME;

} else {

// determining if it is an integer number

char *address;

cid = strtol(chl, &address, 10);

if (*address == '\0') {

// this is valid number

// loop through the array to get the correct name for the channel

int i;

for (i=0; i<channels_len; i++) {

if (channels[i].id == cid) {

channel = channels[i].name;

break;

}

}

if (channel[0] == '\0') {

channel = "未知兆赫";

}

} else {

channel = chl;

}

}

sprintf(kbps, "%d", json_object_get_int(json_object_object_get(obj, "kbps")));

if (strcmp(status, "stop") != 0) {

time_str(json_object_get_int(json_object_object_get(obj, "pos")), pos);

time_str(json_object_get_int(json_object_object_get(obj, "len")), len);

like = json_object_get_int(json_object_object_get(obj, "like")) ? "1" : "0";

artist = strdup(json_object_get_string(json_object_object_get(obj, "artist")));

title = strdup(json_object_get_string(json_object_object_get(obj, "title")));

album = strdup(json_object_get_string(json_object_object_get(obj, "album")));

sprintf(year, "%d", json_object_get_int(json_object_object_get(obj, "year")));

cover = strdup(json_object_get_string(json_object_object_get(obj, "cover")));

douban_url = strdup(json_object_get_string(json_object_object_get(obj, "url")));

}

}

if (output_format[0] == '\0') {

if (strcmp(status, "error") == 0) {

if (obj)

printf("%s\n", json_object_get_string(json_object_object_get(obj, "message")));

else

printf("Unkown error with buf content %s\n", output_buf);

} else if (open_webpage) {

// open the designated webpage

char sh[1280];

// the url can become rather long after escaping

char url[1024];

if (escapesh(url, douban_url)[0] == '\0') {

// we need to make a custom url to open

sprintf(url, "%s %s", artist, album);

// first obtain a curl instance to escape the query

CURL *curl = curl_easy_init();

char *st = curl_easy_escape(curl, url, 0);

sprintf(url, "https://music.douban.com/subject_search?search_text=%s&cat=1003", st);

curl_free(st);

curl_easy_cleanup(curl);

}

/* printf("url is %s\n", url); */

sprintf(sh, "%s $'%s' &", "$BROWSER", url);

/* printf("cmd is %s\n", sh); */

system(sh);

} else {

printf("RPD %s - %s / %s kbps\n", strcmp(status, "play") == 0? "Playing": (strcmp(status, "pause") == 0? "Paused": "Stopped"), channel, kbps);

if (strcmp(status, "stop") != 0) {

printf("%s%s - %s\n%s / %s\n",

like[0] == '1' ? "[Like] ": "", artist, title,

pos, len);

}

}

} else {

char info[1024], *arg = "";

int l = strlen(output_format) + 1;

// we must print out the information the user wants

// trim the space in front

int i = 1, pi = 0;

while ( i < l) {

char ch = output_format[i++];

if (i < l - 1 && ch == '%') {

char spec = output_format[i++];

switch (spec) {

case 'a': arg = artist; break;

case 't': arg = title; break;

case 'b': arg = album; break;

case 'y': arg = year; break;

case 'i': arg = cover; break;

case 'd': arg = douban_url; break;

case 'c': arg = channel; break;

case 'p': arg = pos; break;

case 'l': arg = len; break;

case 'u': arg = status; break;

case 'k': arg = kbps; break;

case 'r': arg = like; break;

case '%': arg = "%%"; break;

default:

printf("Unknown specifier %c. Try help\n", spec);

return 1;

}

// loop through the arg and copy the chars

while (*arg != '\0') {

info[pi++] = *arg;

arg++;

}

} else

info[pi++] = ch;

}

printf(info);

}

json_object_put(obj);

// free the channel list

free_channels(channels, channels_len);

return 0;

}