import Controller, { inject as controller } from "@ember/controller"; import { action } from "@ember/object"; import { alias, empty, sort } from "@ember/object/computed"; import { next } from "@ember/runloop"; import { service } from "@ember/service"; import { popupAjaxError } from "discourse/lib/ajax-error"; import { grantableBadges } from "discourse/lib/grant-badge-utils"; import UserBadge from "discourse/models/user-badge"; import discourseComputed from "discourse-common/utils/decorators"; import { i18n } from "discourse-i18n"; import AdminUser from "admin/models/admin-user"; export default class AdminUserBadgesController extends Controller { @service dialog; @controller adminUser; @alias("adminUser.model") user; @alias("model") userBadges; @alias("badges") allBadges; @sort("model", "badgeSortOrder") sortedBadges; @empty("availableBadges") noAvailableBadges; badgeSortOrder = ["granted_at:desc"]; @discourseComputed("allBadges.[]", "userBadges.[]") availableBadges() { return grantableBadges(this.get("allBadges"), this.get("userBadges")); } @discourseComputed("model", "model.[]", "model.expandedBadges.[]") groupedBadges() { const allBadges = this.model; let grouped = {}; allBadges.forEach((b) => { grouped[b.badge_id] = grouped[b.badge_id] || []; grouped[b.badge_id].push(b); }); let expanded = []; const expandedBadges = allBadges.get("expandedBadges") || []; Object.values(grouped).forEach(function (badges) { let lastGranted = badges[0].granted_at; badges.forEach((badge) => { lastGranted = lastGranted < badge.granted_at ? badge.granted_at : lastGranted; }); if (badges.length === 1 || expandedBadges.includes(badges[0].badge.id)) { badges.forEach((badge) => expanded.push(badge)); return; } let result = { badge: badges[0].badge, granted_at: lastGranted, badges, count: badges.length, grouped: true, }; expanded.push(result); }); expanded.forEach((badgeGroup) => { const user = badgeGroup.granted_by; if (user) { badgeGroup.granted_by = AdminUser.create(user); } }); return expanded.sortBy("granted_at").reverse(); } @action expandGroup(userBadge) { const model = this.model; model.set("expandedBadges", model.get("expandedBadges") || []); model.get("expandedBadges").pushObject(userBadge.badge.id); } @action performGrantBadge() { UserBadge.grant( this.selectedBadgeId, this.get("user.username"), this.badgeReason ).then( (newBadge) => { this.set("badgeReason", ""); this.userBadges.pushObject(newBadge); next(() => { // Update the selected badge ID after the combobox has re-rendered. const newSelectedBadge = this.availableBadges[0]; if (newSelectedBadge) { this.set("selectedBadgeId", newSelectedBadge.get("id")); } }); }, function (error) { popupAjaxError(error); } ); } @action revokeBadge(userBadge) { return this.dialog.yesNoConfirm({ message: i18n("admin.badges.revoke_confirm"), didConfirm: () => { return userBadge.revoke().then(() => { this.model.removeObject(userBadge); }); }, }); } }