#define _DEFAULT_SOURCE
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <errno.h>
#include <fcntl.h>
#include <poll.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <unistd.h>
#include <dlfcn.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/time.h>
#include <stdarg.h>
#include <time.h>
#include "histogram.h"
#include "blackbars.h"
#include "color_boost.h"
#include "interpolation.h"
#include "layout.h"
#include "autostart.h"
#define SERVICE "org.webosbrew.histogramhyperion.service"
#define PRIORITY 149
typedef struct {int code;char *message;const char *file;int line;const char *func;void *padding;unsigned long magic;} LSError;
typedef bool (*Callback)(void*,void*,void*);
typedef struct {const char *name;Callback function;int flags;} Method;
static bool (*reply)(void*,void*,const char*,LSError*);
static const char *(*payload)(void*);
static bool (*error_init)(LSError*);
static void (*error_free)(LSError*);
static void *(*vt_init)(bool(*)(void));
static int (*vt_preset)(void*,int),(*vt_get)(void*,Histogram*),(*vt_finish)(void*);
static void *capture;
static char lg_diagnostic[256];
/* Process-local LG diagnostics; no changes to the loaded LG library. */
void printLog(const char *context,int level,const char *message_id,
              const char *file,const char *function,int line,const char *format,...) {
 (void)context;(void)message_id;(void)file;
 if(capture && level>3)return;
 if(format){va_list args;va_start(args,format);vsnprintf(lg_diagnostic,sizeof(lg_diagnostic),format,args);va_end(args);}
 fprintf(stderr,"LG %s:%d: ",function?function:"?",line);
 if(format){va_list args;va_start(args,format);vfprintf(stderr,format,args);va_end(args);}
 fputc('\n',stderr);
}
static int output=-1;
static unsigned long frames;
static unsigned char rgb[16][3];
static unsigned char sent_rgb[3];
static unsigned black_percent[16];
static unsigned capture_preset;
static unsigned letterbox_mode,settling;
static int pending_letterbox=-1;
static uint64_t candidate_since,next_border_probe;
static bool color_valid;
static unsigned led_count;
static LedBlend *led_blends;
static bool interpolation_enabled;
static unsigned recovery_attempts;
static uint64_t retry_at;
static uint64_t milliseconds(void){struct timespec t;clock_gettime(CLOCK_MONOTONIC,&t);return (uint64_t)t.tv_sec*1000+t.tv_nsec/1000000;}
static char last_error[256];
static volatile sig_atomic_t revoked,quitting;
static int guard_pipe=-1;
static pid_t guard_pid=-1;
#define DEV_MODE "/var/luna/preferences/vt_devmode"
/* LG's developer gate is enabled only during initialization. A separate
 * guardian removes our own marker even if this process aborts or is killed.
 * An existing marker is preserved, as are files replaced by another owner. */
