!7

static int id = 1100000;

concept ImageMeta {

  int imageID;

  bool pornFlag; // TODO: set

  S name;

  S md5; // md5 of RGBImage

  S format = "png";

  *() {}

  *(int *imageID, S *name) { _doneLoading(); change(); }

  void _doneLoading() {

    //print("_doneLoading " + imageID);

    metaMap.put(imageID, this);

  }

  S mimeType() {

    ret "image/" + or2(format, "png");

  }

}

static Map<Int, ImageMeta> metaMap = synchroMap();

p {

  load("id");

  db();

  indexConceptFieldIC(ImageMeta, 'name);

  indexConceptField(ImageMeta, 'imageID);

  for (ImageMeta meta)

    setMD5(meta);

}

svoid setMD5(ImageMeta meta) {

  if (nempty(meta.md5)) ret;

  S md5 = "error";

  pcall {

    md5 = md5OfRGBImage(new RGBImage(loadPNG(imageFile(meta.imageID))));

  }

  cset(meta, +md5);

}

static File imagesDir() {

  ret prepareProgramDir("images");

}

synchronized html {

  print("image server uri: " + uri + ", https: " + subBot_isHttps());

  uri = dropPrefix("/", uri);

  new Matches m;

  if (eq(uri, "setp0rn")) {

    setPornFlag(parseInt(params.get("id")), true);

    ret "OK";

  }

  if (eq(uri, "unp0rn")) {

    setPornFlag(parseInt(params.get("id")), false);

    ret "OK";

  }

  bool p0rnOK = false;

  // TODO: should use uri2?

  if (uri.endsWith("p") && isInteger(dropLast(uri))) {

    p0rnOK = true;

    uri = dropLast(uri);

  }

  S uri2 = dropAfterSlash(uri);

  if (eq(uri, "random-id")) ret random(list(ImageMeta)).imageID;

  if (eq(uri, "highest-id")) ret id-1;

  if (uri.startsWith("title/")) {

    int id = parseFirstInt(uri);

    ImageMeta meta = findConcept(ImageMeta, imageID := id);

    ret meta == null ? "" : meta.name;

  }

  if (isInteger(uri2)) {

    // serve image

    int id = parseInt(uri2);

    File file = imageFile(id);

    ImageMeta meta = findConcept(ImageMeta, imageID := id);

    if (meta == null)

      ret subBot_serve404("Not found");

    if (!p0rnOK && isPornFlag(id))

      ret subBot_serve404("n0 p0rn");

    ret call(getMainBot(), "serveFile_maxCache", imageFile(meta.imageID), meta.mimeType());

  }

  if (uri.startsWith("md5/")) {

    uri = dropPrefix("md5/", uri);

    bool jpg = uri.startsWith("jpg/");

    if (jpg) uri = dropPrefix("jpg/", uri);

    ImageMeta meta = findConcept(ImageMeta, md5 := uri);

    if (meta == null)

      ret "Not found";

    if (jpg && neq(meta.format, "jpeg"))

      ret call(getMainBot(), "serveByteArray_maxCache", toJPEG(imageFile(meta.imageID)), "image/jpeg");

    ret call(getMainBot(), "serveFile_maxCache", imageFile(meta.imageID), meta.mimeType());

  }

  if (uri.startsWith("checkmd5/")) {

    uri = dropPrefix("checkmd5/", uri);

    ImageMeta meta = findConcept(ImageMeta, md5 := uri);

    ret meta == null ? "Not found" : str(meta.imageID);

  }

  if (eq(uri, "upload") || swic(uri, "upload/")) {

    print("upload called.");

    pnlStruct(params);

    S name = params.get("name");

    SS files = getUploadFiles();

    print("got files: " + l(files));

    int id;

    S data = params.get("data");

    if (data != null) {

      print("got data: " + l(data));

      id = newImageID();

      saveBinaryFile(imageFile(id), hexToBytes(data));

    } else {

      String tmpfile = files.get("thefile");

      String originalName = params.get("thefile");

      name = or2(name, originalName);

      assertNotNull(tmpfile);

      File tmp = new File(tmpfile);

      long l = tmp.length();

      if (l == 0) ret "Empty file, exiting";

      id = newImageID();

      copyFile(tmp, imageFile(id));

    }

    ImageMeta meta = cregister ImageMeta(id, unnull(name));

    byte[] bytes = loadBeginningOfBinaryFile(imageFile(id), 16);

    cset(meta, format := isJPEG(bytes) ? "jpeg" /* correct for mime type! */ : isPNG(bytes) ? "png" : null);

    setMD5(meta);

    S imageID = fsI(meta.imageID);

    ret htitle(imageID)

      + p(ahref(myLink_nonRaw("/"), "Home.") + " " + ahref(myLink_nonRaw("/uploadform"), "Upload more."))

      + "OK (ID=" + imageID + "):<br>" + himg(imageLink(id))

      + uploadForm();

  }

  if (eq(uri, "uploadform"))

    ret htitle("Upload")

      + hmobilefix()

      + uploadForm();

  if (eqic(uri, "bot/list")) {

    bool withName = not0_gen(params.get('withName));

    bool withDate = not0_gen(params.get('withDate));

    Cl<S> names = subBot_paramsAsMultiMap().get('name);

    Cl<ImageMeta> l;

    if (empty(names)) l = list(ImageMeta);

    else l = (Set) concatListsAsSet(map(names, name -> conceptsWhereIC(ImageMeta, +name)));

    ret subBot_serveJSON(map(sortedByFieldDesc created(objectsWhere(l, pornFlag := false)), im -> {

      new LinkedHashMap<S, O> map;

      map.put(id := fsI(im.imageID));

      if (withName) map.put(name := im.name);

      if (withDate) map.put(uploaded := im.created);

      map.put(md5 := im.md5);

      ret map;

    } ));

  }

  if (eq(uri, "")) {

    // list images

    L<Int> ids = collect imageID(conceptsWhere(ImageMeta, pornFlag := false));

    sortListDesc(ids);

    ret htitle("Image Server")

      + p(ahref(pageLink("/uploadform"), "Upload."))

      + p(n(ids, "N0n-p0rn image") + ":") + ul(map(ids, func(Int id) -> S {

      ImageMeta meta = metaMap.get(id);

      S title = or2(getString(meta, "name"), "Untitled");

      S linkText = "#" + id + " - " + htmlencode(title);

      S s = ahref(snippetImageURL(id), linkText);

      if (meta != null)

        s += " [md5: " + ahref(rawLink("md5/" + meta.md5), meta.md5) + "]";

      ret s;

    }));

  }

  if (!webAuthed(params)) null;

  if (eq(uri, "setPornFlags")) {

    for (ImageMeta meta)

      if (isPornFlag(meta.imageID))

        cset(meta, pornFlag := true);

    ret "Yo";

  }

  if (startsWith(uri, "delete/", m)) {

    S s = m.rest();

    ImageMeta meta = isMD5(s)

      ? findConcept(ImageMeta, md5 := s)

      : findConcept(ImageMeta, imageID := parseInt(s));

    if (meta == null) ret "Not found";

    deleteFile(imageFile(meta));

    removeConcept(meta);

    ret "Deleted!";

  }

  null;

}

static synchronized int newImageID() {

  while (imageFile(id).exists())

    ++id;

  int i = id++;

  save("id");

  ret i;

}

static File imageFile(ImageMeta meta) {

  ret meta == null ? null : imageFile(meta.imageID);

}

static File imageFile(int id) {

  ret new File(imagesDir(), id + ".png");

}

static void setPornFlag(int id, bool flag) {

  saveTextFile(new File(imagesDir(), id + ".p0rn"), flag ? "t" : null);

}

static bool isPornFlag(int id) {

  ret eq("t", readTextFile(new File(imagesDir(), id + ".p0rn")));

}

sS uploadForm() {

  ret huploadform(

      "Image file: " + hfileupload("accept", "image/png,image/jpeg,image/gif") + " "

    + "<br>"

    + "Optional name: " + hinputfield("name")

    + hsubmit(),

    "action", rawLink("upload"));

}

sS imageLink(long id) {

  //ret rawLink(str(id));

  ret "https://botcompany.de/images/" + id;

}