!7

static int id = 1400000;

concept FileMeta {

  int fileID;

  S name;

  S md5;

  bool authed;

  int redirectTo;

  *() {}

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

  void _doneLoading() {

    //print("_doneLoading " + fileID);

    metaMap.put(fileID, this);

  }

}

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

p {

  load("id");

  dbIndexing(FileMeta, 'fileID, FileMeta, 'name, FileMeta, 'md5);

}

svoid setMD5(FileMeta meta) {

  if (nempty(meta.md5)) ret;

  S md5 = "error";

  pcall {

    md5 = md5(fileFile(meta.fileID));

  }

  cset(meta, +md5);

}

static File filesDir() {

  ret prepareProgramDir("files");

}

synchronized html {

  print("uri: " + uri);

  uri = dropPrefix("/", uri);

  new Matches mm;

  if (eq(uri, "latest-id"))

    ret get fileID(highestByField created(list(FileMeta)));

  if (startsWith(uri, "info/", mm) && isInteger(mm.rest()))

    ret htitle("File #" + mm.rest())

      + fileIDToHTML(parseInt(mm.rest()));

  if (startsWith(uri, "name/", mm) && isInteger(mm.rest())) {

    int id = parseInt(mm.rest());

    FileMeta meta = metaMap.get(id);

    if (meta == null) ret subBot_serve404("ID not found: " + id);

    ret serveText(unnull(meta.name));

  }

  S possibleID = beforeSlash_orAll(uri);

  if (isInteger(possibleID)) {

    // serve file

    int id = parseInt(possibleID);

    FileMeta c = findMetaWithRedirect(id);

    if (c == null) ret subBot_serve404("Not found");

    if (nempty(params.get("remd5"))) {

      cset(c, md5 := null);

      setMD5(c);

    }

    File file = fileFile(c.fileID);

    S givenName = urldecode(afterSlash(uri));

    S serveAs = or2(givenName, c.name);

    S mimeType = or2(params.get("contentType"), params.get("ct"));

    if (empty(mimeType))

      mimeType = or2(extensionToMimeType(c.name), "application/binary");

    print("givenName=" + givenName + ", serveAs=" + serveAs + ", mimeType=" + mimeType);

    ret addHeader("Content-Disposition", "attachment; filename=" + quote(serveAs),

      call(getMainBot(), 'serveFile_maxCache, file, mimeType));

  }

  // Note: returns only authed files

  if (startsWith(uri, "latest-file-named/", mm)) {

    S name = urldecode(mm.get(0));

    FileMeta m = first(findAuthedFiles(name));

    ret struct(m == null ? null : ll(m.fileID, m.md5));

  }

  if (!webAuthed(params)) ret ahref(relativeBotLink(#1002590), "please log in");

  // AUTHED FROM HERE ON

  if (startsWith(uri, "delete/", mm) && isInteger(mm.rest())) {

    FileMeta m = conceptWhere(FileMeta, fileID := parseInt(mm.rest()));

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

    deleteFile(fileFile(m.fileID));

    deleteConcept(m);

    ret "Deleted";

  }

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

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

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

    if (meta == null)

      ret subBot_serve404("Not found");

    ret call(getMainBot(), "serveFile_maxCache", fileFile(meta.fileID), "application/binary");

  }

  // finds a file by md5

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

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

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

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

  }

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

    int from = parseInt(params.get("from"));

    int to = parseInt(params.get("to"));

    FileMeta meta = findConcept(FileMeta, fileID := from);

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

    cset(meta, redirectTo := to);

    ret "Redirected " + str(meta.fileID) + " to " + to;

  }

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

    print("upload called");

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

    SS files = getUploadFiles();

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

    int id = parseInt(params.get('id));

    if (id != 0 && hasConcept(FileMeta, fileID := id))

      ret "File exists: " + id;

    if (id == 0) id = newFileID();

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

    if (data != null) {

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

      saveBinaryFile(fileFile(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";

      copyFile(tmp, fileFile(id));

    }

    FileMeta m = cregister(new FileMeta(id, unnull(name)));

    cset(m, authed := webAuthed(params));

    setMD5(m);

    ret hrefresh(2, rawLink()) + p("OK:<br>" + ahref(rawLink(str(id)), fsI(id) + " - " + or2(name, "?")))

      + ahref(rawLink(), "[back]");

  }

  if (eq(uri, "uploadform")) ret htitle("Upload")

    + hmobilefix()

    + uploadform();

  if (eq(uri, "ids-and-md5s"))

    ret struct(flattenList(map(list(FileMeta), func(FileMeta m) { ll(m.fileID, m.md5) })));

  if (eq(uri, "typical-set")) {

    L<FileMeta> l = findAuthedFiles(

      "Circles From tvejysmllsmz.gz",

      "Webs from program #1011161 on tvejysmllsmz.gz");

    ret struct(flattenList(map(l, func(FileMeta m) { ll(m.fileID, m.md5) })));

  }

  if (eq(uri, "")) {

    // main list

    S msg = "";

    L<int> ids = collect(conceptsSortedByFieldDesc(FileMeta, 'created), 'fileID);

    ret htitle("General File Server")

      + (nempty(msg) ? p(msg) : "")

      + h3("Upload") + uploadform()

      + "<hr>"

      + p(n(ids, "File") + (empty(ids) ? "." : ":"))

      + ul(map(ids, func(int id) -> S { fileIDToHTML(id) })) 

      //+ p(ahref(pageLink("/uploadform"), "Upload a file"))

      ;

  }

  null;

}

static synchronized int newFileID() {

  while (fileFile(id).exists())

    ++id;

  int i = id++;

  save("id");

  ret i;

}

static File fileFile(int id) {

  ret new File(filesDir(), str(id));

}

sS uploadform() {

  ret huploadform(

        p("File to upload: " + hfileupload())

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

      + p(hsubmit("OK")),

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

}

static L<FileMeta> findAuthedFiles(S... names) {

  new L<FileMeta> l;

  for (S name : names)

    addIfNotNull(l, highestByField('id, conceptsWhere(FileMeta, +name, authed := true)));

  ret l;

}

sS fileIDToHTML(int id) {

  FileMeta meta = metaMap.get(id);

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

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

  //S dumbLink = rawLink(str(id));

  S coolLink = "https://botcompany.de/files/" + id + "/" + urlencode(title);

  S s = ahref(coolLink, linkText);

  s += " " + n(fileSize(fileFile(id)), "byte");

  if (meta != null) {

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

    if (meta.authed) s += " " + dottedSpan("*", "Authorized Upload");

    if (meta.redirectTo != 0) s = "redirected to #" + meta.redirectTo + " - " + s;

  }

  ret s;

}

static FileMeta findMetaWithRedirect(int id) {

  FileMeta c = findConcept(FileMeta, fileID := id);

  if (c != null && c.redirectTo != 0)

    c = findConcept(FileMeta, fileID := c.redirectTo);

  ret c;

}