#include <rfb/rfb.h>
#include <fcntl.h>
#include <stdio.h>
#include <errno.h>
#include <sys/mman.h>

#include <xf86drm.h>
#include <libdrm/drm_fourcc.h>
#include <xf86drmMode.h>

int main(int argc, const char **argv)
{
    if (argc < 2) {
        printf("not enough arguments\n");
        return 1;
    }

    const char *card = argv[1];
    const int drmfd = open(card, O_RDONLY);
    if (drmfd < 0) {
        fprintf(stderr, "card %s open failed: %s\n", card, strerror(errno));
        return 1;
    }
    int err;
    int source_plane = 0;
    int source_crtc = 0;
    err = drmSetClientCap(drmfd, DRM_CLIENT_CAP_UNIVERSAL_PLANES, 1);
    if (err < 0) {
        perror("Failed to set universal planes capability: primary planes will not be usable");
	}
    drmModePlane *plane = NULL;
    drmModePlaneRes *plane_res = NULL;
    drmModeFB2 *fb = NULL;
    if (source_plane > 0) {
        plane = drmModeGetPlane(drmfd, source_plane);
        if (!plane) {
            fprintf(stderr, "Failed to get plane %d: %s\n", source_plane, strerror(errno));
            goto cleanup;
        }
        if (plane->fb_id == 0) {
            fprintf(stderr, "Place %d does not have an attached framebuffer\n", source_plane);
        }
    }
    else {
        plane_res = drmModeGetPlaneResources(drmfd);
        if (!plane_res) {
            perror("Failed to get plane resources");
            goto cleanup;
        }
        int i;
        for (i = 0; i < plane_res->count_planes; i++) {
            plane = drmModeGetPlane(drmfd, plane_res->planes[i]);
            if (!plane) {
                fprintf(stderr, "Failed to get plane %u: %s\n", plane_res->planes[i], strerror(errno));
                continue;
            }
            printf("Plane %u CRTC %u FB %u\n", plane->plane_id, plane->crtc_id, plane->fb_id);
            if ((source_crtc > 0 && plane->crtc_id != source_crtc) || plane->fb_id == 0) {
                // Either not connected to the target source CRTC
                // or not active.
                drmModeFreePlane(plane);
                plane = NULL;
                continue;
            }
            break;
        }
        if (i == plane_res->count_planes) {
            if (source_crtc > 0) {
                fprintf(stderr, "No usable planes found on CRTC %d\n", source_crtc);
            } else {
                fprintf(stderr, "No usable planes found\n");
            }
            goto cleanup;
        }
        printf("Using plane %u to locate framebuffers\n", plane->plane_id);
    }
    uint32_t plane_id = plane->plane_id;

    fb = drmModeGetFB2(drmfd, plane->fb_id);
    if (!fb) {
        fprintf(stderr, "Failed to get framebuffer %u: %s\n", plane->fb_id, strerror(errno));
        goto cleanup;
    }
    //printf("Template framebuffer is %u: %ux%u %ubpp %ub depth %u pitch\n", fb->fb_id, fb->width, fb->height, fb->bpp, fb->depth, fb->pitch);
    printf("Template framebuffer is %u: %ux%u fourcc:%u mod:%u flags:%u\n", fb->fb_id, fb->width, fb->height, fb->pixel_format, fb->modifier, fb->flags);
    printf("handles %u %u %u %u\n", fb->handles[0], fb->handles[1], fb->handles[2], fb->handles[3]);
    printf("offsets %u %u %u %u\n", fb->offsets[0], fb->offsets[1], fb->offsets[2], fb->offsets[3]);
    printf("pitches %u %u %u %u\n", fb->pitches[0], fb->pitches[1], fb->pitches[2], fb->pitches[3]);
    printf("format %s, modifier %s:%s\n", drmGetFormatName(fb->pixel_format), drmGetFormatModifierVendor(fb->modifier), drmGetFormatModifierName(fb->modifier));

    if (!fb->handles[0]) {
        fprintf(stderr, "No handle set on framebuffer: maybe you need some additional capabilities?\n");
        goto cleanup;
    }

    int fd = -1;
    err = drmPrimeHandleToFD(drmfd, fb->handles[0], O_RDONLY, &fd);
    if (err < 0 || fd < 0) {
        perror("Failed to get PRIME fd from framebuffer handle");
    }

    //size_t buflen = fb->width * fb->height * fb->bpp / 8;
    size_t buflen = fb->width * fb->height * 32 / 8;
    char *buf = malloc(buflen);
    char *mapped = mmap(NULL, buflen, PROT_READ, MAP_SHARED, fd, 0);
    if (mapped == MAP_FAILED) {
        perror("mmap");
        goto cleanup;
    }
    memcpy(buf, mapped, buflen);

    //unlink("/tmp/1.raw");
    //write(open("/tmp/1.raw", O_WRONLY | O_CREAT, 00644), buf, buflen);

    rfbScreenInfoPtr server=rfbGetScreen(0,NULL,fb->width,fb->height,8,3,32/8);
    if(!server)
        return 1;
    server->frameBuffer=buf+8;
    rfbInitServer(server);           
    rfbRunEventLoop(server,-1,FALSE);
    return(0);

    cleanup:
        close(drmfd);
        return 1;

}