static bool enter_developer_mode(void){
 int marker=open(DEV_MODE,O_CREAT|O_EXCL|O_WRONLY|O_CLOEXEC,0600);
 if(marker<0)return errno==EEXIST;
 struct stat owned;
 if(fstat(marker,&owned)!=0){close(marker);unlink(DEV_MODE);return false;}
 close(marker);int p[2];
 if(pipe(p)!=0){unlink(DEV_MODE);return false;}
 pid_t child=fork();
 if(child<0){close(p[0]);close(p[1]);unlink(DEV_MODE);return false;}
 if(child==0){
  close(p[1]);struct pollfd fd={p[0],POLLIN|POLLHUP,0};
  /* Bounded cleanup also covers an indefinitely blocked initialization. */
  poll(&fd,1,10000);close(p[0]);struct stat current;
  int rc=0;
  if(lstat(DEV_MODE,&current)==0 && current.st_dev==owned.st_dev && current.st_ino==owned.st_ino)rc=unlink(DEV_MODE);
  _exit(rc==0?0:1);
 }
 close(p[0]);guard_pipe=p[1];guard_pid=child;return true;
}
static bool leave_developer_mode(void){
 if(guard_pipe<0)return true;
 close(guard_pipe);guard_pipe=-1;int status;
 while(waitpid(guard_pid,&status,0)<0){if(errno!=EINTR)return false;}
 guard_pid=-1;return WIFEXITED(status)&&WEXITSTATUS(status)==0;
}
static void quit_signal(int s){(void)s;quitting=1;}
static bool policy(void){revoked=1;return true;}
static void set_error(const char *s){snprintf(last_error,sizeof(last_error),"%s",s);fprintf(stderr,"%s\n",s);}
static int send_line(const char *line){
 size_t left=strlen(line);const char *p=line;
 while(left){ssize_t n=send(output,p,left,MSG_NOSIGNAL);if(n<=0)return -1;p+=n;left-=n;}
 return 0;
}
static bool read_layout(void){
 const size_t capacity=2*1024*1024;char *data=malloc(capacity+1);if(!data)return false;
 bool complete=false;size_t used=0;
 if(send_line("{\"command\":\"serverinfo\"}\n")==0){
  while(used<capacity){ssize_t n=recv(output,data+used,capacity-used,0);if(n<=0)break;used+=(size_t)n;data[used]=0;char *end=strchr(data,'\n');if(end){*end=0;complete=true;break;}}
 }
 bool ok=complete && layout_from_json(data,&led_blends,&led_count);free(data);interpolation_enabled=ok;return ok;
}
static void stop_capture(void){
 free(led_blends);led_blends=NULL;led_count=0;interpolation_enabled=false;
 color_valid=false;memset(rgb,0,sizeof(rgb));memset(sent_rgb,0,sizeof(sent_rgb));
 if(output>=0){if(frames){char line[80];snprintf(line,sizeof(line),"{\"command\":\"clear\",\"priority\":%d}\n",PRIORITY);send_line(line);}close(output);output=-1;}
 if(capture){vt_finish(capture);capture=NULL;}
}
static bool respond(void *h,void *m,const char *s){if(!h||!m)return true;LSError e;error_init(&e);bool ok=reply(h,m,s,&e);error_free(&e);return ok;}
static bool start(void *h,void *m,void *ctx){
 (void)ctx;const char *p=m?payload(m):NULL;
 if(p && strcmp(p,"{}")!=0)return respond(h,m,"{\"returnValue\":false,\"errorText\":\"Start expects an empty object; endpoint is localhost:19444\"}");
 if(capture)return respond(h,m,"{\"returnValue\":true,\"running\":true}");
 if(m){retry_at=0;recovery_attempts=0;}
 frames=0;revoked=0;last_error[0]=0;lg_diagnostic[0]=0;
 if(!enter_developer_mode()){set_error("Developer initialization needs root. Elevate this own service through Homebrew Channel.");return respond(h,m,"{\"returnValue\":false,\"errorText\":\"Root access is required for temporary LG developer initialization\"}");}
 capture=vt_init(policy);
 if(!leave_developer_mode()){stop_capture();set_error("Could not restore LG developer marker; check vt_devmode before retrying");return respond(h,m,"{\"returnValue\":false,\"errorText\":\"Developer marker cleanup failed\"}");}
 if(!capture){set_error("LG could not acquire the histogram capture resource. Check for another active capture source; no colors are being sent.");return respond(h,m,"{\"returnValue\":false,\"errorCode\":1,\"errorText\":\"LG refused histogram initialization. Another capture source may hold VT_CAPTURE. No colors are being sent.\"}");}
 if(vt_preset(capture,0)!=0){stop_capture();set_error("LG histogram preset initialization failed");return respond(h,m,"{\"returnValue\":false,\"errorText\":\"Histogram preset failed\"}");}
 capture_preset=letterbox_mode=settling=0;pending_letterbox=-1;next_border_probe=0;
 output=socket(AF_INET,SOCK_STREAM,0);
 struct timeval timeout={1,0};
 if(output>=0){setsockopt(output,SOL_SOCKET,SO_SNDTIMEO,&timeout,sizeof(timeout));setsockopt(output,SOL_SOCKET,SO_RCVTIMEO,&timeout,sizeof(timeout));}
 struct sockaddr_in a={.sin_family=AF_INET,.sin_port=htons(19444),.sin_addr={htonl(INADDR_LOOPBACK)}};
 if(output<0||connect(output,(void*)&a,sizeof(a))!=0){stop_capture();set_error("Hyperion JSON server is unavailable on localhost:19444");return respond(h,m,"{\"returnValue\":false,\"errorText\":\"Cannot connect to local Hyperion\"}");}
 if(!read_layout()){stop_capture();set_error("Cannot read a valid LED layout from Hyperion");return respond(h,m,"{\"returnValue\":false,\"errorText\":\"Cannot read Hyperion LED layout\"}");}
 return respond(h,m,"{\"returnValue\":true,\"running\":true}");
}
static bool stop(void *h,void *m,void *ctx){(void)ctx;retry_at=0;recovery_attempts=0;stop_capture();return respond(h,m,"{\"returnValue\":true,\"running\":false}");}
static bool autostart(void *h,void *m,void *ctx){
 (void)ctx;const char *p=m?payload(m):NULL;bool enabled;
 if(p && strcmp(p,"{\"enabled\":true}")==0)enabled=true;
 else if(p && strcmp(p,"{\"enabled\":false}")==0)enabled=false;
 else return respond(h,m,"{\"returnValue\":false,\"errorText\":\"Expected enabled boolean\"}");
 if(geteuid()!=0 || !set_boot(enabled))return respond(h,m,"{\"returnValue\":false,\"errorText\":\"Autostart could not be saved. Check own service root elevation and boot hook.\"}");
 return respond(h,m,enabled?"{\"returnValue\":true,\"autostart\":true}":"{\"returnValue\":true,\"autostart\":false}");
}
static bool status(void *h,void *m,void *ctx){
 (void)ctx;char json[4096],safe[512];size_t n=0;
 for(size_t i=0;last_error[i] && n+2<sizeof(safe);i++){char c=last_error[i];if(c=='"'||c=='\\')safe[n++]='\\';if((unsigned char)c>=32)safe[n++]=c;}safe[n]=0;
 int used=snprintf(json,sizeof(json),"{\"returnValue\":true,\"version\":\"0.6.3\",\"running\":%s,\"colorValid\":%s,\"frames\":%lu,\"endpoint\":\"127.0.0.1:19444\",\"priority\":%d,\"uid\":%u,\"drmVerified\":false,\"mappingVerified\":false,\"outputColor\":[%u,%u,%u],\"error\":\"%s\",\"regions\":[",capture?"true":"false",color_valid?"true":"false",frames,PRIORITY,(unsigned)geteuid(),sent_rgb[0],sent_rgb[1],sent_rgb[2],safe);
 for(int i=0;i<16;i++)used+=snprintf(json+used,sizeof(json)-used,"%s[%u,%u,%u]",i?",":"",rgb[i][0],rgb[i][1],rgb[i][2]);
 used+=snprintf(json+used,sizeof(json)-used,"],\"ledCount\":%u,\"spatialOutput\":%s,\"blackPercent\":[",led_count,led_count?"true":"false");
 for(int i=0;i<16;i++)used+=snprintf(json+used,sizeof(json)-used,"%s%u",i?",":"",black_percent[i]);
 snprintf(json+used,sizeof(json)-used,"],\"preset\":%u,\"saturationGain\":2.0,\"brightnessGain\":1.33,\"spatialInterpolation\":%s,\"blackBarDetection\":true,\"autostart\":%s,\"letterboxDetected\":%s}",capture_preset,interpolation_enabled?"true":"false",boot_enabled()?"true":"false",capture&&letterbox_mode?"true":"false");return respond(h,m,json);
}
static int tick(void *unused){
 (void)unused;
 if(!capture){
  if(retry_at && milliseconds()>=retry_at){
   retry_at=0;recovery_attempts++;start(NULL,NULL,NULL);
   if(!capture && recovery_attempts<6)retry_at=milliseconds()+5000;
  }
  return 1;
 }
 if(revoked){set_error("LG revoked capture; retrying initialization after source transition");stop_capture();recovery_attempts=0;retry_at=milliseconds()+5000;return 1;}
 if(settling){settling--;return 1;}
 if(letterbox_mode && capture_preset==1 && milliseconds()>=next_border_probe){
  if(vt_preset(capture,0)!=0){set_error("Border probe preset failed");stop_capture();return 1;}
  capture_preset=0;settling=2;return 1;
 }
 Histogram hist[16]={0};int rc=vt_get(capture,hist);
 if(rc){set_error("LG histogram read failed; streaming stopped");stop_capture();return 1;}
 unsigned aggregate[3]={0};
 for(int i=0;i<16;i++){
  black_percent[i]=100;
  for(int c=0;c<3;c++){
   uint64_t total=0,black=0;
   for(int b=0;b<256;b++){total+=hist[i].bins[c][b];if(b<=12)black+=hist[i].bins[c][b];}
   unsigned fraction=total?(unsigned)(100*black/total):0;
   if(fraction<black_percent[i])black_percent[i]=fraction;
  }
 }
 for(int i=0;i<16;i++)for(int c=0;c<3;c++){
  rgb[i][c]=histogram_mean(hist[i].bins[c]);aggregate[c]+=rgb[i][c];
 }
 if(capture_preset==0){
  int candidate=letterbox_candidate(black_percent,rgb);uint64_t now=milliseconds();
  if(candidate<0)pending_letterbox=-1;
  else if(candidate!=pending_letterbox){pending_letterbox=candidate;candidate_since=now;}
  else if((unsigned)candidate!=letterbox_mode && now-candidate_since>=2000)letterbox_mode=(unsigned)candidate;
  if(letterbox_mode){
   if(vt_preset(capture,1)!=0){set_error("Letterbox preset failed");stop_capture();return 1;}
   capture_preset=1;settling=2;next_border_probe=now+2000;return 1;
  }
 }
 for(int c=0;c<3;c++)sent_rgb[c]=(unsigned char)(aggregate[c]/16);
 unsigned char boosted_mean[3];boost_color(sent_rgb,boosted_mean);memcpy(sent_rgb,boosted_mean,3);
 size_t capacity=256+(size_t)led_count*12;char *line=malloc(capacity);
 if(!line){set_error("Color buffer allocation failed");stop_capture();return 1;}
 size_t written=(size_t)snprintf(line,capacity,"{\"command\":\"color\",\"priority\":%d,\"duration\":1500,\"origin\":\"HyperDRM\",\"color\":[",PRIORITY);
 for(unsigned i=0;i<led_count;i++){
  unsigned char interpolated[3],boosted[3];LedBlend blend=led_blends[i];
  blend_color(rgb[blend.a],rgb[blend.b],blend.weight,interpolated);boost_color(interpolated,boosted);
  written+=(size_t)snprintf(line+written,capacity-written,"%s%u,%u,%u",i?",":"",boosted[0],boosted[1],boosted[2]);
 }
 snprintf(line+written,capacity-written,"]}\n");
 int failed=send_line(line);free(line);
 if(failed){set_error("Hyperion connection lost");stop_capture();return 1;}
 char answer[4096];size_t used=0;bool complete=false;
 while(used+1<sizeof(answer)){ssize_t len=recv(output,answer+used,sizeof(answer)-used-1,0);if(len<=0)break;used+=(size_t)len;answer[used]=0;if(strchr(answer,'\n')){complete=true;break;}}
 if(!complete){set_error("Hyperion did not acknowledge color data");stop_capture();return 1;}
 if(!strstr(answer,"\"success\":true")&&!strstr(answer,"\"success\": true")){set_error("Hyperion rejected the color command");stop_capture();return 1;}
 color_valid=true;frames++;return 1;
}
static void *symbol(void *lib,const char *name){void *s=dlsym(lib,name);if(!s){fprintf(stderr,"Missing symbol %s\n",name);exit(2);}return s;}
int main(void){
 signal(SIGTERM,quit_signal);signal(SIGINT,quit_signal);
 if(boot_enabled() && !set_boot(true))fprintf(stderr,"Own legacy boot hook migration failed\n");
 void *vt=dlopen("libvtcapture.so.1",RTLD_NOW),*ls=dlopen("libluna-service2.so.3",RTLD_NOW),*glib=dlopen("libglib-2.0.so.0",RTLD_NOW);
 if(!vt||!ls||!glib){fprintf(stderr,"Missing LG library: %s\n",dlerror());return 2;}
 vt_init=symbol(vt,"vtCapture_initHistogram");vt_preset=symbol(vt,"vtCapture_setPresetHistogram");vt_get=symbol(vt,"vtCapture_getHistogram");vt_finish=symbol(vt,"vtCapture_finalizeHistogram");
 reply=symbol(ls,"LSMessageReply");payload=symbol(ls,"LSMessageGetPayload");error_init=symbol(ls,"LSErrorInit");error_free=symbol(ls,"LSErrorFree");
 bool (*reg)(const char*,void**,LSError*)=symbol(ls,"LSRegister");
 bool (*category)(void*,const char*,Method*,void*,void*,LSError*)=symbol(ls,"LSRegisterCategory");
 bool (*attach)(void*,void*,LSError*)=symbol(ls,"LSGmainAttach");
 bool (*unreg)(void*,LSError*)=symbol(ls,"LSUnregister");
 void *(*loop_new)(void*,int)=symbol(glib,"g_main_loop_new");
 int (*iterate)(void*,int)=symbol(glib,"g_main_context_iteration");
 unsigned (*timer)(unsigned,int(*)(void*),void*)=symbol(glib,"g_timeout_add");
 void (*loop_unref)(void*)=symbol(glib,"g_main_loop_unref");
 void *handle=NULL,*loop=loop_new(NULL,0);LSError e;error_init(&e);
 Method methods[]={{"start",start,0},{"stop",stop,0},{"status",status,0},{"autostart",autostart,0},{NULL,NULL,0}};
 if(!reg(SERVICE,&handle,&e)||!category(handle,"/",methods,NULL,NULL,&e)||!attach(handle,loop,&e)){fprintf(stderr,"Luna registration failed: %s\n",e.message?e.message:"unknown");error_free(&e);return 3;}
 timer(100,tick,NULL);
 while(!quitting){iterate(NULL,0);usleep(10000);}
 stop_capture();unreg(handle,&e);error_free(&e);loop_unref(loop);return 0;
}