1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
import { BaseAction } from "resource://normandy/actions/BaseAction.sys.mjs";
const lazy = {};
ChromeUtils.defineESModuleGetters(lazy, {
ActionSchemas: "resource://normandy/actions/schemas/index.sys.mjs",
PrefUtils: "resource://normandy/lib/PrefUtils.sys.mjs",
PreferenceRollouts: "resource://normandy/lib/PreferenceRollouts.sys.mjs",
TelemetryEnvironment: "resource://gre/modules/TelemetryEnvironment.sys.mjs",
TelemetryEvents: "resource://normandy/lib/TelemetryEvents.sys.mjs",
});
export class PreferenceRollbackAction extends BaseAction {
get schema() {
return lazy.ActionSchemas["preference-rollback"];
}
async _run(recipe) {
const { rolloutSlug } = recipe.arguments;
const rollout = await lazy.PreferenceRollouts.get(rolloutSlug);
if (lazy.PreferenceRollouts.GRADUATION_SET.has(rolloutSlug)) {
// graduated rollouts can't be rolled back
lazy.TelemetryEvents.sendEvent(
"unenrollFailed",
"preference_rollback",
rolloutSlug,
{
reason: "in-graduation-set",
enrollmentId:
rollout?.enrollmentId ??
lazy.TelemetryEvents.NO_ENROLLMENT_ID_MARKER,
}
);
throw new Error(
`Cannot rollback rollout in graduation set "${rolloutSlug}".`
);
}
if (!rollout) {
this.log.debug(`Rollback ${rolloutSlug} not applicable, skipping`);
return;
}
switch (rollout.state) {
case lazy.PreferenceRollouts.STATE_ACTIVE: {
this.log.info(`Rolling back ${rolloutSlug}`);
rollout.state = lazy.PreferenceRollouts.STATE_ROLLED_BACK;
for (const { preferenceName, previousValue } of rollout.preferences) {
lazy.PrefUtils.setPref(preferenceName, previousValue, {
branch: "default",
});
}
await lazy.PreferenceRollouts.update(rollout);
lazy.TelemetryEvents.sendEvent(
"unenroll",
"preference_rollback",
rolloutSlug,
{
reason: "rollback",
enrollmentId:
rollout.enrollmentId ||
lazy.TelemetryEvents.NO_ENROLLMENT_ID_MARKER,
}
);
lazy.TelemetryEnvironment.setExperimentInactive(rolloutSlug);
break;
}
case lazy.PreferenceRollouts.STATE_ROLLED_BACK: {
// The rollout has already been rolled back, so nothing to do here.
break;
}
case lazy.PreferenceRollouts.STATE_GRADUATED: {
// graduated rollouts can't be rolled back
lazy.TelemetryEvents.sendEvent(
"unenrollFailed",
"preference_rollback",
rolloutSlug,
{
reason: "graduated",
enrollmentId:
rollout.enrollmentId ||
lazy.TelemetryEvents.NO_ENROLLMENT_ID_MARKER,
}
);
throw new Error(
`Cannot rollback already graduated rollout ${rolloutSlug}`
);
}
default: {
throw new Error(
`Unexpected state when rolling back ${rolloutSlug}: ${rollout.state}`
);
}
}
}
async _finalize() {
await lazy.PreferenceRollouts.saveStartupPrefs();
}
}
|
