1use gh_workflow::*;
2use indoc::indoc;
3
4use crate::tasks::workflows::{
5 runners,
6 steps::{self, CommonJobConditions, NamedJob, RepositoryTarget, generate_token, named},
7 vars::{self, StepOutput},
8};
9
10pub fn publish_extension_cli() -> Workflow {
11 let publish = publish_job();
12 let update_sha_in_zed = update_sha_in_zed(&publish);
13 let update_sha_in_extensions = update_sha_in_extensions(&publish);
14
15 named::workflow()
16 .on(Event::default().push(Push::default().tags(vec!["extension-cli".to_string()])))
17 .add_env(("CARGO_TERM_COLOR", "always"))
18 .add_env(("CARGO_INCREMENTAL", 0))
19 .add_job(publish.name, publish.job)
20 .add_job(update_sha_in_zed.name, update_sha_in_zed.job)
21 .add_job(update_sha_in_extensions.name, update_sha_in_extensions.job)
22}
23
24fn publish_job() -> NamedJob {
25 fn build_extension_cli() -> Step<Run> {
26 named::bash("cargo build --release --package extension_cli")
27 }
28
29 fn upload_binary() -> Step<Run> {
30 named::bash(r#"script/upload-extension-cli "$GITHUB_SHA""#)
31 .add_env((
32 "DIGITALOCEAN_SPACES_ACCESS_KEY",
33 vars::DIGITALOCEAN_SPACES_ACCESS_KEY,
34 ))
35 .add_env((
36 "DIGITALOCEAN_SPACES_SECRET_KEY",
37 vars::DIGITALOCEAN_SPACES_SECRET_KEY,
38 ))
39 }
40
41 named::job(
42 Job::default()
43 .with_repository_owner_guard()
44 .runs_on(runners::LINUX_DEFAULT)
45 .add_step(steps::checkout_repo())
46 .add_step(steps::cache_rust_dependencies_namespace())
47 .add_step(steps::setup_linux())
48 .add_step(build_extension_cli())
49 .add_step(upload_binary()),
50 )
51}
52
53fn update_sha_in_zed(publish_job: &NamedJob) -> NamedJob {
54 let (generate_token, generated_token) =
55 generate_token(vars::ZED_ZIPPY_APP_ID, vars::ZED_ZIPPY_APP_PRIVATE_KEY).into();
56
57 fn replace_sha() -> Step<Run> {
58 named::bash(indoc! {r#"
59 sed -i "s/ZED_EXTENSION_CLI_SHA: &str = \"[a-f0-9]*\"/ZED_EXTENSION_CLI_SHA: \&str = \"$GITHUB_SHA\"/" \
60 tooling/xtask/src/tasks/workflows/extension_tests.rs
61 "#})
62 }
63
64 fn regenerate_workflows() -> Step<Run> {
65 named::bash("cargo xtask workflows")
66 }
67
68 let (get_short_sha_step, short_sha) = get_short_sha();
69
70 named::job(
71 Job::default()
72 .with_repository_owner_guard()
73 .needs(vec![publish_job.name.clone()])
74 .runs_on(runners::LINUX_LARGE)
75 .add_step(generate_token)
76 .add_step(steps::checkout_repo())
77 .add_step(steps::cache_rust_dependencies_namespace())
78 .add_step(get_short_sha_step)
79 .add_step(replace_sha())
80 .add_step(regenerate_workflows())
81 .add_step(create_pull_request_zed(&generated_token, &short_sha)),
82 )
83}
84
85fn create_pull_request_zed(generated_token: &StepOutput, short_sha: &StepOutput) -> Step<Use> {
86 let title = format!(
87 "extension_ci: Bump extension CLI version to `{}`",
88 short_sha
89 );
90
91 steps::CreatePrStep::new(title, "update-extension-cli-sha", generated_token)
92 .with_body(indoc::indoc! {r#"
93 This PR bumps the extension CLI version used in the extension workflows to `${{ github.sha }}`.
94
95 Release Notes:
96
97 - N/A
98 "#})
99 .into()
100}
101
102fn update_sha_in_extensions(publish_job: &NamedJob) -> NamedJob {
103 let extensions_repo = RepositoryTarget::new("zed-industries", &["extensions"]);
104 let (generate_token, generated_token) =
105 generate_token(vars::ZED_ZIPPY_APP_ID, vars::ZED_ZIPPY_APP_PRIVATE_KEY)
106 .for_repository(extensions_repo)
107 .into();
108
109 fn checkout_extensions_repo(token: &StepOutput) -> Step<Use> {
110 named::uses(
111 "actions",
112 "checkout",
113 "11bd71901bbe5b1630ceea73d27597364c9af683", // v4
114 )
115 .add_with(("repository", "zed-industries/extensions"))
116 .add_with(("token", token.to_string()))
117 }
118
119 fn replace_sha() -> Step<Run> {
120 named::bash(indoc! {r#"
121 sed -i "s/ZED_EXTENSION_CLI_SHA: [a-f0-9]*/ZED_EXTENSION_CLI_SHA: $GITHUB_SHA/" \
122 .github/workflows/ci.yml
123 "#})
124 }
125
126 let (get_short_sha_step, short_sha) = get_short_sha();
127
128 named::job(
129 Job::default()
130 .with_repository_owner_guard()
131 .needs(vec![publish_job.name.clone()])
132 .runs_on(runners::LINUX_SMALL)
133 .add_step(generate_token)
134 .add_step(get_short_sha_step)
135 .add_step(checkout_extensions_repo(&generated_token))
136 .add_step(replace_sha())
137 .add_step(create_pull_request_extensions(&generated_token, &short_sha)),
138 )
139}
140
141fn create_pull_request_extensions(
142 generated_token: &StepOutput,
143 short_sha: &StepOutput,
144) -> Step<Use> {
145 let title = format!("Bump extension CLI version to `{}`", short_sha);
146
147 steps::CreatePrStep::new(title, "update-extension-cli-sha", generated_token)
148 .with_body(indoc::indoc! {r#"
149 This PR bumps the extension CLI version to https://github.com/zed-industries/zed/commit/${{ github.sha }}.
150 "#})
151 .with_labels("allow-no-extension")
152 .into()
153}
154
155fn get_short_sha() -> (Step<Run>, StepOutput) {
156 let step = named::bash(indoc::indoc! {r#"
157 echo "sha_short=$(echo "$GITHUB_SHA" | cut -c1-7)" >> "$GITHUB_OUTPUT"
158 "#})
159 .id("short-sha");
160
161 let step_output = vars::StepOutput::new(&step, "sha_short");
162
163 (step, step_output)
164}