#include "app.h"

#include <math.h></math.h>

#include <sys time.h=""></sys>

#include <sys types.h=""></sys>

#include <unistd.h></unistd.h>

#include <x11 xutil.h=""></x11>

#include <x11 xatom.h=""></x11>

#include <x11 extensions="" xrandr.h=""></x11>

int appInit(App *app) {

  LOGDBG("app: init");

  CLEARFROM(app, dpy);

  LOGDBG("app: init: connect to xcb");

  app->dpy = XOpenDisplay(NULL);

  if (!app->dpy)

    return LOGERR("app: init: cannot connect to xcb");

  LOGDBG("app: init: get screen");

  app->screen = DefaultScreen(app->dpy);

  app->sw = DisplayWidth(app->dpy, app->screen);

  app->sh = DisplayHeight(app->dpy, app->screen);

  LOGDBG("app: init: screen=%d sw=%d sh=%d", app->screen, app->sw, app->sh);

  LOGDBG("app: init: get root window");

  app->root = RootWindow(app->dpy, app->screen);

  LOGDBG("app: init: create window");

  app->x = 0;

  app->w = app->sw * WIDTH_SCALE;

  app->h = app->sh * HEIGHT_SCALE;

  #if defined(WIDTH_SCALE_P) && defined(HEIGHT_SCALE_P)

  if (app->sw < app->sh) {

    app->w = app->sw * WIDTH_SCALE_P;

    app->h = app->sh * HEIGHT_SCALE_P;

  }

  #endif

  #ifdef TOP

  app->y = 0;

  #else

  app->y = app->sh - app->h;

  #endif

  XSetWindowAttributes attr = {};

  #ifdef NOBORDER

  attr.override_redirect = 1;

  #endif

  app->win = XCreateWindow(

    app->dpy, app->root,               // display and parent window

    app->x, app->y, app->w, app->h, 0, // position, size and border width

    CopyFromParent, CopyFromParent, CopyFromParent, // depth, class and visual

    CWOverrideRedirect, &attr);        // mask with attributes

  XSelectInput(

    app->dpy, app->win,

      StructureNotifyMask

    | ExposureMask

    | ButtonPressMask

    | ButtonReleaseMask

    | Button1MotionMask );

  #ifndef NOBORDER

  LOGDBG("app: init: set window title");

  char *title = TITLE;

  XTextProperty wtitle;

  if (!XStringListToTextProperty(&title, 1, &wtitle)) {

    LOGWRN("app: init: XStringListToTextProperty error for title: %s", title);

  } else {

    XSetWMName(app->dpy, app->win, &wtitle);

    XFree(wtitle.value);

  }

  #endif

  #if !defined(NOBORDER) && defined(DOCK)

  LOGDBG("app: init: set window as dock");

  Atom awt = XInternAtom(app->dpy, "_NET_WM_WINDOW_TYPE", False);

  Atom awtv = XInternAtom(app->dpy, "_NET_WM_WINDOW_TYPE_TOOLBAR", False);

  if (awt != None && awtv != None)

    XChangeProperty(app->dpy, app->win, awt, XA_ATOM, 32, PropModeReplace, (unsigned char*)&awtv, 1);

  #endif

  Atom ahps = XInternAtom(app->dpy, "_HILDON_PORTRAIT_MODE_SUPPORT", False);

  if (ahps) {

    LOGDBG("app: init: set portrait mode support for hildon");

    unsigned int v = 1;

    XChangeProperty(app->dpy, app->win, ahps, XA_CARDINAL, 32, PropModeReplace, (unsigned char*)&v, 1);

  }

  LOGDBG("app: init: set window minimum size");

  XSizeHints *wsh = XAllocSizeHints();

  if (!wsh) {

    LOGWRN("app: init: XAllocSizeHints error");

  } else {

    #ifdef LOCK_SIZE

    wsh->flags      = PMinSize | PMaxSize;

    wsh->min_width  = wsh->max_width  = app->w;

    wsh->min_height = wsh->max_height = app->h;

    #else

    wsh->flags      = PMinSize;

    wsh->min_width  = MIN_WIDTH;

    wsh->min_height = MIN_HEIGHT;

    #endif

    XSetSizeHints(app->dpy, app->win, wsh, XA_WM_NORMAL_HINTS);

    XFree(wsh);

  }

  // window must not grab the focus

  LOGDBG("app: init: set window input hint");

  XWMHints *wmh = XAllocWMHints();

  if (!wmh) {

    LOGWRN("app: init: XAllocWMHints error");

  } else {

    wmh->flags = InputHint;

    wmh->input = False;

    XSetWMHints(app->dpy, app->win, wmh);

    XFree(wmh);

  }

  // subscribe to windows delete message from window manager

  app->aWmDel = XInternAtom(app->dpy, "WM_DELETE_WINDOW", False);

  XSetWMProtocols(app->dpy, app->win, &app->aWmDel, 1);

  // subscribe to keyboard state and layout changes

  XkbSelectEvents(app->dpy, XkbUseCoreKbd, XkbMapNotifyMask | XkbStateNotifyMask, XkbMapNotifyMask | XkbStateNotifyMask);

  #ifdef SCREEN_EVENTS

  // subscribe to xrandr events

  if (!XRRQueryExtension(app->dpy, &app->xrev, &app->xrerr)) {

    LOGWRN("app: init: XRRQueryExtension error");

  } else {

    app->xron = 1;

    XRRSelectInput(app->dpy, app->win, RRScreenChangeNotifyMask);

  }

  #endif

  // init submodules

  if (graphInit(&app->graph, app)) {

    graphResize(&app->graph);

    if (inputInit(&app->input, app)) {

      if (keyboardInit(&app->keyboard, app)) {

        return 1;

      }

      inputDeinit(&app->input);

    }

    graphDeinit(&app->graph);

  } else {

    XCloseDisplay(app->dpy);

  }

  return 0;

}

