#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <sys/mman.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <stdarg.h>
#include <sys/ioctl.h>
#include <sys/vt.h>
#include "get-edid.h"

int verbose = 0;

int main(int argc, char **argv)
{
  char edid[256];
  int try_in_console = 0;
  int port = 0;
  int i;

  for (i = 1; i < argc; i++) {
	const char *arg = argv[i];
    if (strcmp(arg, "-v") == 0) verbose = 1;
    else if (strcmp(arg, "--port") == 0 && i+1 < argc) port = atoi(argv[++i]);
    else if (strcmp(arg, "--try-in-console") == 0) try_in_console = 1;
    else if (strcmp(arg, "-h") == 0 ||
	     strcmp(arg, "--help") == 0) {
      printf("usage: monitor-get-edid [-v] [--port <0-3>] [--try-in-console]\n");
      exit(1);
    }
  }

  int size = get_edid(edid, port);

  if (!size && try_in_console) {
    int non_X_console = 1;
    struct vt_stat current;
    int fd = open("/dev/console", O_RDWR);

    if (getenv("DISPLAY") != NULL &&
	ioctl(fd, VT_GETSTATE, &current) == 0 && 
	current.v_active != non_X_console &&	
	ioctl(fd, VT_ACTIVATE, non_X_console) == 0 &&
	ioctl(fd, VT_WAITACTIVE, non_X_console) == 0) {
      /* retrying */
      size = get_edid(edid, port);

      /* restore */
      ioctl(fd, VT_ACTIVATE, current.v_active) == 0 &&
	ioctl(fd, VT_WAITACTIVE, current.v_active) == 0;
    }
    close(fd);
  }

  if (size) write(1, edid, size);

  return size ? 0 : 1;
}