mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2024-11-23 14:20:08 +00:00
create the circleci plugin
This commit is contained in:
parent
5bce72ef57
commit
a86e5876eb
3 changed files with 106 additions and 0 deletions
23
plugins/circleci/README.md
Normal file
23
plugins/circleci/README.md
Normal file
|
@ -0,0 +1,23 @@
|
|||
# CircleCi Plugin
|
||||
|
||||
This plugin provides easy to use cli commands to query circle ci job statuses
|
||||
|
||||
To use it, add `circleci` to the plugins array in your zshrc file:
|
||||
|
||||
```zsh
|
||||
plugins=(... circleci)
|
||||
```
|
||||
|
||||
## Prerequisites
|
||||
You need to have the `CIRCLECI_API_TOKEN` and `CIRCLECI_ORG_SLUG` as environment
|
||||
variables before calling the `circleci_status` function <br>
|
||||
You can learn how to add a circleci api token [here](https://circleci.com/docs/managing-api-tokens/) <br>
|
||||
The org slug takes the format of `{vcs}/{org_name}`
|
||||
|
||||
## Usage
|
||||
```shell
|
||||
> circleci_status
|
||||
```
|
||||
The above command would list down all the jobs (with their run status) on the
|
||||
repository and branch that you are currently on <br>
|
||||
You can also use the `cis` alias to run the above function
|
12
plugins/circleci/circleci.plugin.zsh
Normal file
12
plugins/circleci/circleci.plugin.zsh
Normal file
|
@ -0,0 +1,12 @@
|
|||
# Handle $0 according to the standard:
|
||||
# https://zdharma-continuum.github.io/Zsh-100-Commits-Club/Zsh-Plugin-Standard.html
|
||||
0="${${ZERO:-${0:#$ZSH_ARGZERO}}:-${(%):-%N}}"
|
||||
0="${${(M)0:#/*}:-$PWD/$0}"
|
||||
__CI_ZSH_DIR="${0:h:A}"
|
||||
|
||||
alias cis='circleci_status'
|
||||
|
||||
|
||||
function circleci_status() {
|
||||
python3 "$__CI_ZSH_DIR"/circleci_status.py "$(git_current_branch)" "$(git_repo_name)" | less
|
||||
}
|
71
plugins/circleci/circleci_status.py
Normal file
71
plugins/circleci/circleci_status.py
Normal file
|
@ -0,0 +1,71 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import requests
|
||||
import os
|
||||
import sys
|
||||
|
||||
CIRCLECI_API = "https://circleci.com/api/v2"
|
||||
org_slug = os.environ.get("CIRCLECI_ORG_SLUG")
|
||||
args = sys.argv
|
||||
|
||||
|
||||
class COLORS:
|
||||
reset = '\033[0m'
|
||||
|
||||
class FG:
|
||||
red = '\033[31m'
|
||||
green = '\033[32m'
|
||||
yellow = '\033[93m'
|
||||
|
||||
|
||||
def get_resp_items(resp):
|
||||
if resp.status_code != 200:
|
||||
sys.exit(0)
|
||||
items = resp.json()["items"]
|
||||
if len(items) == 0:
|
||||
sys.exit(0)
|
||||
return items
|
||||
|
||||
|
||||
def get_status_text(status):
|
||||
if status == "success":
|
||||
return COLORS.FG.green + "✓" + COLORS.reset
|
||||
elif status in ("running", "not_run", "retried", "queued"):
|
||||
return COLORS.FG.yellow + "·" + COLORS.reset
|
||||
else:
|
||||
return COLORS.FG.red + "˟" + COLORS.reset
|
||||
|
||||
|
||||
if not org_slug or len(args) != 3:
|
||||
sys.exit(0)
|
||||
|
||||
branch_name = args[1]
|
||||
repo_name = args[2]
|
||||
|
||||
# Here we query for all the pipelines belonging to the current branch in the current repo
|
||||
headers = {"Circle-token": os.environ.get("CIRCLECI_API_TOKEN")}
|
||||
url = f"{CIRCLECI_API}/project/{org_slug}/{repo_name}/pipeline"
|
||||
params = {"branch": branch_name}
|
||||
response = requests.get(url, headers=headers, params=params)
|
||||
pipelines = get_resp_items(response)
|
||||
# use the latest pipeline
|
||||
pipeline_id = pipelines[0]["id"]
|
||||
|
||||
# Now fetch the workflows for the selected pipeline
|
||||
url = f"{CIRCLECI_API}/pipeline/{pipeline_id}/workflow"
|
||||
response = requests.get(url, headers=headers)
|
||||
workflows = get_resp_items(response)
|
||||
# use the latest workflow
|
||||
workflow_id = workflows[0]["id"]
|
||||
|
||||
url = f"{CIRCLECI_API}/workflow/{workflow_id}/job"
|
||||
response = requests.get(url, headers=headers)
|
||||
jobs = get_resp_items(response)
|
||||
|
||||
for job in jobs:
|
||||
status = job["status"]
|
||||
name = job["name"]
|
||||
project_slug = job["project_slug"]
|
||||
job_number = job["job_number"]
|
||||
url = f"https://circleci.com/{project_slug}/{job_number}"
|
||||
print("{} {:<50} {}".format(get_status_text(status), name, url))
|
Loading…
Reference in a new issue