void appDeinit(App *app) {

  LOGDBG("app: deinit");

  keyboardDeinit(&app->keyboard);

  inputDeinit(&app->input);

  graphDeinit(&app->graph);

  XCloseDisplay(app->dpy);

}

int appRun(App *app) {

  LOGDBG("app: run");

  if (app->run)

    return LOGERR("app: run: seems already run");

  app->run = 1;

  LOGDBG("app: run: show window");

  XMapWindow(app->dpy, app->win);

  appUpdateStrut(app);

  LOGDBG("app: run: get connection file descriptor");

  int fd = ConnectionNumber(app->dpy);

  if (fd < 0)

    LOGERR("app: run: cannot get XLib file descruptor");

  LOGDBG("app: run: event loop");

  int buttonDown = 0;

  unsigned int buttonDownMs = 0;

  fd_set fds = {};

  int maxFd = fd > app->stopFd ? fd : app->stopFd;

  XFlush(app->dpy);

  while(1) {

    int hasEvent = 0;

    XEvent event = {};

    FD_ZERO(&fds);

    FD_SET(fd, &fds);

    FD_SET(app->stopFd, &fds);

    // wait for next event

    if (XPending(app->dpy)) {

      XNextEvent(app->dpy, &event);

      hasEvent = 1;

    } else

    if (buttonDown != 1) {

      // just wait for fds

      select(maxFd + 1, &fds, NULL, NULL, NULL);

    } else {

      // manually call select for the X-socket to use timeout

      unsigned int dt = monotonicMs() - buttonDownMs;

      if (dt < LONGPRESS_MS) {

        dt = LONGPRESS_MS - dt;

        struct timeval t = {};

        t.tv_sec = dt/1000;

        t.tv_usec = dt%1000*1000;

        LOGDBG("app: run: use select fd=%d, timeout=%ums, sec=%ld, usec=%ld", fd, dt, t.tv_sec, t.tv_usec);

        XFlush(app->dpy);

        select(maxFd + 1, &fds, NULL, NULL, &t);

      }

    }

    if (!hasEvent && XPending(app->dpy)) {

      XNextEvent(app->dpy, &event);

      hasEvent = 1;

    }

    if (hasEvent) {

      switch(event.type) {

      case Expose: {

        LOGDBG( "app: expose: x=%d, y=%d, w=%d, h=%d",

                event.xexpose.x, event.xexpose.y, event.xexpose.width, event.xexpose.height );

        appInvalidateRect(app, event.xexpose.x, event.xexpose.y, event.xexpose.width, event.xexpose.height);

        break;

      }

      case ConfigureNotify: {

        if ( event.xconfigure.width

          && event.xconfigure.height

          && ( app->x != event.xconfigure.x

            || app->y != event.xconfigure.y

            || app->w != event.xconfigure.width

            || app->h != event.xconfigure.height ))

        {

          LOGDBG( "app: moved: x=%d y=%d w=%d, h=%d",

                  event.xconfigure.x, event.xconfigure.y, event.xconfigure.width, event.xconfigure.height );

          int resized = app->w != event.xconfigure.width || app->h != event.xconfigure.height;

          app->x = event.xconfigure.x;

          app->y = event.xconfigure.y;

          app->w = event.xconfigure.width;

          app->h = event.xconfigure.height;

          appUpdateStrut(app);

          if (resized) graphResize(&app->graph);

        }

        break;

      }

      case ButtonPress: {

        if (event.xbutton.button == 1) {

          LOGDBG("app: mouse down: x=%d, y=%d", event.xbutton.x, event.xbutton.y);

          if (!buttonDown) { buttonDown = 1; buttonDownMs = monotonicMs(); }

          keyboardMouseDown(&app->keyboard, event.xbutton.x, event.xbutton.y);

        }

        break;

      }

      case ButtonRelease: {

        if (event.xbutton.button == 1) {

          LOGDBG("app: mouse up");

          buttonDown = 0;

          keyboardMouseUp(&app->keyboard);

        }

        break;

      }

      case MotionNotify: {

        if (event.xmotion.state & Button1Mask) {

          LOGDBG("app: mouse motion: x=%d, y=%d", event.xmotion.x, event.xmotion.y);

          keyboardMouseMotion(&app->keyboard, event.xmotion.x, event.xmotion.y);

        }

        break;

      }

      case MappingNotify: {

        if ( event.xmapping.request == MappingModifier

          || event.xmapping.request == MappingKeyboard )

        {

          LOGDBG("app: kayboard layout changed");

          XRefreshKeyboardMapping(&event.xmapping);

          inputUpdateLayout(

            &app->input,

            event.xmapping.first_keycode,

            event.xmapping.first_keycode + event.xmapping.count );

        }

        break;

      }

      case ClientMessage: {

        if (event.xclient.data.l[0] == app->aWmDel) {

          LOGDBG("app: delete window event");

          appStop(app, 0);

        }

        break;

      }

      default:

        if (app->xron && event.type == app->xrev + RRScreenChangeNotify) {

          LOGDBG("app: screen change event");

          XRRScreenChangeNotifyEvent *ev = (XRRScreenChangeNotifyEvent*)&event;

          appUpdateScreenSize(app, ev->width, ev->height);

        }

        break;

      }

    }

    if (buttonDown == 1 && monotonicMs() - buttonDownMs >= LONGPRESS_MS) {

        LOGDBG("app: long press");

        buttonDown = 2;

        keyboardMouseLongDown(&app->keyboard);

    }

    if (app->run != 1) break;

    inputUpdateModifiers(&app->input);

    keyboardResize(&app->keyboard);

    keyboardUpdateModifiers(&app->keyboard);

    if (app->irw > 0 && app->irh > 0) {

      LOGDBG("app: draw: x=%d, y=%d, w=%d, h=%d", app->irx, app->iry, app->irw, app->irh);

      graphDrawBackgound(&app->graph, app->irx, app->iry, app->irw, app->irh);

      keyboardDraw(&app->keyboard, app->irx, app->iry, app->irw, app->irh);

      app->irw = 0;

    }

  }

  int err = app->run == -2;

  app->run = 0;

  LOGDBG("app: run: done err=%d", err);

  return !err;

}

