12 Commits

Author SHA1 Message Date
Morten Olsen
d3ad83ddf7 qa: add e2e tests 2024-01-17 08:14:19 +01:00
Morten Olsen
e5064ca905 fix: include runner in dependencies and await context create (#43)
Co-authored-by: Simple Systemer <simplesystemer@users.noreply.github.com>
Co-authored-by: denissb <spawnenius@gmail.com>
2024-01-15 21:54:33 +01:00
Morten Olsen
1c3b993ab2 docs: completed v1 docs (#36) 2024-01-15 15:10:19 +01:00
Morten Olsen
161a098c9f ci: docker fix (#37) 2024-01-15 15:09:51 +01:00
Morten Olsen
a08f9e1c91 chore: docker multi arch (#35) 2024-01-15 13:50:06 +01:00
Morten Olsen
e0c41d9220 fix: docker build (#34) 2024-01-14 13:04:11 +01:00
Morten Olsen
028b65587e fix: insecure tmp path (#33)
Fixes #13
2024-01-14 13:00:17 +01:00
Morten Olsen
7436b3439c feat: improved configuration (#31) 2024-01-14 12:49:54 +01:00
Morten Olsen
2109bc3af9 feat: add scheduler (#30) 2024-01-14 12:30:39 +01:00
Morten Olsen
eeaad68f6e feat: initial policy system and version (#29)
Resolves #22 #24 and #26
2024-01-14 09:45:42 +01:00
Morten Olsen
c7ca97f041 feat: local secrets (#28) 2024-01-14 09:45:10 +01:00
Morten Olsen
c8e02d8da4 ci: fix docker tag (#17) 2024-01-13 14:07:33 +01:00
98 changed files with 2297 additions and 558 deletions

View File

@@ -3,7 +3,7 @@ name: Node.js Package
on: on:
push: push:
pull_request: pull_request:
types: [opened] types: [opened, synchronize, reopened]
# release: # release:
# types: [created] # types: [created]
@@ -18,6 +18,7 @@ env:
permissions: permissions:
contents: read contents: read
packages: read packages: read
pull-requests: write
jobs: jobs:
build: build:
@@ -34,6 +35,11 @@ jobs:
- run: pnpm install - run: pnpm install
- run: pnpm run test - run: pnpm run test
- name: 'Report Coverage'
# Set if: always() to also generate the report if tests are failing
# Only works if you set `reportOnFailure: true` in your vite config as specified above
if: always()
uses: davelosert/vitest-coverage-report-action@v2
- run: pnpm run build - run: pnpm run build
- uses: actions/upload-artifact@v3 - uses: actions/upload-artifact@v3
with: with:
@@ -71,24 +77,41 @@ jobs:
steps: steps:
- name: Checkout repository - name: Checkout repository
uses: actions/checkout@v4 uses: actions/checkout@v4
- name: Log in to the Container registry - name: Log in to the Container registry
uses: docker/login-action@65b78e6e13532edd9afa3aa52ac7964289d1a9c1 uses: docker/login-action@65b78e6e13532edd9afa3aa52ac7964289d1a9c1
with: with:
registry: ${{ env.REGISTRY }} registry: ${{ env.REGISTRY }}
username: ${{ github.actor }} username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }} password: ${{ secrets.GITHUB_TOKEN }}
- name: Set up QEMU
uses: docker/setup-qemu-action@v3
- name: Set up Docker Buildx
uses: docker/setup-buildx-action@v3
- name: Retrieve version
run: |
echo "TAG_NAME=$(git describe --tag --abbrev=0)" >> $GITHUB_OUTPUT
id: version
- name: Extract metadata (tags, labels) for Docker - name: Extract metadata (tags, labels) for Docker
id: meta id: meta
uses: docker/metadata-action@9ec57ed1fcdbf14dcef7dfbe97b2010124a938b7 uses: docker/metadata-action@9ec57ed1fcdbf14dcef7dfbe97b2010124a938b7
with: with:
images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }} images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
tags: | tags: |
${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:latest latest
${{ steps.version.outputs.TAG_NAME }}
- name: Build and push Docker image - name: Build and push Docker image
uses: docker/build-push-action@f2a1d5e99d037542a71f64918e516c093c6f3fc4 uses: docker/build-push-action@f2a1d5e99d037542a71f64918e516c093c6f3fc4
with: with:
context: . context: .
file: ./docker/Dockerfile file: ./docker/Dockerfile
platforms: linux/amd64,linux/arm64
cache-from: type=gha
cache-to: type=gha,mode=max
push: true push: true
tags: ${{ steps.meta.outputs.tags }} tags: ${{ steps.meta.outputs.tags }}
labels: ${{ steps.meta.outputs.labels }} labels: ${{ steps.meta.outputs.labels }}

2
.gitignore vendored
View File

@@ -2,3 +2,5 @@
.turbo/ .turbo/
/.pnpm-store/ /.pnpm-store/
/out/ /out/
/coverage/
/e2e-tmp/

View File

@@ -1,10 +1,8 @@
![banner](./assets/banner.png) ![banner](./assets/banner.png)
# Welcome to Mini Loader! 🌐 # Welcome to Mini Loader! 🌐
Welcome to mini loader, a lightweight, Docker-based server solution for managing and executing workloads with ease. Designed for developers, small teams, and anyone in need of a simple yet powerful tool for running tasks, hosting API servers, or scheduling routine jobs. Welcome to mini loader, a lightweight server solution for managing and executing workloads with ease. Designed for developers, small teams, and anyone in need of a simple yet powerful tool for running tasks, hosting API servers, or scheduling routine jobs.
## Features ## Features
@@ -15,20 +13,37 @@ Welcome to mini loader, a lightweight, Docker-based server solution for managing
- **Task Scheduling**: Built-in support for cron-like job scheduling. - **Task Scheduling**: Built-in support for cron-like job scheduling.
- **HTTP Gateway**: Expose a HTTP server from your workloads - **HTTP Gateway**: Expose a HTTP server from your workloads
Also see [anti-features and limitations](./docs/anti-features.md) Also see [anti-features and limitations](./docs/02-anti-features.md)
:construction: This project is under active development and has not reached v1.0 yet. Expect some bugs and potential breaking changes in APIs. We appreciate your patience and welcome your feedback as we work towards a stable release!
For an overview of what's coming next, check out our roadmap at [GitHub Milestones](https://github.com/morten-olsen/mini-loader/milestones).
## Quick Start ## Quick Start
Get up and running with mini loader in just a few steps: Get up and running with mini loader in just a few steps:
1. **Install the CLI**: `npm install -g @morten-olsen/mini-loader-cli` ```bash
2. **Deploy the Server**: `docker run -p 4500:4500 —-name mini-loader ghcr.io/morten-olsen/mini-loader`. # Install the CLI and the server
3. **Get your access token**: `docker exec mini-loader mini-loader-server create-token` npm i -g @morten-olsen/mini-loader-cli @morten-olsen/mini-loader-server
4. **Login**: `mini-loader auth login http://localhost:4500`
5. **Push Your First Load**: `mini-loader loads push script.mjs -r -i first`
6. **See the logs**: `mini-loader logs ls -l first`
For a detailed guide on getting started, please refer to the [Getting Started Tutorial](./docs/getting-started.md). # Start the server
mini-loader-server start &
# Get your access token
mini-loader-server create-token
# Authenticate the CLI
mini-loader auth login
# Push your first workload
mini-loader loads push -r -ai my-script.js -i first
# See the output logs
mini-loader logs ls -l first
```
For a detailed guide on getting started, please refer to the [Getting Started Tutorial](./docs/01-getting-started.md).
## Support and Contributions ## Support and Contributions
@@ -43,4 +58,4 @@ mini loader is open-source software licensed under the [GPL-3 License](./LICENSE
## Let's Get Started! ## Let's Get Started!
Dive into the world of simplified workload management with mini loader. Start with our [Getting Started Tutorial](./docs/getting-started.md) and unleash the full potential of your tasks and applications! Dive into the world of simplified workload management with mini loader. Start with our [Getting Started Tutorial](./docs/01-getting-started.md) and unleash the full potential of your tasks and applications!

View File

@@ -27,6 +27,10 @@ COPY --from=builder /app/out/full/ .
RUN pnpm turbo run build --filter=@morten-olsen/mini-loader-server RUN pnpm turbo run build --filter=@morten-olsen/mini-loader-server
FROM base AS runner FROM base AS runner
ENV \
NODE_ENV=production \
DATA_DIR=/data \
CACHE_DIR=/cache
RUN apk add --no-cache jq curl RUN apk add --no-cache jq curl
WORKDIR /app WORKDIR /app
@@ -39,7 +43,7 @@ RUN chmod +x /entrypoint.sh
COPY --from=installer /app . COPY --from=installer /app .
EXPOSE 4500 EXPOSE 4500
VOLUME /app/data VOLUME /data
HEALTHCHECK \ HEALTHCHECK \
--interval=10s \ --interval=10s \

View File

@@ -7,6 +7,8 @@ GID=${GID:-1001}
addgroup --system --gid ${GID} nodejs && \ addgroup --system --gid ${GID} nodejs && \
adduser --system --uid ${UID} -G nodejs miniloader && \ adduser --system --uid ${UID} -G nodejs miniloader && \
mkdir -p /app/data mkdir -p ${DATA_DIR}
chown -R miniloader:nodejs /app/data mkdir -p ${CACHE_DIR}
chown -R miniloader:nodejs ${DATA_DIR}
chown -R miniloader:nodejs ${CACHE_DIR}
su miniloader -s /bin/sh -c "$CMD" su miniloader -s /bin/sh -c "$CMD"

View File

@@ -15,20 +15,18 @@ Before diving into mini loader, ensure you have the following:
## Contents ## Contents
- [Creating you first workload](./installation.md): Learn how to write workloads and execute them locally with the mini loader CLI - [Creating you first workload](./03-tutorial/01-first-workload.md): Learn how to write workloads and execute them locally with the mini loader CLI
- [Running the server](./pushing-managing-loads.md): Instructions on how to run the server locally using docker. - [Running the server](./03-tutorial/02-setup-server.md): Instructions on how to run the server locally using docker.
- [Interacting with the server](./interacting-with-server.md): Learn the basic commands used to manage workloads. - [Interacting with the server](./03-tutorial/03-interacting-with-server.md): Learn the basic commands used to manage workloads.
- [Managing secrets](./managing-secrets.md): Upload secrets to the server that can be used inside your scripts. - [Managing secrets](./03-tutorial/04-managing-secrets.md): Upload secrets to the server that can be used inside your scripts.
- [Authorization](./setting-up-oidc.md): Extend the authorization using OIDC - [Create an API](./03-tutorial/05-creating-an-api.md): Create a workload which exposes a HTTP api
- [Create an API](./creating-an-api.md): Create a workload which exposes a HTTP api
## Getting Help ## Getting Help
If you encounter any issues or have questions, please refer to the [FAQs](./faqs.md) If you encounter any issues or have questions, please refer to the [FAQs](./04-faqs.md)
## Let's Get Started! ## Let's Get Started!
Ready to streamline your workload management? Let's jump right into [creating your first workload](./first-workload.md) and set up the mini loader CLI! Ready to streamline your workload management? Let's jump right into [creating your first workload](./03-tutorial/01-first-workload.md) and set up the mini loader CLI!
[Next: create a workload](./03-tutorial/01-first-workload.md)
[Next: create a workload](./first-workload.md)

View File

@@ -46,4 +46,4 @@ After running the command, you should see an output confirming that a new artifa
Congratulations on setting up and running your first script with mini loader! You're now ready to take the next step. Congratulations on setting up and running your first script with mini loader! You're now ready to take the next step.
[Next: Setting Up the Server](./setup-server.md) [Next: Setting Up the Server](./02-setup-server.md)

View File

@@ -1,7 +1,8 @@
Certainly! Here's a revised version of your documentation page to make it Certainly! Here's a revised version of your documentation page to make it
## Quick Start with mini loader using Docker ## Quick Start with mini loader using Docker
This guide will help you quickly set up and run a mini loader server using Docker. Follow these simple steps to deploy your server and start interacting with it using the [mini-loader CLI](./first-workload.md). This guide will help you quickly set up and run a mini loader server using Docker. Follow these simple steps to deploy your server and start interacting with it using the [mini-loader CLI](./01-first-workload.md).
### Step 1: Deploy the mini loader Container ### Step 1: Deploy the mini loader Container
@@ -57,4 +58,4 @@ This command lists all the loads currently on your server, confirming that the C
You've successfully deployed and configured your mini loader server using Docker! You're now ready to start interacting with the server. You've successfully deployed and configured your mini loader server using Docker! You're now ready to start interacting with the server.
[Next: Interacting with the Server](./interacting-with-server.md) [Next: Interacting with the Server](./03-interacting-with-server.md)

View File

@@ -67,4 +67,4 @@ Replace `<id>` with the identifier of the artifact you wish to download.
You're now equipped to manage loads, runs, logs, and artifacts using the mini loader CLI. For advanced usage, such as managing secrets, proceed to the next section. You're now equipped to manage loads, runs, logs, and artifacts using the mini loader CLI. For advanced usage, such as managing secrets, proceed to the next section.
[Next: Managing Secrets](./managing-secrets.md) [Next: Managing Secrets](./04-managing-secrets.md)

View File

@@ -0,0 +1,81 @@
## Managing Secrets
### Introduction
In many workflows, accessing sensitive data such as API tokens or credentials is essential. To handle this securely, you can use secrets management. This tutorial demonstrates how to manage secrets using the CLI and implement them in a simple Node.js workload.
### Creating Secrets with the CLI
To create a new secret, use the `mini-loader` CLI as follows:
```bash
mini-loader secrets set <id>
```
For example, to store a GitHub personal access token, you would use:
```bash
mini-loader secrets set githubtoken
```
Upon execution, you'll be prompted to enter your access token.
### Implementing Secrets in Your Workload
Next, let's create a Node.js script (`github.js`) that uses this token to fetch your GitHub username and saves it as an artifact.
1. **Create `github.js` File:**
```javascript
import { secrets, artifacts } from '@morten-olsen/mini-loader';
import { Octokit } from '@octokit/rest';
// Retrieve the secret
const accessToken = secrets.get('githubtoken');
// Main async function to fetch and save GitHub username
async function run() {
const octokit = new Octokit({ auth: accessToken });
const user = await octokit.users.getAuthenticated();
await artifacts.create('user', JSON.stringify(user.data.login));
}
// Execute the function
run().catch(console.error);
```
This script initializes the Octokit client with the access token, fetches the authenticated user's data, and then saves the username as an artifact.
2. **Run the Script:**
Execute your script with `mini-loader`:
```bash
mini-loader loads push github.js -r -ai
```
### Managing Local Secrets
If you're running the script locally, you can manage secrets either by using a `.secrets` file or setting an environment variable.
1. **Using a `.secrets` File:**
Create a file named `.secrets` and add your token:
```
githubtoken=<your-token>
```
2. **Using Environment Variables:**
Prefix your environment variable with `ML_S_` and run the script:
```bash
ML_S_githubtoken=<your-token> mini-loader local run github.js -ai
```
### Conclusion
By following these steps, you can securely manage and use secrets within your workloads, enhancing the security and integrity of your applications.
[Next: Creating an API](./05-creating-an-api.md)

View File

@@ -0,0 +1,52 @@
## Creating an API Inside Your Workload
Workloads in mini loader can set up simple HTTP servers by connecting to a socket file, a feature supported by many JavaScript server libraries.
### Binding Your Workload to an HTTP Endpoint
To expose your workload as an HTTP server, specify the path parameter using the `getPath()` method provided by the `@morten-olsen/mini-loader` package. This method dynamically assigns a path for your API.
### Important Note
Please be aware that the gateway provided by mini loader isn't fully featured. As such, certain functionalities like streaming and WebSockets may not be supported.
### Example: Setting Up a Server with Fastify
Here's how you can create a simple API server using Fastify in TypeScript:
```typescript
import { http } from '@morten-olsen/mini-loader';
import fastify from 'fastify';
const server = fastify();
// Handling all requests and returning the requested URL
server.all('*', async (req) => {
return req.url;
});
// Listening on the path provided by mini loader
server.listen({
path: http.getPath(),
});
```
With this setup, your server will respond to all incoming requests by returning the requested URL.
### Deploying Your Workload
Now, you can push and run your workload just like any other script:
```bash
mini-loader loads push -r my-script.ts
```
### Accessing Your Server
After pushing your workload, mini loader will display the run ID. You can use this ID to access your server. For example, to make a request to your server, you can use `curl`:
```bash
curl http://localhost:4500/gateway/{your-run-id}
```
Replace `{your-run-id}` with the actual run ID provided by mini loader.

View File

@@ -8,21 +8,25 @@
"build": "turbo build", "build": "turbo build",
"build:dev": "tsc --build --watch", "build:dev": "tsc --build --watch",
"test:lint": "eslint ./packages/*/src", "test:lint": "eslint ./packages/*/src",
"test": "pnpm run test:lint" "test:code": "vitest --coverage",
"test": "pnpm run test:lint && pnpm run test:code"
}, },
"keywords": [], "keywords": [],
"author": "", "author": "",
"devDependencies": { "devDependencies": {
"@react-native-community/eslint-config": "^3.2.0", "@react-native-community/eslint-config": "^3.2.0",
"@vitest/coverage-v8": "^1.2.0",
"eslint": "^8.53.0", "eslint": "^8.53.0",
"prettier": "^2.8.8", "prettier": "^2.8.8",
"turbo": "^1.10.16", "turbo": "^1.10.16",
"typescript": "^5.3.3" "typescript": "^5.3.3",
"vite": "^5.0.11"
}, },
"dependencies": { "dependencies": {
"@pnpm/find-workspace-packages": "^6.0.9", "@pnpm/find-workspace-packages": "^6.0.9",
"@types/node": "^20.10.8", "@types/node": "^20.10.8",
"ts-node": "^10.9.2" "ts-node": "^10.9.2",
"vitest": "^1.2.0"
}, },
"homepage": "https://github.com/morten-olsen/mini-loader", "homepage": "https://github.com/morten-olsen/mini-loader",
"repository": { "repository": {

View File

@@ -1,4 +1,4 @@
#!/usr/bin/env node #!/usr/bin/env node
import 'source-map-support/register.js'; import 'source-map-support/register.js';
import '../dist/esm/index.js'; import '../dist/esm/bin.js';

View File

@@ -14,6 +14,9 @@
"files": [ "files": [
"./dist" "./dist"
], ],
"imports": {
"#pkg": "./package.json"
},
"exports": { "exports": {
".": { ".": {
"import": "./dist/esm/index.js" "import": "./dist/esm/index.js"
@@ -21,6 +24,7 @@
}, },
"dependencies": { "dependencies": {
"@morten-olsen/mini-loader-runner": "workspace:^", "@morten-olsen/mini-loader-runner": "workspace:^",
"@morten-olsen/mini-loader-server": "workspace:^",
"@rollup/plugin-auto-install": "^3.0.5", "@rollup/plugin-auto-install": "^3.0.5",
"@rollup/plugin-commonjs": "^25.0.7", "@rollup/plugin-commonjs": "^25.0.7",
"@rollup/plugin-json": "^6.1.0", "@rollup/plugin-json": "^6.1.0",
@@ -29,17 +33,18 @@
"@rollup/plugin-sucrase": "^5.0.2", "@rollup/plugin-sucrase": "^5.0.2",
"@trpc/client": "^10.45.0", "@trpc/client": "^10.45.0",
"commander": "^11.1.0", "commander": "^11.1.0",
"dotenv": "^16.3.1",
"env-paths": "^3.0.0", "env-paths": "^3.0.0",
"inquirer": "^9.2.12", "inquirer": "^9.2.12",
"ora": "^8.0.1", "ora": "^8.0.1",
"rollup": "^4.9.4", "rollup": "^4.9.4",
"rollup-plugin-node-polyfills": "^0.2.1", "rollup-plugin-node-polyfills": "^0.2.1",
"source-map-support": "^0.5.21", "source-map-support": "^0.5.21",
"superjson": "^2.2.1" "superjson": "^2.2.1",
"typedi": "^0.10.0"
}, },
"devDependencies": { "devDependencies": {
"@morten-olsen/mini-loader-configs": "workspace:^", "@morten-olsen/mini-loader-configs": "workspace:^",
"@morten-olsen/mini-loader-server": "workspace:^",
"@types/inquirer": "^9.0.7", "@types/inquirer": "^9.0.7",
"typescript": "^5.3.3" "typescript": "^5.3.3"
}, },

View File

@@ -0,0 +1,46 @@
import { ContainerInstance, Service } from 'typedi';
import { Context } from '../context/context.js';
import { Config } from '../config/config.js';
import { Client, createClient } from '../client/client.js';
import type { Runtime } from '@morten-olsen/mini-loader-server';
import { Terminal } from './output.js';
@Service()
class CliApi {
#container: ContainerInstance;
#client?: Client;
constructor(container: ContainerInstance) {
this.#container = container;
}
public get log() {
return this.#container.get(Terminal).log;
}
public get step() {
return this.#container.get(Terminal).step;
}
public get output() {
return this.#container.get(Terminal).output;
}
public get config() {
return this.#container.get(Config);
}
public get client() {
if (!this.#client) {
this.#client = createClient(this.context);
}
return this.#client;
}
public get context() {
return new Context(this.config);
}
}
export type { Runtime };
export { CliApi };

View File

@@ -0,0 +1,27 @@
import ora from 'ora';
import { Service } from 'typedi';
@Service()
class Terminal {
public log = (message: string) => {
console.log(message);
};
public output = (data: unknown) => {
console.table(data);
};
public step = async <T>(message: string, action: () => Promise<T>) => {
const spinner = ora(message).start();
try {
const result = await action();
await spinner.succeed();
return result;
} catch (err) {
await spinner.fail();
throw err;
}
};
}
export { Terminal };

7
packages/cli/src/bin.ts Normal file
View File

@@ -0,0 +1,7 @@
import { ContainerInstance } from 'typedi';
import { createClientCli } from './index.js';
const container = new ContainerInstance('client');
const program = createClientCli(container);
await program.parseAsync();

View File

@@ -1,8 +1,5 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const list = new Command('list'); const list = new Command('list');
@@ -22,11 +19,7 @@ list
.option('-a, --limit <limit>', 'Limit', '1000') .option('-a, --limit <limit>', 'Limit', '1000')
.action(async () => { .action(async () => {
const { runId, loadId, offset, limit } = list.opts(); const { runId, loadId, offset, limit } = list.opts();
const config = new Config(); const { step, client } = getApi(list);
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const artifacts = await step('Getting artifacts', async () => { const artifacts = await step('Getting artifacts', async () => {
return await client.artifacts.find.query({ return await client.artifacts.find.query({
runId, runId,

View File

@@ -1,10 +1,7 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { dirname, resolve } from 'path'; import { dirname, resolve } from 'path';
import { mkdir, writeFile } from 'fs/promises'; import { mkdir, writeFile } from 'fs/promises';
import { Config } from '../../config/config.js'; import { getApi } from '../../utils/command.js';
const pull = new Command('pull'); const pull = new Command('pull');
@@ -13,12 +10,8 @@ pull
.argument('<artifact-id>', 'Artifact ID') .argument('<artifact-id>', 'Artifact ID')
.argument('<file>', 'File to save') .argument('<file>', 'File to save')
.action(async (id, file) => { .action(async (id, file) => {
const config = new Config(); const { step, client } = getApi(pull);
const context = new Context(config.context);
const target = resolve(file); const target = resolve(file);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const artifact = await step('Getting artifact', async () => { const artifact = await step('Getting artifact', async () => {
const result = await client.artifacts.get.query(id); const result = await client.artifacts.get.query(id);
if (!result) { if (!result) {

View File

@@ -1,9 +1,6 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import inquirer from 'inquirer'; import inquirer from 'inquirer';
import { Config } from '../../config/config.js'; import { getApi } from '../../utils/command.js';
const remove = new Command('remove'); const remove = new Command('remove');
@@ -22,12 +19,8 @@ remove
.option('-o, --offset <offset>', 'Offset') .option('-o, --offset <offset>', 'Offset')
.option('-a, --limit <limit>', 'Limit', '1000') .option('-a, --limit <limit>', 'Limit', '1000')
.action(async () => { .action(async () => {
const { step, client } = getApi(remove);
const { runId, loadId, offset, limit } = remove.opts(); const { runId, loadId, offset, limit } = remove.opts();
const config = new Config();
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const response = await step('Preparing to delete', async () => { const response = await step('Preparing to delete', async () => {
return await client.artifacts.prepareRemove.query({ return await client.artifacts.prepareRemove.query({
runId, runId,

View File

@@ -1,28 +1,37 @@
import { Command } from 'commander'; import { Command } from 'commander';
import inquerer from 'inquirer'; import inquerer from 'inquirer';
import { Context } from '../../context/context.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Config } from '../../config/config.js';
const login = new Command('login'); const login = new Command('login');
login.description('Login to your account'); login.description('Login to your account');
login.option('--host <host>', 'The host of your server');
login.option('--token <token>', 'The token of your account');
login.action(async () => { login.action(async () => {
const config = new Config(); let { host, token } = login.opts();
const context = new Context(config.context); const { step, context } = getApi(login);
const { host, token } = await inquerer.prompt([ if (!host) {
{ const answers = await inquerer.prompt([
type: 'input', {
name: 'host', type: 'input',
message: 'Enter the host of your server', name: 'host',
default: context.host ?? 'http://localhost:4500', message: 'Enter the host of your server',
}, default: context.host || 'http://localhost:4500',
{ },
type: 'password', ]);
name: 'token', host = answers.host;
message: 'Enter your token', }
},
]); if (!token) {
const answers = await inquerer.prompt([
{
type: 'password',
name: 'token',
message: 'Enter your token',
},
]);
token = answers.token;
}
const healthResponse = await step('Getting auth status', async () => { const healthResponse = await step('Getting auth status', async () => {
return await fetch(`${host}/health`, { return await fetch(`${host}/health`, {

View File

@@ -1,9 +1,9 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { Config } from '../../config/config.js'; import { getApi } from '../../utils/command.js';
const current = new Command('current'); const current = new Command('current');
current.action(async () => { current.action(async () => {
const config = new Config(); const { config } = getApi(current);
console.log(config.context); console.log(config.context);
}); });

View File

@@ -1,10 +1,10 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { Config } from '../../config/config.js'; import { getApi } from '../../utils/command.js';
const use = new Command('use'); const use = new Command('use');
use.argument('<name>').action(async (name) => { use.argument('<name>').action(async (name) => {
const config = new Config(); const { config } = getApi(use);
await config.setContext(name); await config.setContext(name);
}); });

View File

@@ -1,8 +1,5 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const list = new Command('list'); const list = new Command('list');
@@ -10,15 +7,11 @@ list
.alias('ls') .alias('ls')
.description('List loads') .description('List loads')
.action(async () => { .action(async () => {
const config = new Config(); const { output, step, client } = getApi(list);
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const loads = await step('Getting data', async () => { const loads = await step('Getting data', async () => {
return await client.loads.find.query({}); return await client.loads.find.query({});
}); });
console.table(loads); await output(loads);
}); });
export { list }; export { list };

View File

@@ -1,10 +1,7 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { resolve } from 'path'; import { resolve } from 'path';
import { createClient } from '../../client/client.js';
import { bundle } from '../../bundler/bundler.js'; import { bundle } from '../../bundler/bundler.js';
import { step } from '../../utils/step.js'; import { getApi } from '../../utils/command.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const push = new Command('push'); const push = new Command('push');
@@ -16,12 +13,8 @@ push
.option('-ai, --auto-install', 'Auto install dependencies', false) .option('-ai, --auto-install', 'Auto install dependencies', false)
.action(async (script) => { .action(async (script) => {
const opts = push.opts(); const opts = push.opts();
const config = new Config(); const { step, log, client } = getApi(push);
const context = new Context(config.context);
const location = resolve(script); const location = resolve(script);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const code = await step('Bundling', async () => { const code = await step('Bundling', async () => {
return await bundle({ entry: location, autoInstall: opts.autoInstall }); return await bundle({ entry: location, autoInstall: opts.autoInstall });
}); });
@@ -32,12 +25,12 @@ push
script: code, script: code,
}); });
}); });
console.log('created load with id', id); await log(`created load with id ${id}`);
if (opts.run) { if (opts.run) {
const runId = await step('Creating run', async () => { const runId = await step('Creating run', async () => {
return await client.runs.create.mutate({ loadId: id }); return await client.runs.create.mutate({ loadId: id });
}); });
console.log('created run with id', runId); await log(`created run with id ${runId}`);
} }
}); });

View File

@@ -2,7 +2,8 @@ import { Command } from 'commander';
import { resolve } from 'path'; import { resolve } from 'path';
import { run as runLoad } from '@morten-olsen/mini-loader-runner'; import { run as runLoad } from '@morten-olsen/mini-loader-runner';
import { bundle } from '../../bundler/bundler.js'; import { bundle } from '../../bundler/bundler.js';
import { step } from '../../utils/step.js'; import { readSecrets } from './local.utils.js';
import { getApi } from '../../utils/command.js';
const run = new Command('run'); const run = new Command('run');
@@ -10,14 +11,18 @@ run
.option('-ai, --auto-install', 'Auto install dependencies', false) .option('-ai, --auto-install', 'Auto install dependencies', false)
.argument('script') .argument('script')
.action(async (script) => { .action(async (script) => {
const { step, config } = getApi(run);
const location = resolve(script); const location = resolve(script);
const { autoInstall } = run.opts(); const { autoInstall } = run.opts();
const secrets = await readSecrets();
const code = await step('Bundling', async () => { const code = await step('Bundling', async () => {
return await bundle({ entry: location, autoInstall }); return await bundle({ entry: location, autoInstall });
}); });
const { promise, emitter } = await runLoad({ const { promise, emitter } = await runLoad({
script: code, script: code,
secrets,
cacheLocation: config.cacheLocation,
}); });
emitter.addListener('message', (message) => { emitter.addListener('message', (message) => {
switch (message.type) { switch (message.type) {

View File

@@ -0,0 +1,25 @@
import dotenv from 'dotenv';
import { existsSync } from 'fs';
import { readFile } from 'fs/promises';
import { join } from 'path';
const ENV_PREFIX = 'ML_S_';
const readSecrets = async () => {
let secretLocation = join(process.cwd(), '.secrets');
let secrets: Record<string, string> = {};
if (existsSync(secretLocation)) {
const content = await readFile(secretLocation, 'utf-8');
secrets = dotenv.parse(content);
}
for (const key in process.env) {
if (key.startsWith(ENV_PREFIX)) {
secrets[key.replace(ENV_PREFIX, '')] = process.env[key]!;
}
}
return secrets;
};
export { readSecrets };

View File

@@ -1,8 +1,5 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const list = new Command('list'); const list = new Command('list');
@@ -24,11 +21,7 @@ list
.option('-s, --sort <order>', 'Sort', 'desc') .option('-s, --sort <order>', 'Sort', 'desc')
.action(async () => { .action(async () => {
const { runId, loadId, severities, offset, limit, order } = list.opts(); const { runId, loadId, severities, offset, limit, order } = list.opts();
const config = new Config(); const { step, output, client } = getApi(list);
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const logs = await step('Getting logs', async () => { const logs = await step('Getting logs', async () => {
return await client.logs.find.query({ return await client.logs.find.query({
runId, runId,
@@ -39,7 +32,7 @@ list
order, order,
}); });
}); });
console.table(logs); output(logs);
}); });
export { list }; export { list };

View File

@@ -1,9 +1,6 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import inquirer from 'inquirer'; import inquirer from 'inquirer';
import { Config } from '../../config/config.js'; import { getApi } from '../../utils/command.js';
const remove = new Command('remove'); const remove = new Command('remove');
@@ -24,12 +21,8 @@ remove
.option('-a, --limit <limit>', 'Limit', '1000') .option('-a, --limit <limit>', 'Limit', '1000')
.option('-s, --sort <order>', 'Sort', 'desc') .option('-s, --sort <order>', 'Sort', 'desc')
.action(async () => { .action(async () => {
const { step, client } = getApi(remove);
const { runId, loadId, severities, offset, limit, order } = remove.opts(); const { runId, loadId, severities, offset, limit, order } = remove.opts();
const config = new Config();
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const response = await step('Preparing to delete', async () => { const response = await step('Preparing to delete', async () => {
return await client.logs.prepareRemove.query({ return await client.logs.prepareRemove.query({
runId, runId,

View File

@@ -1,8 +1,5 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const create = new Command('create'); const create = new Command('create');
@@ -10,11 +7,7 @@ create
.description('Create a new run') .description('Create a new run')
.argument('load-id', 'Load ID') .argument('load-id', 'Load ID')
.action(async (loadId) => { .action(async (loadId) => {
const config = new Config(); const { step, client } = getApi(create);
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
await step('Creating run', async () => { await step('Creating run', async () => {
await client.runs.create.mutate({ loadId }); await client.runs.create.mutate({ loadId });
}); });

View File

@@ -1,8 +1,5 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const list = new Command('list'); const list = new Command('list');
@@ -11,15 +8,11 @@ list
.description('Find a run') .description('Find a run')
.argument('[load-id]', 'Load ID') .argument('[load-id]', 'Load ID')
.action(async (loadId) => { .action(async (loadId) => {
const config = new Config(); const { step, output, client } = getApi(list);
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const runs = await step('Getting runs', async () => { const runs = await step('Getting runs', async () => {
return await client.runs.find.query({ loadId }); return await client.runs.find.query({ loadId });
}); });
console.table(runs); await output(runs);
}); });
export { list }; export { list };

View File

@@ -1,9 +1,6 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import inquirer from 'inquirer'; import inquirer from 'inquirer';
import { Config } from '../../config/config.js'; import { getApi } from '../../utils/command.js';
const remove = new Command('remove'); const remove = new Command('remove');
@@ -21,12 +18,8 @@ remove
.option('-o, --offset <offset>', 'Offset') .option('-o, --offset <offset>', 'Offset')
.option('-a, --limit <limit>', 'Limit', '1000') .option('-a, --limit <limit>', 'Limit', '1000')
.action(async () => { .action(async () => {
const { step, client } = getApi(remove);
const { loadId, offset, limit } = remove.opts(); const { loadId, offset, limit } = remove.opts();
const config = new Config();
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const response = await step('Preparing to delete', async () => { const response = await step('Preparing to delete', async () => {
return await client.runs.prepareRemove.query({ return await client.runs.prepareRemove.query({
loadId, loadId,

View File

@@ -1,8 +1,5 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const terminate = new Command('terminate'); const terminate = new Command('terminate');
@@ -10,11 +7,7 @@ terminate
.description('Terminate an in progress run') .description('Terminate an in progress run')
.argument('run-id', 'Run ID') .argument('run-id', 'Run ID')
.action(async (runId) => { .action(async (runId) => {
const config = new Config(); const { step, client } = getApi(terminate);
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
await step('Terminating run', async () => { await step('Terminating run', async () => {
await client.runs.terminate.mutate(runId); await client.runs.terminate.mutate(runId);
}); });

View File

@@ -0,0 +1,25 @@
import { Command } from 'commander';
import { getApi } from '../../utils/command.js';
const add = new Command('add');
add
.description('Add schedule')
.argument('<load-id>', 'Load ID')
.argument('<cron>', 'Cron')
.option('-n, --name <name>', 'Name')
.action(async (loadId, cron) => {
const { step, client } = getApi(add);
const { name } = add.opts();
const id = await step('Adding schedule', async () => {
return await client.schedules.add.mutate({
name,
load: loadId,
cron,
});
});
console.log(`Schedule added with ID ${id}`);
});
export { add };

View File

@@ -0,0 +1,32 @@
import { Command } from 'commander';
import { getApi } from '../../utils/command.js';
const list = new Command('list');
const toInt = (value?: string) => {
if (!value) {
return undefined;
}
return parseInt(value, 10);
};
list
.alias('ls')
.description('List schedules')
.option('-l, --load-ids <loadIds...>', 'Load ID')
.option('-o, --offset <offset>', 'Offset')
.option('-a, --limit <limit>', 'Limit', '1000')
.action(async () => {
const { step, output, client } = getApi(list);
const { loadIds, offset, limit } = list.opts();
const schedules = await step('Getting schedules', async () => {
return await client.schedules.find.query({
loadIds,
offset: toInt(offset),
limit: toInt(limit),
});
});
output(schedules);
});
export { list };

View File

@@ -0,0 +1,54 @@
import { Command } from 'commander';
import inquirer from 'inquirer';
import { getApi } from '../../utils/command.js';
const remove = new Command('remove');
const toInt = (value?: string) => {
if (!value) {
return undefined;
}
return parseInt(value, 10);
};
remove
.alias('ls')
.description('LRemove schedules')
.option('-i, --ids <ids...>', 'Load IDs')
.option('-l, --load-ids <loadIds...>', 'Load IDs')
.option('-o, --offset <offset>', 'Offset')
.option('-a, --limit <limit>', 'Limit', '1000')
.action(async () => {
const { step, client } = getApi(remove);
const { ids, loadIds, offset, limit } = remove.opts();
const response = await step('Preparing to delete', async () => {
return await client.schedules.prepareRemove.query({
ids,
loadIds,
offset: toInt(offset),
limit: toInt(limit),
});
});
if (!response.ids.length) {
console.log('No logs to delete');
return;
}
const { confirm } = await inquirer.prompt([
{
type: 'confirm',
name: 'confirm',
message: `Are you sure you want to delete ${response.ids.length} schedules?`,
},
]);
if (!confirm) {
return;
}
await step('Deleting artifacts', async () => {
await client.artifacts.remove.mutate(response);
});
});
export { remove };

View File

@@ -0,0 +1,11 @@
import { Command } from 'commander';
import { list } from './schedules.list.js';
import { remove } from './schedules.remove.js';
import { add } from './schedules.add.js';
const schedules = new Command('schedules');
schedules.addCommand(list);
schedules.addCommand(remove);
schedules.addCommand(add);
export { schedules };

View File

@@ -1,8 +1,5 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const list = new Command('list'); const list = new Command('list');
@@ -15,16 +12,12 @@ const toInt = (value?: string) => {
list list
.alias('ls') .alias('ls')
.description('List logs') .description('List secrets')
.option('-o, --offset <offset>', 'Offset') .option('-o, --offset <offset>', 'Offset')
.option('-a, --limit <limit>', 'Limit', '1000') .option('-a, --limit <limit>', 'Limit', '1000')
.action(async () => { .action(async () => {
const { step, client } = getApi(list);
const { offset, limit } = list.opts(); const { offset, limit } = list.opts();
const config = new Config();
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
const secrets = await step('Getting secrets', async () => { const secrets = await step('Getting secrets', async () => {
return await client.secrets.find.query({ return await client.secrets.find.query({
offset: toInt(offset), offset: toInt(offset),

View File

@@ -1,8 +1,5 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const remove = new Command('remove'); const remove = new Command('remove');
@@ -10,11 +7,7 @@ remove
.alias('rm') .alias('rm')
.argument('<id>') .argument('<id>')
.action(async (id) => { .action(async (id) => {
const config = new Config(); const { step, client } = getApi(remove);
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
await step('Removing', async () => { await step('Removing', async () => {
await client.secrets.remove.mutate({ await client.secrets.remove.mutate({
id, id,

View File

@@ -1,8 +1,5 @@
import { Command } from 'commander'; import { Command } from 'commander';
import { createClient } from '../../client/client.js'; import { getApi } from '../../utils/command.js';
import { step } from '../../utils/step.js';
import { Context } from '../../context/context.js';
import { Config } from '../../config/config.js';
const set = new Command('set'); const set = new Command('set');
@@ -10,11 +7,7 @@ set
.argument('<id>') .argument('<id>')
.argument('[value]') .argument('[value]')
.action(async (id, value) => { .action(async (id, value) => {
const config = new Config(); const { step, client } = getApi(set);
const context = new Context(config.context);
const client = await step('Connecting to server', async () => {
return createClient(context);
});
await step('Setting secret', async () => { await step('Setting secret', async () => {
await client.secrets.set.mutate({ await client.secrets.set.mutate({
id, id,

View File

@@ -1,19 +1,22 @@
import envPaths from 'env-paths'; import { existsSync, readFileSync } from 'fs';
import { existsSync, readFileSync, writeFileSync } from 'fs'; import { mkdir, writeFile } from 'fs/promises';
import { mkdir } from 'fs/promises';
import { join, dirname } from 'path'; import { join, dirname } from 'path';
import { ContainerInstance, Service } from 'typedi';
import { Paths } from '../paths/paths.js';
type ConfigValues = { type ConfigValues = {
context?: string; context?: string;
}; };
@Service()
class Config { class Config {
#paths: Paths;
#location: string; #location: string;
#config?: ConfigValues; #config?: ConfigValues;
constructor() { constructor(contianer: ContainerInstance) {
const paths = envPaths('mini-loader'); this.#paths = contianer.get(Paths);
this.#location = join(paths.config, 'config.json'); this.#location = join(this.#paths.config, 'config.json');
if (existsSync(this.#location)) { if (existsSync(this.#location)) {
this.#config = JSON.parse(readFileSync(this.#location, 'utf-8')); this.#config = JSON.parse(readFileSync(this.#location, 'utf-8'));
} }
@@ -23,6 +26,14 @@ class Config {
return this.#config?.context || 'default'; return this.#config?.context || 'default';
} }
public get location() {
return this.#paths.config;
}
public get cacheLocation() {
return join(this.#paths.cache, this.context);
}
public setContext = (context: string) => { public setContext = (context: string) => {
this.#config = { this.#config = {
...(this.#config || {}), ...(this.#config || {}),
@@ -37,7 +48,7 @@ class Config {
} }
const json = JSON.stringify(this.#config); const json = JSON.stringify(this.#config);
mkdir(dirname(this.#location), { recursive: true }); mkdir(dirname(this.#location), { recursive: true });
writeFileSync(this.#location, json); writeFile(this.#location, json);
}; };
} }

View File

@@ -1,7 +1,8 @@
import envPaths from 'env-paths'; import envPaths from 'env-paths';
import { existsSync, readFileSync, writeFileSync } from 'fs'; import { existsSync, readFileSync } from 'fs';
import { mkdir, readdir } from 'fs/promises'; import { mkdir, readdir, writeFile } from 'fs/promises';
import { dirname, join } from 'path'; import { dirname, join } from 'path';
import { Config } from '../config/config.js';
type ContextValues = { type ContextValues = {
host: string; host: string;
@@ -12,9 +13,8 @@ class Context {
#location: string; #location: string;
#config?: ContextValues; #config?: ContextValues;
constructor(name: string) { constructor(config: Config) {
const paths = envPaths('mini-loader'); this.#location = join(config.location, 'contexts', config.context);
this.#location = join(paths.config, 'contexts', name);
if (existsSync(this.#location)) { if (existsSync(this.#location)) {
this.#config = JSON.parse(readFileSync(this.#location, 'utf-8')); this.#config = JSON.parse(readFileSync(this.#location, 'utf-8'));
} }
@@ -28,13 +28,13 @@ class Context {
return this.#config?.token; return this.#config?.token;
} }
public saveLogin = (host: string, token: string) => { public saveLogin = async (host: string, token: string) => {
this.#config = { this.#config = {
...(this.#config || {}), ...(this.#config || {}),
host, host,
token, token,
}; };
this.save(); await this.save();
}; };
public save = async () => { public save = async () => {
@@ -42,8 +42,8 @@ class Context {
return; return;
} }
const json = JSON.stringify(this.#config); const json = JSON.stringify(this.#config);
mkdir(dirname(this.#location), { recursive: true }); await mkdir(dirname(this.#location), { recursive: true });
writeFileSync(this.#location, json); await writeFile(this.#location, json);
}; };
public static list = async () => { public static list = async () => {

View File

@@ -1,4 +1,4 @@
import { program } from 'commander'; import { Command } from 'commander';
import { loads } from './commands/loads/loads.js'; import { loads } from './commands/loads/loads.js';
import { runs } from './commands/runs/runs.js'; import { runs } from './commands/runs/runs.js';
import { logs } from './commands/logs/logs.js'; import { logs } from './commands/logs/logs.js';
@@ -7,14 +7,28 @@ import { secrets } from './commands/secrets/secrets.js';
import { local } from './commands/local/local.js'; import { local } from './commands/local/local.js';
import { auth } from './commands/auth/auth.js'; import { auth } from './commands/auth/auth.js';
import { contexts } from './commands/contexts/contexts.js'; import { contexts } from './commands/contexts/contexts.js';
import { schedules } from './commands/schedules/schedules.js';
import { ContainerInstance } from 'typedi';
program.addCommand(loads); const createClientCli = (container: ContainerInstance) => {
program.addCommand(runs); const program = new Command();
program.addCommand(logs); program.exitOverride();
program.addCommand(artifacts); program.setOptionValue('_container', container);
program.addCommand(secrets); program.addCommand(loads);
program.addCommand(local); program.addCommand(runs);
program.addCommand(auth); program.addCommand(logs);
program.addCommand(contexts); program.addCommand(artifacts);
program.addCommand(secrets);
program.addCommand(local);
program.addCommand(auth);
program.addCommand(contexts);
program.addCommand(schedules);
await program.parseAsync(); return program;
};
export { CliApi } from './api/api.js';
export { Context } from './context/context.js';
export { Terminal } from './api/output.js';
export { Paths } from './paths/paths.js';
export { createClientCli };

View File

@@ -0,0 +1,17 @@
import envPaths from 'env-paths';
import { Service } from 'typedi';
const paths = envPaths('mini-loader');
@Service()
class Paths {
public get config() {
return paths.config;
}
public get cache() {
return paths.cache;
}
}
export { Paths };

View File

@@ -0,0 +1,11 @@
import { Command } from 'commander';
import { ContainerInstance } from 'typedi';
import { CliApi } from '../api/api.js';
const getApi = (command: Command) => {
const { _container } = command.optsWithGlobals() as { _container: ContainerInstance };
const api = _container.get(CliApi);
return api;
};
export { getApi };

View File

@@ -1,15 +0,0 @@
import ora from 'ora';
const step = async <T>(message: string, fn: () => Promise<T>): Promise<T> => {
const spinner = ora(message).start();
try {
const result = await fn();
await spinner.succeed();
return result;
} catch (err) {
await spinner.fail();
throw err;
}
};
export { step };

View File

@@ -2,6 +2,7 @@
"extends": "@morten-olsen/mini-loader-configs/tsconfig.esm.json", "extends": "@morten-olsen/mini-loader-configs/tsconfig.esm.json",
"compilerOptions": { "compilerOptions": {
"outDir": "dist/esm", "outDir": "dist/esm",
"rootDir": "src"
}, },
"include": [ "include": [
"./src/**/*.ts" "./src/**/*.ts"

View File

@@ -7,7 +7,9 @@
"declarationMap": true, "declarationMap": true,
"sourceMap": true, "sourceMap": true,
"esModuleInterop": true, "esModuleInterop": true,
"experimentalDecorators": true,
"strict": true, "strict": true,
"resolveJsonModule": true,
"allowSyntheticDefaultImports": true, "allowSyntheticDefaultImports": true,
"jsx": "react" "jsx": "react"
}, },

View File

@@ -0,0 +1 @@
demo=foobar

View File

@@ -0,0 +1,3 @@
import { secrets } from '@morten-olsen/mini-loader';
console.log(secrets.get('demo'));

View File

@@ -3,7 +3,6 @@ import { artifacts, logger } from '@morten-olsen/mini-loader';
const run = async () => { const run = async () => {
await logger.info('Hello world'); await logger.info('Hello world');
await artifacts.create('foo', 'bar'); await artifacts.create('foo', 'bar');
process.exit(0);
}; };
run(); run();

View File

@@ -1,4 +1,5 @@
import { Socket, createConnection } from 'net'; import { Socket, createConnection } from 'net';
import { Event } from './index.js';
const connect = () => const connect = () =>
new Promise<Socket>((resolve, reject) => { new Promise<Socket>((resolve, reject) => {
@@ -12,7 +13,7 @@ const connect = () =>
}); });
}); });
const send = async (data: any) => const send = async (data: Event) =>
new Promise<void>(async (resolve, reject) => { new Promise<void>(async (resolve, reject) => {
const connection = await connect(); const connection = await connect();
const cleaned = JSON.parse(JSON.stringify(data)); const cleaned = JSON.parse(JSON.stringify(data));

View File

@@ -5,10 +5,11 @@ type RunOptions = {
script: string; script: string;
input?: Buffer | string; input?: Buffer | string;
secrets?: Record<string, string>; secrets?: Record<string, string>;
cacheLocation: string;
}; };
const run = async ({ script, input, secrets }: RunOptions) => { const run = async ({ script, input, secrets, cacheLocation }: RunOptions) => {
const info = await setup({ script, input, secrets }); const info = await setup({ script, input, secrets, cacheLocation });
const worker = new Worker(info.scriptLocation, { const worker = new Worker(info.scriptLocation, {
stdin: false, stdin: false,

View File

@@ -1,5 +1,4 @@
import { join } from 'path'; import { join } from 'path';
import os from 'os';
import { nanoid } from 'nanoid'; import { nanoid } from 'nanoid';
import { chmod, mkdir, rm, writeFile } from 'fs/promises'; import { chmod, mkdir, rm, writeFile } from 'fs/promises';
import { createServer } from 'net'; import { createServer } from 'net';
@@ -9,6 +8,7 @@ type SetupOptions = {
input?: Buffer | string; input?: Buffer | string;
script: string; script: string;
secrets?: Record<string, string>; secrets?: Record<string, string>;
cacheLocation: string;
}; };
type RunEvents = { type RunEvents = {
@@ -20,7 +20,7 @@ type RunEvents = {
const setup = async (options: SetupOptions) => { const setup = async (options: SetupOptions) => {
const { input, script, secrets } = options; const { input, script, secrets } = options;
const emitter = new EventEmitter<RunEvents>(); const emitter = new EventEmitter<RunEvents>();
const dataDir = join(os.tmpdir(), 'mini-loader', nanoid()); const dataDir = join(options.cacheLocation, nanoid());
await mkdir(dataDir, { recursive: true }); await mkdir(dataDir, { recursive: true });
await chmod(dataDir, 0o700); await chmod(dataDir, 0o700);

View File

@@ -1,4 +1,4 @@
#!/usr/bin/env node #!/usr/bin/env node
import 'source-map-support/register.js'; import 'source-map-support/register.js';
import '../dist/esm/index.js'; import '../dist/esm/bin.js';

View File

@@ -14,6 +14,9 @@
"files": [ "files": [
"./dist" "./dist"
], ],
"imports": {
"#pkg": "./package.json"
},
"exports": { "exports": {
".": { ".": {
"import": "./dist/esm/index.js" "import": "./dist/esm/index.js"
@@ -21,24 +24,28 @@
}, },
"devDependencies": { "devDependencies": {
"@morten-olsen/mini-loader-configs": "workspace:^", "@morten-olsen/mini-loader-configs": "workspace:^",
"@morten-olsen/mini-loader-runner": "workspace:^",
"@types/jsonwebtoken": "^9.0.5", "@types/jsonwebtoken": "^9.0.5",
"@types/node": "^20.10.8", "@types/node": "^20.10.8",
"typescript": "^5.3.3" "typescript": "^5.3.3"
}, },
"dependencies": { "dependencies": {
"@fastify/reply-from": "^9.7.0", "@fastify/reply-from": "^9.7.0",
"@morten-olsen/mini-loader-runner": "workspace:^",
"@trpc/client": "^10.45.0", "@trpc/client": "^10.45.0",
"@trpc/server": "^10.45.0", "@trpc/server": "^10.45.0",
"commander": "^11.1.0", "commander": "^11.1.0",
"cron": "^3.1.6",
"env-paths": "^3.0.0",
"eventemitter3": "^5.0.1", "eventemitter3": "^5.0.1",
"fastify": "^4.25.2", "fastify": "^4.25.2",
"jsonwebtoken": "^9.0.2", "jsonwebtoken": "^9.0.2",
"knex": "^3.1.0", "knex": "^3.1.0",
"nanoid": "^5.0.4", "nanoid": "^5.0.4",
"reflect-metadata": "^0.2.1",
"source-map-support": "^0.5.21", "source-map-support": "^0.5.21",
"sqlite3": "^5.1.7", "sqlite3": "^5.1.7",
"superjson": "^2.2.1", "superjson": "^2.2.1",
"typedi": "^0.10.0",
"zod": "^3.22.4" "zod": "^3.22.4"
}, },
"homepage": "https://github.com/morten-olsen/mini-loader", "homepage": "https://github.com/morten-olsen/mini-loader",

View File

@@ -4,26 +4,30 @@ import { existsSync } from 'fs';
import { mkdir, readFile, writeFile } from 'fs/promises'; import { mkdir, readFile, writeFile } from 'fs/promises';
import jwt from 'jsonwebtoken'; import jwt from 'jsonwebtoken';
import { Config } from '../config/config.js'; import { Config } from '../config/config.js';
import { ContainerInstance, Service } from 'typedi';
type AuthOptions = { type AuthOptions = {
config: Config; config: Config;
}; };
@Service()
class Auth { class Auth {
#options: AuthOptions; #options: AuthOptions;
#data: Promise<{ secret: string }>; #data: Promise<{ secret: string }>;
constructor(options: AuthOptions) { constructor(container: ContainerInstance) {
this.#options = options; this.#options = {
config: container.get(Config),
};
this.#data = this.#setup(); this.#data = this.#setup();
} }
#setup = async () => { #setup = async () => {
const { config } = this.#options; const { config } = this.#options;
const secretLocation = resolve(config.files.location, 'secret'); const secretLocation = resolve(config.files.data, 'secret');
let secret = ''; let secret = '';
await mkdir(config.files.data, { recursive: true });
if (!existsSync(secretLocation)) { if (!existsSync(secretLocation)) {
await mkdir(config.files.location, { recursive: true });
secret = nanoid(); secret = nanoid();
await writeFile(secretLocation, secret); await writeFile(secretLocation, secret);
} else { } else {

View File

@@ -0,0 +1,14 @@
import { ContainerInstance } from 'typedi';
import { createServerCli } from './index.js';
const program = createServerCli(new ContainerInstance('server'));
program.setOptionValue('output', (data: unknown) => {
console.log('got data', data);
});
await program.parseAsync(process.argv);
process.on('unhandledRejection', (reason, p) => {
console.log('Unhandled Rejection at: Promise', p, 'reason:', reason);
});

View File

@@ -1,9 +1,22 @@
import { Knex } from 'knex'; import { Knex } from 'knex';
import { Service } from 'typedi';
import envPaths from 'env-paths';
import { join } from 'path';
type Config = { const paths = envPaths('mini-loader-server');
database: Omit<Knex.Config, 'migrations'>;
files: { @Service()
location: string; class Config {
database: Omit<Knex.Config, 'migrations'> = {
client: 'sqlite3',
connection: {
filename: join(paths.data, 'db.sqlite'),
},
useNullAsDefault: true,
};
files = {
data: paths.data,
cache: paths.cache,
}; };
auth?: { auth?: {
oidc?: { oidc?: {
@@ -14,6 +27,6 @@ type Config = {
}; };
}; };
}; };
}; }
export type { Config }; export { Config };

View File

@@ -1,18 +1,24 @@
import knex, { Knex } from 'knex'; import knex, { Knex } from 'knex';
import { Service, ContainerInstance } from 'typedi';
import { Config } from '../config/config.js';
import { source } from './migrations/migrations.source.js'; import { source } from './migrations/migrations.source.js';
import { mkdir } from 'fs/promises';
import { dirname } from 'path';
const tableNames = { const tableNames = {
loads: 'loads', loads: 'loads',
}; };
@Service()
class Database { class Database {
#instance?: Promise<Knex>; #instance?: Promise<Knex>;
#config: Knex.Config; #config: Knex.Config;
constructor(config: Knex.Config) { constructor(container: ContainerInstance) {
const config = container.get(Config);
this.#config = { this.#config = {
...config, ...config.database,
migrations: { migrations: {
migrationSource: source, migrationSource: source,
}, },
@@ -20,6 +26,15 @@ class Database {
} }
#setup = async (config: Knex.Config) => { #setup = async (config: Knex.Config) => {
if (
config.connection &&
typeof config.connection !== 'string' &&
'filename' in config.connection &&
typeof config.connection.filename === 'string' &&
config.connection.filename !== ':memory:'
) {
await mkdir(dirname(config.connection.filename), { recursive: true });
}
const db = knex(config); const db = knex(config);
await db.migrate.latest(); await db.migrate.latest();
return db; return db;

View File

@@ -0,0 +1,22 @@
import { Knex } from 'knex';
const name = 'schedule-support';
const up = async (knex: Knex) => {
await knex.schema.createTable('schedules', (table) => {
table.string('id').primary();
table.string('name').nullable();
table.string('description').nullable();
table.string('load').notNullable();
table.string('cron').notNullable();
table.string('input').nullable();
table.timestamp('createdAt').notNullable();
table.timestamp('updatedAt').notNullable();
});
};
const down = async (knex: Knex) => {
await knex.schema.dropTable('schedule');
};
export { name, up, down };

View File

@@ -1,6 +1,7 @@
import { Knex } from 'knex'; import { Knex } from 'knex';
import * as init from './migration.init.js'; import * as init from './migration.init.js';
import * as scheduleSupport from './migration.schedule.js';
type Migration = { type Migration = {
name: string; name: string;
@@ -8,7 +9,7 @@ type Migration = {
down: (knex: Knex) => Promise<void>; down: (knex: Knex) => Promise<void>;
}; };
const migrations = [init] satisfies Migration[]; const migrations = [init, scheduleSupport] satisfies Migration[];
const source: Knex.MigrationSource<Migration> = { const source: Knex.MigrationSource<Migration> = {
getMigrations: async () => migrations, getMigrations: async () => migrations,

View File

@@ -0,0 +1,9 @@
import { nanoid } from 'nanoid';
import { Service } from 'typedi';
@Service()
class IdGenerator {
public generate = () => nanoid();
}
export { IdGenerator };

View File

@@ -1,35 +1,42 @@
import { program, Command } from 'commander'; import 'reflect-metadata';
import { Command } from 'commander';
import { Runtime } from './runtime/runtime.js'; import { Runtime } from './runtime/runtime.js';
import { createServer } from './server/server.js'; import { createServer } from './server/server.js';
import { ContainerInstance } from 'typedi';
const start = new Command('start'); const createServerCli = (container: ContainerInstance) => {
start.action(async () => { const program = new Command();
const port = 4500;
const runtime = await Runtime.create(); const start = new Command('start');
const server = await createServer(runtime); start.action(async () => {
await server.listen({ const port = 4500;
port, const runtime = container.get(Runtime);
host: '0.0.0.0', await runtime.scheduler.start();
const server = await createServer(container);
await server.listen({
port,
host: '0.0.0.0',
});
console.log(`Server listening on port ${port}`);
}); });
console.log(`Server listening on port ${port}`); const createToken = new Command('create-token');
}); createToken.action(async () => {
const runtime = container.get(Runtime);
const token = await runtime.auth.createToken({
policy: {
'*:*': ['*'],
},
});
console.log(token);
});
const createToken = new Command('create-token'); program.addCommand(start);
createToken.action(async () => { program.addCommand(createToken);
const runtime = await Runtime.create(); return program;
const token = await runtime.auth.createToken({}); };
console.log(token);
});
program.addCommand(start); export { createServerCli, createServer, Runtime };
program.addCommand(createToken); export { Config } from './config/config.js';
await program.parseAsync(process.argv);
process.on('unhandledRejection', (reason, p) => {
console.log('Unhandled Rejection at: Promise', p, 'reason:', reason);
});
export type { Runtime } from './runtime/runtime.js';
export type { RootRouter } from './router/router.js'; export type { RootRouter } from './router/router.js';

View File

@@ -43,5 +43,15 @@ declare module 'knex/types/tables.js' {
createdAt: Date; createdAt: Date;
updatedAt: Date; updatedAt: Date;
}; };
schedules: {
id: string;
name?: string;
description?: string;
load: string;
cron: string;
input?: string;
createdAt: Date;
updatedAt: Date;
};
} }
} }

View File

@@ -1,21 +1,27 @@
import { EventEmitter } from 'eventemitter3'; import { EventEmitter } from 'eventemitter3';
import { Database } from '../../database/database.js'; import { Database } from '../../database/database.js';
import { nanoid } from 'nanoid';
import { AddArtifactOptions, FindArtifactsOptions } from './artifacts.schemas.js'; import { AddArtifactOptions, FindArtifactsOptions } from './artifacts.schemas.js';
import { createHash } from 'crypto'; import { createHash } from 'crypto';
import { ContainerInstance, Service } from 'typedi';
import { IdGenerator } from '../../id/id.js';
type ArtifactRepoEvents = {}; type ArtifactRepoEvents = {};
type ArtifactRepoOptions = { type ArtifactRepoOptions = {
database: Database; database: Database;
idGenerator: IdGenerator;
}; };
@Service()
class ArtifactRepo extends EventEmitter<ArtifactRepoEvents> { class ArtifactRepo extends EventEmitter<ArtifactRepoEvents> {
#options: ArtifactRepoOptions; #options: ArtifactRepoOptions;
constructor(options: ArtifactRepoOptions) { constructor(container: ContainerInstance) {
super(); super();
this.#options = options; this.#options = {
database: container.get(Database),
idGenerator: container.get(IdGenerator),
};
} }
public get = async (id: string) => { public get = async (id: string) => {
@@ -26,9 +32,9 @@ class ArtifactRepo extends EventEmitter<ArtifactRepoEvents> {
}; };
public add = async (options: AddArtifactOptions) => { public add = async (options: AddArtifactOptions) => {
const { database } = this.#options; const { database, idGenerator } = this.#options;
const db = await database.instance; const db = await database.instance;
const id = nanoid(); const id = idGenerator.generate();
await db('artifacts').insert({ await db('artifacts').insert({
id, id,

View File

@@ -1,11 +1,12 @@
import { EventEmitter } from 'eventemitter3'; import { EventEmitter } from 'eventemitter3';
import { Database } from '../../database/database.js'; import { Database } from '../../database/database.js';
import { FindLoadsOptions, SetLoadOptions } from './loads.schemas.js'; import { FindLoadsOptions, SetLoadOptions } from './loads.schemas.js';
import { nanoid } from 'nanoid';
import { createHash } from 'crypto'; import { createHash } from 'crypto';
import { Config } from '../../config/config.js'; import { Config } from '../../config/config.js';
import { mkdir, writeFile } from 'fs/promises'; import { mkdir, writeFile } from 'fs/promises';
import { resolve } from 'path'; import { resolve } from 'path';
import { ContainerInstance, Service } from 'typedi';
import { IdGenerator } from '../../id/id.js';
type LoadRepoEvents = { type LoadRepoEvents = {
created: (id: string) => void; created: (id: string) => void;
@@ -16,14 +17,20 @@ type LoadRepoEvents = {
type LoadRepoOptions = { type LoadRepoOptions = {
database: Database; database: Database;
config: Config; config: Config;
idGenerator: IdGenerator;
}; };
@Service()
class LoadRepo extends EventEmitter<LoadRepoEvents> { class LoadRepo extends EventEmitter<LoadRepoEvents> {
#options: LoadRepoOptions; #options: LoadRepoOptions;
constructor(options: LoadRepoOptions) { constructor(container: ContainerInstance) {
super(); super();
this.#options = options; this.#options = {
database: container.get(Database),
config: container.get(Config),
idGenerator: container.get(IdGenerator),
};
} }
public getById = async (id: string) => { public getById = async (id: string) => {
@@ -58,11 +65,11 @@ class LoadRepo extends EventEmitter<LoadRepoEvents> {
}; };
public set = async (options: SetLoadOptions) => { public set = async (options: SetLoadOptions) => {
const { database } = this.#options; const { database, idGenerator } = this.#options;
const db = await database.instance; const db = await database.instance;
const id = options.id || nanoid(); const id = options.id || idGenerator.generate();
const script = createHash('sha256').update(options.script).digest('hex'); const script = createHash('sha256').update(options.script).digest('hex');
const scriptDir = resolve(this.#options.config.files.location, 'scripts'); const scriptDir = resolve(this.#options.config.files.data, 'scripts');
await mkdir(scriptDir, { recursive: true }); await mkdir(scriptDir, { recursive: true });
await writeFile(resolve(scriptDir, `${script}.js`), options.script); await writeFile(resolve(scriptDir, `${script}.js`), options.script);

View File

@@ -1,27 +1,33 @@
import { EventEmitter } from 'eventemitter3'; import { EventEmitter } from 'eventemitter3';
import { Database } from '../../database/database.js'; import { Database } from '../../database/database.js';
import { AddLogOptions, FindLogsOptions } from './logs.schemas.js'; import { AddLogOptions, FindLogsOptions } from './logs.schemas.js';
import { nanoid } from 'nanoid';
import { createHash } from 'crypto'; import { createHash } from 'crypto';
import { ContainerInstance, Service } from 'typedi';
import { IdGenerator } from '../../id/id.js';
type LogRepoEvents = {}; type LogRepoEvents = {};
type LogRepoOptions = { type LogRepoOptions = {
database: Database; database: Database;
idGenerator: IdGenerator;
}; };
@Service()
class LogRepo extends EventEmitter<LogRepoEvents> { class LogRepo extends EventEmitter<LogRepoEvents> {
#options: LogRepoOptions; #options: LogRepoOptions;
constructor(options: LogRepoOptions) { constructor(container: ContainerInstance) {
super(); super();
this.#options = options; this.#options = {
database: container.get(Database),
idGenerator: container.get(IdGenerator),
};
} }
public add = async (options: AddLogOptions) => { public add = async (options: AddLogOptions) => {
const { database } = this.#options; const { database, idGenerator } = this.#options;
const db = await database.instance; const db = await database.instance;
const id = nanoid(); const id = idGenerator.generate();
await db('logs').insert({ await db('logs').insert({
id, id,

View File

@@ -1,64 +1,45 @@
import { Config } from '../config/config.js'; import { ContainerInstance, Service } from 'typedi';
import { Database } from '../database/database.js';
import { ArtifactRepo } from './artifacts/artifacts.js'; import { ArtifactRepo } from './artifacts/artifacts.js';
import { LoadRepo } from './loads/loads.js'; import { LoadRepo } from './loads/loads.js';
import { LogRepo } from './logs/logs.js'; import { LogRepo } from './logs/logs.js';
import { RunRepo } from './runs/runs.js'; import { RunRepo } from './runs/runs.js';
import { ScheduleRepo } from './schedules/schedules.js';
import { SecretRepo } from './secrets/secrets.js'; import { SecretRepo } from './secrets/secrets.js';
type ReposOptions = { @Service()
database: Database;
config: Config;
};
class Repos { class Repos {
#loads: LoadRepo; #container: ContainerInstance;
#runs: RunRepo;
#logs: LogRepo;
#artifacts: ArtifactRepo;
#secrets: SecretRepo;
constructor({ database, config }: ReposOptions) { constructor(container: ContainerInstance) {
this.#loads = new LoadRepo({ this.#container = container;
database,
config,
});
this.#runs = new RunRepo({
database,
loads: this.#loads,
});
this.#logs = new LogRepo({
database,
});
this.#artifacts = new ArtifactRepo({
database,
});
this.#secrets = new SecretRepo({
database,
});
} }
public get loads() { public get loads() {
return this.#loads; return this.#container.get(LoadRepo);
} }
public get runs() { public get runs() {
return this.#runs; return this.#container.get(RunRepo);
} }
public get logs() { public get logs() {
return this.#logs; return this.#container.get(LogRepo);
} }
public get artifacts() { public get artifacts() {
return this.#artifacts; return this.#container.get(ArtifactRepo);
} }
public get secrets() { public get secrets() {
return this.#secrets; return this.#container.get(SecretRepo);
}
public get schedules() {
return this.#container.get(ScheduleRepo);
} }
} }
export { findSchedulesSchema, addScheduleSchema } from './schedules/schedules.js';
export { findLogsSchema, addLogSchema } from './logs/logs.js'; export { findLogsSchema, addLogSchema } from './logs/logs.js';
export { setLoadSchema, findLoadsSchema } from './loads/loads.js'; export { setLoadSchema, findLoadsSchema } from './loads/loads.js';
export { createRunSchema, findRunsSchema } from './runs/runs.js'; export { createRunSchema, findRunsSchema } from './runs/runs.js';

View File

@@ -1,9 +1,10 @@
import { nanoid } from 'nanoid';
import { EventEmitter } from 'eventemitter3'; import { EventEmitter } from 'eventemitter3';
import { Database } from '../../database/database.js'; import { Database } from '../../database/database.js';
import { CreateRunOptions, FindRunsOptions, UpdateRunOptions } from './runs.schemas.js'; import { CreateRunOptions, FindRunsOptions, UpdateRunOptions } from './runs.schemas.js';
import { LoadRepo } from '../loads/loads.js'; import { LoadRepo } from '../loads/loads.js';
import { createHash } from 'crypto'; import { createHash } from 'crypto';
import { ContainerInstance, Service } from 'typedi';
import { IdGenerator } from '../../id/id.js';
type RunRepoEvents = { type RunRepoEvents = {
created: (args: { id: string; loadId: string }) => void; created: (args: { id: string; loadId: string }) => void;
@@ -15,16 +16,21 @@ type RunRepoEvents = {
type RunRepoOptions = { type RunRepoOptions = {
database: Database; database: Database;
loads: LoadRepo; loads: LoadRepo;
idGenerator: IdGenerator;
}; };
@Service()
class RunRepo extends EventEmitter<RunRepoEvents> { class RunRepo extends EventEmitter<RunRepoEvents> {
#options: RunRepoOptions; #options: RunRepoOptions;
#isReady: Promise<void>; #isSetup?: Promise<void>;
constructor(options: RunRepoOptions) { constructor(container: ContainerInstance) {
super(); super();
this.#options = options; this.#options = {
this.#isReady = this.#setup(); database: container.get(Database),
loads: container.get(LoadRepo),
idGenerator: container.get(IdGenerator),
};
} }
#setup = async () => { #setup = async () => {
@@ -33,6 +39,13 @@ class RunRepo extends EventEmitter<RunRepoEvents> {
await db('runs').update({ status: 'failed', error: 'server was shut down' }).where({ status: 'running' }); await db('runs').update({ status: 'failed', error: 'server was shut down' }).where({ status: 'running' });
}; };
get #isReady() {
if (!this.#isSetup) {
this.#isSetup = this.#setup();
}
return this.#isSetup;
}
public getById = async (id: string) => { public getById = async (id: string) => {
await this.#isReady; await this.#isReady;
const { database } = this.#options; const { database } = this.#options;
@@ -150,8 +163,8 @@ class RunRepo extends EventEmitter<RunRepoEvents> {
public create = async (options: CreateRunOptions) => { public create = async (options: CreateRunOptions) => {
await this.#isReady; await this.#isReady;
const { database, loads } = this.#options; const { database, loads, idGenerator } = this.#options;
const id = nanoid(); const id = idGenerator.generate();
const db = await database.instance; const db = await database.instance;
const script = await loads.getScript(options.loadId); const script = await loads.getScript(options.loadId);

View File

@@ -0,0 +1,22 @@
import { z } from 'zod';
const addScheduleSchema = z.object({
name: z.string().optional(),
description: z.string().optional(),
load: z.string(),
cron: z.string(),
input: z.string().optional(),
});
const findSchedulesSchema = z.object({
ids: z.array(z.string()).optional(),
loadIds: z.array(z.string()).optional(),
offset: z.number().optional(),
limit: z.number().optional(),
});
type AddScheduleOptions = z.infer<typeof addScheduleSchema>;
type FindSchedulesOptions = z.infer<typeof findSchedulesSchema>;
export type { AddScheduleOptions, FindSchedulesOptions };
export { addScheduleSchema, findSchedulesSchema };

View File

@@ -0,0 +1,124 @@
import { EventEmitter } from 'eventemitter3';
import { Database } from '../../database/database.js';
import { AddScheduleOptions, FindSchedulesOptions } from './schedules.schemas.js';
import { createHash } from 'crypto';
import { ContainerInstance, Service } from 'typedi';
import { IdGenerator } from '../../id/id.js';
type ScheduleRepoEvents = {
added: (id: string) => void;
removed: (id: string) => void;
};
type ScheduleRepoOptions = {
database: Database;
idGenerator: IdGenerator;
};
@Service()
class ScheduleRepo extends EventEmitter<ScheduleRepoEvents> {
#options: ScheduleRepoOptions;
constructor(container: ContainerInstance) {
super();
this.#options = {
database: container.get(Database),
idGenerator: container.get(IdGenerator),
};
}
public get = async (id: string) => {
const { database } = this.#options;
const db = await database.instance;
const result = await db('schedules').where('id', id).first();
return result;
};
public add = async (options: AddScheduleOptions) => {
const { database, idGenerator } = this.#options;
const db = await database.instance;
const id = idGenerator.generate();
await db('schedules').insert({
id,
name: options.name,
description: options.description,
cron: options.cron,
createdAt: new Date(),
updatedAt: new Date(),
});
this.emit('added', id);
return id;
};
public prepareRemove = async (options: FindSchedulesOptions) => {
const { database } = this.#options;
const db = await database.instance;
const query = db('schedules').select('id');
if (options.ids) {
query.whereIn('id', options.ids);
}
if (options.loadIds) {
query.whereIn('loadId', options.loadIds);
}
const result = await query;
const ids = result.map((row) => row.id);
const token = ids.map((id) => Buffer.from(id).toString('base64')).join('|');
const hash = createHash('sha256').update(token).digest('hex');
return {
ids,
hash,
};
};
public remove = async (hash: string, ids: string[]) => {
const { database } = this.#options;
const db = await database.instance;
const token = ids.map((id) => Buffer.from(id).toString('base64')).join('|');
const actualHash = createHash('sha256').update(token).digest('hex');
if (hash !== actualHash) {
throw new Error('Invalid hash');
}
await db('schedules').whereIn('id', ids).delete();
ids.forEach((id) => {
this.emit('removed', id);
});
};
public find = async (options: FindSchedulesOptions) => {
const { database } = this.#options;
const db = await database.instance;
const query = db('schedules');
if (options.ids) {
query.whereIn('id', options.ids);
}
if (options.loadIds) {
query.whereIn('loadId', options.loadIds);
}
if (options.offset) {
query.offset(options.offset);
}
if (options.limit) {
query.limit(options.limit);
}
const results = await query;
return results;
};
}
export { addScheduleSchema, findSchedulesSchema } from './schedules.schemas.js';
export { ScheduleRepo };

View File

@@ -1,6 +1,7 @@
import { EventEmitter } from 'eventemitter3'; import { EventEmitter } from 'eventemitter3';
import { Database } from '../../database/database.js'; import { Database } from '../../database/database.js';
import { FindSecretOptions, SetSecretOptions } from './secrets.schemas.js'; import { FindSecretOptions, SetSecretOptions } from './secrets.schemas.js';
import { ContainerInstance, Service } from 'typedi';
type LogRepoEvents = {}; type LogRepoEvents = {};
@@ -8,12 +9,15 @@ type LogRepoOptions = {
database: Database; database: Database;
}; };
@Service()
class SecretRepo extends EventEmitter<LogRepoEvents> { class SecretRepo extends EventEmitter<LogRepoEvents> {
#options: LogRepoOptions; #options: LogRepoOptions;
constructor(options: LogRepoOptions) { constructor(container: ContainerInstance) {
super(); super();
this.#options = options; this.#options = {
database: container.get(Database),
};
} }
public set = async (options: SetSecretOptions) => { public set = async (options: SetSecretOptions) => {

View File

@@ -0,0 +1,53 @@
import { z } from 'zod';
import { addScheduleSchema, findSchedulesSchema } from '../repos/repos.js';
import { publicProcedure, router } from './router.utils.js';
const add = publicProcedure.input(addScheduleSchema).mutation(async ({ input, ctx }) => {
const { runtime } = ctx;
const { repos } = runtime;
const { schedules } = repos;
const result = await schedules.add(input);
return result;
});
const find = publicProcedure.input(findSchedulesSchema).query(async ({ input, ctx }) => {
const { runtime } = ctx;
const { repos } = runtime;
const { schedules } = repos;
const result = await schedules.find(input);
return result;
});
const prepareRemove = publicProcedure.input(findSchedulesSchema).query(async ({ input, ctx }) => {
const { runtime } = ctx;
const { repos } = runtime;
const { schedules } = repos;
return await schedules.prepareRemove(input);
});
const remove = publicProcedure
.input(
z.object({
hash: z.string(),
ids: z.array(z.string()),
}),
)
.mutation(async ({ input, ctx }) => {
const { runtime } = ctx;
const { repos } = runtime;
const { artifacts } = repos;
await artifacts.remove(input.hash, input.ids);
});
const schedulesRouter = router({
add,
find,
remove,
prepareRemove,
});
export { schedulesRouter };

View File

@@ -2,6 +2,7 @@ import { artifactsRouter } from './router.artifacts.js';
import { loadsRouter } from './router.loads.js'; import { loadsRouter } from './router.loads.js';
import { logsRouter } from './router.logs.js'; import { logsRouter } from './router.logs.js';
import { runsRouter } from './router.runs.js'; import { runsRouter } from './router.runs.js';
import { schedulesRouter } from './router.schedules.js';
import { secretsRouter } from './router.secrets.js'; import { secretsRouter } from './router.secrets.js';
import { router } from './router.utils.js'; import { router } from './router.utils.js';
@@ -11,6 +12,7 @@ const rootRouter = router({
logs: logsRouter, logs: logsRouter,
artifacts: artifactsRouter, artifacts: artifactsRouter,
secrets: secretsRouter, secrets: secretsRouter,
schedules: schedulesRouter,
}); });
type RootRouter = typeof rootRouter; type RootRouter = typeof rootRouter;

View File

@@ -2,13 +2,15 @@ import { initTRPC } from '@trpc/server';
import { CreateFastifyContextOptions } from '@trpc/server/adapters/fastify'; import { CreateFastifyContextOptions } from '@trpc/server/adapters/fastify';
import superjson from 'superjson'; import superjson from 'superjson';
import { Runtime } from '../runtime/runtime.js'; import { Runtime } from '../runtime/runtime.js';
import { ContainerInstance } from 'typedi';
type ContextOptions = { type ContextOptions = {
runtime: Runtime; container: ContainerInstance;
}; };
const createContext = async ({ runtime }: ContextOptions) => { const createContext = async ({ container }: ContextOptions) => {
return async ({ req }: CreateFastifyContextOptions) => { return async ({ req }: CreateFastifyContextOptions) => {
const runtime = container.get(Runtime);
const { authorization } = req.headers; const { authorization } = req.headers;
const { auth } = runtime; const { auth } = runtime;
if (!authorization) { if (!authorization) {
@@ -18,6 +20,7 @@ const createContext = async ({ runtime }: ContextOptions) => {
await auth.validateToken(token); await auth.validateToken(token);
return { return {
runtime, runtime,
container,
}; };
}; };
}; };

View File

@@ -59,7 +59,7 @@ class RunnerInstance extends EventEmitter<RunnerInstanceEvents> {
const { runs, secrets } = repos; const { runs, secrets } = repos;
try { try {
const { script: scriptHash, input } = await runs.getById(id); const { script: scriptHash, input } = await runs.getById(id);
const scriptLocation = resolve(config.files.location, 'scripts', `${scriptHash}.js`); const scriptLocation = resolve(config.files.data, 'scripts', `${scriptHash}.js`);
const script = await readFile(scriptLocation, 'utf-8'); const script = await readFile(scriptLocation, 'utf-8');
const allSecrets = await secrets.getAll(); const allSecrets = await secrets.getAll();
await runs.started(id); await runs.started(id);
@@ -67,6 +67,7 @@ class RunnerInstance extends EventEmitter<RunnerInstanceEvents> {
script, script,
secrets: allSecrets, secrets: allSecrets,
input, input,
cacheLocation: config.files.cache,
}); });
this.#run = current; this.#run = current;
const { promise, emitter } = current; const { promise, emitter } = current;

View File

@@ -1,3 +1,4 @@
import { ContainerInstance, Service } from 'typedi';
import { Config } from '../config/config.js'; import { Config } from '../config/config.js';
import { Repos } from '../repos/repos.js'; import { Repos } from '../repos/repos.js';
import { RunnerInstance } from './runner.instance.js'; import { RunnerInstance } from './runner.instance.js';
@@ -7,13 +8,17 @@ type RunnerOptions = {
config: Config; config: Config;
}; };
@Service()
class Runner { class Runner {
#options: RunnerOptions; #options: RunnerOptions;
#instances: Map<string, RunnerInstance> = new Map(); #instances: Map<string, RunnerInstance> = new Map();
constructor(options: RunnerOptions) { constructor(container: ContainerInstance) {
this.#options = options; this.#options = {
const { repos } = options; repos: container.get(Repos),
config: container.get(Config),
};
const { repos } = this.#options;
repos.runs.on('created', this.#start); repos.runs.on('created', this.#start);
} }

View File

@@ -1,20 +1,21 @@
import { Database } from '../database/database.js'; import { ContainerInstance, Service } from 'typedi';
import { Repos } from '../repos/repos.js'; import { Repos } from '../repos/repos.js';
import { Runner } from '../runner/runner.js'; import { Runner } from '../runner/runner.js';
import { Config } from '../config/config.js';
import { Auth } from '../auth/auth.js'; import { Auth } from '../auth/auth.js';
import { resolve } from 'path'; import { Scheduler } from '../scheduler/scheduler.js';
@Service()
class Runtime { class Runtime {
#repos: Repos; #repos: Repos;
#runner: Runner; #runner: Runner;
#auth: Auth; #auth: Auth;
#scheduler: Scheduler;
constructor(options: Config) { constructor(container: ContainerInstance) {
const database = new Database(options.database); this.#repos = container.get(Repos);
this.#repos = new Repos({ database, config: options }); this.#runner = container.get(Runner);
this.#runner = new Runner({ repos: this.#repos, config: options }); this.#auth = container.get(Auth);
this.#auth = new Auth({ config: options }); this.#scheduler = container.get(Scheduler);
} }
public get repos() { public get repos() {
@@ -29,22 +30,9 @@ class Runtime {
return this.#auth; return this.#auth;
} }
public static create = async () => { public get scheduler() {
const runtime = new Runtime({ return this.#scheduler;
database: { }
client: 'sqlite3',
connection: {
filename: resolve(process.cwd(), 'data', 'database.sqlite'),
},
useNullAsDefault: true,
},
files: {
location: resolve(process.cwd(), 'data', 'files'),
},
});
return runtime;
};
} }
export { Runtime }; export { Runtime };

View File

@@ -0,0 +1,78 @@
import { CronJob } from 'cron';
import { ScheduleRepo } from '../repos/schedules/schedules.js';
import { RunRepo } from '../repos/runs/runs.js';
import { ContainerInstance, Service } from 'typedi';
type SchedulerOptions = {
runs: RunRepo;
schedules: ScheduleRepo;
};
type RunningSchedule = {
id: string;
job: CronJob;
stop: () => Promise<void>;
};
@Service()
class Scheduler {
#running: RunningSchedule[] = [];
#options: SchedulerOptions;
constructor(container: ContainerInstance) {
this.#options = {
runs: container.get(RunRepo),
schedules: container.get(ScheduleRepo),
};
const { schedules } = this.#options;
schedules.on('added', this.#add);
schedules.on('removed', this.#remove);
}
#remove = async (id: string) => {
const current = this.#running.filter((r) => r.id === id);
await Promise.all(current.map((r) => r.stop()));
this.#running = this.#running.filter((r) => r.id !== id);
};
#add = async (id: string) => {
const { schedules, runs } = this.#options;
const current = this.#running.filter((r) => r.id === id);
await Promise.all(current.map((r) => r.stop()));
const schedule = await schedules.get(id);
if (!schedule) {
return;
}
const job = new CronJob(schedule.cron, async () => {
await runs.create({
loadId: schedule.load,
});
});
const stop = async () => {
job.stop();
};
this.#running.push({
id: schedule.id,
job,
stop,
});
};
public stop = async () => {
for (const running of this.#running) {
await running.stop();
this.#running = this.#running.filter((r) => r !== running);
}
};
public start = async () => {
const { schedules } = this.#options;
await this.stop();
const all = await schedules.find({});
for (const schedule of all) {
await this.#add(schedule.id);
}
};
}
export { Scheduler };

View File

@@ -2,10 +2,12 @@ import { fastifyTRPCPlugin, FastifyTRPCPluginOptions } from '@trpc/server/adapte
import fastify from 'fastify'; import fastify from 'fastify';
import { RootRouter, rootRouter } from '../router/router.js'; import { RootRouter, rootRouter } from '../router/router.js';
import { createContext } from '../router/router.utils.js'; import { createContext } from '../router/router.utils.js';
import { Runtime } from '../runtime/runtime.js';
import { gateway } from '../gateway/gateway.js'; import { gateway } from '../gateway/gateway.js';
import { ContainerInstance } from 'typedi';
import { Runtime } from '../runtime/runtime.js';
const createServer = async (runtime: Runtime) => { const createServer = async (container: ContainerInstance) => {
const runtime = container.get(Runtime);
const server = fastify({ const server = fastify({
maxParamLength: 10000, maxParamLength: 10000,
bodyLimit: 30 * 1024 * 1024, bodyLimit: 30 * 1024 * 1024,
@@ -13,9 +15,6 @@ const createServer = async (runtime: Runtime) => {
level: 'warn', level: 'warn',
}, },
}); });
server.get('/', async () => {
return { hello: 'world' };
});
server.get('/health', async (req) => { server.get('/health', async (req) => {
let authorized = false; let authorized = false;
@@ -34,7 +33,7 @@ const createServer = async (runtime: Runtime) => {
prefix: '/trpc', prefix: '/trpc',
trpcOptions: { trpcOptions: {
router: rootRouter, router: rootRouter,
createContext: await createContext({ runtime }), createContext: await createContext({ container }),
onError({ error }) { onError({ error }) {
console.error(error); console.error(error);
}, },

View File

@@ -2,6 +2,7 @@
"extends": "@morten-olsen/mini-loader-configs/tsconfig.esm.json", "extends": "@morten-olsen/mini-loader-configs/tsconfig.esm.json",
"compilerOptions": { "compilerOptions": {
"outDir": "dist/esm", "outDir": "dist/esm",
"rootDir": "src"
}, },
"include": [ "include": [
"src" "src"

4
packages/tests/.gitignore vendored Normal file
View File

@@ -0,0 +1,4 @@
/dist/
/node_modules/
/coverage/
/e2e-tmp/

1
packages/tests/README.md Normal file
View File

@@ -0,0 +1 @@
[Go to documentation](https://github.com/morten-olsen/mini-loader)

View File

@@ -0,0 +1 @@
console.log('simple');

View File

@@ -0,0 +1,40 @@
{
"name": "@morten-olsen/mini-loader-tests",
"private": "true",
"version": "1.0.0",
"license": "GPL-3.0",
"main": "./dist/esm/index.js",
"types": "./dist/esm/index.d.ts",
"type": "module",
"files": [
"./dist"
],
"exports": {
".": {
"import": "./dist/esm/index.js"
}
},
"devDependencies": {
"@morten-olsen/mini-loader-configs": "workspace:^",
"@types/node": "^20.10.8",
"fastify": "^4.25.2",
"sqlite3": "^5.1.7",
"typescript": "^5.3.3",
"vite": "^5.0.11"
},
"dependencies": {
"@morten-olsen/mini-loader": "workspace:^",
"@morten-olsen/mini-loader-cli": "workspace:^",
"@morten-olsen/mini-loader-server": "workspace:^",
"get-port": "^7.0.0",
"memfs": "^4.6.0",
"nanoid": "^5.0.4",
"typedi": "^0.10.0",
"vitest": "^1.2.0"
},
"homepage": "https://github.com/morten-olsen/mini-loader",
"repository": {
"type": "git",
"url": "https://github.com/morten-olsen/mini-loader"
}
}

View File

@@ -0,0 +1,56 @@
import { describe, it, beforeAll, afterAll, vi } from 'vitest';
import { TestEnv, createEnv } from './utils/env.js';
import { writeFile } from 'fs/promises';
import { resolve } from 'path';
const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));
describe('Quick start', () => {
let env: TestEnv;
beforeAll(async () => {
vi.mock('nanoid', () => {
let currentId = 0;
return {
nanoid: () => `id-${currentId++}`,
};
});
env = await createEnv();
});
afterAll(async () => {
vi.clearAllMocks();
await env.cleanup();
});
it('should be able to start the server', async () => {
const { server, port } = env;
await server.listen({ port });
});
it('should be able to login', async () => {
const { clientRun, port, token } = env;
await clientRun(['auth', 'login', '--host', `http://localhost:${port}`, '--token', token]);
});
it('should be able to push a load', async () => {
const { clientRun, location } = env;
const scriptPath = resolve(location, 'simple.js');
await writeFile(scriptPath, 'console.log("hello world")', 'utf-8');
await clientRun(['loads', 'push', scriptPath, '-i', 'demo', '-r']);
await sleep(100);
});
it('should be able to get the run', async () => {
const { clientRun } = env;
await clientRun(['runs', 'ls', 'demo']);
});
it('should be able to get logs', async () => {
const { clientRun, clientOutput } = env;
await clientRun(['logs', 'ls', '-l', 'demo']);
console.log(JSON.stringify(clientOutput, null, 2));
});
});

View File

@@ -0,0 +1,136 @@
import { Paths, Terminal, createClientCli } from '@morten-olsen/mini-loader-cli';
import { Runtime, createServer, Config } from '@morten-olsen/mini-loader-server';
import { FastifyInstance } from 'fastify';
import { ContainerInstance } from 'typedi';
import { join, resolve } from 'path';
import getPort from 'get-port';
import { rm } from 'fs/promises';
const cacheLocation = './e2e-tmp';
const getUniqueIdentifier = () => {
return Math.random().toString(36).substr(2, 9);
};
type CliActionOutput =
| {
type: 'stdout';
data: string;
}
| {
type: 'stderr';
data: string;
}
| {
type: 'step';
message: string;
state: 'pending' | 'success' | 'error';
}
| {
type: 'output';
data: unknown;
};
const createEnv = async () => {
const location = resolve(cacheLocation, getUniqueIdentifier());
const port = await getPort();
let clientOutput: CliActionOutput[] = [];
const serverContainer = new ContainerInstance(getUniqueIdentifier());
const clientRun = async (params: string[]) => {
const clientContainer = new ContainerInstance(getUniqueIdentifier());
clientContainer.set(Paths, {
config: join(location, 'client', 'config'),
cache: join(location, 'client', 'cache'),
});
clientContainer.set(Terminal, {
log: async (message: any) => {
clientOutput.push({
type: 'stdout',
data: message,
});
},
output: async (data: any) => {
clientOutput.push({
type: 'output',
data,
});
},
step: async (message: any, action: any) => {
clientOutput.push({
type: 'step',
message,
state: 'pending',
});
try {
const result = await action();
clientOutput.push({
type: 'step',
message,
state: 'success',
});
return result;
} catch (err) {
clientOutput.push({
type: 'step',
message,
state: 'error',
});
throw err;
}
},
});
const client = createClientCli(clientContainer);
client.configureOutput({
writeOut: (data) => {
clientOutput.push({
type: 'stdout',
data,
});
},
writeErr: (data) => {
clientOutput.push({
type: 'stderr',
data,
});
throw new Error(`Error in client ${data}`);
},
});
await client.parseAsync(['', '', ...params]);
};
serverContainer.set(Config, {
database: {
client: 'sqlite3',
connection: {
filename: join(location, 'server', 'data', 'db.sqlite'),
},
useNullAsDefault: true,
},
files: {
data: join(location, 'server', 'data'),
cache: join(location, 'server', 'cache'),
},
});
const server = await createServer(serverContainer);
const runtime = serverContainer.get(Runtime);
const token = await runtime.auth.createToken({
policy: {
'*:*': ['*'],
},
});
const cleanup = async () => {
await server.close();
await rm(location, { recursive: true, force: true });
};
return { clientRun, clientOutput, runtime, server, token, location, port, cleanup };
};
type TestEnv = Awaited<ReturnType<typeof createEnv>>;
export type { TestEnv, FastifyInstance };
export { createEnv };

View File

@@ -0,0 +1,9 @@
{
"extends": "@morten-olsen/mini-loader-configs/tsconfig.esm.json",
"compilerOptions": {
"outDir": "dist/esm",
},
"include": [
"src"
],
}

View File

@@ -0,0 +1,22 @@
/// <reference types="vitest" />
import { createRequire } from 'module';
import { defineConfig } from 'vite';
const require = createRequire(import.meta.url);
const server = require.resolve('../server/src/index.ts');
const cli = require.resolve('../cli/src/index.ts');
const runner = require.resolve('../runner/src/index.ts');
const sdk = require.resolve('../mini-loader/src/index.ts');
export default defineConfig({
test: {
alias: {
'@morten-olsen/mini-loader-server': server,
'@morten-olsen/mini-loader-cli': cli,
'@morten-olsen/mini-loader-runner': runner,
'@morten-olsen/mini-loader': sdk,
},
},
});

1040
pnpm-lock.yaml generated

File diff suppressed because it is too large Load Diff

14
vitest.config.ts Normal file
View File

@@ -0,0 +1,14 @@
/// <reference types="vitest" />
import { defineConfig } from 'vite';
export default defineConfig({
test: {
coverage: {
reporter: ['text', 'json-summary', 'json', 'html'],
reportOnFailure: true,
include: ['packages/*/src/**/*.ts'],
exclude: ['packages/examples/**/*', 'packages/tests/**/*'],
},
},
});

1
vitest.workspace.ts Normal file
View File

@@ -0,0 +1 @@
export default ['packages/*'];