Compare commits
21
Commits
416e9de7ce
...
test
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9513eb488b | ||
|
|
a51c49e8f4 | ||
|
|
dba476e78a | ||
|
|
87a738461c | ||
|
|
40c9b7442d | ||
|
|
c1aa6f3c41 | ||
|
|
9102d03ef4 | ||
|
|
5c21aa767f | ||
|
|
3187248fb9 | ||
|
|
ed05a8eca5 | ||
|
|
02ec54fd16 | ||
|
|
9fffa861e7 | ||
|
|
3a73c9cf6e | ||
|
|
387e739a07 | ||
|
|
39d1ac1474 | ||
|
|
2aabc9ab37 | ||
|
|
2f964dabcc | ||
|
|
47576392a9 | ||
|
|
31e771eb9f | ||
|
|
935adf98ad | ||
|
|
39b2afaf1f |
@@ -28,6 +28,7 @@ FROM dvcorg/cml:0-dvc2-base1-gpu
|
|||||||
ENV APP_HOME = /home/app
|
ENV APP_HOME = /home/app
|
||||||
RUN mkdir -p $APP_HOME
|
RUN mkdir -p $APP_HOME
|
||||||
WORKDIR $APP_HOME
|
WORKDIR $APP_HOME
|
||||||
|
RUN mkdir -p /home/app/repo
|
||||||
|
|
||||||
# install packages from builder
|
# install packages from builder
|
||||||
COPY --from=builder /usr/src/app/wheels /wheels
|
COPY --from=builder /usr/src/app/wheels /wheels
|
||||||
|
|||||||
+46
-22
@@ -1,50 +1,74 @@
|
|||||||
import uvicorn
|
import uvicorn
|
||||||
from fastapi import FastAPI, Request
|
from fastapi import FastAPI, Request, HTTPException
|
||||||
|
from fastapi.responses import HTMLResponse
|
||||||
from dotenv import load_dotenv
|
from dotenv import load_dotenv
|
||||||
import logging
|
import logging
|
||||||
from pydantic import BaseModel
|
|
||||||
from typing import List
|
from typing import List
|
||||||
import os
|
import os
|
||||||
|
|
||||||
from utils import clone_repository
|
from utils import clone_repository, load_runner_script
|
||||||
from setup_logging import setup_logging
|
from setup_logging import setup_logging
|
||||||
|
from gitea_request import GiteaRequest
|
||||||
class GiteaRequest(BaseModel):
|
|
||||||
ref: str
|
|
||||||
before: str
|
|
||||||
after: str
|
|
||||||
compare_url: str
|
|
||||||
commits: List[dict]
|
|
||||||
total_commits: int
|
|
||||||
head_commit: dict
|
|
||||||
repository: dict
|
|
||||||
pusher: dict
|
|
||||||
sender: dict
|
|
||||||
|
|
||||||
app = FastAPI()
|
app = FastAPI()
|
||||||
|
|
||||||
@app.get('/')
|
@app.get('/', response_class=HTMLResponse)
|
||||||
async def root():
|
async def root():
|
||||||
return '<h1>Welcome to Gitea Runsner</h1>'
|
content_str = """
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<title>Gitea Runner</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>Welcome to Gitea Runner</h1>
|
||||||
|
<p>see /docs for more information</p>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
"""
|
||||||
|
return HTMLResponse(content=content_str, status_code=200)
|
||||||
|
|
||||||
@app.post('/test')
|
@app.post('/test')
|
||||||
async def test(request: Request):
|
async def test(request: Request):
|
||||||
""" Test connection to server. """
|
""" Test connection to server. """
|
||||||
logging.debug('Content-Type: ', request.headers.get('content-type'))
|
logging.debug('Content-Type: ', request.headers.get('content-type'))
|
||||||
return {'status': 'success'}
|
raise HTTPException(
|
||||||
|
status_code=200,
|
||||||
|
detail='test completed'
|
||||||
|
)
|
||||||
|
|
||||||
@app.post('/event')
|
@app.post('/event')
|
||||||
def handle_event(request: GiteaRequest):
|
def handle_event(request: GiteaRequest):
|
||||||
""" Handle event from webhook. """
|
""" Handle event from webhook. """
|
||||||
# stop early if allowed
|
# stop early if asked
|
||||||
commit_msg = request.head_commit['message']
|
commit_msg = request.head_commit['message']
|
||||||
if commit_msg.lower().startswith('[skip ci]'):
|
if commit_msg.lower().startswith('[skip ci]'):
|
||||||
return {'status': 'received task'}
|
raise HTTPException(
|
||||||
|
status_code=200,
|
||||||
|
detail='skipping ci'
|
||||||
|
)
|
||||||
# clone repository
|
# clone repository
|
||||||
repo_url = request.repository['clone_url']
|
repo_url = request.repository['clone_url']
|
||||||
|
logging.info('cloning repository: {repo_url}')
|
||||||
local_repo_dir = clone_repository(repo_url)
|
local_repo_dir = clone_repository(repo_url)
|
||||||
print('local_repo_dir: ', local_repo_dir)
|
logging.info(f'local_repo_dir: {local_repo_dir}')
|
||||||
return {'status': 'success'}
|
# locate runner script
|
||||||
|
runner_script_path = local_repo_dir / 'runner_script.yml'
|
||||||
|
if not runner_script_path.exists():
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=200,
|
||||||
|
detail="no runner_script.yml in repository root"
|
||||||
|
)
|
||||||
|
# load runner commands
|
||||||
|
runner_script = load_runner_script(path=runner_script_path)
|
||||||
|
for pipeline_name, pipeline_step in runner_script.pipelines.items():
|
||||||
|
print(pipeline_name)
|
||||||
|
print(pipeline_step)
|
||||||
|
|
||||||
|
# send response
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=202,
|
||||||
|
detail='received task'
|
||||||
|
)
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
load_dotenv('dev.env')
|
load_dotenv('dev.env')
|
||||||
|
|||||||
@@ -0,0 +1,14 @@
|
|||||||
|
from pydantic import BaseModel
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
class GiteaRequest(BaseModel):
|
||||||
|
ref: str
|
||||||
|
before: str
|
||||||
|
after: str
|
||||||
|
compare_url: str
|
||||||
|
commits: List[dict]
|
||||||
|
total_commits: int
|
||||||
|
head_commit: dict
|
||||||
|
repository: dict
|
||||||
|
pusher: dict
|
||||||
|
sender: dict
|
||||||
@@ -0,0 +1,22 @@
|
|||||||
|
from pydantic import BaseModel
|
||||||
|
from typing import List, Dict
|
||||||
|
import yaml
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
class Pipeline(BaseModel):
|
||||||
|
name: str
|
||||||
|
requirements: str
|
||||||
|
script: List[str]
|
||||||
|
|
||||||
|
class RunnerScript(BaseModel):
|
||||||
|
pipelines: Dict[str, Pipeline]
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
path = Path(__file__).parent.parent / 'runner_script.yml'
|
||||||
|
with open(path, 'r') as fh:
|
||||||
|
script = yaml.safe_load(fh)
|
||||||
|
runner_script = RunnerScript.parse_obj(script)
|
||||||
|
print(runner_script)
|
||||||
|
for k, v in runner_script.pipelines.items():
|
||||||
|
print(k)
|
||||||
|
print(v)
|
||||||
@@ -1,8 +1,12 @@
|
|||||||
from pydantic import AnyHttpUrl
|
from pydantic import AnyHttpUrl
|
||||||
|
from typing import List
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from git import Repo
|
from git import Repo
|
||||||
|
import logging
|
||||||
import shutil
|
import shutil
|
||||||
import os
|
import os
|
||||||
|
import yaml
|
||||||
|
from runner_script import RunnerScript
|
||||||
|
|
||||||
def clone_repository(repo_url: AnyHttpUrl) -> Path:
|
def clone_repository(repo_url: AnyHttpUrl) -> Path:
|
||||||
# extract repo name
|
# extract repo name
|
||||||
@@ -16,4 +20,13 @@ def clone_repository(repo_url: AnyHttpUrl) -> Path:
|
|||||||
dst_dir.mkdir()
|
dst_dir.mkdir()
|
||||||
# git clone repo
|
# git clone repo
|
||||||
Repo.clone_from(url=repo_url, to_path=dst_dir)
|
Repo.clone_from(url=repo_url, to_path=dst_dir)
|
||||||
|
logging.debug('finished')
|
||||||
return dst_dir
|
return dst_dir
|
||||||
|
|
||||||
|
def load_runner_script(path: Path) -> RunnerScript:
|
||||||
|
assert path.exists()
|
||||||
|
with open(path, 'r') as fh:
|
||||||
|
script = yaml.safe_load(fh)
|
||||||
|
runner_script = RunnerScript.parse_obj(script)
|
||||||
|
return runner_script
|
||||||
|
|
||||||
+5
-1
@@ -4,6 +4,10 @@ services:
|
|||||||
build: .
|
build: .
|
||||||
container_name: gitea-runner
|
container_name: gitea-runner
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
env_file: prod.env
|
|
||||||
ports:
|
ports:
|
||||||
- "8000:8000"
|
- "8000:8000"
|
||||||
|
environment:
|
||||||
|
- LOGGER_LEVEL=debug
|
||||||
|
- LISTEN_IP=0.0.0.0
|
||||||
|
- LISTEN_PORT=8000
|
||||||
|
- REPO_DIR=/home/app/repo/
|
||||||
@@ -1,4 +0,0 @@
|
|||||||
LOGGER_LEVEL = info
|
|
||||||
LISTEN_IP = 0.0.0.0
|
|
||||||
LISTEN_PORT = 8000
|
|
||||||
REPO_DIR = /usr/src/app/repo/
|
|
||||||
@@ -37,6 +37,7 @@ pydantic==1.10.11
|
|||||||
Pygments==2.15.1
|
Pygments==2.15.1
|
||||||
python-dateutil==2.8.2
|
python-dateutil==2.8.2
|
||||||
python-dotenv==1.0.0
|
python-dotenv==1.0.0
|
||||||
|
PyYAML==6.0
|
||||||
pyzmq==25.1.0
|
pyzmq==25.1.0
|
||||||
six==1.16.0
|
six==1.16.0
|
||||||
smmap==5.0.0
|
smmap==5.0.0
|
||||||
|
|||||||
@@ -2,3 +2,4 @@ uvicorn==0.16.0
|
|||||||
fastapi==0.99.1
|
fastapi==0.99.1
|
||||||
GitPython==3.1.31
|
GitPython==3.1.31
|
||||||
python-dotenv==1.0.0
|
python-dotenv==1.0.0
|
||||||
|
PyYAML==6.0
|
||||||
@@ -0,0 +1,12 @@
|
|||||||
|
definitions:
|
||||||
|
steps:
|
||||||
|
step: &step-test
|
||||||
|
name: test
|
||||||
|
requirements: req-prod.txt
|
||||||
|
script:
|
||||||
|
- command arg1 arg2
|
||||||
|
- command2 arg1 arg2
|
||||||
|
|
||||||
|
pipelines:
|
||||||
|
default:
|
||||||
|
<<: *step-test
|
||||||
Reference in New Issue
Block a user