void appStopBySignal(App *app) {

  char x = 0;

  write(app->stopFd, &x, 1);

}

void appStop(App *app, int err) {

  LOGDBG("app: stop: err=%d", err);

  if (!app->run) {

    LOGERR("app: stop: seems not started");

    return;

  }

  if (app->run == 1) app->run = -1;

  if (err) app->run = -2;

}

void appUpdateStrut(App *app) {

  unsigned int v[4] = {};

  #ifdef DOCK

  if (app->y == 0) v[2] = app->h; else

    if (app->y + app->h == app->sh) v[3] = app->h;

  #endif

  if (app->dockt == v[2] && app->dockb == v[3])

    return;

  app->dockt = v[2];

  app->dockb = v[3];

  Atom k = XInternAtom(app->dpy, "_NET_WM_STRUT", False);

  if (!k) return;

  LOGDBG("app: update strut: %u %u %u %u", v[0], v[1], v[2], v[3]);

  XChangeProperty(app->dpy, app->win, k, XA_CARDINAL, 32, PropModeReplace, (unsigned char*)v, 4);

}

void appUpdateScreenSize(App *app, int sw, int sh) {

  if (sw <= 0 || sh <= 0) return;

  if (sw == app->sw && sh == app->sh) return;

  LOGDBG("app: update screen size: w=%d, h=%d", sw, sh);

  #if defined(DOCK) || (defined(LOCK_SIZE) && defined(NOBORDER) && defined(NOTITLE))

  // static size case

  int w = sw * WIDTH_SCALE;

  int h = sh * HEIGHT_SCALE;

  #if defined(WIDTH_SCALE_P) && defined(HEIGHT_SCALE_P)

  if (sw < sh) {

    w = sw * WIDTH_SCALE_P;

    h = sh * HEIGHT_SCALE_P;

  }

  #endif

  int x0 = 0;

  #ifdef TOP

  int y0 = 0;

  #else

  int y0 = sh - h;

  #endif

  #else

  // dynamic size case

  double kx = sw/(double)app->sw;

  double ky = sh/(double)app->sh;

  #if defined(WIDTH_SCALE_P) && defined(HEIGHT_SCALE_P)

  // rescale for portrait/landscape switching

  if (app->sw < app->sh && sw >= sh) {

    // portrait -> landscape

    kx *= ((double)WIDTH_SCALE)/((double)WIDTH_SCALE_P);

    ky *= ((double)HEIGHT_SCALE)/((double)HEIGHT_SCALE_P);

  } else

  if (app->sw >= app->sh && sw < sh) {

    // landscape -> portrait

    kx *= ((double)WIDTH_SCALE_P)/((double)WIDTH_SCALE);

    ky *= ((double)HEIGHT_SCALE_P)/((double)HEIGHT_SCALE);

  }

  #endif

  // scale window corners

  int x0 = (int)round( app->x*kx );

  int y0 = (int)round( app->y*ky );

  int x1 = (int)round( (app->x + app->w)*kx );

  int y1 = (int)round( (app->y + app->h)*ky );

  int w = x1 - x0;

  int h = y1 - y0;

  #endif

  app->sw = sw;

  app->sh = sh;

  appMove(app, x0, y0, w, h);

}

