!7 // TODO: check user's rights when accepting an upload !include once #1026298 // DynTalkBot2 for JDA 4.0 standardBot1 RolesExporter { init { if (myName == null) setField(myName := "HyperCubes"); preprocessAtSelfToMyName = true; escapeAtEveryone = true; } sync S processSimplifiedLine(S s, O... _) null { try answer super.processSimplifiedLine(s, _); optPar long channelID; if (channelID == 0) null; optPar Message msg; Message.Attachment attachment = msg == null ? null : first(msg.getAttachments()); if (attachment != null) { File file = downloadFileForChannel(channelID); if (checkPerms(msg) != null) file = appendToFileName(file, ".temp"); // don't save for further operations print("Downloading attachment " + attachment.getFileName() + " to " + file); deleteFile(file); attachment.downloadToFile(file) .exceptionally(error -> { temp enter(); print("download fail"); ret null with postInChannel(channelID, "Couldn't download attachment"); }) .thenAccept(file2 -> postInChannel(channelID, handleAttachment(file2))); } if null (s = dropMyPrefixOrNull(s)) null; try { if "help" ret trimAllLines([[ @myName export - export roles + members as JSON @myName rights - check bot's rights @myName diff - diff last upload against server Upload a JSON file to get it diffed with current server ]].replace("@myName", myName); if "export" ret uploadFileInChannel(channelID, toUtf8(jsonEncode_breakAtNLevels(3, makeData())), "members-and-roles-" + guild.getIdLong() + ".json.txt", null, null); if "rights" ret "Manage roles: " + yesNo(canManageRoles()); if "diff" { File f = downloadFileForChannel(channelID); if (!fileExists(f)) ret "No past upload found in this channel"; ret pairA(diff(f)); } if "add roles" { try answer checkPerms(msg); Either p = diffWithLastUpload(channelID); if (!isB(p)) ret eitherGetA(p); Map diff = cast p.b(); Map rolesDiff = cast mapGet(diff, "roles"); print(+rolesDiff); new LS buf; for (JsonDiff.GatheredKey roleName, O roleDiff : unnull(rolesDiff)) { //buf.add(roleName + ": " + className(roleDiff)); if (roleDiff cast JsonDiff.Added) { //buf.add("Add role " + quote(roleName) + ": " + roleDiff); //buf.add(className(roleDiff.b)); Map roleMap = cast roleDiff.b; print(+roleMap); S name = getString name(roleMap); long permissions = getLong permissions(roleMap); queue(guild.createRole() .setName(name) .setPermissions(permissions)); buf.add("Added role " + quote(roleName)); } } ret or2(lines(buf), "Nothing to do"); } if "unused roles" ret or2(lines(unusedRoles()), "No unused roles"); if "delete unused roles" { try answer checkPerms(msg); ret or2(mapToLines(unusedRoles(), role -> { queue(role.delete()); ret "Deleted role: " + role.getName(); }), "No unused roles"); } if "missing member roles" ret missingMemberRoles(false, _); if "add member roles" ret missingMemberRoles(true, _); } catch print e { ret "Internal error: " + e; } } S missingMemberRoles(bool doIt, O... _) { optPar long channelID; Either p = diffWithLastUpload(channelID); if (!isB(p)) ret eitherGetA(p); Map diff = cast p.b(); Map membersDiff = cast mapGet(diff, "members"); new LS buf; for (JsonDiff.GatheredKey memberID, O memberDiff : unnull(membersDiff)) { long _memberID = (long) memberID!; Member member = guild.getMemberById(_memberID); if (member == null) continue; //buf.add(roleName + ": " + className(roleDiff)); if (memberDiff cast Map) { Cl roleDiffs = values((Map) get roles(memberDiff)); for (O roleDiff : roleDiffs) if (roleDiff cast JsonDiff.Added) { S roleName = getString name(roleDiff.b); buf.add(discordAt(_memberID) + " needs role " + roleName); if (doIt) { L roles = guild.getRolesByName(name); if (empty(roles)) buf.add("Role " + quote(roleName) + " not found"); for (Role role : roles) { queue(guild.addRoleToMember(member, role)); buf.add(" Added role " + role); } } } } } ret or2(lines(buf), "Nothing to do"); } L unusedRoles() { ret filter(guild.getRoles(), r -> empty(guild.getMembersWithRoles(r))); } S checkPerms(Message msg) { if (!msg.getMember().hasPermission(Permission.MANAGE_ROLES)) ret "You have no permission to do that you sneaky boy"; null; } File downloadFileForChannel(long channelID) { ret prepareCacheProgramFile(guildID + "/" + channelID); } bool canManageRoles() { ret guild.getSelfMember().hasPermission(Permission.MANAGE_ROLES); } Map makeData() { L roles = map(guild.getRoles(), r -> litorderedmap( "name" := r.getName(), "id" := r.getIdLong(), "permissions" := r.getPermissionsRaw())); L members = map(guild.getMembers(), m -> litorderedmap( "nickName" := m.getNickname(), "effectiveName" := m.getEffectiveName(), "userName" := m.getUser().getName(), "id" := m.getIdLong(), "isOwner" := trueOrNull(m.isOwner()), "roles" := map(m.getRoles(), r -> litorderedmap("name" := r.getName(), "id" := r.getIdLong())))); ret litorderedmap( dataType := "members and roles for guild", dateExported := dateWithSecondsGMT(), guild := litorderedmap( name := guild.getName(), id := guild.getIdLong()), +roles, +members); } S handleAttachment(File file) enter { try { print("Handling attachment " + f2s(file)); ret pairA(diff(file)); } catch print e { ret "Internal error"; } } // returns null if error // else: result for user, diff Pair diff(File file) { O json; try { json = decodeJson(loadTextFile(file)); } catch e { ret null with print("Not a JSON file: " + f2s(file)); } fixImportedData(json); new JsonDiff differ; // identify roles by names and members by ID differ.getListKey = path -> eq(last(path), 'roles) ? 'name : 'id; O diff = differ.diff(makeData(), json); //ret "Differences: " + structForUser(diff); ret pair("Differences:\n" + nicelyFormatJsonDiff(diff), diff); } void fixImportedData(O data) { changeKey((Map) data, 'allRoles, 'roles); } // either error or diff Either diffWithLastUpload(long channelID) { File f = downloadFileForChannel(channelID); if (!fileExists(f)) ret eitherA("No past upload found in this channel"); Pair p = diff(f); if (p == null) ret eitherA("There was an error with the last upload"); ret eitherB(p.b); } }