void appMove(App *app, int x, int y, int w, int h) {

  LOGDBG("app: move: x=%d, y=%d, w=%d, h=%d", x, y, w, h);

  if (w < MIN_WIDTH)  w = MIN_WIDTH;

  if (h < MIN_HEIGHT) h = MIN_HEIGHT;

  unsigned int mask = 0;

  XWindowChanges ch = {};

  if (app->x != x) ch.x = x, mask |= CWX;

  if (app->y != y) ch.y = y, mask |= CWY;

  if (app->w != w) ch.width  = w, mask |= CWWidth;

  if (app->h != h) ch.height = h, mask |= CWHeight;

  if (!mask) return;

  XConfigureWindow(app->dpy, app->win, mask, &ch);

  XFlush(app->dpy);

}

void appInvalidateRect(App *app, int x, int y, int w, int h) {

  LOGDBG("app: invalidate rect: x=%d, y=%d, w=%d, h=%d", x, y, w, h);

  rectIntersect(&x, &y, &w, &h, 0, 0, app->w, app->h);

  rectMerge(&app->irx, &app->iry, &app->irw, &app->irh, x, y, w, h);

  LOGDBG("app: invalidate rect: summary x=%d, y=%d, w=%d, h=%d", app->irx, app->iry, app->irw, app->irh);

}