mirror of
https://github.com/pesde-pkg/pesde.git
synced 2025-04-18 10:53:56 +01:00
Compare commits
No commits in common. "v0.6.0-rc.3+registry.0.2.0-rc.1" and "0.5" have entirely different histories.
v0.6.0-rc.
...
0.5
153 changed files with 10775 additions and 15788 deletions
|
@ -1,2 +1,2 @@
|
||||||
PUBLIC_REGISTRY_URL= # url of the registry API, this must have a trailing slash and include the version
|
PUBLIC_REGISTRY_URL= # url of the registry API, this must have a trailing slash and include the version
|
||||||
# example: https://registry.pesde.daimond113.com/v1/
|
# example: https://registry.pesde.daimond113.com/v0/
|
|
@ -1,3 +0,0 @@
|
||||||
# .git-blame-ignore-revs
|
|
||||||
# Enabled the `hard_tabs` option in rustfmt.toml
|
|
||||||
0ceb2f6653b12e8261533ef528d78e3dde7ed757
|
|
7
.github/workflows/debug.yml
vendored
7
.github/workflows/debug.yml
vendored
|
@ -40,11 +40,6 @@ jobs:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
artifact-name: pesde-debug-${{ needs.get-version.outputs.version }}-linux-x86_64
|
artifact-name: pesde-debug-${{ needs.get-version.outputs.version }}-linux-x86_64
|
||||||
|
|
||||||
- job-name: linux-aarch64
|
|
||||||
target: aarch64-unknown-linux-gnu
|
|
||||||
runs-on: ubuntu-24.04-arm
|
|
||||||
artifact-name: pesde-debug-${{ needs.get-version.outputs.version }}-linux-aarch64
|
|
||||||
|
|
||||||
- job-name: macos-x86_64
|
- job-name: macos-x86_64
|
||||||
target: x86_64-apple-darwin
|
target: x86_64-apple-darwin
|
||||||
runs-on: macos-13
|
runs-on: macos-13
|
||||||
|
@ -63,7 +58,7 @@ jobs:
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
- name: Install Linux build dependencies
|
- name: Install Linux build dependencies
|
||||||
if: ${{ startsWith(matrix.runs-on, 'ubuntu') }}
|
if: ${{ matrix.runs-on == 'ubuntu-latest' }}
|
||||||
run: |
|
run: |
|
||||||
sudo apt-get update
|
sudo apt-get update
|
||||||
sudo apt-get install libdbus-1-dev pkg-config
|
sudo apt-get install libdbus-1-dev pkg-config
|
||||||
|
|
13
.github/workflows/release.yaml
vendored
13
.github/workflows/release.yaml
vendored
|
@ -51,11 +51,6 @@ jobs:
|
||||||
arch: x86_64
|
arch: x86_64
|
||||||
target: x86_64-unknown-linux-gnu
|
target: x86_64-unknown-linux-gnu
|
||||||
|
|
||||||
- os: ubuntu-24.04-arm
|
|
||||||
host: linux
|
|
||||||
arch: aarch64
|
|
||||||
target: aarch64-unknown-linux-gnu
|
|
||||||
|
|
||||||
- os: windows-latest
|
- os: windows-latest
|
||||||
host: windows
|
host: windows
|
||||||
arch: x86_64
|
arch: x86_64
|
||||||
|
@ -101,9 +96,11 @@ jobs:
|
||||||
if [ ${{ matrix.host }} = "windows" ]; then
|
if [ ${{ matrix.host }} = "windows" ]; then
|
||||||
mv target/${{ matrix.target }}/release/${{ env.BIN_NAME }}.exe ${{ env.BIN_NAME }}.exe
|
mv target/${{ matrix.target }}/release/${{ env.BIN_NAME }}.exe ${{ env.BIN_NAME }}.exe
|
||||||
7z a ${{ env.ARCHIVE_NAME }}.zip ${{ env.BIN_NAME }}.exe
|
7z a ${{ env.ARCHIVE_NAME }}.zip ${{ env.BIN_NAME }}.exe
|
||||||
|
tar -czf ${{ env.ARCHIVE_NAME }}.tar.gz ${{ env.BIN_NAME }}.exe
|
||||||
else
|
else
|
||||||
mv target/${{ matrix.target }}/release/${{ env.BIN_NAME }} ${{ env.BIN_NAME }}
|
mv target/${{ matrix.target }}/release/${{ env.BIN_NAME }} ${{ env.BIN_NAME }}
|
||||||
zip -r ${{ env.ARCHIVE_NAME }}.zip ${{ env.BIN_NAME }}
|
zip -r ${{ env.ARCHIVE_NAME }}.zip ${{ env.BIN_NAME }}
|
||||||
|
tar -czf ${{ env.ARCHIVE_NAME }}.tar.gz ${{ env.BIN_NAME }}
|
||||||
fi
|
fi
|
||||||
|
|
||||||
- name: Upload zip artifact
|
- name: Upload zip artifact
|
||||||
|
@ -112,6 +109,12 @@ jobs:
|
||||||
name: ${{ env.ARCHIVE_NAME }}.zip
|
name: ${{ env.ARCHIVE_NAME }}.zip
|
||||||
path: ${{ env.ARCHIVE_NAME }}.zip
|
path: ${{ env.ARCHIVE_NAME }}.zip
|
||||||
|
|
||||||
|
- name: Upload tar.gz artifact
|
||||||
|
uses: actions/upload-artifact@v4
|
||||||
|
with:
|
||||||
|
name: ${{ env.ARCHIVE_NAME }}.tar.gz
|
||||||
|
path: ${{ env.ARCHIVE_NAME }}.tar.gz
|
||||||
|
|
||||||
publish:
|
publish:
|
||||||
name: Publish to crates.io
|
name: Publish to crates.io
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
|
|
4
.gitignore
vendored
4
.gitignore
vendored
|
@ -4,6 +4,4 @@
|
||||||
cobertura.xml
|
cobertura.xml
|
||||||
tarpaulin-report.html
|
tarpaulin-report.html
|
||||||
build_rs_cov.profraw
|
build_rs_cov.profraw
|
||||||
registry/data
|
registry/data
|
||||||
data
|
|
||||||
manifest.schema.json
|
|
56
CHANGELOG.md
56
CHANGELOG.md
|
@ -5,59 +5,6 @@ All notable changes to this project will be documented in this file.
|
||||||
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
||||||
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
||||||
|
|
||||||
## [0.6.0-rc.3] - 2025-02-08
|
|
||||||
### Fixed
|
|
||||||
- Fix `self-upgrade` using the wrong path when doing a fresh download by @daimond113
|
|
||||||
- Fix types not being re-exported by @daimond113
|
|
||||||
|
|
||||||
## [0.6.0-rc.2] - 2025-02-07
|
|
||||||
### Fixed
|
|
||||||
- Colour deprecate output to match yank output by @daimond113
|
|
||||||
- Fix zbus panic on Linux by @daimond113
|
|
||||||
|
|
||||||
## [0.6.0-rc.1] - 2025-02-06
|
|
||||||
### Added
|
|
||||||
- Improve installation experience by @lukadev-0
|
|
||||||
- Support using aliases of own dependencies for overrides by @daimond113
|
|
||||||
- Support ignoring parse errors in Luau files by @daimond113
|
|
||||||
- Add path dependencies by @daimond113
|
|
||||||
- Inherit pesde-managed scripts from workspace root by @daimond113
|
|
||||||
- Allow using binaries from workspace root in member packages by @daimond113
|
|
||||||
- Add yanking & deprecating by @daimond113
|
|
||||||
- Add engines as a form of managing runtimes by @daimond113
|
|
||||||
- Modify existing installed packages instead of always reinstalling by @daimond113
|
|
||||||
- Add `cas prune` command to remove unused CAS files & packages by @daimond113
|
|
||||||
- Add `list` and `remove` commands to manage packages in the manifest by @daimond113
|
|
||||||
|
|
||||||
### Fixed
|
|
||||||
- Install dev packages in prod mode and remove them after use to allow them to be used in scripts by @daimond113
|
|
||||||
- Fix infinite loop in the resolver in packages depending on themselves by @daimond113
|
|
||||||
- Do Git operations inside spawn_blocking to avoid performance issues by @daimond113
|
|
||||||
- Scope CAS package indices to the source by @daimond113
|
|
||||||
- Do not copy `default.project.json` in workspace dependencies by @daimond113
|
|
||||||
|
|
||||||
### Changed
|
|
||||||
- Change handling of graphs to a flat structure by @daimond113
|
|
||||||
- Store dependency over downloaded graphs in the lockfile by @daimond113
|
|
||||||
- Improve linking process by @daimond113
|
|
||||||
- Use a proper url encoding library to ensure compatibility with all characters by @daimond113
|
|
||||||
- The `*` specifier now matches all versions, even prereleases by @daimond113
|
|
||||||
- Switch CLI dependencies to ones used by other dependencies to optimize the binary size by @daimond113
|
|
||||||
- Reorder the `help` command by @daimond113
|
|
||||||
- Ignore submodules instead of failing when using Git dependencies with submodules by @daimond113
|
|
||||||
- Exit with code 1 from invalid directory binary linkers by @daimond113
|
|
||||||
|
|
||||||
### Removed
|
|
||||||
- Remove old includes format compatibility by @daimond113
|
|
||||||
- Remove data redundancy for workspace package references by @daimond113
|
|
||||||
- Remove dependency checks from CLI in publish command in favor of registry checks by @daimond113
|
|
||||||
|
|
||||||
### Performance
|
|
||||||
- Use `Arc` for more efficient cloning of multiple structs by @daimond113
|
|
||||||
- Avoid cloning where possible by @daimond113
|
|
||||||
- Remove unnecessary mutex in Wally package download by @daimond113
|
|
||||||
- Lazily format error messages by @daimond113
|
|
||||||
|
|
||||||
## [0.5.3] - 2024-12-30
|
## [0.5.3] - 2024-12-30
|
||||||
### Added
|
### Added
|
||||||
- Add meta field in index files to preserve compatibility with potential future changes by @daimond113
|
- Add meta field in index files to preserve compatibility with potential future changes by @daimond113
|
||||||
|
@ -165,9 +112,6 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
|
||||||
- Asyncify dependency linking by @daimond113
|
- Asyncify dependency linking by @daimond113
|
||||||
- Use `exec` in Unix bin linking to reduce the number of processes by @daimond113
|
- Use `exec` in Unix bin linking to reduce the number of processes by @daimond113
|
||||||
|
|
||||||
[0.6.0-rc.3]: https://github.com/daimond113/pesde/compare/v0.6.0-rc.2%2Bregistry.0.2.0-rc.1..v0.6.0-rc.3%2Bregistry.0.2.0-rc.1
|
|
||||||
[0.6.0-rc.2]: https://github.com/daimond113/pesde/compare/v0.6.0-rc.1%2Bregistry.0.2.0-rc.1..v0.6.0-rc.2%2Bregistry.0.2.0-rc.1
|
|
||||||
[0.6.0-rc.1]: https://github.com/daimond113/pesde/compare/v0.5.3%2Bregistry.0.1.2..v0.6.0-rc.1%2Bregistry.0.2.0-rc.1
|
|
||||||
[0.5.3]: https://github.com/daimond113/pesde/compare/v0.5.2%2Bregistry.0.1.1..v0.5.3%2Bregistry.0.1.2
|
[0.5.3]: https://github.com/daimond113/pesde/compare/v0.5.2%2Bregistry.0.1.1..v0.5.3%2Bregistry.0.1.2
|
||||||
[0.5.2]: https://github.com/daimond113/pesde/compare/v0.5.1%2Bregistry.0.1.0..v0.5.2%2Bregistry.0.1.1
|
[0.5.2]: https://github.com/daimond113/pesde/compare/v0.5.1%2Bregistry.0.1.0..v0.5.2%2Bregistry.0.1.1
|
||||||
[0.5.1]: https://github.com/daimond113/pesde/compare/v0.5.0%2Bregistry.0.1.0..v0.5.1%2Bregistry.0.1.0
|
[0.5.1]: https://github.com/daimond113/pesde/compare/v0.5.0%2Bregistry.0.1.0..v0.5.1%2Bregistry.0.1.0
|
||||||
|
|
1112
Cargo.lock
generated
1112
Cargo.lock
generated
File diff suppressed because it is too large
Load diff
67
Cargo.toml
67
Cargo.toml
|
@ -1,6 +1,6 @@
|
||||||
[package]
|
[package]
|
||||||
name = "pesde"
|
name = "pesde"
|
||||||
version = "0.6.0-rc.3"
|
version = "0.5.3"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
license = "MIT"
|
license = "MIT"
|
||||||
authors = ["daimond113 <contact@daimond113.com>"]
|
authors = ["daimond113 <contact@daimond113.com>"]
|
||||||
|
@ -10,29 +10,28 @@ repository = "https://github.com/pesde-pkg/pesde"
|
||||||
include = ["src/**/*", "Cargo.toml", "Cargo.lock", "README.md", "LICENSE", "CHANGELOG.md"]
|
include = ["src/**/*", "Cargo.toml", "Cargo.lock", "README.md", "LICENSE", "CHANGELOG.md"]
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
default = ["wally-compat", "patches"]
|
|
||||||
bin = [
|
bin = [
|
||||||
"dep:clap",
|
"dep:clap",
|
||||||
"dep:dirs",
|
"dep:dirs",
|
||||||
"dep:tracing-subscriber",
|
"dep:tracing-subscriber",
|
||||||
|
"reqwest/json",
|
||||||
"dep:indicatif",
|
"dep:indicatif",
|
||||||
|
"dep:tracing-indicatif",
|
||||||
"dep:inquire",
|
"dep:inquire",
|
||||||
"dep:toml_edit",
|
"dep:toml_edit",
|
||||||
"dep:console",
|
"dep:colored",
|
||||||
"dep:anyhow",
|
"dep:anyhow",
|
||||||
"dep:keyring",
|
"dep:keyring",
|
||||||
"dep:open",
|
"dep:open",
|
||||||
"dep:paste",
|
|
||||||
"dep:serde_json",
|
|
||||||
"dep:windows-registry",
|
|
||||||
"dep:windows",
|
|
||||||
"gix/worktree-mutation",
|
"gix/worktree-mutation",
|
||||||
|
"dep:serde_json",
|
||||||
|
"dep:winreg",
|
||||||
"fs-err/expose_original_error",
|
"fs-err/expose_original_error",
|
||||||
"tokio/rt",
|
"tokio/rt",
|
||||||
"tokio/rt-multi-thread",
|
"tokio/rt-multi-thread",
|
||||||
"tokio/macros",
|
"tokio/macros",
|
||||||
]
|
]
|
||||||
wally-compat = ["dep:serde_json"]
|
wally-compat = ["dep:async_zip", "dep:serde_json"]
|
||||||
patches = ["dep:git2"]
|
patches = ["dep:git2"]
|
||||||
version-management = ["bin"]
|
version-management = ["bin"]
|
||||||
|
|
||||||
|
@ -45,54 +44,50 @@ required-features = ["bin"]
|
||||||
uninlined_format_args = "warn"
|
uninlined_format_args = "warn"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
serde = { version = "1.0.217", features = ["derive"] }
|
serde = { version = "1.0.216", features = ["derive"] }
|
||||||
toml = "0.8.19"
|
toml = "0.8.19"
|
||||||
gix = { version = "0.70.0", default-features = false, features = ["blocking-http-transport-reqwest-rust-tls", "revparse-regex", "credentials", "parallel"] }
|
serde_with = "3.11.0"
|
||||||
|
gix = { version = "0.68.0", default-features = false, features = ["blocking-http-transport-reqwest-rust-tls", "revparse-regex", "credentials", "parallel"] }
|
||||||
semver = { version = "1.0.24", features = ["serde"] }
|
semver = { version = "1.0.24", features = ["serde"] }
|
||||||
reqwest = { version = "0.12.12", default-features = false, features = ["rustls-tls", "stream", "json"] }
|
reqwest = { version = "0.12.9", default-features = false, features = ["rustls-tls"] }
|
||||||
tokio-tar = "0.3.1"
|
tokio-tar = "0.3.1"
|
||||||
async-compression = { version = "0.4.18", features = ["tokio", "gzip"] }
|
async-compression = { version = "0.4.18", features = ["tokio", "gzip"] }
|
||||||
pathdiff = "0.2.3"
|
pathdiff = "0.2.3"
|
||||||
relative-path = { version = "1.9.3", features = ["serde"] }
|
relative-path = { version = "1.9.3", features = ["serde"] }
|
||||||
tracing = { version = "0.1.41", features = ["attributes"] }
|
tracing = { version = "0.1.41", features = ["attributes"] }
|
||||||
thiserror = "2.0.11"
|
thiserror = "2.0.7"
|
||||||
tokio = { version = "1.43.0", features = ["process", "macros"] }
|
tokio = { version = "1.42.0", features = ["process"] }
|
||||||
tokio-util = "0.7.13"
|
tokio-util = "0.7.13"
|
||||||
async-stream = "0.3.6"
|
async-stream = "0.3.6"
|
||||||
futures = "0.3.31"
|
futures = "0.3.31"
|
||||||
full_moon = { version = "1.2.0", features = ["luau"] }
|
full_moon = { version = "1.1.2", features = ["luau"] }
|
||||||
url = { version = "2.5.4", features = ["serde"] }
|
url = { version = "2.5.4", features = ["serde"] }
|
||||||
jiff = { version = "0.1.28", default-features = false, features = ["serde", "std"] }
|
chrono = { version = "0.4.39", features = ["serde"] }
|
||||||
sha2 = "0.10.8"
|
sha2 = "0.10.8"
|
||||||
tempfile = "3.15.0"
|
tempfile = "3.14.0"
|
||||||
wax = { version = "0.6.0", default-features = false }
|
wax = { version = "0.6.0", default-features = false }
|
||||||
fs-err = { version = "3.0.0", features = ["tokio"] }
|
fs-err = { version = "3.0.0", features = ["tokio"] }
|
||||||
urlencoding = "2.1.3"
|
|
||||||
async_zip = { version = "0.0.17", features = ["tokio", "deflate", "deflate64", "tokio-fs"] }
|
|
||||||
|
|
||||||
# TODO: remove this when gitoxide adds support for: committing, pushing, adding
|
# TODO: remove this when gitoxide adds support for: committing, pushing, adding
|
||||||
git2 = { version = "0.20.0", optional = true }
|
git2 = { version = "0.19.0", optional = true }
|
||||||
|
|
||||||
serde_json = { version = "1.0.136", optional = true }
|
async_zip = { version = "0.0.17", features = ["tokio", "deflate", "deflate64", "tokio-fs"], optional = true }
|
||||||
|
serde_json = { version = "1.0.133", optional = true }
|
||||||
|
|
||||||
anyhow = { version = "1.0.95", optional = true }
|
anyhow = { version = "1.0.94", optional = true }
|
||||||
open = { version = "5.3.2", optional = true }
|
open = { version = "5.3.1", optional = true }
|
||||||
keyring = { version = "3.6.1", features = ["crypto-rust", "windows-native", "apple-native", "sync-secret-service"], optional = true }
|
keyring = { version = "3.6.1", features = ["crypto-rust", "windows-native", "apple-native", "async-secret-service", "async-io"], optional = true }
|
||||||
console = { version = "0.15.10", optional = true }
|
colored = { version = "2.1.0", optional = true }
|
||||||
toml_edit = { version = "0.22.22", optional = true }
|
toml_edit = { version = "0.22.22", optional = true }
|
||||||
clap = { version = "4.5.26", features = ["derive"], optional = true }
|
clap = { version = "4.5.23", features = ["derive"], optional = true }
|
||||||
dirs = { version = "6.0.0", optional = true }
|
dirs = { version = "5.0.1", optional = true }
|
||||||
tracing-subscriber = { version = "0.3.19", features = ["env-filter"], optional = true }
|
tracing-subscriber = { version = "0.3.19", features = ["env-filter"], optional = true }
|
||||||
indicatif = { version = "0.17.9", optional = true }
|
indicatif = { version = "0.17.9", optional = true }
|
||||||
inquire = { version = "0.7.5", default-features = false, features = ["console", "one-liners"], optional = true }
|
tracing-indicatif = { version = "0.3.8", optional = true }
|
||||||
paste = { version = "1.0.15", optional = true }
|
inquire = { version = "0.7.5", optional = true }
|
||||||
|
|
||||||
[target.'cfg(target_os = "windows")'.dependencies]
|
[target.'cfg(target_os = "windows")'.dependencies]
|
||||||
windows-registry = { version = "0.4.0", optional = true }
|
winreg = { version = "0.52.0", optional = true }
|
||||||
windows = { version = "0.59.0", features = ["Win32_Storage", "Win32_Storage_FileSystem", "Win32_Security"], optional = true }
|
|
||||||
|
|
||||||
[dev-dependencies]
|
|
||||||
schemars = { git = "https://github.com/daimond113/schemars", rev = "bc7c7d6", features = ["semver1", "url2"] }
|
|
||||||
|
|
||||||
[workspace]
|
[workspace]
|
||||||
resolver = "2"
|
resolver = "2"
|
||||||
|
@ -101,16 +96,12 @@ members = ["registry"]
|
||||||
[profile.dev.package.full_moon]
|
[profile.dev.package.full_moon]
|
||||||
opt-level = 3
|
opt-level = 3
|
||||||
|
|
||||||
[profile.dev.package.miniz_oxide]
|
|
||||||
opt-level = 3
|
|
||||||
|
|
||||||
[profile.release]
|
[profile.release]
|
||||||
opt-level = "s"
|
opt-level = "s"
|
||||||
lto = true
|
lto = true
|
||||||
incremental = true
|
incremental = true
|
||||||
codegen-units = 1
|
codegen-units = 1
|
||||||
panic = "abort"
|
|
||||||
|
|
||||||
[profile.release.package.pesde-registry]
|
[profile.release.package.pesde-registry]
|
||||||
# add debug symbols for Sentry stack traces
|
# add debug symbols for Sentry stack traces
|
||||||
debug = "full"
|
debug = "full"
|
BIN
docs/bun.lockb
BIN
docs/bun.lockb
Binary file not shown.
|
@ -10,20 +10,20 @@
|
||||||
"astro": "astro"
|
"astro": "astro"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@astrojs/check": "0.9.4",
|
"@astrojs/check": "^0.9.3",
|
||||||
"@astrojs/starlight": "0.30.6",
|
"@astrojs/starlight": "^0.28.2",
|
||||||
"@astrojs/starlight-tailwind": "3.0.0",
|
"@astrojs/starlight-tailwind": "^2.0.3",
|
||||||
"@astrojs/tailwind": "5.1.4",
|
"@astrojs/tailwind": "^5.1.1",
|
||||||
"@fontsource-variable/nunito-sans": "^5.1.1",
|
"@fontsource-variable/nunito-sans": "^5.1.0",
|
||||||
"@shikijs/rehype": "^1.26.2",
|
"@shikijs/rehype": "^1.21.0",
|
||||||
"astro": "5.1.5",
|
"astro": "^4.15.9",
|
||||||
"sharp": "^0.33.5",
|
"sharp": "^0.33.5",
|
||||||
"shiki": "^1.26.2",
|
"shiki": "^1.21.0",
|
||||||
"tailwindcss": "^3.4.17",
|
"tailwindcss": "^3.4.13",
|
||||||
"typescript": "^5.7.3"
|
"typescript": "^5.6.2"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"prettier-plugin-astro": "^0.14.1",
|
"prettier-plugin-astro": "^0.14.1",
|
||||||
"prettier-plugin-tailwindcss": "^0.6.9"
|
"prettier-plugin-tailwindcss": "^0.6.8"
|
||||||
}
|
}
|
||||||
}
|
}
|
|
@ -3,7 +3,12 @@
|
||||||
href="https://pesde.daimond113.com/"
|
href="https://pesde.daimond113.com/"
|
||||||
class="flex text-[var(--sl-color-text-accent)] hover:opacity-80"
|
class="flex text-[var(--sl-color-text-accent)] hover:opacity-80"
|
||||||
>
|
>
|
||||||
<svg viewBox="0 0 56 28" class="h-7" fill="none" xmlns="http://www.w3.org/2000/svg">
|
<svg
|
||||||
|
viewBox="0 0 56 28"
|
||||||
|
class="h-7"
|
||||||
|
fill="none"
|
||||||
|
xmlns="http://www.w3.org/2000/svg"
|
||||||
|
>
|
||||||
<title>pesde</title>
|
<title>pesde</title>
|
||||||
<path
|
<path
|
||||||
d="M0 28V26.3156H2.25652V12.2361H0.0635639V10.5517H4.44947L4.48125 11.9819L3.78205 12.3315C4.41769 11.6746 5.16986 11.1661 6.03857 10.8059C6.92846 10.4245 7.82895 10.2338 8.74003 10.2338C9.863 10.2338 10.88 10.4775 11.7911 10.9648C12.7234 11.4522 13.4544 12.1726 13.9841 13.126C14.5349 14.0795 14.8104 15.2448 14.8104 16.6221C14.8104 18.0416 14.5138 19.26 13.9205 20.277C13.3272 21.2728 12.5327 22.0356 11.5368 22.5653C10.5622 23.095 9.5028 23.3598 8.35865 23.3598C7.72301 23.3598 7.11916 23.2751 6.54708 23.1056C5.99619 22.9361 5.50887 22.7242 5.08511 22.4699C4.66135 22.1945 4.34353 21.8873 4.13165 21.5483L4.60838 21.4529L4.5766 26.3156H7.02381V28H0ZM7.94549 21.6118C9.19558 21.6118 10.2444 21.2092 11.0919 20.4041C11.9394 19.5778 12.3632 18.3807 12.3632 16.8127C12.3632 15.2872 11.9606 14.1113 11.1555 13.2849C10.3503 12.4586 9.3333 12.0454 8.1044 12.0454C7.72301 12.0454 7.26747 12.1196 6.73777 12.2679C6.20807 12.395 5.67837 12.6069 5.14867 12.9035C4.61898 13.2002 4.17403 13.5922 3.81383 14.0795L4.5766 12.7446L4.60838 20.7219L3.8774 19.7367C4.42828 20.3299 5.06392 20.7961 5.78431 21.1351C6.5047 21.4529 7.2251 21.6118 7.94549 21.6118Z"
|
d="M0 28V26.3156H2.25652V12.2361H0.0635639V10.5517H4.44947L4.48125 11.9819L3.78205 12.3315C4.41769 11.6746 5.16986 11.1661 6.03857 10.8059C6.92846 10.4245 7.82895 10.2338 8.74003 10.2338C9.863 10.2338 10.88 10.4775 11.7911 10.9648C12.7234 11.4522 13.4544 12.1726 13.9841 13.126C14.5349 14.0795 14.8104 15.2448 14.8104 16.6221C14.8104 18.0416 14.5138 19.26 13.9205 20.277C13.3272 21.2728 12.5327 22.0356 11.5368 22.5653C10.5622 23.095 9.5028 23.3598 8.35865 23.3598C7.72301 23.3598 7.11916 23.2751 6.54708 23.1056C5.99619 22.9361 5.50887 22.7242 5.08511 22.4699C4.66135 22.1945 4.34353 21.8873 4.13165 21.5483L4.60838 21.4529L4.5766 26.3156H7.02381V28H0ZM7.94549 21.6118C9.19558 21.6118 10.2444 21.2092 11.0919 20.4041C11.9394 19.5778 12.3632 18.3807 12.3632 16.8127C12.3632 15.2872 11.9606 14.1113 11.1555 13.2849C10.3503 12.4586 9.3333 12.0454 8.1044 12.0454C7.72301 12.0454 7.26747 12.1196 6.73777 12.2679C6.20807 12.395 5.67837 12.6069 5.14867 12.9035C4.61898 13.2002 4.17403 13.5922 3.81383 14.0795L4.5766 12.7446L4.60838 20.7219L3.8774 19.7367C4.42828 20.3299 5.06392 20.7961 5.78431 21.1351C6.5047 21.4529 7.2251 21.6118 7.94549 21.6118Z"
|
||||||
|
@ -22,7 +27,8 @@
|
||||||
fill="currentColor"></path>
|
fill="currentColor"></path>
|
||||||
</svg>
|
</svg>
|
||||||
</a>
|
</a>
|
||||||
<span class="-mt-px ml-2.5 mr-2 text-xl text-[var(--sl-color-gray-5)]">/</span>
|
<span class="-mt-px ml-2.5 mr-2 text-xl text-[var(--sl-color-gray-5)]">/</span
|
||||||
|
>
|
||||||
<a
|
<a
|
||||||
class="font-medium text-[var(--sl-color-gray-2)] no-underline hover:opacity-80 md:text-lg"
|
class="font-medium text-[var(--sl-color-gray-2)] no-underline hover:opacity-80 md:text-lg"
|
||||||
href="/">docs</a
|
href="/">docs</a
|
||||||
|
|
|
@ -1,7 +1,6 @@
|
||||||
import { defineCollection } from "astro:content"
|
import { defineCollection } from "astro:content"
|
||||||
import { docsLoader } from "@astrojs/starlight/loaders"
|
|
||||||
import { docsSchema } from "@astrojs/starlight/schema"
|
import { docsSchema } from "@astrojs/starlight/schema"
|
||||||
|
|
||||||
export const collections = {
|
export const collections = {
|
||||||
docs: defineCollection({ loader: docsLoader(), schema: docsSchema() }),
|
docs: defineCollection({ schema: docsSchema() }),
|
||||||
}
|
}
|
||||||
|
|
|
@ -42,9 +42,6 @@ hello
|
||||||
# Hello, pesde! (pesde/hello@1.0.0, lune)
|
# Hello, pesde! (pesde/hello@1.0.0, lune)
|
||||||
```
|
```
|
||||||
|
|
||||||
Note that they are scoped to the nearest `pesde.toml` file. However, you can use
|
|
||||||
binaries of the workspace root from member packages.
|
|
||||||
|
|
||||||
## Making a binary package
|
## Making a binary package
|
||||||
|
|
||||||
To make a binary package you must use a target compatible with binary exports.
|
To make a binary package you must use a target compatible with binary exports.
|
||||||
|
|
|
@ -137,24 +137,6 @@ pesde add workspace:acme/bar
|
||||||
href="/guides/workspaces/"
|
href="/guides/workspaces/"
|
||||||
/>
|
/>
|
||||||
|
|
||||||
## Path Dependencies
|
|
||||||
|
|
||||||
Path dependencies are dependencies found anywhere available to the operating system.
|
|
||||||
They are useful for local development, but are forbidden in published packages.
|
|
||||||
|
|
||||||
The path must be absolute and point to a directory containing a `pesde.toml` file.
|
|
||||||
|
|
||||||
```toml title="pesde.toml"
|
|
||||||
[dependencies]
|
|
||||||
foo = { path = "/home/user/foo" }
|
|
||||||
```
|
|
||||||
|
|
||||||
You can also add a path dependency by running the following command:
|
|
||||||
|
|
||||||
```sh
|
|
||||||
pesde add path:/home/user/foo
|
|
||||||
```
|
|
||||||
|
|
||||||
## Peer Dependencies
|
## Peer Dependencies
|
||||||
|
|
||||||
Peer dependencies are dependencies that are not installed automatically when
|
Peer dependencies are dependencies that are not installed automatically when
|
||||||
|
|
|
@ -32,29 +32,6 @@ foo = { name = "acme/foo", version = "^1.0.0" }
|
||||||
|
|
||||||
Now, when you run `pesde install`, `bar` 2.0.0 will be used instead of 1.0.0.
|
Now, when you run `pesde install`, `bar` 2.0.0 will be used instead of 1.0.0.
|
||||||
|
|
||||||
Overrides are also able to use aliases to share the specifier you use for your
|
|
||||||
own dependencies:
|
|
||||||
|
|
||||||
```toml title="pesde.toml"
|
|
||||||
[dependencies]
|
|
||||||
foo = { name = "acme/foo", version = "^1.0.0" }
|
|
||||||
bar = { name = "acme/bar", version = "^2.0.0" }
|
|
||||||
|
|
||||||
[overrides]
|
|
||||||
"foo>bar" = "bar"
|
|
||||||
```
|
|
||||||
|
|
||||||
This is the same as if you had written:
|
|
||||||
|
|
||||||
```toml title="pesde.toml"
|
|
||||||
[dependencies]
|
|
||||||
foo = { name = "acme/foo", version = "^1.0.0" }
|
|
||||||
bar = { name = "acme/bar", version = "^2.0.0" }
|
|
||||||
|
|
||||||
[overrides]
|
|
||||||
"foo>bar" = { name = "acme/bar", version = "^2.0.0" }
|
|
||||||
```
|
|
||||||
|
|
||||||
You can learn more about the syntax for dependency overrides in the
|
You can learn more about the syntax for dependency overrides in the
|
||||||
[reference](/reference/manifest#overrides).
|
[reference](/reference/manifest#overrides).
|
||||||
|
|
||||||
|
|
|
@ -91,13 +91,6 @@ For example, you may publish a package that can be used in both Roblox and
|
||||||
Luau environments by publishing two versions of the package, one for each
|
Luau environments by publishing two versions of the package, one for each
|
||||||
environment.
|
environment.
|
||||||
|
|
||||||
<Aside type="caution">
|
|
||||||
|
|
||||||
Packages for different targets but on the same version must have
|
|
||||||
the same description.
|
|
||||||
|
|
||||||
</Aside>
|
|
||||||
|
|
||||||
## Documentation
|
## Documentation
|
||||||
|
|
||||||
The `README.md` file in the root of the package will be displayed on the
|
The `README.md` file in the root of the package will be displayed on the
|
||||||
|
|
|
@ -1,56 +0,0 @@
|
||||||
---
|
|
||||||
title: Removing Packages
|
|
||||||
description: Learn how to remove packages from the registry.
|
|
||||||
---
|
|
||||||
|
|
||||||
pesde doesn't support removing packages from the registry. This is to ensure
|
|
||||||
that the registry remains a reliable source of packages for everyone. However,
|
|
||||||
pesde provides other mechanisms to handle packages that are no longer needed.
|
|
||||||
|
|
||||||
## Yanking
|
|
||||||
|
|
||||||
Yanking is limited to a specific version (and target) of a package. It is used
|
|
||||||
to mark a version as broken or deprecated. Yanked versions are unavailable
|
|
||||||
to download fresh, but they can still be installed if they are present in the
|
|
||||||
lockfile of a project.
|
|
||||||
|
|
||||||
To yank a package, you can use the `pesde yank` command:
|
|
||||||
|
|
||||||
```sh
|
|
||||||
pesde yank <PACKAGE>@<VERSION> <TARGET>
|
|
||||||
```
|
|
||||||
|
|
||||||
You can leave out the target if you want to yank all targets of the version:
|
|
||||||
|
|
||||||
```sh
|
|
||||||
pesde yank <PACKAGE>@<VERSION>
|
|
||||||
```
|
|
||||||
|
|
||||||
## Deprecating
|
|
||||||
|
|
||||||
On the other hand, deprecating a package is used to mark a package as deprecated
|
|
||||||
in the registry. This is useful when you want to discourage users from using
|
|
||||||
a package, but don't want to break existing projects that depend on it. Unlike
|
|
||||||
yanking, your package will still be able to be installed fresh. However, when it
|
|
||||||
is installed, a warning will be shown to the user.
|
|
||||||
|
|
||||||
To deprecate a package, you can use the `pesde deprecate` command:
|
|
||||||
|
|
||||||
```sh
|
|
||||||
pesde deprecate <PACKAGE> [REASON]
|
|
||||||
```
|
|
||||||
|
|
||||||
You must provide a non-empty reason when deprecating a package. This is to
|
|
||||||
inform users why the package is deprecated. For example, if your package
|
|
||||||
has been replaced by another package, you can provide a reason like:
|
|
||||||
|
|
||||||
```sh
|
|
||||||
pesde deprecate acme/old-package "This package has been replaced by acme/new-package."
|
|
||||||
```
|
|
||||||
|
|
||||||
## Other Options
|
|
||||||
|
|
||||||
There are other situations in which you might want to remove a package from
|
|
||||||
the registry. Please refer to the policies of the registry you are using for
|
|
||||||
more information on how to handle these situations. The process for the official
|
|
||||||
registry is described [here](/registry/policies/#package-removal).
|
|
|
@ -188,13 +188,10 @@ This will cause the `src` directory to be directly synced into Roblox.
|
||||||
|
|
||||||
In pesde, you should not have a `default.project.json` file in your package.
|
In pesde, you should not have a `default.project.json` file in your package.
|
||||||
Instead, you are required to use the `build_files` field to specify a 1:1 match
|
Instead, you are required to use the `build_files` field to specify a 1:1 match
|
||||||
between Roblox and the file system. These are given to the
|
between Roblox and the file system. pesde forbids `default.project.json` to be
|
||||||
`roblox_sync_config_generator` script to generate the configuration for the sync
|
part of a published package, and regenerates it when installing a pesde git
|
||||||
tool the user is using. pesde forbids `default.project.json` to be part of a
|
dependency. This allows the consumer of your package to choose the sync tool
|
||||||
published package, as well as ignoring them from Git dependencies. This allows
|
they want to use, instead of being constrained to only using Rojo.
|
||||||
the consumer of your package to choose the sync tool they want to use, instead
|
|
||||||
of being constrained to only using Rojo as well as preventing broken packages
|
|
||||||
from being published (for example, if the project is configured as a DataModel).
|
|
||||||
|
|
||||||
This has the effect that the structure of the files in the file system ends up
|
This has the effect that the structure of the files in the file system ends up
|
||||||
being reflected inside Roblox.
|
being reflected inside Roblox.
|
||||||
|
|
|
@ -23,7 +23,7 @@ the following content:
|
||||||
api = "https://registry.acme.local/"
|
api = "https://registry.acme.local/"
|
||||||
|
|
||||||
# package download URL (optional)
|
# package download URL (optional)
|
||||||
download = "{API_URL}/v1/packages/{PACKAGE}/{PACKAGE_VERSION}/{PACKAGE_TARGET}/archive"
|
download = "{API_URL}/v0/packages/{PACKAGE}/{PACKAGE_VERSION}/{PACKAGE_TARGET}"
|
||||||
|
|
||||||
# the client ID of the GitHub OAuth app (optional)
|
# the client ID of the GitHub OAuth app (optional)
|
||||||
github_oauth_client_id = "a1d648966fdfbdcd9295"
|
github_oauth_client_id = "a1d648966fdfbdcd9295"
|
||||||
|
@ -58,7 +58,7 @@ scripts_packages = ["pesde/scripts_rojo"]
|
||||||
- `{PACKAGE_VERSION}`: The package version.
|
- `{PACKAGE_VERSION}`: The package version.
|
||||||
- `{PACKAGE_TARGET}`: The package target.
|
- `{PACKAGE_TARGET}`: The package target.
|
||||||
|
|
||||||
Defaults to `{API_URL}/v1/packages/{PACKAGE}/{PACKAGE_VERSION}/{PACKAGE_TARGET}/archive`.
|
Defaults to `{API_URL}/v0/packages/{PACKAGE}/{PACKAGE_VERSION}/{PACKAGE_TARGET}`.
|
||||||
|
|
||||||
- **github_oauth_client_id**: This is required if you use GitHub OAuth for
|
- **github_oauth_client_id**: This is required if you use GitHub OAuth for
|
||||||
authentication. See below for more information.
|
authentication. See below for more information.
|
||||||
|
@ -115,11 +115,11 @@ for this purpose.
|
||||||
`GITHUB_USERNAME`. This is required.
|
`GITHUB_USERNAME`. This is required.
|
||||||
|
|
||||||
- **COMMITTER_GIT_NAME**: The name to use for the committer when updating the
|
- **COMMITTER_GIT_NAME**: The name to use for the committer when updating the
|
||||||
index repository. This is required.\
|
index repository.\
|
||||||
Example: `pesde index updater`
|
Example: `pesde index updater`
|
||||||
|
|
||||||
- **COMMITTER_GIT_EMAIL**: The email to use for the committer when updating the
|
- **COMMITTER_GIT_EMAIL**: The email to use for the committer when updating the
|
||||||
index repository. This is required.\
|
index repository.\
|
||||||
Example: `pesde@localhost`
|
Example: `pesde@localhost`
|
||||||
|
|
||||||
- **DATA_DIR**: The directory where the registry stores miscellaneous data.
|
- **DATA_DIR**: The directory where the registry stores miscellaneous data.
|
||||||
|
|
|
@ -5,11 +5,22 @@ description: Install pesde
|
||||||
|
|
||||||
import { Aside, Steps, TabItem, Tabs } from "@astrojs/starlight/components"
|
import { Aside, Steps, TabItem, Tabs } from "@astrojs/starlight/components"
|
||||||
|
|
||||||
|
## Prerequisites
|
||||||
|
|
||||||
|
pesde requires [Lune](https://lune-org.github.io/docs) to be installed on your
|
||||||
|
system in order to function properly.
|
||||||
|
|
||||||
|
You can follow the installation instructions in the
|
||||||
|
[Lune documentation](https://lune-org.github.io/docs/getting-started/1-installation).
|
||||||
|
|
||||||
|
## Installing pesde
|
||||||
|
|
||||||
<Steps>
|
<Steps>
|
||||||
|
|
||||||
1. Go to the [GitHub releases page](https://github.com/pesde-pkg/pesde/releases/latest).
|
1. Go to the [GitHub releases page](https://github.com/pesde-pkg/pesde/releases/latest).
|
||||||
|
|
||||||
2. Download the corresponding archive for your operating system.
|
2. Download the corresponding archive for your operating system. You can choose
|
||||||
|
whether to use the `.zip` or `.tar.gz` files.
|
||||||
|
|
||||||
3. Extract the downloaded archive to a folder on your computer.
|
3. Extract the downloaded archive to a folder on your computer.
|
||||||
|
|
||||||
|
@ -65,7 +76,6 @@ import { Aside, Steps, TabItem, Tabs } from "@astrojs/starlight/components"
|
||||||
|
|
||||||
</TabItem>
|
</TabItem>
|
||||||
</Tabs>
|
</Tabs>
|
||||||
<br />
|
|
||||||
|
|
||||||
5. Verify that pesde is installed by running the following command:
|
5. Verify that pesde is installed by running the following command:
|
||||||
|
|
||||||
|
@ -82,8 +92,8 @@ import { Aside, Steps, TabItem, Tabs } from "@astrojs/starlight/components"
|
||||||
It is not recommended to use toolchain managers (such as Rokit or Aftman) to
|
It is not recommended to use toolchain managers (such as Rokit or Aftman) to
|
||||||
install pesde. You can use `pesde self-upgrade` if you need to update pesde.
|
install pesde. You can use `pesde self-upgrade` if you need to update pesde.
|
||||||
|
|
||||||
If you need everyone to use a compatible version of pesde, you can use the
|
If you need everyone to use the same version of pesde, you can use the
|
||||||
`[engines.pesde]` field in `pesde.toml` to specify the version of pesde to use
|
`pesde_version` field in `pesde.toml` to specify the version of pesde to use
|
||||||
for the current project.
|
for the current project.
|
||||||
|
|
||||||
</Aside>
|
</Aside>
|
||||||
|
|
|
@ -33,7 +33,7 @@ pesde init
|
||||||
# what is the repository URL of this project?
|
# what is the repository URL of this project?
|
||||||
# what is the license of this project? MIT
|
# what is the license of this project? MIT
|
||||||
# what environment are you targeting for your package? luau
|
# what environment are you targeting for your package? luau
|
||||||
# would you like to setup Roblox compatibility scripts? No
|
# would you like to setup default Roblox compatibility scripts? No
|
||||||
```
|
```
|
||||||
|
|
||||||
The command will create a `pesde.toml` file in the current folder. Go ahead
|
The command will create a `pesde.toml` file in the current folder. Go ahead
|
||||||
|
|
|
@ -55,83 +55,10 @@ is printed.
|
||||||
|
|
||||||
The default index is [`pesde-index`](https://github.com/pesde-pkg/index).
|
The default index is [`pesde-index`](https://github.com/pesde-pkg/index).
|
||||||
|
|
||||||
## `pesde cas`
|
|
||||||
|
|
||||||
Content-addressable storage (CAS) related commands.
|
|
||||||
|
|
||||||
### `pesde cas prune`
|
|
||||||
|
|
||||||
Removes unused CAS files and packages.
|
|
||||||
|
|
||||||
## `pesde init`
|
## `pesde init`
|
||||||
|
|
||||||
Initializes a new pesde project in the current directory.
|
Initializes a new pesde project in the current directory.
|
||||||
|
|
||||||
## `pesde add`
|
|
||||||
|
|
||||||
```sh
|
|
||||||
pesde add <PACKAGE>
|
|
||||||
```
|
|
||||||
|
|
||||||
Adds a package to the dependencies of the current project.
|
|
||||||
|
|
||||||
- `-i, --index <INDEX>`: The index in which to search for the package.
|
|
||||||
- `-t, --target <TARGET>`: The target environment for the package.
|
|
||||||
- `-a, --alias <ALIAS>`: The alias to use for the package, defaults to the
|
|
||||||
package name.
|
|
||||||
- `-p, --peer`: Adds the package as a peer dependency.
|
|
||||||
- `-d, --dev`: Adds the package as a dev dependency.
|
|
||||||
|
|
||||||
The following formats are supported:
|
|
||||||
|
|
||||||
```sh
|
|
||||||
pesde add pesde/hello
|
|
||||||
pesde add pesde/hello@1.2.3
|
|
||||||
pesde add gh#acme/package#main
|
|
||||||
pesde add https://git.acme.local/package.git#aeff6
|
|
||||||
pesde add workspace:pesde/hello
|
|
||||||
pesde add workspace:pesde/hello@1.2.3
|
|
||||||
pesde add path:/home/user/package
|
|
||||||
```
|
|
||||||
|
|
||||||
## `pesde remove`
|
|
||||||
|
|
||||||
```sh
|
|
||||||
pesde remove <ALIAS>
|
|
||||||
```
|
|
||||||
|
|
||||||
Removes a package from the dependencies of the current project.
|
|
||||||
|
|
||||||
## `pesde install`
|
|
||||||
|
|
||||||
Installs dependencies for the current project.
|
|
||||||
|
|
||||||
- `--locked`: Whether to error if the lockfile is out of date.
|
|
||||||
- `--prod`: Whether to not linking dev dependencies.
|
|
||||||
- `--network-concurrency <CONCURRENCY>`: The number of concurrent network
|
|
||||||
requests to make at most. Defaults to 16.
|
|
||||||
- `--force`: Whether to force reinstall all packages even if they are already
|
|
||||||
installed (useful if there is any issue with the current installation).
|
|
||||||
|
|
||||||
## `pesde update`
|
|
||||||
|
|
||||||
Updates the dependencies of the current project.
|
|
||||||
|
|
||||||
- `--no-install`: Whether to only update the lockfile without installing the
|
|
||||||
dependencies.
|
|
||||||
- `--network-concurrency <CONCURRENCY>`: The number of concurrent network
|
|
||||||
requests to make at most. Defaults to 16.
|
|
||||||
- `--force`: Whether to force reinstall all packages even if they are already
|
|
||||||
installed (useful if there is any issue with the current installation).
|
|
||||||
|
|
||||||
## `pesde outdated`
|
|
||||||
|
|
||||||
Lists outdated dependencies of the current project.
|
|
||||||
|
|
||||||
## `pesde list`
|
|
||||||
|
|
||||||
Lists the dependencies of the current project.
|
|
||||||
|
|
||||||
## `pesde run`
|
## `pesde run`
|
||||||
|
|
||||||
Runs a script from the current project using Lune.
|
Runs a script from the current project using Lune.
|
||||||
|
@ -156,6 +83,13 @@ Arguments can be passed to the script by using `--` followed by the arguments.
|
||||||
pesde run foo -- --arg1 --arg2
|
pesde run foo -- --arg1 --arg2
|
||||||
```
|
```
|
||||||
|
|
||||||
|
## `pesde install`
|
||||||
|
|
||||||
|
Installs dependencies for the current project.
|
||||||
|
|
||||||
|
- `--locked`: Whether to error if the lockfile is out of date.
|
||||||
|
- `--prod`: Whether to skip installing dev dependencies.
|
||||||
|
|
||||||
## `pesde publish`
|
## `pesde publish`
|
||||||
|
|
||||||
Publishes the current project to the pesde registry.
|
Publishes the current project to the pesde registry.
|
||||||
|
@ -165,26 +99,18 @@ Publishes the current project to the pesde registry.
|
||||||
publish it.
|
publish it.
|
||||||
- `-y, --yes`: Whether to skip the confirmation prompt.
|
- `-y, --yes`: Whether to skip the confirmation prompt.
|
||||||
- `-i, --index`: Name of the index to publish to. Defaults to `default`.
|
- `-i, --index`: Name of the index to publish to. Defaults to `default`.
|
||||||
- `--no-verify`: Whether to skip syntax validation of the exports of the
|
|
||||||
package.
|
|
||||||
|
|
||||||
## `pesde yank`
|
## `pesde self-install`
|
||||||
|
|
||||||
Yanks a version of a package from the registry.
|
Performs the pesde installation process. This should be the first command run
|
||||||
|
after downloading the pesde binary.
|
||||||
|
|
||||||
- `--undo`: Whether to unyank the package.
|
## `pesde self-upgrade`
|
||||||
- `-i, --index`: Name of the index to yank from. Defaults to `default`.
|
|
||||||
|
|
||||||
## `pesde deprecate`
|
Upgrades the pesde binary to the latest version.
|
||||||
|
|
||||||
```sh
|
- `--use-cached`: Whether to use the version displayed in the "upgrade available"
|
||||||
pesde deprecate <PACKAGE> [REASON]
|
message instead of checking for the latest version.
|
||||||
```
|
|
||||||
|
|
||||||
Deprecates a package in the registry. A non-empty reason must be provided.
|
|
||||||
|
|
||||||
- `--undo`: Whether to undepricate the package.
|
|
||||||
- `-i, --index`: Name of the index to deprecate from. Defaults to `default`.
|
|
||||||
|
|
||||||
## `pesde patch`
|
## `pesde patch`
|
||||||
|
|
||||||
|
@ -211,6 +137,33 @@ pesde patch-commit <PATH>
|
||||||
|
|
||||||
Applies the changes made in the patching environment created by `pesde patch`.
|
Applies the changes made in the patching environment created by `pesde patch`.
|
||||||
|
|
||||||
|
## `pesde add`
|
||||||
|
|
||||||
|
```sh
|
||||||
|
pesde add <PACKAGE>
|
||||||
|
```
|
||||||
|
|
||||||
|
Adds a package to the dependencies of the current project.
|
||||||
|
|
||||||
|
- `-i, --index <INDEX>`: The index in which to search for the package.
|
||||||
|
- `-t, --target <TARGET>`: The target environment for the package.
|
||||||
|
- `-a, --alias <ALIAS>`: The alias to use for the package, defaults to the
|
||||||
|
package name.
|
||||||
|
- `-p, --peer`: Adds the package as a peer dependency.
|
||||||
|
- `-d, --dev`: Adds the package as a dev dependency.
|
||||||
|
|
||||||
|
The following formats are supported:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
pesde add pesde/hello
|
||||||
|
pesde add gh#acme/package#main
|
||||||
|
pesde add https://git.acme.local/package.git#aeff6
|
||||||
|
```
|
||||||
|
|
||||||
|
## `pesde update`
|
||||||
|
|
||||||
|
Updates the dependencies of the current project.
|
||||||
|
|
||||||
## `pesde x`
|
## `pesde x`
|
||||||
|
|
||||||
Runs a one-off binary package.
|
Runs a one-off binary package.
|
||||||
|
@ -225,15 +178,3 @@ a pesde project.
|
||||||
```sh
|
```sh
|
||||||
pesde x pesde/hello
|
pesde x pesde/hello
|
||||||
```
|
```
|
||||||
|
|
||||||
## `pesde self-install`
|
|
||||||
|
|
||||||
Performs the pesde installation process. This should be the first command run
|
|
||||||
after downloading the pesde binary.
|
|
||||||
|
|
||||||
## `pesde self-upgrade`
|
|
||||||
|
|
||||||
Upgrades the pesde binary to the latest version.
|
|
||||||
|
|
||||||
- `--use-cached`: Whether to use the version displayed in the "upgrade available"
|
|
||||||
message instead of checking for the latest version.
|
|
||||||
|
|
|
@ -84,6 +84,11 @@ includes = [
|
||||||
]
|
]
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### `pesde_version`
|
||||||
|
|
||||||
|
The version of pesde to use within this project. The `pesde` CLI will look at
|
||||||
|
this field and run the correct version of pesde for this project.
|
||||||
|
|
||||||
### `workspace_members`
|
### `workspace_members`
|
||||||
|
|
||||||
A list of globs containing the members of this workspace.
|
A list of globs containing the members of this workspace.
|
||||||
|
@ -268,27 +273,10 @@ version `1.0.0`, and the `bar` and `baz` dependencies of the `foo` package with
|
||||||
version `2.0.0`.
|
version `2.0.0`.
|
||||||
|
|
||||||
Each key in the overrides table is a comma-separated list of package paths. The
|
Each key in the overrides table is a comma-separated list of package paths. The
|
||||||
path is a list of aliases separated by `>`. For example, `foo>bar>baz`
|
path is a list of package names separated by `>`. For example, `foo>bar>baz`
|
||||||
refers to the `baz` dependency of the `bar` package, which is a dependency of
|
refers to the `baz` dependency of the `bar` package, which is a dependency of
|
||||||
the `foo` package.
|
the `foo` package.
|
||||||
|
|
||||||
The value of an override entry can be either a specifier or an alias. If it is an
|
|
||||||
alias (a string), it will be equivalent to putting the specifier of the dependency
|
|
||||||
under that alias. For example, the following two overrides are equivalent:
|
|
||||||
|
|
||||||
```toml
|
|
||||||
[dependencies]
|
|
||||||
bar = { name = "acme/bar", version = "2.0.0" }
|
|
||||||
|
|
||||||
[overrides]
|
|
||||||
"foo>bar" = "bar"
|
|
||||||
```
|
|
||||||
|
|
||||||
```toml
|
|
||||||
[overrides]
|
|
||||||
"foo>bar" = { name = "acme/bar", version = "2.0.0" }
|
|
||||||
```
|
|
||||||
|
|
||||||
<LinkCard
|
<LinkCard
|
||||||
title="Overrides"
|
title="Overrides"
|
||||||
description="Learn more about overriding and patching packages."
|
description="Learn more about overriding and patching packages."
|
||||||
|
@ -411,19 +399,18 @@ foo = { workspace = "acme/foo", version = "^" }
|
||||||
href="/guides/workspaces/#workspace-dependencies"
|
href="/guides/workspaces/#workspace-dependencies"
|
||||||
/>
|
/>
|
||||||
|
|
||||||
### Path
|
## `[peer_dependencies]`
|
||||||
|
|
||||||
|
The `[peer_dependencies]` section contains a list of peer dependencies for the
|
||||||
|
package. These are dependencies that are required by the package, but are not
|
||||||
|
installed automatically. Instead, they must be installed by the user of the
|
||||||
|
package.
|
||||||
|
|
||||||
```toml
|
```toml
|
||||||
[dependencies]
|
[peer_dependencies]
|
||||||
foo = { path = "/home/user/foo" }
|
foo = { name = "acme/foo", version = "1.2.3" }
|
||||||
```
|
```
|
||||||
|
|
||||||
**Path dependencies** contain the following fields:
|
|
||||||
|
|
||||||
- `path`: The path to the package on the local filesystem.
|
|
||||||
|
|
||||||
Path dependencies are forbidden in published packages.
|
|
||||||
|
|
||||||
## `[dev_dependencies]`
|
## `[dev_dependencies]`
|
||||||
|
|
||||||
The `[dev_dependencies]` section contains a list of development dependencies for
|
The `[dev_dependencies]` section contains a list of development dependencies for
|
||||||
|
@ -443,31 +430,3 @@ foo = { name = "acme/foo", version = "1.2.3" }
|
||||||
description="Learn more about specifying dependencies in pesde."
|
description="Learn more about specifying dependencies in pesde."
|
||||||
href="/guides/dependencies/"
|
href="/guides/dependencies/"
|
||||||
/>
|
/>
|
||||||
|
|
||||||
## `[peer_dependencies]`
|
|
||||||
|
|
||||||
The `[peer_dependencies]` section contains a list of peer dependencies for the
|
|
||||||
package. These are dependencies that are required by the package, but are not
|
|
||||||
installed automatically. Instead, they must be installed by the user of the
|
|
||||||
package.
|
|
||||||
|
|
||||||
```toml
|
|
||||||
[peer_dependencies]
|
|
||||||
foo = { name = "acme/foo", version = "1.2.3" }
|
|
||||||
```
|
|
||||||
|
|
||||||
## `[engines]`
|
|
||||||
|
|
||||||
The `[engines]` section contains a list of engines that the package is compatible
|
|
||||||
with.
|
|
||||||
|
|
||||||
```toml
|
|
||||||
[engines]
|
|
||||||
pesde = "^0.6.0"
|
|
||||||
lune = "^0.8.9"
|
|
||||||
```
|
|
||||||
|
|
||||||
Currently, the only engines that can be specified are `pesde` and `lune`.
|
|
||||||
Additionally, the engines you declared in your project will be installed when
|
|
||||||
you run `pesde install`. Then, a version of the engine that satisfies the
|
|
||||||
specified version range will be used when you run the engine.
|
|
||||||
|
|
|
@ -5,32 +5,18 @@ All notable changes to this project will be documented in this file.
|
||||||
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
||||||
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
||||||
|
|
||||||
## [0.2.0-rc.1] - 2025-02-06
|
## [0.1.2]
|
||||||
### Added
|
|
||||||
- Support deprecating and yanking packages by @daimond113
|
|
||||||
- Add yanking & deprecating to registry by @daimond113
|
|
||||||
- Log more information about configured auth & storage by @daimond113
|
|
||||||
- Add individual endpoints for package data over using `Accept` header conditional returns by @daimond113
|
|
||||||
- Set `Content-Length` header for FS storage backend by @daimond113
|
|
||||||
|
|
||||||
### Performance
|
|
||||||
- Switch to using a `RwLock` over a `Mutex` to store repository data by @daimond113
|
|
||||||
- Asyncify blocking operations by @daimond113
|
|
||||||
- Asyncify reading of package data of top search results by @daimond113
|
|
||||||
|
|
||||||
## [0.1.2] - 2024-12-30
|
|
||||||
### Changed
|
### Changed
|
||||||
- Update to pesde lib API changes by @daimond113
|
- Update to pesde lib API changes by @daimond113
|
||||||
|
|
||||||
## [0.1.1] - 2024-12-19
|
## [0.1.1] - 2024-12-19
|
||||||
### Changed
|
### Changed
|
||||||
- Switch to tracing for logging by @daimond113
|
- Switch to traccing for logging by @daimond113
|
||||||
|
|
||||||
## [0.1.0] - 2024-12-14
|
## [0.1.0] - 2024-12-14
|
||||||
### Added
|
### Added
|
||||||
- Rewrite registry for pesde v0.5.0 by @daimond113
|
- Rewrite registry for pesde v0.5.0 by @daimond113
|
||||||
|
|
||||||
[0.2.0-rc.1]: https://github.com/daimond113/pesde/compare/v0.5.3%2Bregistry.0.1.2..v0.6.0-rc.1%2Bregistry.0.2.0-rc.1
|
|
||||||
[0.1.2]: https://github.com/daimond113/pesde/compare/v0.5.2%2Bregistry.0.1.1..v0.5.3%2Bregistry.0.1.2
|
[0.1.2]: https://github.com/daimond113/pesde/compare/v0.5.2%2Bregistry.0.1.1..v0.5.3%2Bregistry.0.1.2
|
||||||
[0.1.1]: https://github.com/daimond113/pesde/compare/v0.5.1%2Bregistry.0.1.0..v0.5.2%2Bregistry.0.1.1
|
[0.1.1]: https://github.com/daimond113/pesde/compare/v0.5.1%2Bregistry.0.1.0..v0.5.2%2Bregistry.0.1.1
|
||||||
[0.1.0]: https://github.com/daimond113/pesde/compare/v0.4.7..v0.5.0%2Bregistry.0.1.0
|
[0.1.0]: https://github.com/daimond113/pesde/compare/v0.4.7..v0.5.0%2Bregistry.0.1.0
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
[package]
|
[package]
|
||||||
name = "pesde-registry"
|
name = "pesde-registry"
|
||||||
version = "0.2.0-rc.1"
|
version = "0.1.2"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
repository = "https://github.com/pesde-pkg/index"
|
repository = "https://github.com/pesde-pkg/index"
|
||||||
publish = false
|
publish = false
|
||||||
|
@ -10,32 +10,31 @@ actix-web = "4.9.0"
|
||||||
actix-cors = "0.7.0"
|
actix-cors = "0.7.0"
|
||||||
actix-governor = "0.8.0"
|
actix-governor = "0.8.0"
|
||||||
dotenvy = "0.15.7"
|
dotenvy = "0.15.7"
|
||||||
thiserror = "2.0.11"
|
thiserror = "2.0.7"
|
||||||
tantivy = "0.22.0"
|
tantivy = "0.22.0"
|
||||||
semver = "1.0.24"
|
semver = "1.0.24"
|
||||||
jiff = { version = "0.1.28", features = ["serde"] }
|
chrono = { version = "0.4.39", features = ["serde"] }
|
||||||
futures = "0.3.31"
|
futures = "0.3.31"
|
||||||
tokio = "1.43.0"
|
tokio = "1.42.0"
|
||||||
tokio-util = "0.7.13"
|
tempfile = "3.14.0"
|
||||||
tempfile = "3.15.0"
|
|
||||||
fs-err = { version = "3.0.0", features = ["tokio"] }
|
fs-err = { version = "3.0.0", features = ["tokio"] }
|
||||||
async-stream = "0.3.6"
|
async-stream = "0.3.6"
|
||||||
|
|
||||||
git2 = "0.20.0"
|
git2 = "0.19.0"
|
||||||
gix = { version = "0.70.0", default-features = false, features = [
|
gix = { version = "0.68.0", default-features = false, features = [
|
||||||
"blocking-http-transport-reqwest-rust-tls",
|
"blocking-http-transport-reqwest-rust-tls",
|
||||||
"credentials",
|
"credentials",
|
||||||
] }
|
] }
|
||||||
|
|
||||||
serde = "1.0.217"
|
serde = "1.0.216"
|
||||||
serde_json = "1.0.136"
|
serde_json = "1.0.133"
|
||||||
serde_yaml = "0.9.34"
|
serde_yaml = "0.9.34"
|
||||||
toml = "0.8.19"
|
toml = "0.8.19"
|
||||||
convert_case = "0.7.1"
|
convert_case = "0.6.0"
|
||||||
sha2 = "0.10.8"
|
sha2 = "0.10.8"
|
||||||
|
|
||||||
rusty-s3 = "0.7.0"
|
rusty-s3 = "0.5.0"
|
||||||
reqwest = { version = "0.12.12", features = ["json", "rustls-tls"] }
|
reqwest = { version = "0.12.9", features = ["json", "rustls-tls"] }
|
||||||
constant_time_eq = "0.3.1"
|
constant_time_eq = "0.3.1"
|
||||||
|
|
||||||
tokio-tar = "0.3.1"
|
tokio-tar = "0.3.1"
|
||||||
|
@ -45,7 +44,7 @@ tracing = { version = "0.1.41", features = ["attributes"] }
|
||||||
tracing-subscriber = { version = "0.3.19", features = ["env-filter"] }
|
tracing-subscriber = { version = "0.3.19", features = ["env-filter"] }
|
||||||
tracing-actix-web = "0.7.15"
|
tracing-actix-web = "0.7.15"
|
||||||
|
|
||||||
sentry = { version = "0.36.0", default-features = false, features = ["backtrace", "contexts", "debug-images", "panic", "reqwest", "rustls", "tracing"] }
|
sentry = { version = "0.35.0", default-features = false, features = ["backtrace", "contexts", "debug-images", "panic", "reqwest", "rustls", "tracing"] }
|
||||||
sentry-actix = "0.36.0"
|
sentry-actix = "0.35.0"
|
||||||
|
|
||||||
pesde = { path = "..", default-features = false, features = ["wally-compat"] }
|
pesde = { path = "..", features = ["wally-compat"] }
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
use crate::{
|
use crate::{
|
||||||
auth::{get_token_from_req, AuthImpl, UserId},
|
auth::{get_token_from_req, AuthImpl, UserId},
|
||||||
error::ReqwestErrorExt,
|
error::ReqwestErrorExt,
|
||||||
};
|
};
|
||||||
use actix_web::{dev::ServiceRequest, Error as ActixError};
|
use actix_web::{dev::ServiceRequest, Error as ActixError};
|
||||||
use reqwest::StatusCode;
|
use reqwest::StatusCode;
|
||||||
|
@ -9,79 +9,79 @@ use std::fmt::Display;
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct GitHubAuth {
|
pub struct GitHubAuth {
|
||||||
pub reqwest_client: reqwest::Client,
|
pub reqwest_client: reqwest::Client,
|
||||||
pub client_id: String,
|
pub client_id: String,
|
||||||
pub client_secret: String,
|
pub client_secret: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
#[derive(Debug, Serialize)]
|
||||||
struct TokenRequestBody {
|
struct TokenRequestBody {
|
||||||
access_token: String,
|
access_token: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AuthImpl for GitHubAuth {
|
impl AuthImpl for GitHubAuth {
|
||||||
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
||||||
let token = match get_token_from_req(req) {
|
let token = match get_token_from_req(req) {
|
||||||
Some(token) => token,
|
Some(token) => token,
|
||||||
None => return Ok(None),
|
None => return Ok(None),
|
||||||
};
|
};
|
||||||
|
|
||||||
let response = match self
|
let response = match self
|
||||||
.reqwest_client
|
.reqwest_client
|
||||||
.post(format!(
|
.post(format!(
|
||||||
"https://api.github.com/applications/{}/token",
|
"https://api.github.com/applications/{}/token",
|
||||||
self.client_id
|
self.client_id
|
||||||
))
|
))
|
||||||
.basic_auth(&self.client_id, Some(&self.client_secret))
|
.basic_auth(&self.client_id, Some(&self.client_secret))
|
||||||
.json(&TokenRequestBody {
|
.json(&TokenRequestBody {
|
||||||
access_token: token,
|
access_token: token,
|
||||||
})
|
})
|
||||||
.send()
|
.send()
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(response) => match response.error_for_status_ref() {
|
Ok(response) => match response.error_for_status_ref() {
|
||||||
Ok(_) => response,
|
Ok(_) => response,
|
||||||
Err(e) if e.status().is_some_and(|s| s == StatusCode::NOT_FOUND) => {
|
Err(e) if e.status().is_some_and(|s| s == StatusCode::NOT_FOUND) => {
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
Err(_) => {
|
Err(_) => {
|
||||||
tracing::error!(
|
tracing::error!(
|
||||||
"failed to get user: {}",
|
"failed to get user: {}",
|
||||||
response.into_error().await.unwrap_err()
|
response.into_error().await.unwrap_err()
|
||||||
);
|
);
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
tracing::error!("failed to get user: {e}");
|
tracing::error!("failed to get user: {e}");
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let user_id = match response.json::<UserResponse>().await {
|
let user_id = match response.json::<UserResponse>().await {
|
||||||
Ok(resp) => resp.user.id,
|
Ok(resp) => resp.user.id,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
tracing::error!("failed to get user: {e}");
|
tracing::error!("failed to get user: {e}");
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
Ok(Some(UserId(user_id)))
|
Ok(Some(UserId(user_id)))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for GitHubAuth {
|
impl Display for GitHubAuth {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
write!(f, "GitHub (client id: {})", self.client_id)
|
write!(f, "GitHub")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
struct User {
|
struct User {
|
||||||
id: u64,
|
id: u64,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
struct UserResponse {
|
struct UserResponse {
|
||||||
user: User,
|
user: User,
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,12 +6,12 @@ mod token;
|
||||||
use crate::{benv, make_reqwest, AppState};
|
use crate::{benv, make_reqwest, AppState};
|
||||||
use actix_governor::{KeyExtractor, SimpleKeyExtractionError};
|
use actix_governor::{KeyExtractor, SimpleKeyExtractionError};
|
||||||
use actix_web::{
|
use actix_web::{
|
||||||
body::MessageBody,
|
body::MessageBody,
|
||||||
dev::{ServiceRequest, ServiceResponse},
|
dev::{ServiceRequest, ServiceResponse},
|
||||||
error::Error as ActixError,
|
error::Error as ActixError,
|
||||||
http::header::AUTHORIZATION,
|
http::header::AUTHORIZATION,
|
||||||
middleware::Next,
|
middleware::Next,
|
||||||
web, HttpMessage, HttpResponse,
|
web, HttpMessage, HttpResponse,
|
||||||
};
|
};
|
||||||
use pesde::source::pesde::IndexConfig;
|
use pesde::source::pesde::IndexConfig;
|
||||||
use sentry::add_breadcrumb;
|
use sentry::add_breadcrumb;
|
||||||
|
@ -22,178 +22,178 @@ use std::fmt::Display;
|
||||||
pub struct UserId(pub u64);
|
pub struct UserId(pub u64);
|
||||||
|
|
||||||
impl UserId {
|
impl UserId {
|
||||||
// there isn't any account on GitHub that has the ID 0, so it should be safe to use it
|
// there isn't any account on GitHub that has the ID 0, so it should be safe to use it
|
||||||
pub const DEFAULT: UserId = UserId(0);
|
pub const DEFAULT: UserId = UserId(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct UserIdExtractor;
|
pub struct UserIdExtractor;
|
||||||
|
|
||||||
impl KeyExtractor for UserIdExtractor {
|
impl KeyExtractor for UserIdExtractor {
|
||||||
type Key = UserId;
|
type Key = UserId;
|
||||||
type KeyExtractionError = SimpleKeyExtractionError<&'static str>;
|
type KeyExtractionError = SimpleKeyExtractionError<&'static str>;
|
||||||
|
|
||||||
fn extract(&self, req: &ServiceRequest) -> Result<Self::Key, Self::KeyExtractionError> {
|
fn extract(&self, req: &ServiceRequest) -> Result<Self::Key, Self::KeyExtractionError> {
|
||||||
match req.extensions().get::<UserId>() {
|
match req.extensions().get::<UserId>() {
|
||||||
Some(user_id) => Ok(*user_id),
|
Some(user_id) => Ok(*user_id),
|
||||||
None => Err(SimpleKeyExtractionError::new("UserId not found")),
|
None => Err(SimpleKeyExtractionError::new("UserId not found")),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub enum Auth {
|
pub enum Auth {
|
||||||
GitHub(github::GitHubAuth),
|
GitHub(github::GitHubAuth),
|
||||||
None(none::NoneAuth),
|
None(none::NoneAuth),
|
||||||
Token(token::TokenAuth),
|
Token(token::TokenAuth),
|
||||||
RwToken(rw_token::RwTokenAuth),
|
RwToken(rw_token::RwTokenAuth),
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait AuthImpl: Display {
|
pub trait AuthImpl: Display {
|
||||||
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError>;
|
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError>;
|
||||||
|
|
||||||
async fn for_read_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
async fn for_read_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
||||||
self.for_write_request(req).await
|
self.for_write_request(req).await
|
||||||
}
|
}
|
||||||
|
|
||||||
fn read_needs_auth(&self) -> bool {
|
fn read_needs_auth(&self) -> bool {
|
||||||
benv!("READ_NEEDS_AUTH").is_ok()
|
benv!("READ_NEEDS_AUTH").is_ok()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AuthImpl for Auth {
|
impl AuthImpl for Auth {
|
||||||
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
||||||
match self {
|
match self {
|
||||||
Auth::GitHub(github) => github.for_write_request(req).await,
|
Auth::GitHub(github) => github.for_write_request(req).await,
|
||||||
Auth::None(none) => none.for_write_request(req).await,
|
Auth::None(none) => none.for_write_request(req).await,
|
||||||
Auth::Token(token) => token.for_write_request(req).await,
|
Auth::Token(token) => token.for_write_request(req).await,
|
||||||
Auth::RwToken(rw_token) => rw_token.for_write_request(req).await,
|
Auth::RwToken(rw_token) => rw_token.for_write_request(req).await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn for_read_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
async fn for_read_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
||||||
match self {
|
match self {
|
||||||
Auth::GitHub(github) => github.for_read_request(req).await,
|
Auth::GitHub(github) => github.for_read_request(req).await,
|
||||||
Auth::None(none) => none.for_write_request(req).await,
|
Auth::None(none) => none.for_write_request(req).await,
|
||||||
Auth::Token(token) => token.for_write_request(req).await,
|
Auth::Token(token) => token.for_write_request(req).await,
|
||||||
Auth::RwToken(rw_token) => rw_token.for_read_request(req).await,
|
Auth::RwToken(rw_token) => rw_token.for_read_request(req).await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn read_needs_auth(&self) -> bool {
|
fn read_needs_auth(&self) -> bool {
|
||||||
match self {
|
match self {
|
||||||
Auth::GitHub(github) => github.read_needs_auth(),
|
Auth::GitHub(github) => github.read_needs_auth(),
|
||||||
Auth::None(none) => none.read_needs_auth(),
|
Auth::None(none) => none.read_needs_auth(),
|
||||||
Auth::Token(token) => token.read_needs_auth(),
|
Auth::Token(token) => token.read_needs_auth(),
|
||||||
Auth::RwToken(rw_token) => rw_token.read_needs_auth(),
|
Auth::RwToken(rw_token) => rw_token.read_needs_auth(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for Auth {
|
impl Display for Auth {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
match self {
|
match self {
|
||||||
Auth::GitHub(github) => write!(f, "{}", github),
|
Auth::GitHub(github) => write!(f, "{}", github),
|
||||||
Auth::None(none) => write!(f, "{}", none),
|
Auth::None(none) => write!(f, "{}", none),
|
||||||
Auth::Token(token) => write!(f, "{}", token),
|
Auth::Token(token) => write!(f, "{}", token),
|
||||||
Auth::RwToken(rw_token) => write!(f, "{}", rw_token),
|
Auth::RwToken(rw_token) => write!(f, "{}", rw_token),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn write_mw(
|
pub async fn write_mw(
|
||||||
app_state: web::Data<AppState>,
|
app_state: web::Data<AppState>,
|
||||||
req: ServiceRequest,
|
req: ServiceRequest,
|
||||||
next: Next<impl MessageBody + 'static>,
|
next: Next<impl MessageBody + 'static>,
|
||||||
) -> Result<ServiceResponse<impl MessageBody>, ActixError> {
|
) -> Result<ServiceResponse<impl MessageBody>, ActixError> {
|
||||||
let user_id = match app_state.auth.for_write_request(&req).await? {
|
let user_id = match app_state.auth.for_write_request(&req).await? {
|
||||||
Some(user_id) => user_id,
|
Some(user_id) => user_id,
|
||||||
None => {
|
None => {
|
||||||
return Ok(req
|
return Ok(req
|
||||||
.into_response(HttpResponse::Unauthorized().finish())
|
.into_response(HttpResponse::Unauthorized().finish())
|
||||||
.map_into_right_body())
|
.map_into_right_body())
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
add_breadcrumb(sentry::Breadcrumb {
|
add_breadcrumb(sentry::Breadcrumb {
|
||||||
category: Some("auth".into()),
|
category: Some("auth".into()),
|
||||||
message: Some(format!("write request authorized as {}", user_id.0)),
|
message: Some(format!("write request authorized as {}", user_id.0)),
|
||||||
level: sentry::Level::Info,
|
level: sentry::Level::Info,
|
||||||
..Default::default()
|
..Default::default()
|
||||||
});
|
});
|
||||||
|
|
||||||
req.extensions_mut().insert(user_id);
|
req.extensions_mut().insert(user_id);
|
||||||
|
|
||||||
next.call(req).await.map(|res| res.map_into_left_body())
|
next.call(req).await.map(|res| res.map_into_left_body())
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn read_mw(
|
pub async fn read_mw(
|
||||||
app_state: web::Data<AppState>,
|
app_state: web::Data<AppState>,
|
||||||
req: ServiceRequest,
|
req: ServiceRequest,
|
||||||
next: Next<impl MessageBody + 'static>,
|
next: Next<impl MessageBody + 'static>,
|
||||||
) -> Result<ServiceResponse<impl MessageBody>, ActixError> {
|
) -> Result<ServiceResponse<impl MessageBody>, ActixError> {
|
||||||
if app_state.auth.read_needs_auth() {
|
if app_state.auth.read_needs_auth() {
|
||||||
let user_id = match app_state.auth.for_read_request(&req).await? {
|
let user_id = match app_state.auth.for_read_request(&req).await? {
|
||||||
Some(user_id) => user_id,
|
Some(user_id) => user_id,
|
||||||
None => {
|
None => {
|
||||||
return Ok(req
|
return Ok(req
|
||||||
.into_response(HttpResponse::Unauthorized().finish())
|
.into_response(HttpResponse::Unauthorized().finish())
|
||||||
.map_into_right_body())
|
.map_into_right_body())
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
add_breadcrumb(sentry::Breadcrumb {
|
add_breadcrumb(sentry::Breadcrumb {
|
||||||
category: Some("auth".into()),
|
category: Some("auth".into()),
|
||||||
message: Some(format!("read request authorized as {}", user_id.0)),
|
message: Some(format!("read request authorized as {}", user_id.0)),
|
||||||
level: sentry::Level::Info,
|
level: sentry::Level::Info,
|
||||||
..Default::default()
|
..Default::default()
|
||||||
});
|
});
|
||||||
|
|
||||||
req.extensions_mut().insert(Some(user_id));
|
req.extensions_mut().insert(Some(user_id));
|
||||||
} else {
|
} else {
|
||||||
req.extensions_mut().insert(None::<UserId>);
|
req.extensions_mut().insert(None::<UserId>);
|
||||||
}
|
}
|
||||||
|
|
||||||
next.call(req).await.map(|res| res.map_into_left_body())
|
next.call(req).await.map(|res| res.map_into_left_body())
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_auth_from_env(config: &IndexConfig) -> Auth {
|
pub fn get_auth_from_env(config: &IndexConfig) -> Auth {
|
||||||
if let Ok(token) = benv!("ACCESS_TOKEN") {
|
if let Ok(token) = benv!("ACCESS_TOKEN") {
|
||||||
Auth::Token(token::TokenAuth {
|
Auth::Token(token::TokenAuth {
|
||||||
token: *Sha256::digest(token.as_bytes()).as_ref(),
|
token: *Sha256::digest(token.as_bytes()).as_ref(),
|
||||||
})
|
})
|
||||||
} else if let Ok(client_secret) = benv!("GITHUB_CLIENT_SECRET") {
|
} else if let Ok(client_secret) = benv!("GITHUB_CLIENT_SECRET") {
|
||||||
Auth::GitHub(github::GitHubAuth {
|
Auth::GitHub(github::GitHubAuth {
|
||||||
reqwest_client: make_reqwest(),
|
reqwest_client: make_reqwest(),
|
||||||
client_id: config
|
client_id: config
|
||||||
.github_oauth_client_id
|
.github_oauth_client_id
|
||||||
.clone()
|
.clone()
|
||||||
.expect("index isn't configured for GitHub"),
|
.expect("index isn't configured for GitHub"),
|
||||||
client_secret,
|
client_secret,
|
||||||
})
|
})
|
||||||
} else if let Ok((r, w)) =
|
} else if let Ok((r, w)) =
|
||||||
benv!("READ_ACCESS_TOKEN").and_then(|r| benv!("WRITE_ACCESS_TOKEN").map(|w| (r, w)))
|
benv!("READ_ACCESS_TOKEN").and_then(|r| benv!("WRITE_ACCESS_TOKEN").map(|w| (r, w)))
|
||||||
{
|
{
|
||||||
Auth::RwToken(rw_token::RwTokenAuth {
|
Auth::RwToken(rw_token::RwTokenAuth {
|
||||||
read_token: *Sha256::digest(r.as_bytes()).as_ref(),
|
read_token: *Sha256::digest(r.as_bytes()).as_ref(),
|
||||||
write_token: *Sha256::digest(w.as_bytes()).as_ref(),
|
write_token: *Sha256::digest(w.as_bytes()).as_ref(),
|
||||||
})
|
})
|
||||||
} else {
|
} else {
|
||||||
Auth::None(none::NoneAuth)
|
Auth::None(none::NoneAuth)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_token_from_req(req: &ServiceRequest) -> Option<String> {
|
pub fn get_token_from_req(req: &ServiceRequest) -> Option<String> {
|
||||||
let token = req
|
let token = req
|
||||||
.headers()
|
.headers()
|
||||||
.get(AUTHORIZATION)
|
.get(AUTHORIZATION)
|
||||||
.and_then(|token| token.to_str().ok())?;
|
.and_then(|token| token.to_str().ok())?;
|
||||||
|
|
||||||
let token = if token.to_lowercase().starts_with("bearer ") {
|
let token = if token.to_lowercase().starts_with("bearer ") {
|
||||||
token[7..].to_string()
|
token[7..].to_string()
|
||||||
} else {
|
} else {
|
||||||
token.to_string()
|
token.to_string()
|
||||||
};
|
};
|
||||||
|
|
||||||
Some(token)
|
Some(token)
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,13 +6,13 @@ use std::fmt::Display;
|
||||||
pub struct NoneAuth;
|
pub struct NoneAuth;
|
||||||
|
|
||||||
impl AuthImpl for NoneAuth {
|
impl AuthImpl for NoneAuth {
|
||||||
async fn for_write_request(&self, _req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
async fn for_write_request(&self, _req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
||||||
Ok(Some(UserId::DEFAULT))
|
Ok(Some(UserId::DEFAULT))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for NoneAuth {
|
impl Display for NoneAuth {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
write!(f, "None")
|
write!(f, "None")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,48 +6,48 @@ use std::fmt::Display;
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct RwTokenAuth {
|
pub struct RwTokenAuth {
|
||||||
pub read_token: [u8; 32],
|
pub read_token: [u8; 32],
|
||||||
pub write_token: [u8; 32],
|
pub write_token: [u8; 32],
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AuthImpl for RwTokenAuth {
|
impl AuthImpl for RwTokenAuth {
|
||||||
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
||||||
let token = match get_token_from_req(req) {
|
let token = match get_token_from_req(req) {
|
||||||
Some(token) => token,
|
Some(token) => token,
|
||||||
None => return Ok(None),
|
None => return Ok(None),
|
||||||
};
|
};
|
||||||
|
|
||||||
let token: [u8; 32] = Sha256::digest(token.as_bytes()).into();
|
let token: [u8; 32] = Sha256::digest(token.as_bytes()).into();
|
||||||
|
|
||||||
Ok(if constant_time_eq_32(&self.write_token, &token) {
|
Ok(if constant_time_eq_32(&self.write_token, &token) {
|
||||||
Some(UserId::DEFAULT)
|
Some(UserId::DEFAULT)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn for_read_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
async fn for_read_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
||||||
let token = match get_token_from_req(req) {
|
let token = match get_token_from_req(req) {
|
||||||
Some(token) => token,
|
Some(token) => token,
|
||||||
None => return Ok(None),
|
None => return Ok(None),
|
||||||
};
|
};
|
||||||
|
|
||||||
let token: [u8; 32] = Sha256::digest(token.as_bytes()).into();
|
let token: [u8; 32] = Sha256::digest(token.as_bytes()).into();
|
||||||
|
|
||||||
Ok(if constant_time_eq_32(&self.read_token, &token) {
|
Ok(if constant_time_eq_32(&self.read_token, &token) {
|
||||||
Some(UserId::DEFAULT)
|
Some(UserId::DEFAULT)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn read_needs_auth(&self) -> bool {
|
fn read_needs_auth(&self) -> bool {
|
||||||
true
|
true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for RwTokenAuth {
|
impl Display for RwTokenAuth {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
write!(f, "RwToken")
|
write!(f, "RwToken")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,29 +6,29 @@ use std::fmt::Display;
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct TokenAuth {
|
pub struct TokenAuth {
|
||||||
// needs to be an SHA-256 hash
|
// needs to be an SHA-256 hash
|
||||||
pub token: [u8; 32],
|
pub token: [u8; 32],
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AuthImpl for TokenAuth {
|
impl AuthImpl for TokenAuth {
|
||||||
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
async fn for_write_request(&self, req: &ServiceRequest) -> Result<Option<UserId>, ActixError> {
|
||||||
let token = match get_token_from_req(req) {
|
let token = match get_token_from_req(req) {
|
||||||
Some(token) => token,
|
Some(token) => token,
|
||||||
None => return Ok(None),
|
None => return Ok(None),
|
||||||
};
|
};
|
||||||
|
|
||||||
let token: [u8; 32] = Sha256::digest(token.as_bytes()).into();
|
let token: [u8; 32] = Sha256::digest(token.as_bytes()).into();
|
||||||
|
|
||||||
Ok(if constant_time_eq_32(&self.token, &token) {
|
Ok(if constant_time_eq_32(&self.token, &token) {
|
||||||
Some(UserId::DEFAULT)
|
Some(UserId::DEFAULT)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for TokenAuth {
|
impl Display for TokenAuth {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
write!(f, "Token")
|
write!(f, "Token")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,76 +0,0 @@
|
||||||
use crate::{
|
|
||||||
auth::UserId,
|
|
||||||
error::{ErrorResponse, RegistryError},
|
|
||||||
git::push_changes,
|
|
||||||
package::{read_package, read_scope_info},
|
|
||||||
search::search_version_changed,
|
|
||||||
AppState,
|
|
||||||
};
|
|
||||||
use actix_web::{http::Method, web, HttpRequest, HttpResponse};
|
|
||||||
use pesde::names::PackageName;
|
|
||||||
use std::collections::HashMap;
|
|
||||||
|
|
||||||
pub async fn deprecate_package_version(
|
|
||||||
request: HttpRequest,
|
|
||||||
app_state: web::Data<AppState>,
|
|
||||||
path: web::Path<PackageName>,
|
|
||||||
bytes: web::Bytes,
|
|
||||||
user_id: web::ReqData<UserId>,
|
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
|
||||||
let deprecated = request.method() != Method::DELETE;
|
|
||||||
let reason = if deprecated {
|
|
||||||
match String::from_utf8(bytes.to_vec()).map(|s| s.trim().to_string()) {
|
|
||||||
Ok(reason) if !reason.is_empty() => reason,
|
|
||||||
Err(e) => {
|
|
||||||
return Ok(HttpResponse::BadRequest().json(ErrorResponse {
|
|
||||||
error: format!("invalid utf-8: {e}"),
|
|
||||||
}))
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
return Ok(HttpResponse::BadRequest().json(ErrorResponse {
|
|
||||||
error: "deprecating must have a non-empty reason".to_string(),
|
|
||||||
}))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
String::new()
|
|
||||||
};
|
|
||||||
let name = path.into_inner();
|
|
||||||
let source = app_state.source.write().await;
|
|
||||||
|
|
||||||
let Some(scope_info) = read_scope_info(&app_state, name.scope(), &source).await? else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
if !scope_info.owners.contains(&user_id.0) {
|
|
||||||
return Ok(HttpResponse::Forbidden().finish());
|
|
||||||
}
|
|
||||||
|
|
||||||
let Some(mut file) = read_package(&app_state, &name, &source).await? else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
if file.meta.deprecated == reason {
|
|
||||||
return Ok(HttpResponse::Conflict().finish());
|
|
||||||
}
|
|
||||||
|
|
||||||
file.meta.deprecated = reason;
|
|
||||||
|
|
||||||
let file_string = toml::to_string(&file)?;
|
|
||||||
|
|
||||||
push_changes(
|
|
||||||
&app_state,
|
|
||||||
&source,
|
|
||||||
name.scope().to_string(),
|
|
||||||
HashMap::from([(name.name().to_string(), file_string.into_bytes())]),
|
|
||||||
format!("{}deprecate {name}", if deprecated { "" } else { "un" }),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
search_version_changed(&app_state, &name, &file);
|
|
||||||
|
|
||||||
Ok(HttpResponse::Ok().body(format!(
|
|
||||||
"{}deprecated {name}",
|
|
||||||
if deprecated { "" } else { "un" },
|
|
||||||
)))
|
|
||||||
}
|
|
|
@ -1,9 +1,4 @@
|
||||||
pub mod deprecate_version;
|
|
||||||
pub mod package_archive;
|
|
||||||
pub mod package_doc;
|
|
||||||
pub mod package_readme;
|
|
||||||
pub mod package_version;
|
pub mod package_version;
|
||||||
pub mod package_versions;
|
pub mod package_versions;
|
||||||
pub mod publish_version;
|
pub mod publish_version;
|
||||||
pub mod search;
|
pub mod search;
|
||||||
pub mod yank_version;
|
|
||||||
|
|
|
@ -1,27 +0,0 @@
|
||||||
use actix_web::{web, HttpResponse};
|
|
||||||
|
|
||||||
use crate::{
|
|
||||||
error::RegistryError,
|
|
||||||
package::read_package,
|
|
||||||
request_path::{resolve_version_and_target, AnyOrSpecificTarget, LatestOrSpecificVersion},
|
|
||||||
storage::StorageImpl,
|
|
||||||
AppState,
|
|
||||||
};
|
|
||||||
use pesde::names::PackageName;
|
|
||||||
|
|
||||||
pub async fn get_package_archive(
|
|
||||||
app_state: web::Data<AppState>,
|
|
||||||
path: web::Path<(PackageName, LatestOrSpecificVersion, AnyOrSpecificTarget)>,
|
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
|
||||||
let (name, version, target) = path.into_inner();
|
|
||||||
|
|
||||||
let Some(file) = read_package(&app_state, &name, &*app_state.source.read().await).await? else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
let Some(v_id) = resolve_version_and_target(&file, version, target) else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
app_state.storage.get_package(&name, v_id).await
|
|
||||||
}
|
|
|
@ -1,66 +0,0 @@
|
||||||
use crate::{
|
|
||||||
error::RegistryError,
|
|
||||||
package::read_package,
|
|
||||||
request_path::{resolve_version_and_target, AnyOrSpecificTarget, LatestOrSpecificVersion},
|
|
||||||
storage::StorageImpl,
|
|
||||||
AppState,
|
|
||||||
};
|
|
||||||
use actix_web::{web, HttpResponse};
|
|
||||||
use pesde::{
|
|
||||||
names::PackageName,
|
|
||||||
source::{
|
|
||||||
ids::VersionId,
|
|
||||||
pesde::{DocEntryKind, IndexFile},
|
|
||||||
},
|
|
||||||
};
|
|
||||||
use serde::Deserialize;
|
|
||||||
|
|
||||||
pub fn find_package_doc<'a>(
|
|
||||||
file: &'a IndexFile,
|
|
||||||
v_id: &VersionId,
|
|
||||||
doc_name: &str,
|
|
||||||
) -> Option<&'a str> {
|
|
||||||
let mut queue = file.entries[v_id]
|
|
||||||
.docs
|
|
||||||
.iter()
|
|
||||||
.map(|doc| &doc.kind)
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
while let Some(doc) = queue.pop() {
|
|
||||||
match doc {
|
|
||||||
DocEntryKind::Page { name, hash } if name == doc_name => return Some(hash.as_str()),
|
|
||||||
DocEntryKind::Category { items, .. } => {
|
|
||||||
queue.extend(items.iter().map(|item| &item.kind))
|
|
||||||
}
|
|
||||||
_ => continue,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
None
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
pub struct Query {
|
|
||||||
doc: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_package_doc(
|
|
||||||
app_state: web::Data<AppState>,
|
|
||||||
path: web::Path<(PackageName, LatestOrSpecificVersion, AnyOrSpecificTarget)>,
|
|
||||||
request_query: web::Query<Query>,
|
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
|
||||||
let (name, version, target) = path.into_inner();
|
|
||||||
|
|
||||||
let Some(file) = read_package(&app_state, &name, &*app_state.source.read().await).await? else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
let Some(v_id) = resolve_version_and_target(&file, version, target) else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
let Some(hash) = find_package_doc(&file, v_id, &request_query.doc) else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
app_state.storage.get_doc(hash).await
|
|
||||||
}
|
|
|
@ -1,27 +0,0 @@
|
||||||
use actix_web::{web, HttpResponse};
|
|
||||||
|
|
||||||
use crate::{
|
|
||||||
error::RegistryError,
|
|
||||||
package::read_package,
|
|
||||||
request_path::{resolve_version_and_target, AnyOrSpecificTarget, LatestOrSpecificVersion},
|
|
||||||
storage::StorageImpl,
|
|
||||||
AppState,
|
|
||||||
};
|
|
||||||
use pesde::names::PackageName;
|
|
||||||
|
|
||||||
pub async fn get_package_readme(
|
|
||||||
app_state: web::Data<AppState>,
|
|
||||||
path: web::Path<(PackageName, LatestOrSpecificVersion, AnyOrSpecificTarget)>,
|
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
|
||||||
let (name, version, target) = path.into_inner();
|
|
||||||
|
|
||||||
let Some(file) = read_package(&app_state, &name, &*app_state.source.read().await).await? else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
let Some(v_id) = resolve_version_and_target(&file, version, target) else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
app_state.storage.get_readme(&name, v_id).await
|
|
||||||
}
|
|
|
@ -1,79 +1,171 @@
|
||||||
use actix_web::{http::header::ACCEPT, web, HttpRequest, HttpResponse};
|
use actix_web::{http::header::ACCEPT, web, HttpRequest, HttpResponse, Responder};
|
||||||
use serde::Deserialize;
|
use semver::Version;
|
||||||
|
use serde::{Deserialize, Deserializer};
|
||||||
|
|
||||||
use crate::{
|
use crate::{error::Error, package::PackageResponse, storage::StorageImpl, AppState};
|
||||||
endpoints::package_doc::find_package_doc,
|
use pesde::{
|
||||||
error::RegistryError,
|
manifest::target::TargetKind,
|
||||||
package::{read_package, PackageResponse},
|
names::PackageName,
|
||||||
request_path::{resolve_version_and_target, AnyOrSpecificTarget, LatestOrSpecificVersion},
|
source::{
|
||||||
storage::StorageImpl,
|
git_index::{read_file, root_tree, GitBasedSource},
|
||||||
AppState,
|
pesde::{DocEntryKind, IndexFile},
|
||||||
|
},
|
||||||
};
|
};
|
||||||
use pesde::names::PackageName;
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum VersionRequest {
|
||||||
|
Latest,
|
||||||
|
Specific(Version),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'de> Deserialize<'de> for VersionRequest {
|
||||||
|
fn deserialize<D>(deserializer: D) -> Result<VersionRequest, D::Error>
|
||||||
|
where
|
||||||
|
D: Deserializer<'de>,
|
||||||
|
{
|
||||||
|
let s = String::deserialize(deserializer)?;
|
||||||
|
if s.eq_ignore_ascii_case("latest") {
|
||||||
|
return Ok(VersionRequest::Latest);
|
||||||
|
}
|
||||||
|
|
||||||
|
s.parse()
|
||||||
|
.map(VersionRequest::Specific)
|
||||||
|
.map_err(serde::de::Error::custom)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum TargetRequest {
|
||||||
|
Any,
|
||||||
|
Specific(TargetKind),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'de> Deserialize<'de> for TargetRequest {
|
||||||
|
fn deserialize<D>(deserializer: D) -> Result<TargetRequest, D::Error>
|
||||||
|
where
|
||||||
|
D: Deserializer<'de>,
|
||||||
|
{
|
||||||
|
let s = String::deserialize(deserializer)?;
|
||||||
|
if s.eq_ignore_ascii_case("any") {
|
||||||
|
return Ok(TargetRequest::Any);
|
||||||
|
}
|
||||||
|
|
||||||
|
s.parse()
|
||||||
|
.map(TargetRequest::Specific)
|
||||||
|
.map_err(serde::de::Error::custom)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
pub struct Query {
|
pub struct Query {
|
||||||
doc: Option<String>,
|
doc: Option<String>,
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_package_version_v0(
|
|
||||||
request: HttpRequest,
|
|
||||||
app_state: web::Data<AppState>,
|
|
||||||
path: web::Path<(PackageName, LatestOrSpecificVersion, AnyOrSpecificTarget)>,
|
|
||||||
request_query: web::Query<Query>,
|
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
|
||||||
let (name, version, target) = path.into_inner();
|
|
||||||
|
|
||||||
let Some(file) = read_package(&app_state, &name, &*app_state.source.read().await).await? else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
let Some(v_id) = resolve_version_and_target(&file, version, target) else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
if let Some(doc_name) = request_query.doc.as_deref() {
|
|
||||||
let Some(hash) = find_package_doc(&file, v_id, doc_name) else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
return app_state.storage.get_doc(hash).await;
|
|
||||||
}
|
|
||||||
|
|
||||||
let accept = request
|
|
||||||
.headers()
|
|
||||||
.get(ACCEPT)
|
|
||||||
.and_then(|accept| accept.to_str().ok())
|
|
||||||
.and_then(|accept| match accept.to_lowercase().as_str() {
|
|
||||||
"text/plain" => Some(true),
|
|
||||||
"application/octet-stream" => Some(false),
|
|
||||||
_ => None,
|
|
||||||
});
|
|
||||||
|
|
||||||
if let Some(readme) = accept {
|
|
||||||
return if readme {
|
|
||||||
app_state.storage.get_readme(&name, v_id).await
|
|
||||||
} else {
|
|
||||||
app_state.storage.get_package(&name, v_id).await
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(HttpResponse::Ok().json(PackageResponse::new(&name, v_id, &file)))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn get_package_version(
|
pub async fn get_package_version(
|
||||||
app_state: web::Data<AppState>,
|
request: HttpRequest,
|
||||||
path: web::Path<(PackageName, LatestOrSpecificVersion, AnyOrSpecificTarget)>,
|
app_state: web::Data<AppState>,
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
path: web::Path<(PackageName, VersionRequest, TargetRequest)>,
|
||||||
let (name, version, target) = path.into_inner();
|
query: web::Query<Query>,
|
||||||
|
) -> Result<impl Responder, Error> {
|
||||||
|
let (name, version, target) = path.into_inner();
|
||||||
|
|
||||||
let Some(file) = read_package(&app_state, &name, &*app_state.source.read().await).await? else {
|
let (scope, name_part) = name.as_str();
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
let Some(v_id) = resolve_version_and_target(&file, version, target) else {
|
let file: IndexFile = {
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
let source = app_state.source.lock().await;
|
||||||
};
|
let repo = gix::open(source.path(&app_state.project))?;
|
||||||
|
let tree = root_tree(&repo)?;
|
||||||
|
|
||||||
Ok(HttpResponse::Ok().json(PackageResponse::new(&name, v_id, &file)))
|
match read_file(&tree, [scope, name_part])? {
|
||||||
|
Some(versions) => toml::de::from_str(&versions)?,
|
||||||
|
None => return Ok(HttpResponse::NotFound().finish()),
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let Some((v_id, entry, targets)) = ({
|
||||||
|
let version = match version {
|
||||||
|
VersionRequest::Latest => match file.entries.keys().map(|k| k.version()).max() {
|
||||||
|
Some(latest) => latest.clone(),
|
||||||
|
None => return Ok(HttpResponse::NotFound().finish()),
|
||||||
|
},
|
||||||
|
VersionRequest::Specific(version) => version,
|
||||||
|
};
|
||||||
|
|
||||||
|
let versions = file
|
||||||
|
.entries
|
||||||
|
.iter()
|
||||||
|
.filter(|(v_id, _)| *v_id.version() == version);
|
||||||
|
|
||||||
|
match target {
|
||||||
|
TargetRequest::Any => versions.clone().min_by_key(|(v_id, _)| *v_id.target()),
|
||||||
|
TargetRequest::Specific(kind) => versions
|
||||||
|
.clone()
|
||||||
|
.find(|(_, entry)| entry.target.kind() == kind),
|
||||||
|
}
|
||||||
|
.map(|(v_id, entry)| {
|
||||||
|
(
|
||||||
|
v_id,
|
||||||
|
entry,
|
||||||
|
versions.map(|(_, entry)| (&entry.target).into()).collect(),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
}) else {
|
||||||
|
return Ok(HttpResponse::NotFound().finish());
|
||||||
|
};
|
||||||
|
|
||||||
|
if let Some(doc_name) = query.doc.as_deref() {
|
||||||
|
let hash = 'finder: {
|
||||||
|
let mut hash = entry.docs.iter().map(|doc| &doc.kind).collect::<Vec<_>>();
|
||||||
|
while let Some(doc) = hash.pop() {
|
||||||
|
match doc {
|
||||||
|
DocEntryKind::Page { name, hash } if name == doc_name => {
|
||||||
|
break 'finder hash.clone()
|
||||||
|
}
|
||||||
|
DocEntryKind::Category { items, .. } => {
|
||||||
|
hash.extend(items.iter().map(|item| &item.kind))
|
||||||
|
}
|
||||||
|
_ => continue,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
return Ok(HttpResponse::NotFound().finish());
|
||||||
|
};
|
||||||
|
|
||||||
|
return app_state.storage.get_doc(&hash).await;
|
||||||
|
}
|
||||||
|
|
||||||
|
let accept = request
|
||||||
|
.headers()
|
||||||
|
.get(ACCEPT)
|
||||||
|
.and_then(|accept| accept.to_str().ok())
|
||||||
|
.and_then(|accept| match accept.to_lowercase().as_str() {
|
||||||
|
"text/plain" => Some(true),
|
||||||
|
"application/octet-stream" => Some(false),
|
||||||
|
_ => None,
|
||||||
|
});
|
||||||
|
|
||||||
|
if let Some(readme) = accept {
|
||||||
|
return if readme {
|
||||||
|
app_state.storage.get_readme(&name, v_id).await
|
||||||
|
} else {
|
||||||
|
app_state.storage.get_package(&name, v_id).await
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
let response = PackageResponse {
|
||||||
|
name: name.to_string(),
|
||||||
|
version: v_id.version().to_string(),
|
||||||
|
targets,
|
||||||
|
description: entry.description.clone().unwrap_or_default(),
|
||||||
|
published_at: entry.published_at,
|
||||||
|
license: entry.license.clone().unwrap_or_default(),
|
||||||
|
authors: entry.authors.clone(),
|
||||||
|
repository: entry.repository.clone().map(|url| url.to_string()),
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut value = serde_json::to_value(response)?;
|
||||||
|
value["docs"] = serde_json::to_value(entry.docs.clone())?;
|
||||||
|
value["dependencies"] = serde_json::to_value(entry.dependencies.clone())?;
|
||||||
|
|
||||||
|
Ok(HttpResponse::Ok().json(value))
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,55 +1,54 @@
|
||||||
use crate::{
|
use std::collections::{BTreeMap, BTreeSet};
|
||||||
error::RegistryError,
|
|
||||||
package::{read_package, PackageResponse, PackageVersionsResponse},
|
|
||||||
AppState,
|
|
||||||
};
|
|
||||||
use actix_web::{web, HttpResponse, Responder};
|
use actix_web::{web, HttpResponse, Responder};
|
||||||
use pesde::{names::PackageName, source::ids::VersionId};
|
|
||||||
use semver::Version;
|
|
||||||
use std::collections::{btree_map::Entry, BTreeMap};
|
|
||||||
|
|
||||||
pub async fn get_package_versions_v0(
|
use crate::{error::Error, package::PackageResponse, AppState};
|
||||||
app_state: web::Data<AppState>,
|
use pesde::{
|
||||||
path: web::Path<PackageName>,
|
names::PackageName,
|
||||||
) -> Result<impl Responder, RegistryError> {
|
source::{
|
||||||
let name = path.into_inner();
|
git_index::{read_file, root_tree, GitBasedSource},
|
||||||
|
pesde::IndexFile,
|
||||||
let Some(file) = read_package(&app_state, &name, &*app_state.source.read().await).await? else {
|
},
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
};
|
||||||
};
|
|
||||||
|
|
||||||
let mut versions = BTreeMap::<&Version, &VersionId>::new();
|
|
||||||
|
|
||||||
for v_id in file.entries.keys() {
|
|
||||||
match versions.entry(v_id.version()) {
|
|
||||||
Entry::Vacant(entry) => {
|
|
||||||
entry.insert(v_id);
|
|
||||||
}
|
|
||||||
Entry::Occupied(mut entry) => {
|
|
||||||
if entry.get() < &v_id {
|
|
||||||
entry.insert(v_id);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let responses = versions
|
|
||||||
.into_values()
|
|
||||||
.map(|v_id| PackageResponse::new(&name, v_id, &file))
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
|
|
||||||
Ok(HttpResponse::Ok().json(responses))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_package_versions(
|
pub async fn get_package_versions(
|
||||||
app_state: web::Data<AppState>,
|
app_state: web::Data<AppState>,
|
||||||
path: web::Path<PackageName>,
|
path: web::Path<PackageName>,
|
||||||
) -> Result<impl Responder, RegistryError> {
|
) -> Result<impl Responder, Error> {
|
||||||
let name = path.into_inner();
|
let name = path.into_inner();
|
||||||
|
|
||||||
let Some(file) = read_package(&app_state, &name, &*app_state.source.read().await).await? else {
|
let (scope, name_part) = name.as_str();
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(HttpResponse::Ok().json(PackageVersionsResponse::new(&name, &file)))
|
let file: IndexFile = {
|
||||||
|
let source = app_state.source.lock().await;
|
||||||
|
let repo = gix::open(source.path(&app_state.project))?;
|
||||||
|
let tree = root_tree(&repo)?;
|
||||||
|
|
||||||
|
match read_file(&tree, [scope, name_part])? {
|
||||||
|
Some(versions) => toml::de::from_str(&versions)?,
|
||||||
|
None => return Ok(HttpResponse::NotFound().finish()),
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut responses = BTreeMap::new();
|
||||||
|
|
||||||
|
for (v_id, entry) in file.entries {
|
||||||
|
let info = responses
|
||||||
|
.entry(v_id.version().clone())
|
||||||
|
.or_insert_with(|| PackageResponse {
|
||||||
|
name: name.to_string(),
|
||||||
|
version: v_id.version().to_string(),
|
||||||
|
targets: BTreeSet::new(),
|
||||||
|
description: entry.description.unwrap_or_default(),
|
||||||
|
published_at: entry.published_at,
|
||||||
|
license: entry.license.unwrap_or_default(),
|
||||||
|
authors: entry.authors.clone(),
|
||||||
|
repository: entry.repository.clone().map(|url| url.to_string()),
|
||||||
|
});
|
||||||
|
|
||||||
|
info.targets.insert(entry.target.into());
|
||||||
|
info.published_at = info.published_at.max(entry.published_at);
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(HttpResponse::Ok().json(responses.into_values().collect::<Vec<_>>()))
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,465 +1,507 @@
|
||||||
use crate::{
|
use crate::{
|
||||||
auth::UserId,
|
auth::UserId,
|
||||||
error::{ErrorResponse, RegistryError},
|
benv,
|
||||||
git::push_changes,
|
error::{Error, ErrorResponse},
|
||||||
package::{read_package, read_scope_info},
|
search::update_version,
|
||||||
search::update_search_version,
|
storage::StorageImpl,
|
||||||
storage::StorageImpl,
|
AppState,
|
||||||
AppState,
|
|
||||||
};
|
};
|
||||||
use actix_web::{web, web::Bytes, HttpResponse};
|
use actix_web::{web, web::Bytes, HttpResponse, Responder};
|
||||||
use async_compression::Level;
|
use async_compression::Level;
|
||||||
use convert_case::{Case, Casing};
|
use convert_case::{Case, Casing};
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
|
use futures::{future::join_all, join};
|
||||||
|
use git2::{Remote, Repository, Signature};
|
||||||
use pesde::{
|
use pesde::{
|
||||||
manifest::{DependencyType, Manifest},
|
manifest::Manifest,
|
||||||
source::{
|
source::{
|
||||||
git_index::GitBasedSource,
|
git_index::{read_file, root_tree, GitBasedSource},
|
||||||
ids::VersionId,
|
pesde::{DocEntry, DocEntryKind, IndexFile, IndexFileEntry, ScopeInfo, SCOPE_INFO_FILE},
|
||||||
pesde::{DocEntry, DocEntryKind, IndexFileEntry, ScopeInfo, SCOPE_INFO_FILE},
|
specifiers::DependencySpecifiers,
|
||||||
specifiers::DependencySpecifiers,
|
version_id::VersionId,
|
||||||
traits::RefreshOptions,
|
IGNORED_DIRS, IGNORED_FILES,
|
||||||
ADDITIONAL_FORBIDDEN_FILES, IGNORED_DIRS, IGNORED_FILES,
|
},
|
||||||
},
|
MANIFEST_FILE_NAME,
|
||||||
MANIFEST_FILE_NAME,
|
|
||||||
};
|
};
|
||||||
use sentry::add_breadcrumb;
|
use sentry::add_breadcrumb;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use sha2::{Digest, Sha256};
|
use sha2::{Digest, Sha256};
|
||||||
use std::{
|
use std::{
|
||||||
collections::{BTreeSet, HashMap},
|
collections::{BTreeSet, HashMap},
|
||||||
io::Cursor,
|
io::{Cursor, Write},
|
||||||
};
|
|
||||||
use tokio::{
|
|
||||||
io::{AsyncReadExt, AsyncWriteExt},
|
|
||||||
task::JoinSet,
|
|
||||||
};
|
};
|
||||||
|
use tokio::io::{AsyncReadExt, AsyncWriteExt};
|
||||||
|
|
||||||
|
fn signature<'a>() -> Signature<'a> {
|
||||||
|
Signature::now(
|
||||||
|
&benv!(required "COMMITTER_GIT_NAME"),
|
||||||
|
&benv!(required "COMMITTER_GIT_EMAIL"),
|
||||||
|
)
|
||||||
|
.unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_refspec(repo: &Repository, remote: &mut Remote) -> Result<String, git2::Error> {
|
||||||
|
let upstream_branch_buf = repo.branch_upstream_name(repo.head()?.name().unwrap())?;
|
||||||
|
let upstream_branch = upstream_branch_buf.as_str().unwrap();
|
||||||
|
|
||||||
|
let refspec_buf = remote
|
||||||
|
.refspecs()
|
||||||
|
.find(|r| r.direction() == git2::Direction::Fetch && r.dst_matches(upstream_branch))
|
||||||
|
.unwrap()
|
||||||
|
.rtransform(upstream_branch)?;
|
||||||
|
let refspec = refspec_buf.as_str().unwrap();
|
||||||
|
|
||||||
|
Ok(refspec.to_string())
|
||||||
|
}
|
||||||
|
|
||||||
|
const ADDITIONAL_FORBIDDEN_FILES: &[&str] = &["default.project.json"];
|
||||||
|
|
||||||
#[derive(Debug, Deserialize, Default)]
|
#[derive(Debug, Deserialize, Default)]
|
||||||
struct DocEntryInfo {
|
struct DocEntryInfo {
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
label: Option<String>,
|
label: Option<String>,
|
||||||
#[serde(default, alias = "position")]
|
#[serde(default, alias = "position")]
|
||||||
sidebar_position: Option<usize>,
|
sidebar_position: Option<usize>,
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
collapsed: bool,
|
collapsed: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn publish_package(
|
pub async fn publish_package(
|
||||||
app_state: web::Data<AppState>,
|
app_state: web::Data<AppState>,
|
||||||
bytes: Bytes,
|
bytes: Bytes,
|
||||||
user_id: web::ReqData<UserId>,
|
user_id: web::ReqData<UserId>,
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
) -> Result<impl Responder, Error> {
|
||||||
let source = app_state.source.write().await;
|
let source = app_state.source.lock().await;
|
||||||
source
|
source.refresh(&app_state.project).await.map_err(Box::new)?;
|
||||||
.refresh(&RefreshOptions {
|
let config = source.config(&app_state.project).await?;
|
||||||
project: app_state.project.clone(),
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.map_err(Box::new)?;
|
|
||||||
let config = source.config(&app_state.project).await?;
|
|
||||||
|
|
||||||
let package_dir = tempfile::tempdir()?;
|
let package_dir = tempfile::tempdir()?;
|
||||||
|
|
||||||
{
|
{
|
||||||
let mut decoder = async_compression::tokio::bufread::GzipDecoder::new(Cursor::new(&bytes));
|
let mut decoder = async_compression::tokio::bufread::GzipDecoder::new(Cursor::new(&bytes));
|
||||||
let mut archive = tokio_tar::Archive::new(&mut decoder);
|
let mut archive = tokio_tar::Archive::new(&mut decoder);
|
||||||
|
|
||||||
archive.unpack(package_dir.path()).await?;
|
archive.unpack(package_dir.path()).await?;
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut manifest = None::<Manifest>;
|
let mut manifest = None::<Manifest>;
|
||||||
let mut readme = None::<Vec<u8>>;
|
let mut readme = None::<Vec<u8>>;
|
||||||
let mut docs = BTreeSet::new();
|
let mut docs = BTreeSet::new();
|
||||||
let mut docs_pages = HashMap::new();
|
let mut docs_pages = HashMap::new();
|
||||||
|
|
||||||
let mut read_dir = fs::read_dir(package_dir.path()).await?;
|
let mut read_dir = fs::read_dir(package_dir.path()).await?;
|
||||||
while let Some(entry) = read_dir.next_entry().await? {
|
while let Some(entry) = read_dir.next_entry().await? {
|
||||||
let file_name = entry
|
let file_name = entry
|
||||||
.file_name()
|
.file_name()
|
||||||
.to_str()
|
.to_str()
|
||||||
.ok_or_else(|| {
|
.ok_or_else(|| Error::InvalidArchive("file name contains non UTF-8 characters".into()))?
|
||||||
RegistryError::InvalidArchive("file name contains non UTF-8 characters".into())
|
.to_string();
|
||||||
})?
|
|
||||||
.to_string();
|
|
||||||
|
|
||||||
if entry.file_type().await?.is_dir() {
|
if entry.file_type().await?.is_dir() {
|
||||||
if IGNORED_DIRS.contains(&file_name.as_str()) {
|
if IGNORED_DIRS.contains(&file_name.as_str()) {
|
||||||
return Err(RegistryError::InvalidArchive(format!(
|
return Err(Error::InvalidArchive(format!(
|
||||||
"archive contains forbidden directory: {file_name}"
|
"archive contains forbidden directory: {file_name}"
|
||||||
)));
|
)));
|
||||||
}
|
}
|
||||||
|
|
||||||
if file_name == "docs" {
|
if file_name == "docs" {
|
||||||
let mut stack = vec![(
|
let mut stack = vec![(
|
||||||
BTreeSet::new(),
|
BTreeSet::new(),
|
||||||
fs::read_dir(entry.path()).await?,
|
fs::read_dir(entry.path()).await?,
|
||||||
None::<DocEntryInfo>,
|
None::<DocEntryInfo>,
|
||||||
)];
|
)];
|
||||||
|
|
||||||
'outer: while let Some((set, iter, category_info)) = stack.last_mut() {
|
'outer: while let Some((set, iter, category_info)) = stack.last_mut() {
|
||||||
while let Some(entry) = iter.next_entry().await? {
|
while let Some(entry) = iter.next_entry().await? {
|
||||||
let file_name = entry
|
let file_name = entry
|
||||||
.file_name()
|
.file_name()
|
||||||
.to_str()
|
.to_str()
|
||||||
.ok_or_else(|| {
|
.ok_or_else(|| {
|
||||||
RegistryError::InvalidArchive(
|
Error::InvalidArchive(
|
||||||
"file name contains non UTF-8 characters".into(),
|
"file name contains non UTF-8 characters".into(),
|
||||||
)
|
)
|
||||||
})?
|
})?
|
||||||
.to_string();
|
.to_string();
|
||||||
|
|
||||||
if entry.file_type().await?.is_dir() {
|
if entry.file_type().await?.is_dir() {
|
||||||
stack.push((
|
stack.push((
|
||||||
BTreeSet::new(),
|
BTreeSet::new(),
|
||||||
fs::read_dir(entry.path()).await?,
|
fs::read_dir(entry.path()).await?,
|
||||||
Some(DocEntryInfo {
|
Some(DocEntryInfo {
|
||||||
label: Some(file_name.to_case(Case::Title)),
|
label: Some(file_name.to_case(Case::Title)),
|
||||||
..Default::default()
|
..Default::default()
|
||||||
}),
|
}),
|
||||||
));
|
));
|
||||||
continue 'outer;
|
continue 'outer;
|
||||||
}
|
}
|
||||||
|
|
||||||
if file_name == "_category_.json" {
|
if file_name == "_category_.json" {
|
||||||
let info = fs::read_to_string(entry.path()).await?;
|
let info = fs::read_to_string(entry.path()).await?;
|
||||||
let mut info: DocEntryInfo = serde_json::from_str(&info)?;
|
let mut info: DocEntryInfo = serde_json::from_str(&info)?;
|
||||||
let old_info = category_info.take();
|
let old_info = category_info.take();
|
||||||
info.label = info.label.or(old_info.and_then(|i| i.label));
|
info.label = info.label.or(old_info.and_then(|i| i.label));
|
||||||
*category_info = Some(info);
|
*category_info = Some(info);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
let Some(file_name) = file_name.strip_suffix(".md") else {
|
let Some(file_name) = file_name.strip_suffix(".md") else {
|
||||||
continue;
|
continue;
|
||||||
};
|
};
|
||||||
|
|
||||||
let content = fs::read_to_string(entry.path()).await?;
|
let content = fs::read_to_string(entry.path()).await?;
|
||||||
let content = content.trim();
|
let content = content.trim();
|
||||||
let hash = format!("{:x}", Sha256::digest(content.as_bytes()));
|
let hash = format!("{:x}", Sha256::digest(content.as_bytes()));
|
||||||
|
|
||||||
let mut gz = async_compression::tokio::bufread::GzipEncoder::with_quality(
|
let mut gz = async_compression::tokio::bufread::GzipEncoder::with_quality(
|
||||||
Cursor::new(content.as_bytes().to_vec()),
|
Cursor::new(content.as_bytes().to_vec()),
|
||||||
Level::Best,
|
Level::Best,
|
||||||
);
|
);
|
||||||
let mut bytes = vec![];
|
let mut bytes = vec![];
|
||||||
gz.read_to_end(&mut bytes).await?;
|
gz.read_to_end(&mut bytes).await?;
|
||||||
docs_pages.insert(hash.to_string(), bytes);
|
docs_pages.insert(hash.to_string(), bytes);
|
||||||
|
|
||||||
let mut lines = content.lines().peekable();
|
let mut lines = content.lines().peekable();
|
||||||
let front_matter = if lines.peek().filter(|l| **l == "---").is_some() {
|
let front_matter = if lines.peek().filter(|l| **l == "---").is_some() {
|
||||||
lines.next(); // skip the first `---`
|
lines.next(); // skip the first `---`
|
||||||
|
|
||||||
let front_matter = lines
|
let front_matter = lines
|
||||||
.by_ref()
|
.by_ref()
|
||||||
.take_while(|l| *l != "---")
|
.take_while(|l| *l != "---")
|
||||||
.collect::<Vec<_>>()
|
.collect::<Vec<_>>()
|
||||||
.join("\n");
|
.join("\n");
|
||||||
|
|
||||||
lines.next(); // skip the last `---`
|
lines.next(); // skip the last `---`
|
||||||
|
|
||||||
front_matter
|
front_matter
|
||||||
} else {
|
} else {
|
||||||
"".to_string()
|
"".to_string()
|
||||||
};
|
};
|
||||||
|
|
||||||
let h1 = lines
|
let h1 = lines
|
||||||
.find(|l| !l.trim().is_empty())
|
.find(|l| !l.trim().is_empty())
|
||||||
.and_then(|l| l.strip_prefix("# "))
|
.and_then(|l| l.strip_prefix("# "))
|
||||||
.map(|s| s.to_string());
|
.map(|s| s.to_string());
|
||||||
|
|
||||||
let info: DocEntryInfo =
|
let info: DocEntryInfo =
|
||||||
serde_yaml::from_str(&front_matter).map_err(|_| {
|
serde_yaml::from_str(&front_matter).map_err(|_| {
|
||||||
RegistryError::InvalidArchive(format!(
|
Error::InvalidArchive(format!(
|
||||||
"doc {file_name}'s frontmatter isn't valid YAML"
|
"doc {file_name}'s frontmatter isn't valid YAML"
|
||||||
))
|
))
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
set.insert(DocEntry {
|
set.insert(DocEntry {
|
||||||
label: info.label.or(h1).unwrap_or(file_name.to_case(Case::Title)),
|
label: info.label.or(h1).unwrap_or(file_name.to_case(Case::Title)),
|
||||||
position: info.sidebar_position,
|
position: info.sidebar_position,
|
||||||
kind: DocEntryKind::Page {
|
kind: DocEntryKind::Page {
|
||||||
name: entry
|
name: entry
|
||||||
.path()
|
.path()
|
||||||
.strip_prefix(package_dir.path().join("docs"))
|
.strip_prefix(package_dir.path().join("docs"))
|
||||||
.unwrap()
|
.unwrap()
|
||||||
.with_extension("")
|
.with_extension("")
|
||||||
.to_str()
|
.to_str()
|
||||||
.ok_or_else(|| {
|
.ok_or_else(|| {
|
||||||
RegistryError::InvalidArchive(
|
Error::InvalidArchive(
|
||||||
"file name contains non UTF-8 characters".into(),
|
"file name contains non UTF-8 characters".into(),
|
||||||
)
|
)
|
||||||
})?
|
})?
|
||||||
// ensure that the path is always using forward slashes
|
// ensure that the path is always using forward slashes
|
||||||
.replace("\\", "/"),
|
.replace("\\", "/"),
|
||||||
hash,
|
hash,
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
// should never be None
|
// should never be None
|
||||||
let (popped, _, category_info) = stack.pop().unwrap();
|
let (popped, _, category_info) = stack.pop().unwrap();
|
||||||
docs = popped;
|
docs = popped;
|
||||||
|
|
||||||
if let Some((set, _, _)) = stack.last_mut() {
|
if let Some((set, _, _)) = stack.last_mut() {
|
||||||
let category_info = category_info.unwrap_or_default();
|
let category_info = category_info.unwrap_or_default();
|
||||||
|
|
||||||
set.insert(DocEntry {
|
set.insert(DocEntry {
|
||||||
label: category_info.label.unwrap(),
|
label: category_info.label.unwrap(),
|
||||||
position: category_info.sidebar_position,
|
position: category_info.sidebar_position,
|
||||||
kind: DocEntryKind::Category {
|
kind: DocEntryKind::Category {
|
||||||
items: {
|
items: {
|
||||||
let curr_docs = docs;
|
let curr_docs = docs;
|
||||||
docs = BTreeSet::new();
|
docs = BTreeSet::new();
|
||||||
curr_docs
|
curr_docs
|
||||||
},
|
},
|
||||||
collapsed: category_info.collapsed,
|
collapsed: category_info.collapsed,
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
if IGNORED_FILES.contains(&file_name.as_str())
|
if IGNORED_FILES.contains(&file_name.as_str())
|
||||||
|| ADDITIONAL_FORBIDDEN_FILES.contains(&file_name.as_str())
|
|| ADDITIONAL_FORBIDDEN_FILES.contains(&file_name.as_str())
|
||||||
{
|
{
|
||||||
return Err(RegistryError::InvalidArchive(format!(
|
return Err(Error::InvalidArchive(format!(
|
||||||
"archive contains forbidden file: {file_name}"
|
"archive contains forbidden file: {file_name}"
|
||||||
)));
|
)));
|
||||||
}
|
}
|
||||||
|
|
||||||
if file_name == MANIFEST_FILE_NAME {
|
if file_name == MANIFEST_FILE_NAME {
|
||||||
let content = fs::read_to_string(entry.path()).await?;
|
let content = fs::read_to_string(entry.path()).await?;
|
||||||
|
|
||||||
manifest = Some(toml::de::from_str(&content)?);
|
manifest = Some(toml::de::from_str(&content)?);
|
||||||
} else if file_name
|
} else if file_name
|
||||||
.to_lowercase()
|
.to_lowercase()
|
||||||
.split_once('.')
|
.split_once('.')
|
||||||
.filter(|(file, ext)| *file == "readme" && (*ext == "md" || *ext == "txt"))
|
.filter(|(file, ext)| *file == "readme" && (*ext == "md" || *ext == "txt"))
|
||||||
.is_some()
|
.is_some()
|
||||||
{
|
{
|
||||||
if readme.is_some() {
|
if readme.is_some() {
|
||||||
return Err(RegistryError::InvalidArchive(
|
return Err(Error::InvalidArchive(
|
||||||
"archive contains multiple readme files".into(),
|
"archive contains multiple readme files".into(),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut file = fs::File::open(entry.path()).await?;
|
let mut file = fs::File::open(entry.path()).await?;
|
||||||
|
|
||||||
let mut gz = async_compression::tokio::write::GzipEncoder::new(vec![]);
|
let mut gz = async_compression::tokio::write::GzipEncoder::new(vec![]);
|
||||||
tokio::io::copy(&mut file, &mut gz).await?;
|
tokio::io::copy(&mut file, &mut gz).await?;
|
||||||
gz.shutdown().await?;
|
gz.shutdown().await?;
|
||||||
readme = Some(gz.into_inner());
|
readme = Some(gz.into_inner());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let Some(manifest) = manifest else {
|
let Some(manifest) = manifest else {
|
||||||
return Err(RegistryError::InvalidArchive(
|
return Err(Error::InvalidArchive(
|
||||||
"archive doesn't contain a manifest".into(),
|
"archive doesn't contain a manifest".into(),
|
||||||
));
|
));
|
||||||
};
|
};
|
||||||
|
|
||||||
add_breadcrumb(sentry::Breadcrumb {
|
add_breadcrumb(sentry::Breadcrumb {
|
||||||
category: Some("publish".into()),
|
category: Some("publish".into()),
|
||||||
message: Some(format!(
|
message: Some(format!(
|
||||||
"publish request for {}@{} {}. has readme: {}. docs: {}",
|
"publish request for {}@{} {}. has readme: {}. docs: {}",
|
||||||
manifest.name,
|
manifest.name,
|
||||||
manifest.version,
|
manifest.version,
|
||||||
manifest.target,
|
manifest.target,
|
||||||
readme.is_some(),
|
readme.is_some(),
|
||||||
docs_pages.len()
|
docs_pages.len()
|
||||||
)),
|
)),
|
||||||
level: sentry::Level::Info,
|
level: sentry::Level::Info,
|
||||||
..Default::default()
|
..Default::default()
|
||||||
});
|
});
|
||||||
|
|
||||||
{
|
{
|
||||||
let dependencies = manifest.all_dependencies().map_err(|e| {
|
let dependencies = manifest.all_dependencies().map_err(|e| {
|
||||||
RegistryError::InvalidArchive(format!("manifest has invalid dependencies: {e}"))
|
Error::InvalidArchive(format!("manifest has invalid dependencies: {e}"))
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
for (specifier, ty) in dependencies.values() {
|
for (specifier, _) in dependencies.values() {
|
||||||
// we need not verify dev dependencies, as they won't be installed
|
match specifier {
|
||||||
if *ty == DependencyType::Dev {
|
DependencySpecifiers::Pesde(specifier) => {
|
||||||
continue;
|
if specifier
|
||||||
}
|
.index
|
||||||
|
.as_deref()
|
||||||
|
.filter(|index| match gix::Url::try_from(*index) {
|
||||||
|
Ok(url) => config
|
||||||
|
.other_registries_allowed
|
||||||
|
.is_allowed_or_same(source.repo_url().clone(), url),
|
||||||
|
Err(_) => false,
|
||||||
|
})
|
||||||
|
.is_none()
|
||||||
|
{
|
||||||
|
return Err(Error::InvalidArchive(format!(
|
||||||
|
"invalid index in pesde dependency {specifier}"
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
DependencySpecifiers::Wally(specifier) => {
|
||||||
|
if specifier
|
||||||
|
.index
|
||||||
|
.as_deref()
|
||||||
|
.filter(|index| match gix::Url::try_from(*index) {
|
||||||
|
Ok(url) => config.wally_allowed.is_allowed(url),
|
||||||
|
Err(_) => false,
|
||||||
|
})
|
||||||
|
.is_none()
|
||||||
|
{
|
||||||
|
return Err(Error::InvalidArchive(format!(
|
||||||
|
"invalid index in wally dependency {specifier}"
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
DependencySpecifiers::Git(specifier) => {
|
||||||
|
if !config.git_allowed.is_allowed(specifier.repo.clone()) {
|
||||||
|
return Err(Error::InvalidArchive(
|
||||||
|
"git dependencies are not allowed".into(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
DependencySpecifiers::Workspace(_) => {
|
||||||
|
// workspace specifiers are to be transformed into pesde specifiers by the sender
|
||||||
|
return Err(Error::InvalidArchive(
|
||||||
|
"non-transformed workspace dependency".into(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
match specifier {
|
let repo = Repository::open_bare(source.path(&app_state.project))?;
|
||||||
DependencySpecifiers::Pesde(specifier) => {
|
let gix_repo = gix::open(repo.path())?;
|
||||||
if specifier
|
|
||||||
.index
|
|
||||||
.as_deref()
|
|
||||||
.filter(|index| match gix::Url::try_from(*index) {
|
|
||||||
Ok(url) => config
|
|
||||||
.other_registries_allowed
|
|
||||||
.is_allowed_or_same(source.repo_url().clone(), url),
|
|
||||||
Err(_) => false,
|
|
||||||
})
|
|
||||||
.is_none()
|
|
||||||
{
|
|
||||||
return Err(RegistryError::InvalidArchive(format!(
|
|
||||||
"invalid index in pesde dependency {specifier}"
|
|
||||||
)));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
DependencySpecifiers::Wally(specifier) => {
|
|
||||||
if specifier
|
|
||||||
.index
|
|
||||||
.as_deref()
|
|
||||||
.filter(|index| match gix::Url::try_from(*index) {
|
|
||||||
Ok(url) => config.wally_allowed.is_allowed(url),
|
|
||||||
Err(_) => false,
|
|
||||||
})
|
|
||||||
.is_none()
|
|
||||||
{
|
|
||||||
return Err(RegistryError::InvalidArchive(format!(
|
|
||||||
"invalid index in wally dependency {specifier}"
|
|
||||||
)));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
DependencySpecifiers::Git(specifier) => {
|
|
||||||
if !config.git_allowed.is_allowed(specifier.repo.clone()) {
|
|
||||||
return Err(RegistryError::InvalidArchive(
|
|
||||||
"git dependencies are not allowed".into(),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
DependencySpecifiers::Workspace(_) => {
|
|
||||||
// workspace specifiers are to be transformed into pesde specifiers by the sender
|
|
||||||
return Err(RegistryError::InvalidArchive(
|
|
||||||
"non-transformed workspace dependency".into(),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
DependencySpecifiers::Path(_) => {
|
|
||||||
return Err(RegistryError::InvalidArchive(
|
|
||||||
"path dependencies are not allowed".into(),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut files = HashMap::new();
|
let gix_tree = root_tree(&gix_repo)?;
|
||||||
|
|
||||||
let scope = read_scope_info(&app_state, manifest.name.scope(), &source).await?;
|
let (scope, name) = manifest.name.as_str();
|
||||||
match scope {
|
let mut oids = vec![];
|
||||||
Some(info) => {
|
|
||||||
if !info.owners.contains(&user_id.0) {
|
|
||||||
return Ok(HttpResponse::Forbidden().finish());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
None => {
|
|
||||||
let scope_info = toml::to_string(&ScopeInfo {
|
|
||||||
owners: BTreeSet::from([user_id.0]),
|
|
||||||
})?;
|
|
||||||
|
|
||||||
files.insert(SCOPE_INFO_FILE.to_string(), scope_info.into_bytes());
|
match read_file(&gix_tree, [scope, SCOPE_INFO_FILE])? {
|
||||||
}
|
Some(info) => {
|
||||||
}
|
let info: ScopeInfo = toml::de::from_str(&info)?;
|
||||||
|
if !info.owners.contains(&user_id.0) {
|
||||||
|
return Ok(HttpResponse::Forbidden().finish());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None => {
|
||||||
|
let scope_info = toml::to_string(&ScopeInfo {
|
||||||
|
owners: BTreeSet::from([user_id.0]),
|
||||||
|
})?;
|
||||||
|
|
||||||
let mut file = read_package(&app_state, &manifest.name, &source)
|
let mut blob_writer = repo.blob_writer(None)?;
|
||||||
.await?
|
blob_writer.write_all(scope_info.as_bytes())?;
|
||||||
.unwrap_or_default();
|
oids.push((SCOPE_INFO_FILE, blob_writer.commit()?));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
let new_entry = IndexFileEntry {
|
let mut file: IndexFile =
|
||||||
target: manifest.target.clone(),
|
toml::de::from_str(&read_file(&gix_tree, [scope, name])?.unwrap_or_default())?;
|
||||||
published_at: jiff::Timestamp::now(),
|
|
||||||
engines: manifest.engines.clone(),
|
|
||||||
description: manifest.description.clone(),
|
|
||||||
license: manifest.license.clone(),
|
|
||||||
authors: manifest.authors.clone(),
|
|
||||||
repository: manifest.repository.clone(),
|
|
||||||
yanked: false,
|
|
||||||
docs,
|
|
||||||
|
|
||||||
dependencies,
|
let new_entry = IndexFileEntry {
|
||||||
};
|
target: manifest.target.clone(),
|
||||||
|
published_at: chrono::Utc::now(),
|
||||||
|
description: manifest.description.clone(),
|
||||||
|
license: manifest.license.clone(),
|
||||||
|
authors: manifest.authors.clone(),
|
||||||
|
repository: manifest.repository.clone(),
|
||||||
|
docs,
|
||||||
|
|
||||||
let same_version = file
|
dependencies,
|
||||||
.entries
|
};
|
||||||
.iter()
|
|
||||||
.find(|(v_id, _)| *v_id.version() == manifest.version);
|
|
||||||
if let Some((_, other_entry)) = same_version {
|
|
||||||
// description cannot be different - which one to render in the "Recently published" list?
|
|
||||||
if other_entry.description != new_entry.description {
|
|
||||||
return Ok(HttpResponse::BadRequest().json(ErrorResponse {
|
|
||||||
error: "same versions with different descriptions are forbidden".to_string(),
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if file
|
let this_version = file
|
||||||
.entries
|
.entries
|
||||||
.insert(
|
.keys()
|
||||||
VersionId::new(manifest.version.clone(), manifest.target.kind()),
|
.find(|v_id| *v_id.version() == manifest.version);
|
||||||
new_entry.clone(),
|
if let Some(this_version) = this_version {
|
||||||
)
|
let other_entry = file.entries.get(this_version).unwrap();
|
||||||
.is_some()
|
|
||||||
{
|
|
||||||
return Ok(HttpResponse::Conflict().finish());
|
|
||||||
}
|
|
||||||
|
|
||||||
files.insert(
|
// description cannot be different - which one to render in the "Recently published" list?
|
||||||
manifest.name.name().to_string(),
|
// the others cannot be different because what to return from the versions endpoint?
|
||||||
toml::to_string(&file)?.into_bytes(),
|
if other_entry.description != new_entry.description
|
||||||
);
|
|| other_entry.license != new_entry.license
|
||||||
|
|| other_entry.authors != new_entry.authors
|
||||||
|
|| other_entry.repository != new_entry.repository
|
||||||
|
{
|
||||||
|
return Ok(HttpResponse::BadRequest().json(ErrorResponse {
|
||||||
|
error: "same version with different description or license already exists"
|
||||||
|
.to_string(),
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
push_changes(
|
if file
|
||||||
&app_state,
|
.entries
|
||||||
&source,
|
.insert(
|
||||||
manifest.name.scope().to_string(),
|
VersionId::new(manifest.version.clone(), manifest.target.kind()),
|
||||||
files,
|
new_entry.clone(),
|
||||||
format!(
|
)
|
||||||
"add {}@{} {}",
|
.is_some()
|
||||||
manifest.name, manifest.version, manifest.target
|
{
|
||||||
),
|
return Ok(HttpResponse::Conflict().finish());
|
||||||
)
|
}
|
||||||
.await?;
|
|
||||||
|
|
||||||
update_search_version(&app_state, &manifest.name, &new_entry);
|
let mut remote = repo.find_remote("origin")?;
|
||||||
}
|
let refspec = get_refspec(&repo, &mut remote)?;
|
||||||
|
|
||||||
let version_id = VersionId::new(manifest.version.clone(), manifest.target.kind());
|
let reference = repo.find_reference(&refspec)?;
|
||||||
|
|
||||||
let mut tasks = docs_pages
|
{
|
||||||
.into_iter()
|
let index_content = toml::to_string(&file)?;
|
||||||
.map(|(hash, content)| {
|
let mut blob_writer = repo.blob_writer(None)?;
|
||||||
let app_state = app_state.clone();
|
blob_writer.write_all(index_content.as_bytes())?;
|
||||||
async move { app_state.storage.store_doc(hash, content).await }
|
oids.push((name, blob_writer.commit()?));
|
||||||
})
|
}
|
||||||
.collect::<JoinSet<_>>();
|
|
||||||
|
|
||||||
{
|
let old_root_tree = reference.peel_to_tree()?;
|
||||||
let app_state = app_state.clone();
|
let old_scope_tree = match old_root_tree.get_name(scope) {
|
||||||
let name = manifest.name.clone();
|
Some(entry) => Some(repo.find_tree(entry.id())?),
|
||||||
let version_id = version_id.clone();
|
None => None,
|
||||||
|
};
|
||||||
|
|
||||||
tasks.spawn(async move {
|
let mut scope_tree = repo.treebuilder(old_scope_tree.as_ref())?;
|
||||||
app_state
|
for (file, oid) in oids {
|
||||||
.storage
|
scope_tree.insert(file, oid, 0o100644)?;
|
||||||
.store_package(&name, &version_id, bytes.to_vec())
|
}
|
||||||
.await
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(readme) = readme {
|
let scope_tree_id = scope_tree.write()?;
|
||||||
let app_state = app_state.clone();
|
let mut root_tree = repo.treebuilder(Some(&repo.find_tree(old_root_tree.id())?))?;
|
||||||
let name = manifest.name.clone();
|
root_tree.insert(scope, scope_tree_id, 0o040000)?;
|
||||||
let version_id = version_id.clone();
|
|
||||||
|
|
||||||
tasks.spawn(async move {
|
let tree_oid = root_tree.write()?;
|
||||||
app_state
|
|
||||||
.storage
|
|
||||||
.store_readme(&name, &version_id, readme)
|
|
||||||
.await
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
while let Some(res) = tasks.join_next().await {
|
repo.commit(
|
||||||
res.unwrap()?;
|
Some("HEAD"),
|
||||||
}
|
&signature(),
|
||||||
|
&signature(),
|
||||||
|
&format!(
|
||||||
|
"add {}@{} {}",
|
||||||
|
manifest.name, manifest.version, manifest.target
|
||||||
|
),
|
||||||
|
&repo.find_tree(tree_oid)?,
|
||||||
|
&[&reference.peel_to_commit()?],
|
||||||
|
)?;
|
||||||
|
|
||||||
Ok(HttpResponse::Ok().body(format!("published {}@{version_id}", manifest.name)))
|
let mut push_options = git2::PushOptions::new();
|
||||||
|
let mut remote_callbacks = git2::RemoteCallbacks::new();
|
||||||
|
|
||||||
|
let git_creds = app_state.project.auth_config().git_credentials().unwrap();
|
||||||
|
remote_callbacks.credentials(|_, _, _| {
|
||||||
|
git2::Cred::userpass_plaintext(&git_creds.username, &git_creds.password)
|
||||||
|
});
|
||||||
|
|
||||||
|
push_options.remote_callbacks(remote_callbacks);
|
||||||
|
|
||||||
|
remote.push(&[refspec], Some(&mut push_options))?;
|
||||||
|
|
||||||
|
update_version(&app_state, &manifest.name, new_entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
let version_id = VersionId::new(manifest.version.clone(), manifest.target.kind());
|
||||||
|
|
||||||
|
let (a, b, c) = join!(
|
||||||
|
app_state
|
||||||
|
.storage
|
||||||
|
.store_package(&manifest.name, &version_id, bytes.to_vec()),
|
||||||
|
join_all(
|
||||||
|
docs_pages
|
||||||
|
.into_iter()
|
||||||
|
.map(|(hash, content)| app_state.storage.store_doc(hash, content)),
|
||||||
|
),
|
||||||
|
async {
|
||||||
|
if let Some(readme) = readme {
|
||||||
|
app_state
|
||||||
|
.storage
|
||||||
|
.store_readme(&manifest.name, &version_id, readme)
|
||||||
|
.await
|
||||||
|
} else {
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
);
|
||||||
|
a?;
|
||||||
|
b.into_iter().collect::<Result<(), _>>()?;
|
||||||
|
c?;
|
||||||
|
|
||||||
|
Ok(HttpResponse::Ok().body(format!(
|
||||||
|
"published {}@{} {}",
|
||||||
|
manifest.name, manifest.version, manifest.target
|
||||||
|
)))
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,93 +1,107 @@
|
||||||
use crate::{
|
use std::collections::HashMap;
|
||||||
error::RegistryError,
|
|
||||||
package::{read_package, PackageResponse},
|
use actix_web::{web, HttpResponse, Responder};
|
||||||
search::find_max_searchable,
|
|
||||||
AppState,
|
|
||||||
};
|
|
||||||
use actix_web::{web, HttpResponse};
|
|
||||||
use pesde::names::PackageName;
|
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use std::{collections::HashMap, sync::Arc};
|
|
||||||
use tantivy::{collector::Count, query::AllQuery, schema::Value, DateTime, Order};
|
use tantivy::{collector::Count, query::AllQuery, schema::Value, DateTime, Order};
|
||||||
use tokio::task::JoinSet;
|
|
||||||
|
use crate::{error::Error, package::PackageResponse, AppState};
|
||||||
|
use pesde::{
|
||||||
|
names::PackageName,
|
||||||
|
source::{
|
||||||
|
git_index::{read_file, root_tree, GitBasedSource},
|
||||||
|
pesde::IndexFile,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
pub struct Request {
|
pub struct Request {
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
query: String,
|
query: Option<String>,
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
offset: usize,
|
offset: Option<usize>,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn search_packages(
|
pub async fn search_packages(
|
||||||
app_state: web::Data<AppState>,
|
app_state: web::Data<AppState>,
|
||||||
request_query: web::Query<Request>,
|
request: web::Query<Request>,
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
) -> Result<impl Responder, Error> {
|
||||||
let searcher = app_state.search_reader.searcher();
|
let searcher = app_state.search_reader.searcher();
|
||||||
let schema = searcher.schema();
|
let schema = searcher.schema();
|
||||||
|
|
||||||
let id = schema.get_field("id").unwrap();
|
let id = schema.get_field("id").unwrap();
|
||||||
|
|
||||||
let query = request_query.query.trim();
|
let query = request.query.as_deref().unwrap_or_default().trim();
|
||||||
|
|
||||||
let query = if query.is_empty() {
|
let query = if query.is_empty() {
|
||||||
Box::new(AllQuery)
|
Box::new(AllQuery)
|
||||||
} else {
|
} else {
|
||||||
app_state.query_parser.parse_query(query)?
|
app_state.query_parser.parse_query(query)?
|
||||||
};
|
};
|
||||||
|
|
||||||
let (count, top_docs) = searcher
|
let (count, top_docs) = searcher
|
||||||
.search(
|
.search(
|
||||||
&query,
|
&query,
|
||||||
&(
|
&(
|
||||||
Count,
|
Count,
|
||||||
tantivy::collector::TopDocs::with_limit(50)
|
tantivy::collector::TopDocs::with_limit(50)
|
||||||
.and_offset(request_query.offset)
|
.and_offset(request.offset.unwrap_or_default())
|
||||||
.order_by_fast_field::<DateTime>("published_at", Order::Desc),
|
.order_by_fast_field::<DateTime>("published_at", Order::Desc),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
let source = Arc::new(app_state.source.clone().read_owned().await);
|
let source = app_state.source.lock().await;
|
||||||
|
let repo = gix::open(source.path(&app_state.project))?;
|
||||||
|
let tree = root_tree(&repo)?;
|
||||||
|
|
||||||
let mut results = top_docs
|
let top_docs = top_docs
|
||||||
.iter()
|
.into_iter()
|
||||||
.map(|_| None::<PackageResponse>)
|
.map(|(_, doc_address)| {
|
||||||
.collect::<Vec<_>>();
|
let doc = searcher.doc::<HashMap<_, _>>(doc_address).unwrap();
|
||||||
|
|
||||||
let mut tasks = top_docs
|
let id = doc
|
||||||
.into_iter()
|
.get(&id)
|
||||||
.enumerate()
|
.unwrap()
|
||||||
.map(|(i, (_, doc_address))| {
|
.as_str()
|
||||||
let app_state = app_state.clone();
|
.unwrap()
|
||||||
let doc = searcher.doc::<HashMap<_, _>>(doc_address).unwrap();
|
.parse::<PackageName>()
|
||||||
let source = source.clone();
|
.unwrap();
|
||||||
|
let (scope, name) = id.as_str();
|
||||||
|
|
||||||
async move {
|
let file: IndexFile =
|
||||||
let id = doc
|
toml::de::from_str(&read_file(&tree, [scope, name]).unwrap().unwrap()).unwrap();
|
||||||
.get(&id)
|
|
||||||
.unwrap()
|
|
||||||
.as_str()
|
|
||||||
.unwrap()
|
|
||||||
.parse::<PackageName>()
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let file = read_package(&app_state, &id, &source).await?.unwrap();
|
let (latest_version, entry) = file
|
||||||
|
.entries
|
||||||
|
.iter()
|
||||||
|
.max_by_key(|(v_id, _)| v_id.version())
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
let (version_id, _) = find_max_searchable(&file).unwrap();
|
PackageResponse {
|
||||||
|
name: id.to_string(),
|
||||||
|
version: latest_version.version().to_string(),
|
||||||
|
targets: file
|
||||||
|
.entries
|
||||||
|
.iter()
|
||||||
|
.filter(|(v_id, _)| v_id.version() == latest_version.version())
|
||||||
|
.map(|(_, entry)| (&entry.target).into())
|
||||||
|
.collect(),
|
||||||
|
description: entry.description.clone().unwrap_or_default(),
|
||||||
|
published_at: file
|
||||||
|
.entries
|
||||||
|
.values()
|
||||||
|
.map(|entry| entry.published_at)
|
||||||
|
.max()
|
||||||
|
.unwrap(),
|
||||||
|
license: entry.license.clone().unwrap_or_default(),
|
||||||
|
authors: entry.authors.clone(),
|
||||||
|
repository: entry.repository.clone().map(|url| url.to_string()),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
Ok::<_, RegistryError>((i, PackageResponse::new(&id, version_id, &file)))
|
Ok(HttpResponse::Ok().json(serde_json::json!({
|
||||||
}
|
"data": top_docs,
|
||||||
})
|
"count": count,
|
||||||
.collect::<JoinSet<_>>();
|
})))
|
||||||
|
|
||||||
while let Some(res) = tasks.join_next().await {
|
|
||||||
let (i, res) = res.unwrap()?;
|
|
||||||
results[i] = Some(res);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(HttpResponse::Ok().json(serde_json::json!({
|
|
||||||
"data": results,
|
|
||||||
"count": count,
|
|
||||||
})))
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,83 +0,0 @@
|
||||||
use crate::{
|
|
||||||
auth::UserId,
|
|
||||||
error::RegistryError,
|
|
||||||
git::push_changes,
|
|
||||||
package::{read_package, read_scope_info},
|
|
||||||
request_path::AllOrSpecificTarget,
|
|
||||||
search::search_version_changed,
|
|
||||||
AppState,
|
|
||||||
};
|
|
||||||
use actix_web::{http::Method, web, HttpRequest, HttpResponse};
|
|
||||||
use pesde::names::PackageName;
|
|
||||||
use semver::Version;
|
|
||||||
use std::collections::HashMap;
|
|
||||||
|
|
||||||
pub async fn yank_package_version(
|
|
||||||
request: HttpRequest,
|
|
||||||
app_state: web::Data<AppState>,
|
|
||||||
path: web::Path<(PackageName, Version, AllOrSpecificTarget)>,
|
|
||||||
user_id: web::ReqData<UserId>,
|
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
|
||||||
let yanked = request.method() != Method::DELETE;
|
|
||||||
let (name, version, target) = path.into_inner();
|
|
||||||
let source = app_state.source.write().await;
|
|
||||||
|
|
||||||
let Some(scope_info) = read_scope_info(&app_state, name.scope(), &source).await? else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
if !scope_info.owners.contains(&user_id.0) {
|
|
||||||
return Ok(HttpResponse::Forbidden().finish());
|
|
||||||
}
|
|
||||||
|
|
||||||
let Some(mut file) = read_package(&app_state, &name, &source).await? else {
|
|
||||||
return Ok(HttpResponse::NotFound().finish());
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut targets = vec![];
|
|
||||||
|
|
||||||
for (v_id, entry) in &mut file.entries {
|
|
||||||
if *v_id.version() != version {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
match target {
|
|
||||||
AllOrSpecificTarget::Specific(kind) if entry.target.kind() != kind => continue,
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
|
|
||||||
if entry.yanked == yanked {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
targets.push(entry.target.kind().to_string());
|
|
||||||
entry.yanked = yanked;
|
|
||||||
}
|
|
||||||
|
|
||||||
if targets.is_empty() {
|
|
||||||
return Ok(HttpResponse::Conflict().finish());
|
|
||||||
}
|
|
||||||
|
|
||||||
let file_string = toml::to_string(&file)?;
|
|
||||||
|
|
||||||
push_changes(
|
|
||||||
&app_state,
|
|
||||||
&source,
|
|
||||||
name.scope().to_string(),
|
|
||||||
HashMap::from([(name.name().to_string(), file_string.into_bytes())]),
|
|
||||||
format!(
|
|
||||||
"{}yank {name}@{version} {}",
|
|
||||||
if yanked { "" } else { "un" },
|
|
||||||
targets.join(", "),
|
|
||||||
),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
search_version_changed(&app_state, &name, &file);
|
|
||||||
|
|
||||||
Ok(HttpResponse::Ok().body(format!(
|
|
||||||
"{}yanked {name}@{version} {}",
|
|
||||||
if yanked { "" } else { "un" },
|
|
||||||
targets.join(", "),
|
|
||||||
)))
|
|
||||||
}
|
|
|
@ -4,86 +4,86 @@ use serde::Serialize;
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
||||||
|
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
pub enum RegistryError {
|
pub enum Error {
|
||||||
#[error("failed to parse query")]
|
#[error("failed to parse query")]
|
||||||
Query(#[from] tantivy::query::QueryParserError),
|
Query(#[from] tantivy::query::QueryParserError),
|
||||||
|
|
||||||
#[error("error reading repo file")]
|
#[error("error reading repo file")]
|
||||||
ReadFile(#[from] ReadFile),
|
ReadFile(#[from] ReadFile),
|
||||||
|
|
||||||
#[error("error deserializing file")]
|
#[error("error deserializing file")]
|
||||||
Deserialize(#[from] toml::de::Error),
|
Deserialize(#[from] toml::de::Error),
|
||||||
|
|
||||||
#[error("failed to send request: {1}\nserver response: {0}")]
|
#[error("failed to send request: {1}\nserver response: {0}")]
|
||||||
ReqwestResponse(String, #[source] reqwest::Error),
|
ReqwestResponse(String, #[source] reqwest::Error),
|
||||||
|
|
||||||
#[error("error sending request")]
|
#[error("error sending request")]
|
||||||
Reqwest(#[from] reqwest::Error),
|
Reqwest(#[from] reqwest::Error),
|
||||||
|
|
||||||
#[error("failed to parse archive entries")]
|
#[error("failed to parse archive entries")]
|
||||||
Tar(#[from] std::io::Error),
|
Tar(#[from] std::io::Error),
|
||||||
|
|
||||||
#[error("invalid archive")]
|
#[error("invalid archive")]
|
||||||
InvalidArchive(String),
|
InvalidArchive(String),
|
||||||
|
|
||||||
#[error("failed to read index config")]
|
#[error("failed to read index config")]
|
||||||
Config(#[from] pesde::source::pesde::errors::ConfigError),
|
Config(#[from] pesde::source::pesde::errors::ConfigError),
|
||||||
|
|
||||||
#[error("git error")]
|
#[error("git error")]
|
||||||
Git(#[from] git2::Error),
|
Git(#[from] git2::Error),
|
||||||
|
|
||||||
#[error("failed to refresh source")]
|
#[error("failed to refresh source")]
|
||||||
Refresh(#[from] Box<RefreshError>),
|
Refresh(#[from] Box<RefreshError>),
|
||||||
|
|
||||||
#[error("failed to serialize struct")]
|
#[error("failed to serialize struct")]
|
||||||
Serialize(#[from] toml::ser::Error),
|
Serialize(#[from] toml::ser::Error),
|
||||||
|
|
||||||
#[error("failed to serialize struct")]
|
#[error("failed to serialize struct")]
|
||||||
SerializeJson(#[from] serde_json::Error),
|
SerializeJson(#[from] serde_json::Error),
|
||||||
|
|
||||||
#[error("failed to open git repo")]
|
#[error("failed to open git repo")]
|
||||||
OpenRepo(#[from] gix::open::Error),
|
OpenRepo(#[from] gix::open::Error),
|
||||||
|
|
||||||
#[error("failed to get root tree")]
|
#[error("failed to get root tree")]
|
||||||
RootTree(#[from] TreeError),
|
RootTree(#[from] TreeError),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
#[derive(Debug, Serialize)]
|
||||||
pub struct ErrorResponse {
|
pub struct ErrorResponse {
|
||||||
pub error: String,
|
pub error: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ResponseError for RegistryError {
|
impl ResponseError for Error {
|
||||||
fn error_response(&self) -> HttpResponse<BoxBody> {
|
fn error_response(&self) -> HttpResponse<BoxBody> {
|
||||||
match self {
|
match self {
|
||||||
RegistryError::Query(e) => HttpResponse::BadRequest().json(ErrorResponse {
|
Error::Query(e) => HttpResponse::BadRequest().json(ErrorResponse {
|
||||||
error: format!("failed to parse query: {e}"),
|
error: format!("failed to parse query: {e}"),
|
||||||
}),
|
}),
|
||||||
RegistryError::Tar(_) => HttpResponse::BadRequest().json(ErrorResponse {
|
Error::Tar(_) => HttpResponse::BadRequest().json(ErrorResponse {
|
||||||
error: "corrupt archive".to_string(),
|
error: "corrupt archive".to_string(),
|
||||||
}),
|
}),
|
||||||
RegistryError::InvalidArchive(e) => HttpResponse::BadRequest().json(ErrorResponse {
|
Error::InvalidArchive(e) => HttpResponse::BadRequest().json(ErrorResponse {
|
||||||
error: format!("archive is invalid: {e}"),
|
error: format!("archive is invalid: {e}"),
|
||||||
}),
|
}),
|
||||||
e => {
|
e => {
|
||||||
tracing::error!("unhandled error: {e:?}");
|
tracing::error!("unhandled error: {e:?}");
|
||||||
HttpResponse::InternalServerError().finish()
|
HttpResponse::InternalServerError().finish()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait ReqwestErrorExt {
|
pub trait ReqwestErrorExt {
|
||||||
async fn into_error(self) -> Result<Self, RegistryError>
|
async fn into_error(self) -> Result<Self, Error>
|
||||||
where
|
where
|
||||||
Self: Sized;
|
Self: Sized;
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ReqwestErrorExt for reqwest::Response {
|
impl ReqwestErrorExt for reqwest::Response {
|
||||||
async fn into_error(self) -> Result<Self, RegistryError> {
|
async fn into_error(self) -> Result<Self, Error> {
|
||||||
match self.error_for_status_ref() {
|
match self.error_for_status_ref() {
|
||||||
Ok(_) => Ok(self),
|
Ok(_) => Ok(self),
|
||||||
Err(e) => Err(RegistryError::ReqwestResponse(self.text().await?, e)),
|
Err(e) => Err(Error::ReqwestResponse(self.text().await?, e)),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,98 +0,0 @@
|
||||||
use crate::{benv, error::RegistryError, AppState};
|
|
||||||
use git2::{Remote, Repository, Signature};
|
|
||||||
use pesde::source::{git_index::GitBasedSource, pesde::PesdePackageSource};
|
|
||||||
use std::collections::HashMap;
|
|
||||||
use tokio::task::spawn_blocking;
|
|
||||||
|
|
||||||
fn signature<'a>() -> Signature<'a> {
|
|
||||||
Signature::now(
|
|
||||||
&benv!(required "COMMITTER_GIT_NAME"),
|
|
||||||
&benv!(required "COMMITTER_GIT_EMAIL"),
|
|
||||||
)
|
|
||||||
.unwrap()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_refspec(repo: &Repository, remote: &mut Remote) -> Result<String, git2::Error> {
|
|
||||||
let upstream_branch_buf = repo.branch_upstream_name(repo.head()?.name().unwrap())?;
|
|
||||||
let upstream_branch = upstream_branch_buf.as_str().unwrap();
|
|
||||||
|
|
||||||
let refspec_buf = remote
|
|
||||||
.refspecs()
|
|
||||||
.find(|r| r.direction() == git2::Direction::Fetch && r.dst_matches(upstream_branch))
|
|
||||||
.unwrap()
|
|
||||||
.rtransform(upstream_branch)?;
|
|
||||||
let refspec = refspec_buf.as_str().unwrap();
|
|
||||||
|
|
||||||
Ok(refspec.to_string())
|
|
||||||
}
|
|
||||||
|
|
||||||
const FILE_FILEMODE: i32 = 0o100644;
|
|
||||||
const DIR_FILEMODE: i32 = 0o040000;
|
|
||||||
|
|
||||||
pub async fn push_changes(
|
|
||||||
app_state: &AppState,
|
|
||||||
source: &PesdePackageSource,
|
|
||||||
directory: String,
|
|
||||||
files: HashMap<String, Vec<u8>>,
|
|
||||||
message: String,
|
|
||||||
) -> Result<(), RegistryError> {
|
|
||||||
let path = source.path(&app_state.project);
|
|
||||||
let auth_config = app_state.project.auth_config().clone();
|
|
||||||
|
|
||||||
spawn_blocking(move || {
|
|
||||||
let repo = Repository::open_bare(path)?;
|
|
||||||
let mut oids = HashMap::new();
|
|
||||||
|
|
||||||
let mut remote = repo.find_remote("origin")?;
|
|
||||||
let refspec = get_refspec(&repo, &mut remote)?;
|
|
||||||
|
|
||||||
let reference = repo.find_reference(&refspec)?;
|
|
||||||
|
|
||||||
for (name, contents) in files {
|
|
||||||
let oid = repo.blob(&contents)?;
|
|
||||||
oids.insert(name, oid);
|
|
||||||
}
|
|
||||||
|
|
||||||
let old_root_tree = reference.peel_to_tree()?;
|
|
||||||
let old_dir_tree = match old_root_tree.get_name(&directory) {
|
|
||||||
Some(entry) => Some(repo.find_tree(entry.id())?),
|
|
||||||
None => None,
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut dir_tree = repo.treebuilder(old_dir_tree.as_ref())?;
|
|
||||||
for (file, oid) in oids {
|
|
||||||
dir_tree.insert(file, oid, FILE_FILEMODE)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
let dir_tree_id = dir_tree.write()?;
|
|
||||||
let mut root_tree = repo.treebuilder(Some(&repo.find_tree(old_root_tree.id())?))?;
|
|
||||||
root_tree.insert(directory, dir_tree_id, DIR_FILEMODE)?;
|
|
||||||
|
|
||||||
let tree_oid = root_tree.write()?;
|
|
||||||
|
|
||||||
repo.commit(
|
|
||||||
Some("HEAD"),
|
|
||||||
&signature(),
|
|
||||||
&signature(),
|
|
||||||
&message,
|
|
||||||
&repo.find_tree(tree_oid)?,
|
|
||||||
&[&reference.peel_to_commit()?],
|
|
||||||
)?;
|
|
||||||
|
|
||||||
let mut push_options = git2::PushOptions::new();
|
|
||||||
let mut remote_callbacks = git2::RemoteCallbacks::new();
|
|
||||||
|
|
||||||
let git_creds = auth_config.git_credentials().unwrap();
|
|
||||||
remote_callbacks.credentials(|_, _, _| {
|
|
||||||
git2::Cred::userpass_plaintext(&git_creds.username, &git_creds.password)
|
|
||||||
});
|
|
||||||
|
|
||||||
push_options.remote_callbacks(remote_callbacks);
|
|
||||||
|
|
||||||
remote.push(&[refspec], Some(&mut push_options))?;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.unwrap()
|
|
||||||
}
|
|
|
@ -1,60 +1,55 @@
|
||||||
use crate::{
|
use crate::{
|
||||||
auth::{get_auth_from_env, Auth, UserIdExtractor},
|
auth::{get_auth_from_env, Auth, UserIdExtractor},
|
||||||
search::make_search,
|
search::make_search,
|
||||||
storage::{get_storage_from_env, Storage},
|
storage::{get_storage_from_env, Storage},
|
||||||
};
|
};
|
||||||
use actix_cors::Cors;
|
use actix_cors::Cors;
|
||||||
use actix_governor::{Governor, GovernorConfigBuilder};
|
use actix_governor::{Governor, GovernorConfigBuilder};
|
||||||
use actix_web::{
|
use actix_web::{
|
||||||
middleware::{from_fn, Compress, NormalizePath, TrailingSlash},
|
middleware::{from_fn, Compress, NormalizePath, TrailingSlash},
|
||||||
rt::System,
|
rt::System,
|
||||||
web,
|
web,
|
||||||
web::PayloadConfig,
|
web::PayloadConfig,
|
||||||
App, HttpServer,
|
App, HttpServer,
|
||||||
};
|
};
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use pesde::{
|
use pesde::{
|
||||||
source::{
|
source::{pesde::PesdePackageSource, traits::PackageSource},
|
||||||
pesde::PesdePackageSource,
|
AuthConfig, Project,
|
||||||
traits::{PackageSource, RefreshOptions},
|
|
||||||
},
|
|
||||||
AuthConfig, Project,
|
|
||||||
};
|
};
|
||||||
use std::{env::current_dir, path::PathBuf, sync::Arc};
|
use std::{env::current_dir, path::PathBuf};
|
||||||
use tracing::level_filters::LevelFilter;
|
use tracing::level_filters::LevelFilter;
|
||||||
use tracing_subscriber::{
|
use tracing_subscriber::{
|
||||||
fmt::format::FmtSpan, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter,
|
fmt::format::FmtSpan, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter,
|
||||||
};
|
};
|
||||||
|
|
||||||
mod auth;
|
mod auth;
|
||||||
mod endpoints;
|
mod endpoints;
|
||||||
mod error;
|
mod error;
|
||||||
mod git;
|
|
||||||
mod package;
|
mod package;
|
||||||
mod request_path;
|
|
||||||
mod search;
|
mod search;
|
||||||
mod storage;
|
mod storage;
|
||||||
|
|
||||||
pub fn make_reqwest() -> reqwest::Client {
|
pub fn make_reqwest() -> reqwest::Client {
|
||||||
reqwest::ClientBuilder::new()
|
reqwest::ClientBuilder::new()
|
||||||
.user_agent(concat!(
|
.user_agent(concat!(
|
||||||
env!("CARGO_PKG_NAME"),
|
env!("CARGO_PKG_NAME"),
|
||||||
"/",
|
"/",
|
||||||
env!("CARGO_PKG_VERSION")
|
env!("CARGO_PKG_VERSION")
|
||||||
))
|
))
|
||||||
.build()
|
.build()
|
||||||
.unwrap()
|
.unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub struct AppState {
|
pub struct AppState {
|
||||||
pub source: Arc<tokio::sync::RwLock<PesdePackageSource>>,
|
pub source: tokio::sync::Mutex<PesdePackageSource>,
|
||||||
pub project: Project,
|
pub project: Project,
|
||||||
pub storage: Storage,
|
pub storage: Storage,
|
||||||
pub auth: Auth,
|
pub auth: Auth,
|
||||||
|
|
||||||
pub search_reader: tantivy::IndexReader,
|
pub search_reader: tantivy::IndexReader,
|
||||||
pub search_writer: std::sync::Mutex<tantivy::IndexWriter>,
|
pub search_writer: std::sync::Mutex<tantivy::IndexWriter>,
|
||||||
pub query_parser: tantivy::query::QueryParser,
|
pub query_parser: tantivy::query::QueryParser,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[macro_export]
|
#[macro_export]
|
||||||
|
@ -91,223 +86,157 @@ macro_rules! benv {
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn run() -> std::io::Result<()> {
|
async fn run() -> std::io::Result<()> {
|
||||||
let address = benv!("ADDRESS" => "127.0.0.1");
|
let address = benv!("ADDRESS" => "127.0.0.1");
|
||||||
let port: u16 = benv!(parse "PORT" => "8080");
|
let port: u16 = benv!(parse "PORT" => "8080");
|
||||||
|
|
||||||
let cwd = current_dir().unwrap();
|
let cwd = current_dir().unwrap();
|
||||||
let data_dir =
|
let data_dir =
|
||||||
PathBuf::from(benv!("DATA_DIR" => "{CWD}/data").replace("{CWD}", cwd.to_str().unwrap()));
|
PathBuf::from(benv!("DATA_DIR" => "{CWD}/data").replace("{CWD}", cwd.to_str().unwrap()));
|
||||||
fs::create_dir_all(&data_dir).await.unwrap();
|
fs::create_dir_all(&data_dir).await.unwrap();
|
||||||
|
|
||||||
let project = Project::new(
|
let project = Project::new(
|
||||||
&cwd,
|
&cwd,
|
||||||
None::<PathBuf>,
|
None::<PathBuf>,
|
||||||
data_dir.join("project"),
|
data_dir.join("project"),
|
||||||
&cwd,
|
&cwd,
|
||||||
AuthConfig::new().with_git_credentials(Some(gix::sec::identity::Account {
|
AuthConfig::new().with_git_credentials(Some(gix::sec::identity::Account {
|
||||||
username: benv!(required "GIT_USERNAME"),
|
username: benv!(required "GIT_USERNAME"),
|
||||||
password: benv!(required "GIT_PASSWORD"),
|
password: benv!(required "GIT_PASSWORD"),
|
||||||
})),
|
})),
|
||||||
);
|
);
|
||||||
let source = PesdePackageSource::new(benv!(required "INDEX_REPO_URL").try_into().unwrap());
|
let source = PesdePackageSource::new(benv!(required "INDEX_REPO_URL").try_into().unwrap());
|
||||||
source
|
source
|
||||||
.refresh(&RefreshOptions {
|
.refresh(&project)
|
||||||
project: project.clone(),
|
.await
|
||||||
})
|
.expect("failed to refresh source");
|
||||||
.await
|
let config = source
|
||||||
.expect("failed to refresh source");
|
.config(&project)
|
||||||
let config = source
|
.await
|
||||||
.config(&project)
|
.expect("failed to get index config");
|
||||||
.await
|
|
||||||
.expect("failed to get index config");
|
|
||||||
|
|
||||||
let (search_reader, search_writer, query_parser) = make_search(&project, &source).await;
|
let (search_reader, search_writer, query_parser) = make_search(&project, &source).await;
|
||||||
|
|
||||||
let app_data = web::Data::new(AppState {
|
let app_data = web::Data::new(AppState {
|
||||||
storage: {
|
storage: {
|
||||||
let storage = get_storage_from_env();
|
let storage = get_storage_from_env();
|
||||||
tracing::info!("storage: {storage}");
|
tracing::info!("storage: {storage}");
|
||||||
storage
|
storage
|
||||||
},
|
},
|
||||||
auth: {
|
auth: {
|
||||||
let auth = get_auth_from_env(&config);
|
let auth = get_auth_from_env(&config);
|
||||||
tracing::info!("auth: {auth}");
|
tracing::info!("auth: {auth}");
|
||||||
auth
|
auth
|
||||||
},
|
},
|
||||||
source: Arc::new(tokio::sync::RwLock::new(source)),
|
source: tokio::sync::Mutex::new(source),
|
||||||
project,
|
project,
|
||||||
|
|
||||||
search_reader,
|
search_reader,
|
||||||
search_writer: std::sync::Mutex::new(search_writer),
|
search_writer: std::sync::Mutex::new(search_writer),
|
||||||
query_parser,
|
query_parser,
|
||||||
});
|
});
|
||||||
|
|
||||||
let publish_governor_config = GovernorConfigBuilder::default()
|
let publish_governor_config = GovernorConfigBuilder::default()
|
||||||
.key_extractor(UserIdExtractor)
|
.key_extractor(UserIdExtractor)
|
||||||
.burst_size(12)
|
.burst_size(12)
|
||||||
.seconds_per_request(60)
|
.seconds_per_request(60)
|
||||||
.use_headers()
|
.use_headers()
|
||||||
.finish()
|
.finish()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
let publish_payload_config = PayloadConfig::new(config.max_archive_size);
|
HttpServer::new(move || {
|
||||||
|
App::new()
|
||||||
HttpServer::new(move || {
|
.wrap(sentry_actix::Sentry::with_transaction())
|
||||||
App::new()
|
.wrap(NormalizePath::new(TrailingSlash::Trim))
|
||||||
.wrap(sentry_actix::Sentry::with_transaction())
|
.wrap(Cors::permissive())
|
||||||
.wrap(NormalizePath::new(TrailingSlash::Trim))
|
.wrap(tracing_actix_web::TracingLogger::default())
|
||||||
.wrap(Cors::permissive())
|
.wrap(Compress::default())
|
||||||
.wrap(tracing_actix_web::TracingLogger::default())
|
.app_data(app_data.clone())
|
||||||
.wrap(Compress::default())
|
.route(
|
||||||
.app_data(app_data.clone())
|
"/",
|
||||||
.route(
|
web::get().to(|| async {
|
||||||
"/",
|
concat!(env!("CARGO_PKG_NAME"), "/", env!("CARGO_PKG_VERSION"))
|
||||||
web::get().to(|| async {
|
}),
|
||||||
concat!(env!("CARGO_PKG_NAME"), "/", env!("CARGO_PKG_VERSION"))
|
)
|
||||||
}),
|
.service(
|
||||||
)
|
web::scope("/v0")
|
||||||
.service(
|
.route(
|
||||||
web::scope("/v0")
|
"/search",
|
||||||
.route(
|
web::get()
|
||||||
"/search",
|
.to(endpoints::search::search_packages)
|
||||||
web::get()
|
.wrap(from_fn(auth::read_mw)),
|
||||||
.to(endpoints::search::search_packages)
|
)
|
||||||
.wrap(from_fn(auth::read_mw)),
|
.route(
|
||||||
)
|
"/packages/{name}",
|
||||||
.route(
|
web::get()
|
||||||
"/packages/{name}",
|
.to(endpoints::package_versions::get_package_versions)
|
||||||
web::get()
|
.wrap(from_fn(auth::read_mw)),
|
||||||
.to(endpoints::package_versions::get_package_versions_v0)
|
)
|
||||||
.wrap(from_fn(auth::read_mw)),
|
.route(
|
||||||
)
|
"/packages/{name}/{version}/{target}",
|
||||||
.route(
|
web::get()
|
||||||
"/packages/{name}/{version}/{target}",
|
.to(endpoints::package_version::get_package_version)
|
||||||
web::get()
|
.wrap(from_fn(auth::read_mw)),
|
||||||
.to(endpoints::package_version::get_package_version_v0)
|
)
|
||||||
.wrap(from_fn(auth::read_mw)),
|
.service(
|
||||||
)
|
web::scope("/packages")
|
||||||
.service(
|
.app_data(PayloadConfig::new(config.max_archive_size))
|
||||||
web::scope("/packages")
|
.route(
|
||||||
.app_data(publish_payload_config.clone())
|
"",
|
||||||
.route(
|
web::post()
|
||||||
"",
|
.to(endpoints::publish_version::publish_package)
|
||||||
web::post()
|
.wrap(Governor::new(&publish_governor_config))
|
||||||
.to(endpoints::publish_version::publish_package)
|
.wrap(from_fn(auth::write_mw)),
|
||||||
.wrap(Governor::new(&publish_governor_config))
|
),
|
||||||
.wrap(from_fn(auth::write_mw)),
|
),
|
||||||
),
|
)
|
||||||
),
|
})
|
||||||
)
|
.bind((address, port))?
|
||||||
.service(
|
.run()
|
||||||
web::scope("/v1")
|
.await
|
||||||
.route(
|
|
||||||
"/search",
|
|
||||||
web::get()
|
|
||||||
.to(endpoints::search::search_packages)
|
|
||||||
.wrap(from_fn(auth::read_mw)),
|
|
||||||
)
|
|
||||||
.route(
|
|
||||||
"/packages/{name}",
|
|
||||||
web::get()
|
|
||||||
.to(endpoints::package_versions::get_package_versions)
|
|
||||||
.wrap(from_fn(auth::read_mw)),
|
|
||||||
)
|
|
||||||
.service(
|
|
||||||
web::resource("/packages/{name}/deprecate")
|
|
||||||
.put(endpoints::deprecate_version::deprecate_package_version)
|
|
||||||
.delete(endpoints::deprecate_version::deprecate_package_version)
|
|
||||||
.wrap(from_fn(auth::write_mw)),
|
|
||||||
)
|
|
||||||
.route(
|
|
||||||
"/packages/{name}/{version}/{target}",
|
|
||||||
web::get()
|
|
||||||
.to(endpoints::package_version::get_package_version)
|
|
||||||
.wrap(from_fn(auth::read_mw)),
|
|
||||||
)
|
|
||||||
.route(
|
|
||||||
"/packages/{name}/{version}/{target}/archive",
|
|
||||||
web::get()
|
|
||||||
.to(endpoints::package_archive::get_package_archive)
|
|
||||||
.wrap(from_fn(auth::read_mw)),
|
|
||||||
)
|
|
||||||
.route(
|
|
||||||
"/packages/{name}/{version}/{target}/doc",
|
|
||||||
web::get()
|
|
||||||
.to(endpoints::package_doc::get_package_doc)
|
|
||||||
.wrap(from_fn(auth::read_mw)),
|
|
||||||
)
|
|
||||||
.route(
|
|
||||||
"/packages/{name}/{version}/{target}/readme",
|
|
||||||
web::get()
|
|
||||||
.to(endpoints::package_readme::get_package_readme)
|
|
||||||
.wrap(from_fn(auth::read_mw)),
|
|
||||||
)
|
|
||||||
.service(
|
|
||||||
web::resource("/packages/{name}/{version}/{target}/yank")
|
|
||||||
.put(endpoints::yank_version::yank_package_version)
|
|
||||||
.delete(endpoints::yank_version::yank_package_version)
|
|
||||||
.wrap(from_fn(auth::write_mw)),
|
|
||||||
)
|
|
||||||
.service(
|
|
||||||
web::scope("/packages")
|
|
||||||
.app_data(publish_payload_config.clone())
|
|
||||||
.route(
|
|
||||||
"",
|
|
||||||
web::post()
|
|
||||||
.to(endpoints::publish_version::publish_package)
|
|
||||||
.wrap(Governor::new(&publish_governor_config))
|
|
||||||
.wrap(from_fn(auth::write_mw)),
|
|
||||||
),
|
|
||||||
),
|
|
||||||
)
|
|
||||||
})
|
|
||||||
.bind((address, port))?
|
|
||||||
.run()
|
|
||||||
.await
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// can't use #[actix_web::main] because of Sentry:
|
// can't use #[actix_web::main] because of Sentry:
|
||||||
// "Note: Macros like #[tokio::main] and #[actix_web::main] are not supported. The Sentry client must be initialized before the async runtime is started so that all threads are correctly connected to the Hub."
|
// "Note: Macros like #[tokio::main] and #[actix_web::main] are not supported. The Sentry client must be initialized before the async runtime is started so that all threads are correctly connected to the Hub."
|
||||||
// https://docs.sentry.io/platforms/rust/guides/actix-web/
|
// https://docs.sentry.io/platforms/rust/guides/actix-web/
|
||||||
fn main() -> std::io::Result<()> {
|
fn main() -> std::io::Result<()> {
|
||||||
let _ = dotenvy::dotenv();
|
let _ = dotenvy::dotenv();
|
||||||
|
|
||||||
let tracing_env_filter = EnvFilter::builder()
|
let tracing_env_filter = EnvFilter::builder()
|
||||||
.with_default_directive(LevelFilter::INFO.into())
|
.with_default_directive(LevelFilter::INFO.into())
|
||||||
.from_env_lossy()
|
.from_env_lossy()
|
||||||
.add_directive("reqwest=info".parse().unwrap())
|
.add_directive("reqwest=info".parse().unwrap())
|
||||||
.add_directive("rustls=info".parse().unwrap())
|
.add_directive("rustls=info".parse().unwrap())
|
||||||
.add_directive("tokio_util=info".parse().unwrap())
|
.add_directive("tokio_util=info".parse().unwrap())
|
||||||
.add_directive("goblin=info".parse().unwrap())
|
.add_directive("goblin=info".parse().unwrap())
|
||||||
.add_directive("tower=info".parse().unwrap())
|
.add_directive("tower=info".parse().unwrap())
|
||||||
.add_directive("hyper=info".parse().unwrap())
|
.add_directive("hyper=info".parse().unwrap())
|
||||||
.add_directive("h2=info".parse().unwrap());
|
.add_directive("h2=info".parse().unwrap());
|
||||||
|
|
||||||
tracing_subscriber::registry()
|
tracing_subscriber::registry()
|
||||||
.with(tracing_env_filter)
|
.with(tracing_env_filter)
|
||||||
.with(
|
.with(
|
||||||
tracing_subscriber::fmt::layer()
|
tracing_subscriber::fmt::layer()
|
||||||
.compact()
|
.compact()
|
||||||
.with_span_events(FmtSpan::NEW | FmtSpan::CLOSE),
|
.with_span_events(FmtSpan::NEW | FmtSpan::CLOSE),
|
||||||
)
|
)
|
||||||
.with(sentry::integrations::tracing::layer())
|
.with(sentry::integrations::tracing::layer())
|
||||||
.init();
|
.init();
|
||||||
|
|
||||||
let guard = sentry::init(sentry::ClientOptions {
|
let guard = sentry::init(sentry::ClientOptions {
|
||||||
release: sentry::release_name!(),
|
release: sentry::release_name!(),
|
||||||
dsn: benv!(parse "SENTRY_DSN").ok(),
|
dsn: benv!(parse "SENTRY_DSN").ok(),
|
||||||
session_mode: sentry::SessionMode::Request,
|
session_mode: sentry::SessionMode::Request,
|
||||||
traces_sample_rate: 1.0,
|
traces_sample_rate: 1.0,
|
||||||
debug: true,
|
debug: true,
|
||||||
..Default::default()
|
..Default::default()
|
||||||
});
|
});
|
||||||
|
|
||||||
if guard.is_enabled() {
|
if guard.is_enabled() {
|
||||||
std::env::set_var("RUST_BACKTRACE", "full");
|
std::env::set_var("RUST_BACKTRACE", "full");
|
||||||
tracing::info!("sentry initialized");
|
tracing::info!("sentry initialized");
|
||||||
} else {
|
} else {
|
||||||
tracing::info!("sentry **NOT** initialized");
|
tracing::info!("sentry **NOT** initialized");
|
||||||
}
|
}
|
||||||
|
|
||||||
System::new().block_on(run())
|
System::new().block_on(run())
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,267 +1,61 @@
|
||||||
use crate::AppState;
|
use chrono::{DateTime, Utc};
|
||||||
use pesde::{
|
use pesde::manifest::target::{Target, TargetKind};
|
||||||
manifest::{
|
|
||||||
target::{Target, TargetKind},
|
|
||||||
Alias, DependencyType,
|
|
||||||
},
|
|
||||||
names::PackageName,
|
|
||||||
source::{
|
|
||||||
git_index::{read_file, root_tree, GitBasedSource},
|
|
||||||
ids::VersionId,
|
|
||||||
pesde::{IndexFile, IndexFileEntry, PesdePackageSource, ScopeInfo, SCOPE_INFO_FILE},
|
|
||||||
specifiers::DependencySpecifiers,
|
|
||||||
},
|
|
||||||
};
|
|
||||||
use semver::Version;
|
|
||||||
use serde::Serialize;
|
use serde::Serialize;
|
||||||
use std::collections::{BTreeMap, BTreeSet};
|
use std::collections::BTreeSet;
|
||||||
use tokio::task::spawn_blocking;
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Eq, PartialEq)]
|
|
||||||
struct TargetInfoInner {
|
|
||||||
lib: bool,
|
|
||||||
bin: bool,
|
|
||||||
#[serde(skip_serializing_if = "BTreeSet::is_empty")]
|
|
||||||
scripts: BTreeSet<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl TargetInfoInner {
|
|
||||||
fn new(target: &Target) -> Self {
|
|
||||||
TargetInfoInner {
|
|
||||||
lib: target.lib_path().is_some(),
|
|
||||||
bin: target.bin_path().is_some(),
|
|
||||||
scripts: target
|
|
||||||
.scripts()
|
|
||||||
.map(|scripts| scripts.keys().cloned().collect())
|
|
||||||
.unwrap_or_default(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Eq, PartialEq)]
|
#[derive(Debug, Serialize, Eq, PartialEq)]
|
||||||
pub struct TargetInfo {
|
pub struct TargetInfo {
|
||||||
kind: TargetKind,
|
kind: TargetKind,
|
||||||
#[serde(skip_serializing_if = "std::ops::Not::not")]
|
lib: bool,
|
||||||
yanked: bool,
|
bin: bool,
|
||||||
#[serde(flatten)]
|
#[serde(skip_serializing_if = "BTreeSet::is_empty")]
|
||||||
inner: TargetInfoInner,
|
scripts: BTreeSet<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl TargetInfo {
|
impl From<Target> for TargetInfo {
|
||||||
fn new(target: &Target, yanked: bool) -> Self {
|
fn from(target: Target) -> Self {
|
||||||
TargetInfo {
|
(&target).into()
|
||||||
kind: target.kind(),
|
}
|
||||||
yanked,
|
}
|
||||||
inner: TargetInfoInner::new(target),
|
|
||||||
}
|
impl From<&Target> for TargetInfo {
|
||||||
}
|
fn from(target: &Target) -> Self {
|
||||||
|
TargetInfo {
|
||||||
|
kind: target.kind(),
|
||||||
|
lib: target.lib_path().is_some(),
|
||||||
|
bin: target.bin_path().is_some(),
|
||||||
|
scripts: target
|
||||||
|
.scripts()
|
||||||
|
.map(|scripts| scripts.keys().cloned().collect())
|
||||||
|
.unwrap_or_default(),
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Ord for TargetInfo {
|
impl Ord for TargetInfo {
|
||||||
fn cmp(&self, other: &Self) -> std::cmp::Ordering {
|
fn cmp(&self, other: &Self) -> std::cmp::Ordering {
|
||||||
self.kind.cmp(&other.kind)
|
self.kind.cmp(&other.kind)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl PartialOrd for TargetInfo {
|
impl PartialOrd for TargetInfo {
|
||||||
fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
|
fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
|
||||||
Some(self.cmp(other))
|
Some(self.cmp(other))
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Ord, PartialOrd, Eq, PartialEq)]
|
|
||||||
#[serde(untagged)]
|
|
||||||
pub enum RegistryDocEntryKind {
|
|
||||||
Page {
|
|
||||||
name: String,
|
|
||||||
},
|
|
||||||
Category {
|
|
||||||
#[serde(default, skip_serializing_if = "BTreeSet::is_empty")]
|
|
||||||
items: BTreeSet<RegistryDocEntry>,
|
|
||||||
#[serde(default, skip_serializing_if = "std::ops::Not::not")]
|
|
||||||
collapsed: bool,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Ord, PartialOrd, Eq, PartialEq)]
|
|
||||||
pub struct RegistryDocEntry {
|
|
||||||
label: String,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
position: Option<usize>,
|
|
||||||
#[serde(flatten)]
|
|
||||||
kind: RegistryDocEntryKind,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<pesde::source::pesde::DocEntry> for RegistryDocEntry {
|
|
||||||
fn from(entry: pesde::source::pesde::DocEntry) -> Self {
|
|
||||||
Self {
|
|
||||||
label: entry.label,
|
|
||||||
position: entry.position,
|
|
||||||
kind: match entry.kind {
|
|
||||||
pesde::source::pesde::DocEntryKind::Page { name, .. } => {
|
|
||||||
RegistryDocEntryKind::Page { name }
|
|
||||||
}
|
|
||||||
pesde::source::pesde::DocEntryKind::Category { items, collapsed } => {
|
|
||||||
RegistryDocEntryKind::Category {
|
|
||||||
items: items.into_iter().map(Into::into).collect(),
|
|
||||||
collapsed,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
|
||||||
pub struct PackageResponseInner {
|
|
||||||
published_at: jiff::Timestamp,
|
|
||||||
#[serde(skip_serializing_if = "String::is_empty")]
|
|
||||||
license: String,
|
|
||||||
#[serde(skip_serializing_if = "Vec::is_empty")]
|
|
||||||
authors: Vec<String>,
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
repository: Option<String>,
|
|
||||||
#[serde(skip_serializing_if = "BTreeSet::is_empty")]
|
|
||||||
docs: BTreeSet<RegistryDocEntry>,
|
|
||||||
#[serde(skip_serializing_if = "BTreeMap::is_empty")]
|
|
||||||
dependencies: BTreeMap<Alias, (DependencySpecifiers, DependencyType)>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl PackageResponseInner {
|
|
||||||
pub fn new(entry: &IndexFileEntry) -> Self {
|
|
||||||
PackageResponseInner {
|
|
||||||
published_at: entry.published_at,
|
|
||||||
license: entry.license.clone().unwrap_or_default(),
|
|
||||||
authors: entry.authors.clone(),
|
|
||||||
repository: entry.repository.clone().map(|url| url.to_string()),
|
|
||||||
docs: entry.docs.iter().cloned().map(Into::into).collect(),
|
|
||||||
dependencies: entry.dependencies.clone(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
#[derive(Debug, Serialize)]
|
||||||
pub struct PackageResponse {
|
pub struct PackageResponse {
|
||||||
name: String,
|
pub name: String,
|
||||||
version: String,
|
pub version: String,
|
||||||
targets: BTreeSet<TargetInfo>,
|
pub targets: BTreeSet<TargetInfo>,
|
||||||
#[serde(skip_serializing_if = "String::is_empty")]
|
#[serde(skip_serializing_if = "String::is_empty")]
|
||||||
description: String,
|
pub description: String,
|
||||||
#[serde(skip_serializing_if = "String::is_empty")]
|
pub published_at: DateTime<Utc>,
|
||||||
deprecated: String,
|
#[serde(skip_serializing_if = "String::is_empty")]
|
||||||
#[serde(flatten)]
|
pub license: String,
|
||||||
inner: PackageResponseInner,
|
#[serde(skip_serializing_if = "Vec::is_empty")]
|
||||||
}
|
pub authors: Vec<String>,
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
impl PackageResponse {
|
pub repository: Option<String>,
|
||||||
pub fn new(name: &PackageName, version_id: &VersionId, file: &IndexFile) -> Self {
|
|
||||||
let entry = file.entries.get(version_id).unwrap();
|
|
||||||
|
|
||||||
PackageResponse {
|
|
||||||
name: name.to_string(),
|
|
||||||
version: version_id.version().to_string(),
|
|
||||||
targets: file
|
|
||||||
.entries
|
|
||||||
.iter()
|
|
||||||
.filter(|(ver, _)| ver.version() == version_id.version())
|
|
||||||
.map(|(_, entry)| TargetInfo::new(&entry.target, entry.yanked))
|
|
||||||
.collect(),
|
|
||||||
description: entry.description.clone().unwrap_or_default(),
|
|
||||||
deprecated: file.meta.deprecated.clone(),
|
|
||||||
inner: PackageResponseInner::new(entry),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
|
||||||
struct PackageVersionsResponseVersionInner {
|
|
||||||
target: TargetInfoInner,
|
|
||||||
#[serde(skip_serializing_if = "std::ops::Not::not")]
|
|
||||||
yanked: bool,
|
|
||||||
#[serde(flatten)]
|
|
||||||
inner: PackageResponseInner,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Default)]
|
|
||||||
struct PackageVersionsResponseVersion {
|
|
||||||
#[serde(skip_serializing_if = "String::is_empty")]
|
|
||||||
description: String,
|
|
||||||
targets: BTreeMap<TargetKind, PackageVersionsResponseVersionInner>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Serialize)]
|
|
||||||
pub struct PackageVersionsResponse {
|
|
||||||
name: String,
|
|
||||||
#[serde(skip_serializing_if = "String::is_empty")]
|
|
||||||
deprecated: String,
|
|
||||||
versions: BTreeMap<Version, PackageVersionsResponseVersion>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl PackageVersionsResponse {
|
|
||||||
pub fn new(name: &PackageName, file: &IndexFile) -> Self {
|
|
||||||
let mut versions = BTreeMap::<Version, PackageVersionsResponseVersion>::new();
|
|
||||||
|
|
||||||
for (v_id, entry) in file.entries.iter() {
|
|
||||||
let versions_resp = versions.entry(v_id.version().clone()).or_default();
|
|
||||||
|
|
||||||
versions_resp.description = entry.description.clone().unwrap_or_default();
|
|
||||||
versions_resp.targets.insert(
|
|
||||||
entry.target.kind(),
|
|
||||||
PackageVersionsResponseVersionInner {
|
|
||||||
target: TargetInfoInner::new(&entry.target),
|
|
||||||
yanked: entry.yanked,
|
|
||||||
inner: PackageResponseInner::new(entry),
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
PackageVersionsResponse {
|
|
||||||
name: name.to_string(),
|
|
||||||
deprecated: file.meta.deprecated.clone(),
|
|
||||||
versions,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn read_package(
|
|
||||||
app_state: &AppState,
|
|
||||||
package: &PackageName,
|
|
||||||
source: &PesdePackageSource,
|
|
||||||
) -> Result<Option<IndexFile>, crate::error::RegistryError> {
|
|
||||||
let path = source.path(&app_state.project);
|
|
||||||
let package = package.clone();
|
|
||||||
spawn_blocking(move || {
|
|
||||||
let (scope, name) = package.as_str();
|
|
||||||
let repo = gix::open(path)?;
|
|
||||||
let tree = root_tree(&repo)?;
|
|
||||||
|
|
||||||
let Some(versions) = read_file(&tree, [scope, name])? else {
|
|
||||||
return Ok(None);
|
|
||||||
};
|
|
||||||
|
|
||||||
toml::de::from_str(&versions).map_err(Into::into)
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.unwrap()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn read_scope_info(
|
|
||||||
app_state: &AppState,
|
|
||||||
scope: &str,
|
|
||||||
source: &PesdePackageSource,
|
|
||||||
) -> Result<Option<ScopeInfo>, crate::error::RegistryError> {
|
|
||||||
let path = source.path(&app_state.project);
|
|
||||||
let scope = scope.to_string();
|
|
||||||
spawn_blocking(move || {
|
|
||||||
let repo = gix::open(path)?;
|
|
||||||
let tree = root_tree(&repo)?;
|
|
||||||
|
|
||||||
let Some(versions) = read_file(&tree, [&*scope, SCOPE_INFO_FILE])? else {
|
|
||||||
return Ok(None);
|
|
||||||
};
|
|
||||||
|
|
||||||
toml::de::from_str(&versions).map_err(Into::into)
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.unwrap()
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,99 +0,0 @@
|
||||||
use pesde::{
|
|
||||||
manifest::target::TargetKind,
|
|
||||||
source::{ids::VersionId, pesde::IndexFile},
|
|
||||||
};
|
|
||||||
use semver::Version;
|
|
||||||
use serde::{Deserialize, Deserializer};
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
pub enum LatestOrSpecificVersion {
|
|
||||||
Latest,
|
|
||||||
Specific(Version),
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'de> Deserialize<'de> for LatestOrSpecificVersion {
|
|
||||||
fn deserialize<D>(deserializer: D) -> Result<LatestOrSpecificVersion, D::Error>
|
|
||||||
where
|
|
||||||
D: Deserializer<'de>,
|
|
||||||
{
|
|
||||||
let s = String::deserialize(deserializer)?;
|
|
||||||
if s.eq_ignore_ascii_case("latest") {
|
|
||||||
return Ok(LatestOrSpecificVersion::Latest);
|
|
||||||
}
|
|
||||||
|
|
||||||
s.parse()
|
|
||||||
.map(LatestOrSpecificVersion::Specific)
|
|
||||||
.map_err(serde::de::Error::custom)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
pub enum AnyOrSpecificTarget {
|
|
||||||
Any,
|
|
||||||
Specific(TargetKind),
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'de> Deserialize<'de> for AnyOrSpecificTarget {
|
|
||||||
fn deserialize<D>(deserializer: D) -> Result<AnyOrSpecificTarget, D::Error>
|
|
||||||
where
|
|
||||||
D: Deserializer<'de>,
|
|
||||||
{
|
|
||||||
let s = String::deserialize(deserializer)?;
|
|
||||||
if s.eq_ignore_ascii_case("any") {
|
|
||||||
return Ok(AnyOrSpecificTarget::Any);
|
|
||||||
}
|
|
||||||
|
|
||||||
s.parse()
|
|
||||||
.map(AnyOrSpecificTarget::Specific)
|
|
||||||
.map_err(serde::de::Error::custom)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn resolve_version_and_target(
|
|
||||||
file: &IndexFile,
|
|
||||||
version: LatestOrSpecificVersion,
|
|
||||||
target: AnyOrSpecificTarget,
|
|
||||||
) -> Option<&VersionId> {
|
|
||||||
let version = match version {
|
|
||||||
LatestOrSpecificVersion::Latest => match file.entries.keys().map(|k| k.version()).max() {
|
|
||||||
Some(latest) => latest.clone(),
|
|
||||||
None => return None,
|
|
||||||
},
|
|
||||||
LatestOrSpecificVersion::Specific(version) => version,
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut versions = file
|
|
||||||
.entries
|
|
||||||
.iter()
|
|
||||||
.filter(|(v_id, _)| *v_id.version() == version);
|
|
||||||
|
|
||||||
match target {
|
|
||||||
AnyOrSpecificTarget::Any => versions.min_by_key(|(v_id, _)| v_id.target()),
|
|
||||||
AnyOrSpecificTarget::Specific(kind) => {
|
|
||||||
versions.find(|(_, entry)| entry.target.kind() == kind)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
.map(|(v_id, _)| v_id)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
pub enum AllOrSpecificTarget {
|
|
||||||
All,
|
|
||||||
Specific(TargetKind),
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'de> Deserialize<'de> for AllOrSpecificTarget {
|
|
||||||
fn deserialize<D>(deserializer: D) -> Result<AllOrSpecificTarget, D::Error>
|
|
||||||
where
|
|
||||||
D: Deserializer<'de>,
|
|
||||||
{
|
|
||||||
let s = String::deserialize(deserializer)?;
|
|
||||||
if s.eq_ignore_ascii_case("all") {
|
|
||||||
return Ok(AllOrSpecificTarget::All);
|
|
||||||
}
|
|
||||||
|
|
||||||
s.parse()
|
|
||||||
.map(AllOrSpecificTarget::Specific)
|
|
||||||
.map_err(serde::de::Error::custom)
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -2,188 +2,147 @@ use crate::AppState;
|
||||||
use async_stream::stream;
|
use async_stream::stream;
|
||||||
use futures::{Stream, StreamExt};
|
use futures::{Stream, StreamExt};
|
||||||
use pesde::{
|
use pesde::{
|
||||||
names::PackageName,
|
names::PackageName,
|
||||||
source::{
|
source::{
|
||||||
git_index::{root_tree, GitBasedSource},
|
git_index::{root_tree, GitBasedSource},
|
||||||
ids::VersionId,
|
pesde::{IndexFile, IndexFileEntry, PesdePackageSource, SCOPE_INFO_FILE},
|
||||||
pesde::{IndexFile, IndexFileEntry, PesdePackageSource, SCOPE_INFO_FILE},
|
},
|
||||||
},
|
Project,
|
||||||
Project,
|
|
||||||
};
|
};
|
||||||
use tantivy::{
|
use tantivy::{
|
||||||
doc,
|
doc,
|
||||||
query::QueryParser,
|
query::QueryParser,
|
||||||
schema::{IndexRecordOption, TextFieldIndexing, TextOptions, FAST, STORED, STRING},
|
schema::{IndexRecordOption, TextFieldIndexing, TextOptions, FAST, STORED, STRING},
|
||||||
tokenizer::TextAnalyzer,
|
tokenizer::TextAnalyzer,
|
||||||
DateTime, IndexReader, IndexWriter, Term,
|
DateTime, IndexReader, IndexWriter, Term,
|
||||||
};
|
};
|
||||||
use tokio::pin;
|
use tokio::pin;
|
||||||
|
|
||||||
async fn all_packages(
|
pub async fn all_packages(
|
||||||
source: &PesdePackageSource,
|
source: &PesdePackageSource,
|
||||||
project: &Project,
|
project: &Project,
|
||||||
) -> impl Stream<Item = (PackageName, IndexFile)> {
|
) -> impl Stream<Item = (PackageName, IndexFile)> {
|
||||||
let path = source.path(project);
|
let path = source.path(project);
|
||||||
|
|
||||||
stream! {
|
stream! {
|
||||||
let repo = gix::open(&path).expect("failed to open index");
|
let repo = gix::open(&path).expect("failed to open index");
|
||||||
let tree = root_tree(&repo).expect("failed to get root tree");
|
let tree = root_tree(&repo).expect("failed to get root tree");
|
||||||
|
|
||||||
for entry in tree.iter() {
|
for entry in tree.iter() {
|
||||||
let entry = entry.expect("failed to read entry");
|
let entry = entry.expect("failed to read entry");
|
||||||
let object = entry.object().expect("failed to get object");
|
let object = entry.object().expect("failed to get object");
|
||||||
|
|
||||||
// directories will be trees, and files will be blobs
|
// directories will be trees, and files will be blobs
|
||||||
if !matches!(object.kind, gix::object::Kind::Tree) {
|
if !matches!(object.kind, gix::object::Kind::Tree) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
let package_scope = entry.filename().to_string();
|
let package_scope = entry.filename().to_string();
|
||||||
|
|
||||||
for inner_entry in object.into_tree().iter() {
|
for inner_entry in object.into_tree().iter() {
|
||||||
let inner_entry = inner_entry.expect("failed to read inner entry");
|
let inner_entry = inner_entry.expect("failed to read inner entry");
|
||||||
let object = inner_entry.object().expect("failed to get object");
|
let object = inner_entry.object().expect("failed to get object");
|
||||||
|
|
||||||
if !matches!(object.kind, gix::object::Kind::Blob) {
|
if !matches!(object.kind, gix::object::Kind::Blob) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
let package_name = inner_entry.filename().to_string();
|
let package_name = inner_entry.filename().to_string();
|
||||||
|
|
||||||
if package_name == SCOPE_INFO_FILE {
|
if package_name == SCOPE_INFO_FILE {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
let blob = object.into_blob();
|
let blob = object.into_blob();
|
||||||
let string = String::from_utf8(blob.data.clone()).expect("failed to parse utf8");
|
let string = String::from_utf8(blob.data.clone()).expect("failed to parse utf8");
|
||||||
|
|
||||||
let file: IndexFile = toml::from_str(&string).expect("failed to parse index file");
|
let file: IndexFile = toml::from_str(&string).expect("failed to parse index file");
|
||||||
|
|
||||||
// if this panics, it's an issue with the index.
|
// if this panics, it's an issue with the index.
|
||||||
let name = format!("{package_scope}/{package_name}").parse().unwrap();
|
let name = format!("{package_scope}/{package_name}").parse().unwrap();
|
||||||
|
|
||||||
yield (name, file);
|
yield (name, file);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
pub fn find_max_searchable(file: &IndexFile) -> Option<(&VersionId, &IndexFileEntry)> {
|
|
||||||
file.entries
|
|
||||||
.iter()
|
|
||||||
.filter(|(_, entry)| !entry.yanked)
|
|
||||||
.max_by(|(v_id_a, entry_a), (v_id_b, entry_b)| {
|
|
||||||
v_id_a
|
|
||||||
.version()
|
|
||||||
.cmp(v_id_b.version())
|
|
||||||
.then(entry_a.published_at.cmp(&entry_b.published_at))
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn make_search(
|
pub async fn make_search(
|
||||||
project: &Project,
|
project: &Project,
|
||||||
source: &PesdePackageSource,
|
source: &PesdePackageSource,
|
||||||
) -> (IndexReader, IndexWriter, QueryParser) {
|
) -> (IndexReader, IndexWriter, QueryParser) {
|
||||||
let mut schema_builder = tantivy::schema::SchemaBuilder::new();
|
let mut schema_builder = tantivy::schema::SchemaBuilder::new();
|
||||||
|
|
||||||
let field_options = TextOptions::default().set_indexing_options(
|
let field_options = TextOptions::default().set_indexing_options(
|
||||||
TextFieldIndexing::default()
|
TextFieldIndexing::default()
|
||||||
.set_tokenizer("ngram")
|
.set_tokenizer("ngram")
|
||||||
.set_index_option(IndexRecordOption::WithFreqsAndPositions),
|
.set_index_option(IndexRecordOption::WithFreqsAndPositions),
|
||||||
);
|
);
|
||||||
|
|
||||||
let id_field = schema_builder.add_text_field("id", STRING | STORED);
|
let id_field = schema_builder.add_text_field("id", STRING | STORED);
|
||||||
|
let scope = schema_builder.add_text_field("scope", field_options.clone());
|
||||||
|
let name = schema_builder.add_text_field("name", field_options.clone());
|
||||||
|
let description = schema_builder.add_text_field("description", field_options);
|
||||||
|
let published_at = schema_builder.add_date_field("published_at", FAST);
|
||||||
|
|
||||||
let scope = schema_builder.add_text_field("scope", field_options.clone());
|
let search_index = tantivy::Index::create_in_ram(schema_builder.build());
|
||||||
let name = schema_builder.add_text_field("name", field_options.clone());
|
search_index.tokenizers().register(
|
||||||
let description = schema_builder.add_text_field("description", field_options);
|
"ngram",
|
||||||
let published_at = schema_builder.add_date_field("published_at", FAST);
|
TextAnalyzer::builder(tantivy::tokenizer::NgramTokenizer::all_ngrams(1, 12).unwrap())
|
||||||
|
.filter(tantivy::tokenizer::LowerCaser)
|
||||||
|
.build(),
|
||||||
|
);
|
||||||
|
|
||||||
let search_index = tantivy::Index::create_in_ram(schema_builder.build());
|
let search_reader = search_index
|
||||||
search_index.tokenizers().register(
|
.reader_builder()
|
||||||
"ngram",
|
.reload_policy(tantivy::ReloadPolicy::Manual)
|
||||||
TextAnalyzer::builder(tantivy::tokenizer::NgramTokenizer::all_ngrams(1, 12).unwrap())
|
.try_into()
|
||||||
.filter(tantivy::tokenizer::LowerCaser)
|
.unwrap();
|
||||||
.build(),
|
let mut search_writer = search_index.writer(50_000_000).unwrap();
|
||||||
);
|
|
||||||
|
|
||||||
let search_reader = search_index
|
let stream = all_packages(source, project).await;
|
||||||
.reader_builder()
|
pin!(stream);
|
||||||
.reload_policy(tantivy::ReloadPolicy::Manual)
|
|
||||||
.try_into()
|
|
||||||
.unwrap();
|
|
||||||
let mut search_writer = search_index.writer(50_000_000).unwrap();
|
|
||||||
|
|
||||||
let stream = all_packages(source, project).await;
|
while let Some((pkg_name, mut file)) = stream.next().await {
|
||||||
pin!(stream);
|
let Some((_, latest_entry)) = file.entries.pop_last() else {
|
||||||
|
tracing::error!("no versions found for {pkg_name}");
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
|
||||||
while let Some((pkg_name, file)) = stream.next().await {
|
search_writer.add_document(doc!(
|
||||||
if !file.meta.deprecated.is_empty() {
|
id_field => pkg_name.to_string(),
|
||||||
continue;
|
scope => pkg_name.as_str().0,
|
||||||
}
|
name => pkg_name.as_str().1,
|
||||||
|
description => latest_entry.description.unwrap_or_default(),
|
||||||
|
published_at => DateTime::from_timestamp_secs(latest_entry.published_at.timestamp()),
|
||||||
|
)).unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
let Some((_, latest_entry)) = find_max_searchable(&file) else {
|
search_writer.commit().unwrap();
|
||||||
continue;
|
search_reader.reload().unwrap();
|
||||||
};
|
|
||||||
|
|
||||||
search_writer
|
let mut query_parser = QueryParser::for_index(&search_index, vec![scope, name, description]);
|
||||||
.add_document(doc!(
|
query_parser.set_field_boost(scope, 2.0);
|
||||||
id_field => pkg_name.to_string(),
|
query_parser.set_field_boost(name, 3.5);
|
||||||
scope => pkg_name.scope(),
|
|
||||||
name => pkg_name.name(),
|
|
||||||
description => latest_entry.description.clone().unwrap_or_default(),
|
|
||||||
published_at => DateTime::from_timestamp_nanos(latest_entry.published_at.as_nanosecond() as i64),
|
|
||||||
))
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
search_writer.commit().unwrap();
|
(search_reader, search_writer, query_parser)
|
||||||
search_reader.reload().unwrap();
|
|
||||||
|
|
||||||
let mut query_parser = QueryParser::for_index(&search_index, vec![scope, name, description]);
|
|
||||||
query_parser.set_field_boost(scope, 2.0);
|
|
||||||
query_parser.set_field_boost(name, 3.5);
|
|
||||||
|
|
||||||
(search_reader, search_writer, query_parser)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn update_search_version(app_state: &AppState, name: &PackageName, entry: &IndexFileEntry) {
|
pub fn update_version(app_state: &AppState, name: &PackageName, entry: IndexFileEntry) {
|
||||||
let mut search_writer = app_state.search_writer.lock().unwrap();
|
let mut search_writer = app_state.search_writer.lock().unwrap();
|
||||||
let schema = search_writer.index().schema();
|
let schema = search_writer.index().schema();
|
||||||
let id_field = schema.get_field("id").unwrap();
|
let id_field = schema.get_field("id").unwrap();
|
||||||
|
|
||||||
search_writer.delete_term(Term::from_field_text(id_field, &name.to_string()));
|
search_writer.delete_term(Term::from_field_text(id_field, &name.to_string()));
|
||||||
|
|
||||||
search_writer.add_document(doc!(
|
search_writer.add_document(doc!(
|
||||||
id_field => name.to_string(),
|
id_field => name.to_string(),
|
||||||
schema.get_field("scope").unwrap() => name.scope(),
|
schema.get_field("scope").unwrap() => name.as_str().0,
|
||||||
schema.get_field("name").unwrap() => name.name(),
|
schema.get_field("name").unwrap() => name.as_str().1,
|
||||||
schema.get_field("description").unwrap() => entry.description.clone().unwrap_or_default(),
|
schema.get_field("description").unwrap() => entry.description.unwrap_or_default(),
|
||||||
schema.get_field("published_at").unwrap() => DateTime::from_timestamp_nanos(entry.published_at.as_nanosecond() as i64)
|
schema.get_field("published_at").unwrap() => DateTime::from_timestamp_secs(entry.published_at.timestamp())
|
||||||
)).unwrap();
|
)).unwrap();
|
||||||
|
|
||||||
search_writer.commit().unwrap();
|
search_writer.commit().unwrap();
|
||||||
app_state.search_reader.reload().unwrap();
|
app_state.search_reader.reload().unwrap();
|
||||||
}
|
|
||||||
|
|
||||||
pub fn search_version_changed(app_state: &AppState, name: &PackageName, file: &IndexFile) {
|
|
||||||
let entry = if file.meta.deprecated.is_empty() {
|
|
||||||
find_max_searchable(file)
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
};
|
|
||||||
|
|
||||||
let Some((_, entry)) = entry else {
|
|
||||||
let mut search_writer = app_state.search_writer.lock().unwrap();
|
|
||||||
let schema = search_writer.index().schema();
|
|
||||||
let id_field = schema.get_field("id").unwrap();
|
|
||||||
|
|
||||||
search_writer.delete_term(Term::from_field_text(id_field, &name.to_string()));
|
|
||||||
search_writer.commit().unwrap();
|
|
||||||
app_state.search_reader.reload().unwrap();
|
|
||||||
|
|
||||||
return;
|
|
||||||
};
|
|
||||||
|
|
||||||
update_search_version(app_state, name, entry);
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,131 +1,126 @@
|
||||||
use crate::{error::RegistryError, storage::StorageImpl};
|
use crate::{error::Error, storage::StorageImpl};
|
||||||
use actix_web::{
|
use actix_web::{
|
||||||
http::header::{CONTENT_ENCODING, CONTENT_LENGTH, CONTENT_TYPE},
|
http::header::{CONTENT_ENCODING, CONTENT_TYPE},
|
||||||
HttpResponse,
|
HttpResponse,
|
||||||
};
|
};
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use pesde::{names::PackageName, source::ids::VersionId};
|
use pesde::{names::PackageName, source::version_id::VersionId};
|
||||||
use std::{
|
use std::{
|
||||||
fmt::Display,
|
fmt::Display,
|
||||||
path::{Path, PathBuf},
|
path::{Path, PathBuf},
|
||||||
};
|
};
|
||||||
use tokio_util::io::ReaderStream;
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct FSStorage {
|
pub struct FSStorage {
|
||||||
pub root: PathBuf,
|
pub root: PathBuf,
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn read_file_to_response(
|
async fn read_file_to_response(path: &Path, content_type: &str) -> Result<HttpResponse, Error> {
|
||||||
path: &Path,
|
Ok(match fs::read(path).await {
|
||||||
content_type: &str,
|
Ok(contents) => HttpResponse::Ok()
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
.append_header((CONTENT_TYPE, content_type))
|
||||||
Ok(match fs::File::open(path).await {
|
.append_header((CONTENT_ENCODING, "gzip"))
|
||||||
Ok(file) => HttpResponse::Ok()
|
.body(contents),
|
||||||
.append_header((CONTENT_TYPE, content_type))
|
Err(e) if e.kind() == std::io::ErrorKind::NotFound => HttpResponse::NotFound().finish(),
|
||||||
.append_header((CONTENT_ENCODING, "gzip"))
|
Err(e) => return Err(e.into()),
|
||||||
.append_header((CONTENT_LENGTH, file.metadata().await?.len()))
|
})
|
||||||
.streaming(ReaderStream::new(file)),
|
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => HttpResponse::NotFound().finish(),
|
|
||||||
Err(e) => return Err(e.into()),
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl StorageImpl for FSStorage {
|
impl StorageImpl for FSStorage {
|
||||||
async fn store_package(
|
async fn store_package(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
contents: Vec<u8>,
|
contents: Vec<u8>,
|
||||||
) -> Result<(), RegistryError> {
|
) -> Result<(), Error> {
|
||||||
let (scope, name) = package_name.as_str();
|
let (scope, name) = package_name.as_str();
|
||||||
|
|
||||||
let path = self
|
let path = self
|
||||||
.root
|
.root
|
||||||
.join(scope)
|
.join(scope)
|
||||||
.join(name)
|
.join(name)
|
||||||
.join(version.version().to_string())
|
.join(version.version().to_string())
|
||||||
.join(version.target().to_string());
|
.join(version.target().to_string());
|
||||||
fs::create_dir_all(&path).await?;
|
fs::create_dir_all(&path).await?;
|
||||||
|
|
||||||
fs::write(path.join("pkg.tar.gz"), &contents).await?;
|
fs::write(path.join("pkg.tar.gz"), &contents).await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_package(
|
async fn get_package(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
) -> Result<HttpResponse, Error> {
|
||||||
let (scope, name) = package_name.as_str();
|
let (scope, name) = package_name.as_str();
|
||||||
|
|
||||||
let path = self
|
let path = self
|
||||||
.root
|
.root
|
||||||
.join(scope)
|
.join(scope)
|
||||||
.join(name)
|
.join(name)
|
||||||
.join(version.version().to_string())
|
.join(version.version().to_string())
|
||||||
.join(version.target().to_string());
|
.join(version.target().to_string());
|
||||||
|
|
||||||
read_file_to_response(&path.join("pkg.tar.gz"), "application/gzip").await
|
read_file_to_response(&path.join("pkg.tar.gz"), "application/gzip").await
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn store_readme(
|
async fn store_readme(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
contents: Vec<u8>,
|
contents: Vec<u8>,
|
||||||
) -> Result<(), RegistryError> {
|
) -> Result<(), Error> {
|
||||||
let (scope, name) = package_name.as_str();
|
let (scope, name) = package_name.as_str();
|
||||||
|
|
||||||
let path = self
|
let path = self
|
||||||
.root
|
.root
|
||||||
.join(scope)
|
.join(scope)
|
||||||
.join(name)
|
.join(name)
|
||||||
.join(version.version().to_string())
|
.join(version.version().to_string())
|
||||||
.join(version.target().to_string());
|
.join(version.target().to_string());
|
||||||
fs::create_dir_all(&path).await?;
|
fs::create_dir_all(&path).await?;
|
||||||
|
|
||||||
fs::write(path.join("readme.gz"), &contents).await?;
|
fs::write(path.join("readme.gz"), &contents).await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_readme(
|
async fn get_readme(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
) -> Result<HttpResponse, Error> {
|
||||||
let (scope, name) = package_name.as_str();
|
let (scope, name) = package_name.as_str();
|
||||||
|
|
||||||
let path = self
|
let path = self
|
||||||
.root
|
.root
|
||||||
.join(scope)
|
.join(scope)
|
||||||
.join(name)
|
.join(name)
|
||||||
.join(version.version().to_string())
|
.join(version.version().to_string())
|
||||||
.join(version.target().to_string());
|
.join(version.target().to_string());
|
||||||
|
|
||||||
read_file_to_response(&path.join("readme.gz"), "text/plain").await
|
read_file_to_response(&path.join("readme.gz"), "text/plain").await
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn store_doc(&self, doc_hash: String, contents: Vec<u8>) -> Result<(), RegistryError> {
|
async fn store_doc(&self, doc_hash: String, contents: Vec<u8>) -> Result<(), Error> {
|
||||||
let path = self.root.join("Doc");
|
let path = self.root.join("Doc");
|
||||||
fs::create_dir_all(&path).await?;
|
fs::create_dir_all(&path).await?;
|
||||||
|
|
||||||
fs::write(path.join(format!("{doc_hash}.gz")), &contents).await?;
|
fs::write(path.join(format!("{doc_hash}.gz")), &contents).await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_doc(&self, doc_hash: &str) -> Result<HttpResponse, RegistryError> {
|
async fn get_doc(&self, doc_hash: &str) -> Result<HttpResponse, Error> {
|
||||||
let path = self.root.join("Doc");
|
let path = self.root.join("Doc");
|
||||||
|
|
||||||
read_file_to_response(&path.join(format!("{doc_hash}.gz")), "text/plain").await
|
read_file_to_response(&path.join(format!("{doc_hash}.gz")), "text/plain").await
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for FSStorage {
|
impl Display for FSStorage {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
write!(f, "FS ({})", self.root.display())
|
write!(f, "FS")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
use crate::{benv, error::RegistryError, make_reqwest};
|
use crate::{benv, error::Error, make_reqwest};
|
||||||
use actix_web::HttpResponse;
|
use actix_web::HttpResponse;
|
||||||
use pesde::{names::PackageName, source::ids::VersionId};
|
use pesde::{names::PackageName, source::version_id::VersionId};
|
||||||
use rusty_s3::{Bucket, Credentials, UrlStyle};
|
use rusty_s3::{Bucket, Credentials, UrlStyle};
|
||||||
use std::fmt::Display;
|
use std::fmt::Display;
|
||||||
|
|
||||||
|
@ -9,129 +9,133 @@ mod s3;
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub enum Storage {
|
pub enum Storage {
|
||||||
S3(s3::S3Storage),
|
S3(s3::S3Storage),
|
||||||
FS(fs::FSStorage),
|
FS(fs::FSStorage),
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait StorageImpl: Display {
|
pub trait StorageImpl: Display {
|
||||||
async fn store_package(
|
async fn store_package(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
contents: Vec<u8>,
|
contents: Vec<u8>,
|
||||||
) -> Result<(), RegistryError>;
|
) -> Result<(), crate::error::Error>;
|
||||||
async fn get_package(
|
async fn get_package(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
) -> Result<HttpResponse, RegistryError>;
|
) -> Result<HttpResponse, crate::error::Error>;
|
||||||
|
|
||||||
async fn store_readme(
|
async fn store_readme(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
contents: Vec<u8>,
|
contents: Vec<u8>,
|
||||||
) -> Result<(), RegistryError>;
|
) -> Result<(), crate::error::Error>;
|
||||||
async fn get_readme(
|
async fn get_readme(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
) -> Result<HttpResponse, RegistryError>;
|
) -> Result<HttpResponse, crate::error::Error>;
|
||||||
|
|
||||||
async fn store_doc(&self, doc_hash: String, contents: Vec<u8>) -> Result<(), RegistryError>;
|
async fn store_doc(
|
||||||
async fn get_doc(&self, doc_hash: &str) -> Result<HttpResponse, RegistryError>;
|
&self,
|
||||||
|
doc_hash: String,
|
||||||
|
contents: Vec<u8>,
|
||||||
|
) -> Result<(), crate::error::Error>;
|
||||||
|
async fn get_doc(&self, doc_hash: &str) -> Result<HttpResponse, crate::error::Error>;
|
||||||
}
|
}
|
||||||
|
|
||||||
impl StorageImpl for Storage {
|
impl StorageImpl for Storage {
|
||||||
async fn store_package(
|
async fn store_package(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
contents: Vec<u8>,
|
contents: Vec<u8>,
|
||||||
) -> Result<(), RegistryError> {
|
) -> Result<(), Error> {
|
||||||
match self {
|
match self {
|
||||||
Storage::S3(s3) => s3.store_package(package_name, version, contents).await,
|
Storage::S3(s3) => s3.store_package(package_name, version, contents).await,
|
||||||
Storage::FS(fs) => fs.store_package(package_name, version, contents).await,
|
Storage::FS(fs) => fs.store_package(package_name, version, contents).await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_package(
|
async fn get_package(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
) -> Result<HttpResponse, Error> {
|
||||||
match self {
|
match self {
|
||||||
Storage::S3(s3) => s3.get_package(package_name, version).await,
|
Storage::S3(s3) => s3.get_package(package_name, version).await,
|
||||||
Storage::FS(fs) => fs.get_package(package_name, version).await,
|
Storage::FS(fs) => fs.get_package(package_name, version).await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn store_readme(
|
async fn store_readme(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
contents: Vec<u8>,
|
contents: Vec<u8>,
|
||||||
) -> Result<(), RegistryError> {
|
) -> Result<(), Error> {
|
||||||
match self {
|
match self {
|
||||||
Storage::S3(s3) => s3.store_readme(package_name, version, contents).await,
|
Storage::S3(s3) => s3.store_readme(package_name, version, contents).await,
|
||||||
Storage::FS(fs) => fs.store_readme(package_name, version, contents).await,
|
Storage::FS(fs) => fs.store_readme(package_name, version, contents).await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_readme(
|
async fn get_readme(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
) -> Result<HttpResponse, Error> {
|
||||||
match self {
|
match self {
|
||||||
Storage::S3(s3) => s3.get_readme(package_name, version).await,
|
Storage::S3(s3) => s3.get_readme(package_name, version).await,
|
||||||
Storage::FS(fs) => fs.get_readme(package_name, version).await,
|
Storage::FS(fs) => fs.get_readme(package_name, version).await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn store_doc(&self, doc_hash: String, contents: Vec<u8>) -> Result<(), RegistryError> {
|
async fn store_doc(&self, doc_hash: String, contents: Vec<u8>) -> Result<(), Error> {
|
||||||
match self {
|
match self {
|
||||||
Storage::S3(s3) => s3.store_doc(doc_hash, contents).await,
|
Storage::S3(s3) => s3.store_doc(doc_hash, contents).await,
|
||||||
Storage::FS(fs) => fs.store_doc(doc_hash, contents).await,
|
Storage::FS(fs) => fs.store_doc(doc_hash, contents).await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_doc(&self, doc_hash: &str) -> Result<HttpResponse, RegistryError> {
|
async fn get_doc(&self, doc_hash: &str) -> Result<HttpResponse, Error> {
|
||||||
match self {
|
match self {
|
||||||
Storage::S3(s3) => s3.get_doc(doc_hash).await,
|
Storage::S3(s3) => s3.get_doc(doc_hash).await,
|
||||||
Storage::FS(fs) => fs.get_doc(doc_hash).await,
|
Storage::FS(fs) => fs.get_doc(doc_hash).await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for Storage {
|
impl Display for Storage {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
match self {
|
match self {
|
||||||
Storage::S3(s3) => write!(f, "{}", s3),
|
Storage::S3(s3) => write!(f, "{}", s3),
|
||||||
Storage::FS(fs) => write!(f, "{}", fs),
|
Storage::FS(fs) => write!(f, "{}", fs),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_storage_from_env() -> Storage {
|
pub fn get_storage_from_env() -> Storage {
|
||||||
if let Ok(endpoint) = benv!(parse "S3_ENDPOINT") {
|
if let Ok(endpoint) = benv!(parse "S3_ENDPOINT") {
|
||||||
Storage::S3(s3::S3Storage {
|
Storage::S3(s3::S3Storage {
|
||||||
bucket: Bucket::new(
|
s3_bucket: Bucket::new(
|
||||||
endpoint,
|
endpoint,
|
||||||
UrlStyle::Path,
|
UrlStyle::Path,
|
||||||
benv!(required "S3_BUCKET_NAME"),
|
benv!(required "S3_BUCKET_NAME"),
|
||||||
benv!(required "S3_REGION"),
|
benv!(required "S3_REGION"),
|
||||||
)
|
)
|
||||||
.unwrap(),
|
.unwrap(),
|
||||||
credentials: Credentials::new(
|
s3_credentials: Credentials::new(
|
||||||
benv!(required "S3_ACCESS_KEY"),
|
benv!(required "S3_ACCESS_KEY"),
|
||||||
benv!(required "S3_SECRET_KEY"),
|
benv!(required "S3_SECRET_KEY"),
|
||||||
),
|
),
|
||||||
reqwest_client: make_reqwest(),
|
reqwest_client: make_reqwest(),
|
||||||
})
|
})
|
||||||
} else if let Ok(root) = benv!(parse "FS_STORAGE_ROOT") {
|
} else if let Ok(root) = benv!(parse "FS_STORAGE_ROOT") {
|
||||||
Storage::FS(fs::FSStorage { root })
|
Storage::FS(fs::FSStorage { root })
|
||||||
} else {
|
} else {
|
||||||
panic!("no storage backend configured")
|
panic!("no storage backend configured")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,166 +1,166 @@
|
||||||
use crate::{
|
use crate::{
|
||||||
error::{RegistryError, ReqwestErrorExt},
|
error::{Error, ReqwestErrorExt},
|
||||||
storage::StorageImpl,
|
storage::StorageImpl,
|
||||||
};
|
};
|
||||||
use actix_web::{http::header::LOCATION, HttpResponse};
|
use actix_web::{http::header::LOCATION, HttpResponse};
|
||||||
use pesde::{names::PackageName, source::ids::VersionId};
|
use pesde::{names::PackageName, source::version_id::VersionId};
|
||||||
use reqwest::header::{CONTENT_ENCODING, CONTENT_TYPE};
|
use reqwest::header::{CONTENT_ENCODING, CONTENT_TYPE};
|
||||||
use rusty_s3::{
|
use rusty_s3::{
|
||||||
actions::{GetObject, PutObject},
|
actions::{GetObject, PutObject},
|
||||||
Bucket, Credentials, S3Action,
|
Bucket, Credentials, S3Action,
|
||||||
};
|
};
|
||||||
use std::{fmt::Display, time::Duration};
|
use std::{fmt::Display, time::Duration};
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct S3Storage {
|
pub struct S3Storage {
|
||||||
pub bucket: Bucket,
|
pub s3_bucket: Bucket,
|
||||||
pub credentials: Credentials,
|
pub s3_credentials: Credentials,
|
||||||
pub reqwest_client: reqwest::Client,
|
pub reqwest_client: reqwest::Client,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub const S3_SIGN_DURATION: Duration = Duration::from_secs(60 * 15);
|
pub const S3_SIGN_DURATION: Duration = Duration::from_secs(60 * 15);
|
||||||
|
|
||||||
impl StorageImpl for S3Storage {
|
impl StorageImpl for S3Storage {
|
||||||
async fn store_package(
|
async fn store_package(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
contents: Vec<u8>,
|
contents: Vec<u8>,
|
||||||
) -> Result<(), RegistryError> {
|
) -> Result<(), Error> {
|
||||||
let object_url = PutObject::new(
|
let object_url = PutObject::new(
|
||||||
&self.bucket,
|
&self.s3_bucket,
|
||||||
Some(&self.credentials),
|
Some(&self.s3_credentials),
|
||||||
&format!(
|
&format!(
|
||||||
"{package_name}/{}/{}/pkg.tar.gz",
|
"{package_name}/{}/{}/pkg.tar.gz",
|
||||||
version.version(),
|
version.version(),
|
||||||
version.target()
|
version.target()
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
.sign(S3_SIGN_DURATION);
|
.sign(S3_SIGN_DURATION);
|
||||||
|
|
||||||
self.reqwest_client
|
self.reqwest_client
|
||||||
.put(object_url)
|
.put(object_url)
|
||||||
.header(CONTENT_TYPE, "application/gzip")
|
.header(CONTENT_TYPE, "application/gzip")
|
||||||
.header(CONTENT_ENCODING, "gzip")
|
.header(CONTENT_ENCODING, "gzip")
|
||||||
.body(contents)
|
.body(contents)
|
||||||
.send()
|
.send()
|
||||||
.await?
|
.await?
|
||||||
.into_error()
|
.into_error()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_package(
|
async fn get_package(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
) -> Result<HttpResponse, Error> {
|
||||||
let object_url = GetObject::new(
|
let object_url = GetObject::new(
|
||||||
&self.bucket,
|
&self.s3_bucket,
|
||||||
Some(&self.credentials),
|
Some(&self.s3_credentials),
|
||||||
&format!(
|
&format!(
|
||||||
"{package_name}/{}/{}/pkg.tar.gz",
|
"{package_name}/{}/{}/pkg.tar.gz",
|
||||||
version.version(),
|
version.version(),
|
||||||
version.target()
|
version.target()
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
.sign(S3_SIGN_DURATION);
|
.sign(S3_SIGN_DURATION);
|
||||||
|
|
||||||
Ok(HttpResponse::TemporaryRedirect()
|
Ok(HttpResponse::TemporaryRedirect()
|
||||||
.append_header((LOCATION, object_url.as_str()))
|
.append_header((LOCATION, object_url.as_str()))
|
||||||
.finish())
|
.finish())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn store_readme(
|
async fn store_readme(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
contents: Vec<u8>,
|
contents: Vec<u8>,
|
||||||
) -> Result<(), RegistryError> {
|
) -> Result<(), Error> {
|
||||||
let object_url = PutObject::new(
|
let object_url = PutObject::new(
|
||||||
&self.bucket,
|
&self.s3_bucket,
|
||||||
Some(&self.credentials),
|
Some(&self.s3_credentials),
|
||||||
&format!(
|
&format!(
|
||||||
"{package_name}/{}/{}/readme.gz",
|
"{package_name}/{}/{}/readme.gz",
|
||||||
version.version(),
|
version.version(),
|
||||||
version.target()
|
version.target()
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
.sign(S3_SIGN_DURATION);
|
.sign(S3_SIGN_DURATION);
|
||||||
|
|
||||||
self.reqwest_client
|
self.reqwest_client
|
||||||
.put(object_url)
|
.put(object_url)
|
||||||
.header(CONTENT_TYPE, "text/plain")
|
.header(CONTENT_TYPE, "text/plain")
|
||||||
.header(CONTENT_ENCODING, "gzip")
|
.header(CONTENT_ENCODING, "gzip")
|
||||||
.body(contents)
|
.body(contents)
|
||||||
.send()
|
.send()
|
||||||
.await?
|
.await?
|
||||||
.into_error()
|
.into_error()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_readme(
|
async fn get_readme(
|
||||||
&self,
|
&self,
|
||||||
package_name: &PackageName,
|
package_name: &PackageName,
|
||||||
version: &VersionId,
|
version: &VersionId,
|
||||||
) -> Result<HttpResponse, RegistryError> {
|
) -> Result<HttpResponse, Error> {
|
||||||
let object_url = GetObject::new(
|
let object_url = GetObject::new(
|
||||||
&self.bucket,
|
&self.s3_bucket,
|
||||||
Some(&self.credentials),
|
Some(&self.s3_credentials),
|
||||||
&format!(
|
&format!(
|
||||||
"{package_name}/{}/{}/readme.gz",
|
"{package_name}/{}/{}/readme.gz",
|
||||||
version.version(),
|
version.version(),
|
||||||
version.target()
|
version.target()
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
.sign(S3_SIGN_DURATION);
|
.sign(S3_SIGN_DURATION);
|
||||||
|
|
||||||
Ok(HttpResponse::TemporaryRedirect()
|
Ok(HttpResponse::TemporaryRedirect()
|
||||||
.append_header((LOCATION, object_url.as_str()))
|
.append_header((LOCATION, object_url.as_str()))
|
||||||
.finish())
|
.finish())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn store_doc(&self, doc_hash: String, contents: Vec<u8>) -> Result<(), RegistryError> {
|
async fn store_doc(&self, doc_hash: String, contents: Vec<u8>) -> Result<(), Error> {
|
||||||
let object_url = PutObject::new(
|
let object_url = PutObject::new(
|
||||||
&self.bucket,
|
&self.s3_bucket,
|
||||||
Some(&self.credentials),
|
Some(&self.s3_credentials),
|
||||||
// capitalize Doc to prevent conflicts with scope names
|
// capitalize Doc to prevent conflicts with scope names
|
||||||
&format!("Doc/{}.gz", doc_hash),
|
&format!("Doc/{}.gz", doc_hash),
|
||||||
)
|
)
|
||||||
.sign(S3_SIGN_DURATION);
|
.sign(S3_SIGN_DURATION);
|
||||||
|
|
||||||
self.reqwest_client
|
self.reqwest_client
|
||||||
.put(object_url)
|
.put(object_url)
|
||||||
.header(CONTENT_TYPE, "text/plain")
|
.header(CONTENT_TYPE, "text/plain")
|
||||||
.header(CONTENT_ENCODING, "gzip")
|
.header(CONTENT_ENCODING, "gzip")
|
||||||
.body(contents)
|
.body(contents)
|
||||||
.send()
|
.send()
|
||||||
.await?
|
.await?
|
||||||
.into_error()
|
.into_error()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_doc(&self, doc_hash: &str) -> Result<HttpResponse, RegistryError> {
|
async fn get_doc(&self, doc_hash: &str) -> Result<HttpResponse, Error> {
|
||||||
let object_url = GetObject::new(
|
let object_url = GetObject::new(
|
||||||
&self.bucket,
|
&self.s3_bucket,
|
||||||
Some(&self.credentials),
|
Some(&self.s3_credentials),
|
||||||
&format!("Doc/{}.gz", doc_hash),
|
&format!("Doc/{}.gz", doc_hash),
|
||||||
)
|
)
|
||||||
.sign(S3_SIGN_DURATION);
|
.sign(S3_SIGN_DURATION);
|
||||||
|
|
||||||
Ok(HttpResponse::TemporaryRedirect()
|
Ok(HttpResponse::TemporaryRedirect()
|
||||||
.append_header((LOCATION, object_url.as_str()))
|
.append_header((LOCATION, object_url.as_str()))
|
||||||
.finish())
|
.finish())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for S3Storage {
|
impl Display for S3Storage {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
write!(f, "S3 (bucket name: {})", self.bucket.name())
|
write!(f, "S3")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,2 +1 @@
|
||||||
imports_granularity = "Crate"
|
imports_granularity = "Crate"
|
||||||
hard_tabs = true
|
|
168
src/cli/auth.rs
168
src/cli/auth.rs
|
@ -5,129 +5,115 @@ use keyring::Entry;
|
||||||
use reqwest::header::AUTHORIZATION;
|
use reqwest::header::AUTHORIZATION;
|
||||||
use serde::{ser::SerializeMap, Deserialize, Serialize};
|
use serde::{ser::SerializeMap, Deserialize, Serialize};
|
||||||
use std::collections::BTreeMap;
|
use std::collections::BTreeMap;
|
||||||
use tokio::task::spawn_blocking;
|
|
||||||
use tracing::instrument;
|
use tracing::instrument;
|
||||||
|
|
||||||
#[derive(Debug, Clone, Default)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct Tokens(pub BTreeMap<gix::Url, String>);
|
pub struct Tokens(pub BTreeMap<gix::Url, String>);
|
||||||
|
|
||||||
impl Serialize for Tokens {
|
impl Serialize for Tokens {
|
||||||
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
||||||
where
|
where
|
||||||
S: serde::ser::Serializer,
|
S: serde::ser::Serializer,
|
||||||
{
|
{
|
||||||
let mut map = serializer.serialize_map(Some(self.0.len()))?;
|
let mut map = serializer.serialize_map(Some(self.0.len()))?;
|
||||||
for (k, v) in &self.0 {
|
for (k, v) in &self.0 {
|
||||||
map.serialize_entry(&k.to_bstring().to_string(), v)?;
|
map.serialize_entry(&k.to_bstring().to_string(), v)?;
|
||||||
}
|
}
|
||||||
map.end()
|
map.end()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'de> Deserialize<'de> for Tokens {
|
impl<'de> Deserialize<'de> for Tokens {
|
||||||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||||||
where
|
where
|
||||||
D: serde::de::Deserializer<'de>,
|
D: serde::de::Deserializer<'de>,
|
||||||
{
|
{
|
||||||
Ok(Tokens(
|
Ok(Tokens(
|
||||||
BTreeMap::<String, String>::deserialize(deserializer)?
|
BTreeMap::<String, String>::deserialize(deserializer)?
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|(k, v)| gix::Url::from_bytes(BStr::new(&k)).map(|k| (k, v)))
|
.map(|(k, v)| gix::Url::from_bytes(BStr::new(&k)).map(|k| (k, v)))
|
||||||
.collect::<Result<_, _>>()
|
.collect::<Result<_, _>>()
|
||||||
.map_err(serde::de::Error::custom)?,
|
.map_err(serde::de::Error::custom)?,
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(level = "trace")]
|
#[instrument(level = "trace")]
|
||||||
pub async fn get_tokens() -> anyhow::Result<Tokens> {
|
pub async fn get_tokens() -> anyhow::Result<Tokens> {
|
||||||
let config = read_config().await?;
|
let config = read_config().await?;
|
||||||
if !config.tokens.0.is_empty() {
|
if !config.tokens.0.is_empty() {
|
||||||
tracing::debug!("using tokens from config");
|
tracing::debug!("using tokens from config");
|
||||||
return Ok(config.tokens);
|
return Ok(config.tokens);
|
||||||
}
|
}
|
||||||
|
|
||||||
let keyring_tokens = spawn_blocking(|| match Entry::new("tokens", env!("CARGO_PKG_NAME")) {
|
match Entry::new("tokens", env!("CARGO_PKG_NAME")) {
|
||||||
Ok(entry) => match entry.get_password() {
|
Ok(entry) => match entry.get_password() {
|
||||||
Ok(token) => serde_json::from_str(&token)
|
Ok(token) => {
|
||||||
.map(Some)
|
tracing::debug!("using tokens from keyring");
|
||||||
.context("failed to parse tokens"),
|
return serde_json::from_str(&token).context("failed to parse tokens");
|
||||||
Err(keyring::Error::PlatformFailure(_) | keyring::Error::NoEntry) => Ok(None),
|
}
|
||||||
Err(e) => Err(e.into()),
|
Err(keyring::Error::PlatformFailure(_) | keyring::Error::NoEntry) => {}
|
||||||
},
|
Err(e) => return Err(e.into()),
|
||||||
Err(keyring::Error::PlatformFailure(_)) => Ok(None),
|
},
|
||||||
Err(e) => Err(e.into()),
|
Err(keyring::Error::PlatformFailure(_)) => {}
|
||||||
})
|
Err(e) => return Err(e.into()),
|
||||||
.await
|
}
|
||||||
.unwrap()?;
|
|
||||||
|
|
||||||
if let Some(tokens) = keyring_tokens {
|
Ok(Tokens(BTreeMap::new()))
|
||||||
tracing::debug!("using tokens from keyring");
|
|
||||||
return Ok(tokens);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(Tokens::default())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(level = "trace")]
|
#[instrument(level = "trace")]
|
||||||
pub async fn set_tokens(tokens: Tokens) -> anyhow::Result<()> {
|
pub async fn set_tokens(tokens: Tokens) -> anyhow::Result<()> {
|
||||||
let json = serde_json::to_string(&tokens).context("failed to serialize tokens")?;
|
let entry = Entry::new("tokens", env!("CARGO_PKG_NAME"))?;
|
||||||
|
let json = serde_json::to_string(&tokens).context("failed to serialize tokens")?;
|
||||||
|
|
||||||
let to_keyring = spawn_blocking(move || {
|
match entry.set_password(&json) {
|
||||||
let entry = Entry::new("tokens", env!("CARGO_PKG_NAME"))?;
|
Ok(()) => {
|
||||||
|
tracing::debug!("tokens saved to keyring");
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
Err(keyring::Error::PlatformFailure(_) | keyring::Error::NoEntry) => {}
|
||||||
|
Err(e) => return Err(e.into()),
|
||||||
|
}
|
||||||
|
|
||||||
match entry.set_password(&json) {
|
tracing::debug!("tokens saved to config");
|
||||||
Ok(()) => Ok::<_, anyhow::Error>(true),
|
|
||||||
Err(keyring::Error::PlatformFailure(_) | keyring::Error::NoEntry) => Ok(false),
|
|
||||||
Err(e) => Err(e.into()),
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.unwrap()?;
|
|
||||||
|
|
||||||
if to_keyring {
|
let mut config = read_config().await?;
|
||||||
tracing::debug!("tokens saved to keyring");
|
config.tokens = tokens;
|
||||||
return Ok(());
|
write_config(&config).await.map_err(Into::into)
|
||||||
}
|
|
||||||
|
|
||||||
tracing::debug!("saving tokens to config");
|
|
||||||
|
|
||||||
let mut config = read_config().await?;
|
|
||||||
config.tokens = tokens;
|
|
||||||
write_config(&config).await
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn set_token(repo: &gix::Url, token: Option<&str>) -> anyhow::Result<()> {
|
pub async fn set_token(repo: &gix::Url, token: Option<&str>) -> anyhow::Result<()> {
|
||||||
let mut tokens = get_tokens().await?;
|
let mut tokens = get_tokens().await?;
|
||||||
if let Some(token) = token {
|
if let Some(token) = token {
|
||||||
tokens.0.insert(repo.clone(), token.to_string());
|
tokens.0.insert(repo.clone(), token.to_string());
|
||||||
} else {
|
} else {
|
||||||
tokens.0.remove(repo);
|
tokens.0.remove(repo);
|
||||||
}
|
}
|
||||||
set_tokens(tokens).await
|
set_tokens(tokens).await
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
struct UserResponse {
|
struct UserResponse {
|
||||||
login: String,
|
login: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(level = "trace")]
|
#[instrument(level = "trace")]
|
||||||
pub async fn get_token_login(
|
pub async fn get_token_login(
|
||||||
reqwest: &reqwest::Client,
|
reqwest: &reqwest::Client,
|
||||||
access_token: &str,
|
access_token: &str,
|
||||||
) -> anyhow::Result<String> {
|
) -> anyhow::Result<String> {
|
||||||
let response = reqwest
|
let response = reqwest
|
||||||
.get("https://api.github.com/user")
|
.get("https://api.github.com/user")
|
||||||
.header(AUTHORIZATION, access_token)
|
.header(AUTHORIZATION, access_token)
|
||||||
.send()
|
.send()
|
||||||
.await
|
.await
|
||||||
.context("failed to send user request")?
|
.context("failed to send user request")?
|
||||||
.error_for_status()
|
.error_for_status()
|
||||||
.context("failed to get user")?
|
.context("failed to get user")?
|
||||||
.json::<UserResponse>()
|
.json::<UserResponse>()
|
||||||
.await
|
.await
|
||||||
.context("failed to parse user response")?;
|
.context("failed to parse user response")?;
|
||||||
|
|
||||||
Ok(response.login)
|
Ok(response.login)
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,81 +0,0 @@
|
||||||
local process = require("@lune/process")
|
|
||||||
local fs = require("@lune/fs")
|
|
||||||
local stdio = require("@lune/stdio")
|
|
||||||
local serde = require("@lune/serde")
|
|
||||||
|
|
||||||
local project_root = nil
|
|
||||||
local path_components = string.split(string.gsub(process.cwd, "\\", "/"), "/")
|
|
||||||
if path_components[#path_components] == "" then
|
|
||||||
table.remove(path_components)
|
|
||||||
end
|
|
||||||
|
|
||||||
local function in_lockfile(lockfile)
|
|
||||||
if not lockfile.graph then
|
|
||||||
return false
|
|
||||||
end
|
|
||||||
|
|
||||||
for _, versions in lockfile.graph do
|
|
||||||
for _, node in versions do
|
|
||||||
if node.direct and node.direct[1] == "{alias}" then
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
return false
|
|
||||||
end
|
|
||||||
|
|
||||||
for i = #path_components, 1, -1 do
|
|
||||||
local path = table.concat(path_components, "/", 1, i)
|
|
||||||
if not fs.isFile(path .. "/{MANIFEST_FILE_NAME}") then
|
|
||||||
continue
|
|
||||||
end
|
|
||||||
|
|
||||||
if project_root == nil then
|
|
||||||
project_root = path
|
|
||||||
end
|
|
||||||
|
|
||||||
if project_root and fs.isFile(path .. "/{LOCKFILE_FILE_NAME}") then
|
|
||||||
local lockfile = serde.decode("toml", fs.readFile(path .. "/{LOCKFILE_FILE_NAME}"))
|
|
||||||
if not lockfile.workspace then
|
|
||||||
continue
|
|
||||||
end
|
|
||||||
|
|
||||||
local search_for = string.gsub(project_root, path, "")
|
|
||||||
if string.sub(search_for, 1, 1) == "/" then
|
|
||||||
search_for = string.sub(search_for, 2)
|
|
||||||
end
|
|
||||||
|
|
||||||
if search_for == "" then
|
|
||||||
if in_lockfile(lockfile) then
|
|
||||||
break
|
|
||||||
end
|
|
||||||
|
|
||||||
continue
|
|
||||||
end
|
|
||||||
|
|
||||||
for _, targets in lockfile.workspace do
|
|
||||||
for _, member_path in targets do
|
|
||||||
local path_normalized = string.gsub(member_path, "\\", "/")
|
|
||||||
if path_normalized == search_for and in_lockfile(lockfile) then
|
|
||||||
project_root = path
|
|
||||||
break
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
if project_root ~= nil then
|
|
||||||
for _, packages_folder in {{ {all_folders} }} do
|
|
||||||
local path = `{{project_root}}/{{packages_folder}}/{alias}.bin.luau`
|
|
||||||
|
|
||||||
if fs.isFile(path) then
|
|
||||||
require(path)
|
|
||||||
return
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
stdio.ewrite(stdio.color("red") .. "binary `{alias}` not found. are you in the right directory?" .. stdio.color("reset") .. "\n")
|
|
||||||
process.exit(1)
|
|
|
@ -1,273 +1,250 @@
|
||||||
use std::str::FromStr;
|
use std::{collections::HashSet, str::FromStr};
|
||||||
|
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
|
use colored::Colorize;
|
||||||
use semver::VersionReq;
|
use semver::VersionReq;
|
||||||
|
|
||||||
use crate::cli::{
|
use crate::cli::{config::read_config, AnyPackageIdentifier, VersionedPackageName};
|
||||||
config::read_config, dep_type_to_key, AnyPackageIdentifier, VersionedPackageName,
|
|
||||||
};
|
|
||||||
use pesde::{
|
use pesde::{
|
||||||
manifest::{target::TargetKind, Alias, DependencyType},
|
manifest::target::TargetKind,
|
||||||
names::PackageNames,
|
names::PackageNames,
|
||||||
source::{
|
source::{
|
||||||
git::{specifier::GitDependencySpecifier, GitPackageSource},
|
git::{specifier::GitDependencySpecifier, GitPackageSource},
|
||||||
path::{specifier::PathDependencySpecifier, PathPackageSource},
|
pesde::{specifier::PesdeDependencySpecifier, PesdePackageSource},
|
||||||
pesde::{specifier::PesdeDependencySpecifier, PesdePackageSource},
|
specifiers::DependencySpecifiers,
|
||||||
specifiers::DependencySpecifiers,
|
traits::PackageSource,
|
||||||
traits::{PackageSource, RefreshOptions, ResolveOptions},
|
workspace::WorkspacePackageSource,
|
||||||
workspace::{specifier::WorkspaceDependencySpecifier, WorkspacePackageSource},
|
PackageSources,
|
||||||
PackageSources,
|
},
|
||||||
},
|
Project, DEFAULT_INDEX_NAME,
|
||||||
Project, RefreshedSources, DEFAULT_INDEX_NAME,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct AddCommand {
|
pub struct AddCommand {
|
||||||
/// The package name to add
|
/// The package name to add
|
||||||
#[arg(index = 1)]
|
#[arg(index = 1)]
|
||||||
name: AnyPackageIdentifier<VersionReq>,
|
name: AnyPackageIdentifier<VersionReq>,
|
||||||
|
|
||||||
/// The index in which to search for the package
|
/// The index in which to search for the package
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
index: Option<String>,
|
index: Option<String>,
|
||||||
|
|
||||||
/// The target environment of the package
|
/// The target environment of the package
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
target: Option<TargetKind>,
|
target: Option<TargetKind>,
|
||||||
|
|
||||||
/// The alias to use for the package
|
/// The alias to use for the package
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
alias: Option<Alias>,
|
alias: Option<String>,
|
||||||
|
|
||||||
/// Whether to add the package as a peer dependency
|
/// Whether to add the package as a peer dependency
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
peer: bool,
|
peer: bool,
|
||||||
|
|
||||||
/// Whether to add the package as a dev dependency
|
/// Whether to add the package as a dev dependency
|
||||||
#[arg(short, long, conflicts_with = "peer")]
|
#[arg(short, long, conflicts_with = "peer")]
|
||||||
dev: bool,
|
dev: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AddCommand {
|
impl AddCommand {
|
||||||
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
||||||
let manifest = project
|
let manifest = project
|
||||||
.deser_manifest()
|
.deser_manifest()
|
||||||
.await
|
.await
|
||||||
.context("failed to read manifest")?;
|
.context("failed to read manifest")?;
|
||||||
|
|
||||||
let (source, specifier) = match &self.name {
|
let (source, specifier) = match &self.name {
|
||||||
AnyPackageIdentifier::PackageName(versioned) => match &versioned {
|
AnyPackageIdentifier::PackageName(versioned) => match &versioned {
|
||||||
VersionedPackageName(PackageNames::Pesde(name), version) => {
|
VersionedPackageName(PackageNames::Pesde(name), version) => {
|
||||||
let index = manifest
|
let index = manifest
|
||||||
.indices
|
.indices
|
||||||
.get(self.index.as_deref().unwrap_or(DEFAULT_INDEX_NAME))
|
.get(self.index.as_deref().unwrap_or(DEFAULT_INDEX_NAME))
|
||||||
.cloned();
|
.cloned();
|
||||||
|
|
||||||
if let Some(index) = self.index.as_ref().filter(|_| index.is_none()) {
|
if let Some(index) = self.index.as_ref().filter(|_| index.is_none()) {
|
||||||
anyhow::bail!("index {index} not found");
|
println!("{}: index {index} not found", "error".red().bold());
|
||||||
}
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
let index = match index {
|
let index = match index {
|
||||||
Some(index) => index,
|
Some(index) => index,
|
||||||
None => read_config().await?.default_index,
|
None => read_config().await?.default_index,
|
||||||
};
|
};
|
||||||
|
|
||||||
let source = PackageSources::Pesde(PesdePackageSource::new(index));
|
let source = PackageSources::Pesde(PesdePackageSource::new(index));
|
||||||
let specifier = DependencySpecifiers::Pesde(PesdeDependencySpecifier {
|
let specifier = DependencySpecifiers::Pesde(PesdeDependencySpecifier {
|
||||||
name: name.clone(),
|
name: name.clone(),
|
||||||
version: version.clone().unwrap_or(VersionReq::STAR),
|
version: version.clone().unwrap_or(VersionReq::STAR),
|
||||||
index: self.index,
|
index: self.index,
|
||||||
target: self.target,
|
target: self.target,
|
||||||
});
|
});
|
||||||
|
|
||||||
(source, specifier)
|
(source, specifier)
|
||||||
}
|
}
|
||||||
#[cfg(feature = "wally-compat")]
|
#[cfg(feature = "wally-compat")]
|
||||||
VersionedPackageName(PackageNames::Wally(name), version) => {
|
VersionedPackageName(PackageNames::Wally(name), version) => {
|
||||||
let index = manifest
|
let index = manifest
|
||||||
.wally_indices
|
.wally_indices
|
||||||
.get(self.index.as_deref().unwrap_or(DEFAULT_INDEX_NAME))
|
.get(self.index.as_deref().unwrap_or(DEFAULT_INDEX_NAME))
|
||||||
.cloned();
|
.cloned();
|
||||||
|
|
||||||
if let Some(index) = self.index.as_ref().filter(|_| index.is_none()) {
|
if let Some(index) = self.index.as_ref().filter(|_| index.is_none()) {
|
||||||
anyhow::bail!("wally index {index} not found");
|
println!("{}: wally index {index} not found", "error".red().bold());
|
||||||
}
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
let index = index.context("no wally index found")?;
|
let index = index.context("no wally index found")?;
|
||||||
|
|
||||||
let source =
|
let source =
|
||||||
PackageSources::Wally(pesde::source::wally::WallyPackageSource::new(index));
|
PackageSources::Wally(pesde::source::wally::WallyPackageSource::new(index));
|
||||||
let specifier = DependencySpecifiers::Wally(
|
let specifier = DependencySpecifiers::Wally(
|
||||||
pesde::source::wally::specifier::WallyDependencySpecifier {
|
pesde::source::wally::specifier::WallyDependencySpecifier {
|
||||||
name: name.clone(),
|
name: name.clone(),
|
||||||
version: version.clone().unwrap_or(VersionReq::STAR),
|
version: version.clone().unwrap_or(VersionReq::STAR),
|
||||||
index: self.index,
|
index: self.index,
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
|
|
||||||
(source, specifier)
|
(source, specifier)
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
AnyPackageIdentifier::Url((url, rev)) => (
|
AnyPackageIdentifier::Url((url, rev)) => (
|
||||||
PackageSources::Git(GitPackageSource::new(url.clone())),
|
PackageSources::Git(GitPackageSource::new(url.clone())),
|
||||||
DependencySpecifiers::Git(GitDependencySpecifier {
|
DependencySpecifiers::Git(GitDependencySpecifier {
|
||||||
repo: url.clone(),
|
repo: url.clone(),
|
||||||
rev: rev.to_string(),
|
rev: rev.to_string(),
|
||||||
path: None,
|
path: None,
|
||||||
}),
|
}),
|
||||||
),
|
),
|
||||||
AnyPackageIdentifier::Workspace(VersionedPackageName(name, version)) => (
|
AnyPackageIdentifier::Workspace(VersionedPackageName(name, version)) => (
|
||||||
PackageSources::Workspace(WorkspacePackageSource),
|
PackageSources::Workspace(WorkspacePackageSource),
|
||||||
DependencySpecifiers::Workspace(WorkspaceDependencySpecifier {
|
DependencySpecifiers::Workspace(
|
||||||
name: name.clone(),
|
pesde::source::workspace::specifier::WorkspaceDependencySpecifier {
|
||||||
version: version.clone().unwrap_or_default(),
|
name: name.clone(),
|
||||||
target: self.target,
|
version: version.clone().unwrap_or_default(),
|
||||||
}),
|
target: self.target,
|
||||||
),
|
},
|
||||||
AnyPackageIdentifier::Path(path) => (
|
),
|
||||||
PackageSources::Path(PathPackageSource),
|
),
|
||||||
DependencySpecifiers::Path(PathDependencySpecifier { path: path.clone() }),
|
};
|
||||||
),
|
source
|
||||||
};
|
.refresh(&project)
|
||||||
|
.await
|
||||||
|
.context("failed to refresh package source")?;
|
||||||
|
|
||||||
let refreshed_sources = RefreshedSources::new();
|
let Some(version_id) = source
|
||||||
|
.resolve(
|
||||||
|
&specifier,
|
||||||
|
&project,
|
||||||
|
manifest.target.kind(),
|
||||||
|
&mut HashSet::new(),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to resolve package")?
|
||||||
|
.1
|
||||||
|
.pop_last()
|
||||||
|
.map(|(v_id, _)| v_id)
|
||||||
|
else {
|
||||||
|
println!("{}: no versions found for package", "error".red().bold());
|
||||||
|
|
||||||
refreshed_sources
|
return Ok(());
|
||||||
.refresh(
|
};
|
||||||
&source,
|
|
||||||
&RefreshOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to refresh package source")?;
|
|
||||||
|
|
||||||
let Some(version_id) = source
|
let project_target = manifest.target.kind();
|
||||||
.resolve(
|
let mut manifest = toml_edit::DocumentMut::from_str(
|
||||||
&specifier,
|
&project
|
||||||
&ResolveOptions {
|
.read_manifest()
|
||||||
project: project.clone(),
|
.await
|
||||||
target: manifest.target.kind(),
|
.context("failed to read manifest")?,
|
||||||
refreshed_sources,
|
)
|
||||||
},
|
.context("failed to parse manifest")?;
|
||||||
)
|
let dependency_key = if self.peer {
|
||||||
.await
|
"peer_dependencies"
|
||||||
.context("failed to resolve package")?
|
} else if self.dev {
|
||||||
.1
|
"dev_dependencies"
|
||||||
.pop_last()
|
} else {
|
||||||
.map(|(v_id, _)| v_id)
|
"dependencies"
|
||||||
else {
|
};
|
||||||
anyhow::bail!("no versions found for package");
|
|
||||||
};
|
|
||||||
|
|
||||||
let project_target = manifest.target.kind();
|
let alias = self.alias.unwrap_or_else(|| match self.name.clone() {
|
||||||
let mut manifest = toml_edit::DocumentMut::from_str(
|
AnyPackageIdentifier::PackageName(versioned) => versioned.0.as_str().1.to_string(),
|
||||||
&project
|
AnyPackageIdentifier::Url((url, _)) => url
|
||||||
.read_manifest()
|
.path
|
||||||
.await
|
.to_string()
|
||||||
.context("failed to read manifest")?,
|
.split('/')
|
||||||
)
|
.last()
|
||||||
.context("failed to parse manifest")?;
|
.map(|s| s.to_string())
|
||||||
let dependency_key = dep_type_to_key(if self.peer {
|
.unwrap_or(url.path.to_string()),
|
||||||
DependencyType::Peer
|
AnyPackageIdentifier::Workspace(versioned) => versioned.0.as_str().1.to_string(),
|
||||||
} else if self.dev {
|
});
|
||||||
DependencyType::Dev
|
|
||||||
} else {
|
|
||||||
DependencyType::Standard
|
|
||||||
});
|
|
||||||
|
|
||||||
let alias = match self.alias {
|
let field = &mut manifest[dependency_key]
|
||||||
Some(alias) => alias,
|
.or_insert(toml_edit::Item::Table(toml_edit::Table::new()))[&alias];
|
||||||
None => match &self.name {
|
|
||||||
AnyPackageIdentifier::PackageName(versioned) => versioned.0.name().to_string(),
|
|
||||||
AnyPackageIdentifier::Url((url, _)) => url
|
|
||||||
.path
|
|
||||||
.to_string()
|
|
||||||
.split('/')
|
|
||||||
.next_back()
|
|
||||||
.map(|s| s.to_string())
|
|
||||||
.unwrap_or(url.path.to_string()),
|
|
||||||
AnyPackageIdentifier::Workspace(versioned) => versioned.0.name().to_string(),
|
|
||||||
AnyPackageIdentifier::Path(path) => path
|
|
||||||
.file_name()
|
|
||||||
.map(|s| s.to_string_lossy().to_string())
|
|
||||||
.expect("path has no file name"),
|
|
||||||
}
|
|
||||||
.parse()
|
|
||||||
.context("auto-generated alias is invalid. use --alias to specify one")?,
|
|
||||||
};
|
|
||||||
|
|
||||||
let field = &mut manifest[dependency_key]
|
match specifier {
|
||||||
.or_insert(toml_edit::Item::Table(toml_edit::Table::new()))[alias.as_str()];
|
DependencySpecifiers::Pesde(spec) => {
|
||||||
|
field["name"] = toml_edit::value(spec.name.clone().to_string());
|
||||||
|
field["version"] = toml_edit::value(format!("^{}", version_id.version()));
|
||||||
|
|
||||||
match specifier {
|
if *version_id.target() != project_target {
|
||||||
DependencySpecifiers::Pesde(spec) => {
|
field["target"] = toml_edit::value(version_id.target().to_string());
|
||||||
field["name"] = toml_edit::value(spec.name.clone().to_string());
|
}
|
||||||
field["version"] = toml_edit::value(format!("^{}", version_id.version()));
|
|
||||||
|
|
||||||
if version_id.target() != project_target {
|
if let Some(index) = spec.index.filter(|i| i != DEFAULT_INDEX_NAME) {
|
||||||
field["target"] = toml_edit::value(version_id.target().to_string());
|
field["index"] = toml_edit::value(index);
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(index) = spec.index.filter(|i| i != DEFAULT_INDEX_NAME) {
|
println!(
|
||||||
field["index"] = toml_edit::value(index);
|
"added {}@{} {} to {}",
|
||||||
}
|
spec.name,
|
||||||
|
version_id.version(),
|
||||||
|
version_id.target(),
|
||||||
|
dependency_key
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[cfg(feature = "wally-compat")]
|
||||||
|
DependencySpecifiers::Wally(spec) => {
|
||||||
|
field["wally"] = toml_edit::value(spec.name.clone().to_string());
|
||||||
|
field["version"] = toml_edit::value(format!("^{}", version_id.version()));
|
||||||
|
|
||||||
println!(
|
if let Some(index) = spec.index.filter(|i| i != DEFAULT_INDEX_NAME) {
|
||||||
"added {}@{} {} to {dependency_key}",
|
field["index"] = toml_edit::value(index);
|
||||||
spec.name,
|
}
|
||||||
version_id.version(),
|
|
||||||
version_id.target()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
#[cfg(feature = "wally-compat")]
|
|
||||||
DependencySpecifiers::Wally(spec) => {
|
|
||||||
let name_str = spec.name.to_string();
|
|
||||||
let name_str = name_str.trim_start_matches("wally#");
|
|
||||||
field["wally"] = toml_edit::value(name_str);
|
|
||||||
field["version"] = toml_edit::value(format!("^{}", version_id.version()));
|
|
||||||
|
|
||||||
if let Some(index) = spec.index.filter(|i| i != DEFAULT_INDEX_NAME) {
|
println!(
|
||||||
field["index"] = toml_edit::value(index);
|
"added wally {}@{} to {}",
|
||||||
}
|
spec.name,
|
||||||
|
version_id.version(),
|
||||||
|
dependency_key
|
||||||
|
);
|
||||||
|
}
|
||||||
|
DependencySpecifiers::Git(spec) => {
|
||||||
|
field["repo"] = toml_edit::value(spec.repo.to_bstring().to_string());
|
||||||
|
field["rev"] = toml_edit::value(spec.rev.clone());
|
||||||
|
|
||||||
println!(
|
println!("added git {}#{} to {}", spec.repo, spec.rev, dependency_key);
|
||||||
"added wally {name_str}@{} to {dependency_key}",
|
}
|
||||||
version_id.version()
|
DependencySpecifiers::Workspace(spec) => {
|
||||||
);
|
field["workspace"] = toml_edit::value(spec.name.clone().to_string());
|
||||||
}
|
if let AnyPackageIdentifier::Workspace(versioned) = self.name {
|
||||||
DependencySpecifiers::Git(spec) => {
|
if let Some(version) = versioned.1 {
|
||||||
field["repo"] = toml_edit::value(spec.repo.to_bstring().to_string());
|
field["version"] = toml_edit::value(version.to_string());
|
||||||
field["rev"] = toml_edit::value(spec.rev.clone());
|
}
|
||||||
|
}
|
||||||
|
|
||||||
println!("added git {}#{} to {dependency_key}", spec.repo, spec.rev);
|
println!(
|
||||||
}
|
"added workspace {}@{} to {}",
|
||||||
DependencySpecifiers::Workspace(spec) => {
|
spec.name, spec.version, dependency_key
|
||||||
field["workspace"] = toml_edit::value(spec.name.clone().to_string());
|
);
|
||||||
if let AnyPackageIdentifier::Workspace(versioned) = self.name {
|
}
|
||||||
if let Some(version) = versioned.1 {
|
}
|
||||||
field["version"] = toml_edit::value(version.to_string());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
println!(
|
project
|
||||||
"added workspace {}@{} to {dependency_key}",
|
.write_manifest(manifest.to_string())
|
||||||
spec.name, spec.version
|
.await
|
||||||
);
|
.context("failed to write manifest")?;
|
||||||
}
|
|
||||||
DependencySpecifiers::Path(spec) => {
|
|
||||||
field["path"] = toml_edit::value(spec.path.to_string_lossy().to_string());
|
|
||||||
|
|
||||||
println!("added path {} to {dependency_key}", spec.path.display());
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
project
|
|
||||||
.write_manifest(manifest.to_string())
|
|
||||||
.await
|
|
||||||
.context("failed to write manifest")?;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,200 +1,193 @@
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use console::style;
|
use colored::Colorize;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use std::thread::spawn;
|
use std::thread::spawn;
|
||||||
use tokio::time::sleep;
|
use tokio::time::sleep;
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
|
||||||
use crate::cli::{
|
|
||||||
auth::{get_token_login, set_token},
|
|
||||||
style::URL_STYLE,
|
|
||||||
};
|
|
||||||
use pesde::{
|
use pesde::{
|
||||||
source::{
|
source::{pesde::PesdePackageSource, traits::PackageSource},
|
||||||
pesde::PesdePackageSource,
|
Project,
|
||||||
traits::{PackageSource, RefreshOptions},
|
|
||||||
},
|
|
||||||
Project,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use crate::cli::auth::{get_token_login, set_token};
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct LoginCommand {
|
pub struct LoginCommand {
|
||||||
/// The token to use for authentication, skipping login
|
/// The token to use for authentication, skipping login
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
token: Option<String>,
|
token: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
struct DeviceCodeResponse {
|
struct DeviceCodeResponse {
|
||||||
device_code: String,
|
device_code: String,
|
||||||
user_code: String,
|
user_code: String,
|
||||||
verification_uri: Url,
|
verification_uri: Url,
|
||||||
expires_in: u64,
|
expires_in: u64,
|
||||||
interval: u64,
|
interval: u64,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
#[serde(rename_all = "snake_case", tag = "error")]
|
#[serde(rename_all = "snake_case", tag = "error")]
|
||||||
enum AccessTokenError {
|
enum AccessTokenError {
|
||||||
AuthorizationPending,
|
AuthorizationPending,
|
||||||
SlowDown { interval: u64 },
|
SlowDown { interval: u64 },
|
||||||
ExpiredToken,
|
ExpiredToken,
|
||||||
AccessDenied,
|
AccessDenied,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
#[serde(untagged)]
|
#[serde(untagged)]
|
||||||
enum AccessTokenResponse {
|
enum AccessTokenResponse {
|
||||||
Success { access_token: String },
|
Success { access_token: String },
|
||||||
|
|
||||||
Error(AccessTokenError),
|
Error(AccessTokenError),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl LoginCommand {
|
impl LoginCommand {
|
||||||
pub async fn authenticate_device_flow(
|
pub async fn authenticate_device_flow(
|
||||||
&self,
|
&self,
|
||||||
index_url: &gix::Url,
|
index_url: &gix::Url,
|
||||||
project: &Project,
|
project: &Project,
|
||||||
reqwest: &reqwest::Client,
|
reqwest: &reqwest::Client,
|
||||||
) -> anyhow::Result<String> {
|
) -> anyhow::Result<String> {
|
||||||
println!("logging in into {index_url}");
|
println!("logging in into {index_url}");
|
||||||
|
|
||||||
let source = PesdePackageSource::new(index_url.clone());
|
let source = PesdePackageSource::new(index_url.clone());
|
||||||
source
|
source
|
||||||
.refresh(&RefreshOptions {
|
.refresh(project)
|
||||||
project: project.clone(),
|
.await
|
||||||
})
|
.context("failed to refresh index")?;
|
||||||
.await
|
|
||||||
.context("failed to refresh index")?;
|
|
||||||
|
|
||||||
let config = source
|
let config = source
|
||||||
.config(project)
|
.config(project)
|
||||||
.await
|
.await
|
||||||
.context("failed to read index config")?;
|
.context("failed to read index config")?;
|
||||||
let Some(client_id) = config.github_oauth_client_id else {
|
let Some(client_id) = config.github_oauth_client_id else {
|
||||||
anyhow::bail!("index not configured for Github oauth.");
|
anyhow::bail!("index not configured for Github oauth.");
|
||||||
};
|
};
|
||||||
|
|
||||||
let response = reqwest
|
let response = reqwest
|
||||||
.post(Url::parse_with_params(
|
.post(Url::parse_with_params(
|
||||||
"https://github.com/login/device/code",
|
"https://github.com/login/device/code",
|
||||||
&[("client_id", &client_id)],
|
&[("client_id", &client_id)],
|
||||||
)?)
|
)?)
|
||||||
.send()
|
.send()
|
||||||
.await
|
.await
|
||||||
.context("failed to send device code request")?
|
.context("failed to send device code request")?
|
||||||
.error_for_status()
|
.error_for_status()
|
||||||
.context("failed to get device code response")?
|
.context("failed to get device code response")?
|
||||||
.json::<DeviceCodeResponse>()
|
.json::<DeviceCodeResponse>()
|
||||||
.await
|
.await
|
||||||
.context("failed to parse device code response")?;
|
.context("failed to parse device code response")?;
|
||||||
|
|
||||||
println!(
|
println!(
|
||||||
"copy your one-time code: {}\npress enter to open {} in your browser...",
|
"copy your one-time code: {}\npress enter to open {} in your browser...",
|
||||||
style(response.user_code).bold(),
|
response.user_code.bold(),
|
||||||
URL_STYLE.apply_to(response.verification_uri.as_str())
|
response.verification_uri.as_str().blue()
|
||||||
);
|
);
|
||||||
|
|
||||||
spawn(move || {
|
spawn(move || {
|
||||||
{
|
{
|
||||||
let mut input = String::new();
|
let mut input = String::new();
|
||||||
std::io::stdin()
|
std::io::stdin()
|
||||||
.read_line(&mut input)
|
.read_line(&mut input)
|
||||||
.expect("failed to read input");
|
.expect("failed to read input");
|
||||||
}
|
}
|
||||||
|
|
||||||
match open::that(response.verification_uri.as_str()) {
|
match open::that(response.verification_uri.as_str()) {
|
||||||
Ok(_) => (),
|
Ok(_) => (),
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
eprintln!("failed to open browser: {e}");
|
eprintln!("failed to open browser: {e}");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
let mut time_left = response.expires_in;
|
let mut time_left = response.expires_in;
|
||||||
let mut interval = std::time::Duration::from_secs(response.interval);
|
let mut interval = std::time::Duration::from_secs(response.interval);
|
||||||
|
|
||||||
while time_left > 0 {
|
while time_left > 0 {
|
||||||
sleep(interval).await;
|
sleep(interval).await;
|
||||||
time_left = time_left.saturating_sub(interval.as_secs());
|
time_left = time_left.saturating_sub(interval.as_secs());
|
||||||
|
|
||||||
let response = reqwest
|
let response = reqwest
|
||||||
.post(Url::parse_with_params(
|
.post(Url::parse_with_params(
|
||||||
"https://github.com/login/oauth/access_token",
|
"https://github.com/login/oauth/access_token",
|
||||||
&[
|
&[
|
||||||
("client_id", &client_id),
|
("client_id", &client_id),
|
||||||
("device_code", &response.device_code),
|
("device_code", &response.device_code),
|
||||||
(
|
(
|
||||||
"grant_type",
|
"grant_type",
|
||||||
&"urn:ietf:params:oauth:grant-type:device_code".to_string(),
|
&"urn:ietf:params:oauth:grant-type:device_code".to_string(),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
)?)
|
)?)
|
||||||
.send()
|
.send()
|
||||||
.await
|
.await
|
||||||
.context("failed to send access token request")?
|
.context("failed to send access token request")?
|
||||||
.error_for_status()
|
.error_for_status()
|
||||||
.context("failed to get access token response")?
|
.context("failed to get access token response")?
|
||||||
.json::<AccessTokenResponse>()
|
.json::<AccessTokenResponse>()
|
||||||
.await
|
.await
|
||||||
.context("failed to parse access token response")?;
|
.context("failed to parse access token response")?;
|
||||||
|
|
||||||
match response {
|
match response {
|
||||||
AccessTokenResponse::Success { access_token } => {
|
AccessTokenResponse::Success { access_token } => {
|
||||||
return Ok(access_token);
|
return Ok(access_token);
|
||||||
}
|
}
|
||||||
AccessTokenResponse::Error(e) => match e {
|
AccessTokenResponse::Error(e) => match e {
|
||||||
AccessTokenError::AuthorizationPending => continue,
|
AccessTokenError::AuthorizationPending => continue,
|
||||||
AccessTokenError::SlowDown {
|
AccessTokenError::SlowDown {
|
||||||
interval: new_interval,
|
interval: new_interval,
|
||||||
} => {
|
} => {
|
||||||
interval = std::time::Duration::from_secs(new_interval);
|
interval = std::time::Duration::from_secs(new_interval);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
AccessTokenError::ExpiredToken => {
|
AccessTokenError::ExpiredToken => {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
AccessTokenError::AccessDenied => {
|
AccessTokenError::AccessDenied => {
|
||||||
anyhow::bail!("access denied, re-run the login command");
|
anyhow::bail!("access denied, re-run the login command");
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
anyhow::bail!("code expired, please re-run the login command");
|
anyhow::bail!("code expired, please re-run the login command");
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn run(
|
pub async fn run(
|
||||||
self,
|
self,
|
||||||
index_url: gix::Url,
|
index_url: gix::Url,
|
||||||
project: Project,
|
project: Project,
|
||||||
reqwest: reqwest::Client,
|
reqwest: reqwest::Client,
|
||||||
) -> anyhow::Result<()> {
|
) -> anyhow::Result<()> {
|
||||||
let token_given = self.token.is_some();
|
let token_given = self.token.is_some();
|
||||||
let token = match self.token {
|
let token = match self.token {
|
||||||
Some(token) => token,
|
Some(token) => token,
|
||||||
None => {
|
None => {
|
||||||
self.authenticate_device_flow(&index_url, &project, &reqwest)
|
self.authenticate_device_flow(&index_url, &project, &reqwest)
|
||||||
.await?
|
.await?
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let token = if token_given {
|
let token = if token_given {
|
||||||
println!("set token for {index_url}");
|
println!("set token for {index_url}");
|
||||||
token
|
token
|
||||||
} else {
|
} else {
|
||||||
let token = format!("Bearer {token}");
|
let token = format!("Bearer {token}");
|
||||||
println!(
|
println!(
|
||||||
"logged in as {} for {index_url}",
|
"logged in as {} for {index_url}",
|
||||||
style(get_token_login(&reqwest, &token).await?).bold()
|
get_token_login(&reqwest, &token).await?.bold()
|
||||||
);
|
);
|
||||||
|
|
||||||
token
|
token
|
||||||
};
|
};
|
||||||
|
|
||||||
set_token(&index_url, Some(&token)).await?;
|
set_token(&index_url, Some(&token)).await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -5,11 +5,11 @@ use clap::Args;
|
||||||
pub struct LogoutCommand {}
|
pub struct LogoutCommand {}
|
||||||
|
|
||||||
impl LogoutCommand {
|
impl LogoutCommand {
|
||||||
pub async fn run(self, index_url: gix::Url) -> anyhow::Result<()> {
|
pub async fn run(self, index_url: gix::Url) -> anyhow::Result<()> {
|
||||||
set_token(&index_url, None).await?;
|
set_token(&index_url, None).await?;
|
||||||
|
|
||||||
println!("logged out of {index_url}");
|
println!("logged out of {index_url}");
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
use crate::cli::get_index;
|
use crate::cli::config::read_config;
|
||||||
use clap::{Args, Subcommand};
|
use clap::{Args, Subcommand};
|
||||||
use pesde::Project;
|
use pesde::{errors::ManifestReadError, Project, DEFAULT_INDEX_NAME};
|
||||||
|
|
||||||
mod login;
|
mod login;
|
||||||
mod logout;
|
mod logout;
|
||||||
|
@ -9,36 +9,65 @@ mod whoami;
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct AuthSubcommand {
|
pub struct AuthSubcommand {
|
||||||
/// The index to use. Defaults to `default`, or the configured default index if current directory doesn't have a manifest
|
/// The index to use. Defaults to `default`, or the configured default index if current directory doesn't have a manifest
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
pub index: Option<String>,
|
pub index: Option<String>,
|
||||||
|
|
||||||
#[clap(subcommand)]
|
#[clap(subcommand)]
|
||||||
pub command: AuthCommands,
|
pub command: AuthCommands,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Subcommand)]
|
#[derive(Debug, Subcommand)]
|
||||||
pub enum AuthCommands {
|
pub enum AuthCommands {
|
||||||
/// Sets a token for an index. Optionally gets it from GitHub
|
/// Sets a token for an index. Optionally gets it from GitHub
|
||||||
Login(login::LoginCommand),
|
Login(login::LoginCommand),
|
||||||
/// Removes the stored token
|
/// Removes the stored token
|
||||||
Logout(logout::LogoutCommand),
|
Logout(logout::LogoutCommand),
|
||||||
/// Prints the username of the currently logged-in user
|
/// Prints the username of the currently logged-in user
|
||||||
#[clap(name = "whoami")]
|
#[clap(name = "whoami")]
|
||||||
WhoAmI(whoami::WhoAmICommand),
|
WhoAmI(whoami::WhoAmICommand),
|
||||||
/// Prints the token for an index
|
/// Prints the token for an index
|
||||||
Token(token::TokenCommand),
|
Token(token::TokenCommand),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AuthSubcommand {
|
impl AuthSubcommand {
|
||||||
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
||||||
let index_url = get_index(&project, self.index.as_deref()).await?;
|
let manifest = match project.deser_manifest().await {
|
||||||
|
Ok(manifest) => Some(manifest),
|
||||||
|
Err(e) => match e {
|
||||||
|
ManifestReadError::Io(e) if e.kind() == std::io::ErrorKind::NotFound => None,
|
||||||
|
e => return Err(e.into()),
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
match self.command {
|
let index_url = match self.index.as_deref() {
|
||||||
AuthCommands::Login(login) => login.run(index_url, project, reqwest).await,
|
Some(index) => match index.try_into() {
|
||||||
AuthCommands::Logout(logout) => logout.run(index_url).await,
|
Ok(url) => Some(url),
|
||||||
AuthCommands::WhoAmI(whoami) => whoami.run(index_url, reqwest).await,
|
Err(_) => None,
|
||||||
AuthCommands::Token(token) => token.run(index_url).await,
|
},
|
||||||
}
|
None => match manifest {
|
||||||
}
|
Some(_) => None,
|
||||||
|
None => Some(read_config().await?.default_index),
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
let index_url = match index_url {
|
||||||
|
Some(url) => url,
|
||||||
|
None => {
|
||||||
|
let index_name = self.index.as_deref().unwrap_or(DEFAULT_INDEX_NAME);
|
||||||
|
|
||||||
|
match manifest.unwrap().indices.get(index_name) {
|
||||||
|
Some(index) => index.clone(),
|
||||||
|
None => anyhow::bail!("index {index_name} not found in manifest"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
match self.command {
|
||||||
|
AuthCommands::Login(login) => login.run(index_url, project, reqwest).await,
|
||||||
|
AuthCommands::Logout(logout) => logout.run(index_url).await,
|
||||||
|
AuthCommands::WhoAmI(whoami) => whoami.run(index_url, reqwest).await,
|
||||||
|
AuthCommands::Token(token) => token.run(index_url).await,
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -5,18 +5,18 @@ use clap::Args;
|
||||||
pub struct TokenCommand {}
|
pub struct TokenCommand {}
|
||||||
|
|
||||||
impl TokenCommand {
|
impl TokenCommand {
|
||||||
pub async fn run(self, index_url: gix::Url) -> anyhow::Result<()> {
|
pub async fn run(self, index_url: gix::Url) -> anyhow::Result<()> {
|
||||||
let tokens = get_tokens().await?;
|
let tokens = get_tokens().await?;
|
||||||
let token = match tokens.0.get(&index_url) {
|
let token = match tokens.0.get(&index_url) {
|
||||||
Some(token) => token,
|
Some(token) => token,
|
||||||
None => {
|
None => {
|
||||||
println!("not logged in into {index_url}");
|
println!("not logged in into {index_url}");
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
println!("token for {index_url}: \"{token}\"");
|
println!("token for {index_url}: \"{token}\"");
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,26 +1,26 @@
|
||||||
use crate::cli::auth::{get_token_login, get_tokens};
|
use crate::cli::auth::{get_token_login, get_tokens};
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use console::style;
|
use colored::Colorize;
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct WhoAmICommand {}
|
pub struct WhoAmICommand {}
|
||||||
|
|
||||||
impl WhoAmICommand {
|
impl WhoAmICommand {
|
||||||
pub async fn run(self, index_url: gix::Url, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
pub async fn run(self, index_url: gix::Url, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
||||||
let tokens = get_tokens().await?;
|
let tokens = get_tokens().await?;
|
||||||
let token = match tokens.0.get(&index_url) {
|
let token = match tokens.0.get(&index_url) {
|
||||||
Some(token) => token,
|
Some(token) => token,
|
||||||
None => {
|
None => {
|
||||||
println!("not logged in into {index_url}");
|
println!("not logged in into {index_url}");
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
println!(
|
println!(
|
||||||
"logged in as {} into {index_url}",
|
"logged in as {} into {index_url}",
|
||||||
style(get_token_login(&reqwest, token).await?).bold()
|
get_token_login(&reqwest, token).await?.bold()
|
||||||
);
|
);
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,18 +0,0 @@
|
||||||
use clap::Subcommand;
|
|
||||||
use pesde::Project;
|
|
||||||
|
|
||||||
mod prune;
|
|
||||||
|
|
||||||
#[derive(Debug, Subcommand)]
|
|
||||||
pub enum CasCommands {
|
|
||||||
/// Removes unused files from the CAS
|
|
||||||
Prune(prune::PruneCommand),
|
|
||||||
}
|
|
||||||
|
|
||||||
impl CasCommands {
|
|
||||||
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
|
||||||
match self {
|
|
||||||
CasCommands::Prune(prune) => prune.run(project).await,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,346 +0,0 @@
|
||||||
use crate::{
|
|
||||||
cli::{
|
|
||||||
reporters::run_with_reporter,
|
|
||||||
style::{INFO_STYLE, SUCCESS_STYLE},
|
|
||||||
},
|
|
||||||
util::remove_empty_dir,
|
|
||||||
};
|
|
||||||
use anyhow::Context;
|
|
||||||
use async_stream::try_stream;
|
|
||||||
use clap::Args;
|
|
||||||
use fs_err::tokio as fs;
|
|
||||||
use futures::{future::BoxFuture, FutureExt, Stream, StreamExt};
|
|
||||||
use pesde::{
|
|
||||||
source::fs::{FsEntry, PackageFs},
|
|
||||||
Project,
|
|
||||||
};
|
|
||||||
use std::{
|
|
||||||
collections::{HashMap, HashSet},
|
|
||||||
future::Future,
|
|
||||||
path::{Path, PathBuf},
|
|
||||||
};
|
|
||||||
use tokio::task::JoinSet;
|
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
|
||||||
pub struct PruneCommand {}
|
|
||||||
|
|
||||||
async fn read_dir_stream(
|
|
||||||
dir: &Path,
|
|
||||||
) -> std::io::Result<impl Stream<Item = std::io::Result<fs::DirEntry>>> {
|
|
||||||
let mut read_dir = fs::read_dir(dir).await?;
|
|
||||||
|
|
||||||
Ok(try_stream! {
|
|
||||||
while let Some(entry) = read_dir.next_entry().await? {
|
|
||||||
yield entry;
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
#[allow(unreachable_code)]
|
|
||||||
async fn get_nlinks(path: &Path) -> anyhow::Result<u64> {
|
|
||||||
#[cfg(unix)]
|
|
||||||
{
|
|
||||||
use std::os::unix::fs::MetadataExt;
|
|
||||||
let metadata = fs::metadata(path).await?;
|
|
||||||
return Ok(metadata.nlink());
|
|
||||||
}
|
|
||||||
// life if rust stabilized the nightly feature from 2019
|
|
||||||
#[cfg(windows)]
|
|
||||||
{
|
|
||||||
use std::os::windows::ffi::OsStrExt;
|
|
||||||
use windows::{
|
|
||||||
core::PWSTR,
|
|
||||||
Win32::{
|
|
||||||
Foundation::CloseHandle,
|
|
||||||
Storage::FileSystem::{
|
|
||||||
CreateFileW, GetFileInformationByHandle, FILE_ATTRIBUTE_NORMAL,
|
|
||||||
FILE_GENERIC_READ, FILE_SHARE_READ, OPEN_EXISTING,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
let path = path.to_path_buf();
|
|
||||||
return tokio::task::spawn_blocking(move || unsafe {
|
|
||||||
let handle = CreateFileW(
|
|
||||||
PWSTR(
|
|
||||||
path.as_os_str()
|
|
||||||
.encode_wide()
|
|
||||||
.chain(std::iter::once(0))
|
|
||||||
.collect::<Vec<_>>()
|
|
||||||
.as_mut_ptr(),
|
|
||||||
),
|
|
||||||
FILE_GENERIC_READ.0,
|
|
||||||
FILE_SHARE_READ,
|
|
||||||
None,
|
|
||||||
OPEN_EXISTING,
|
|
||||||
FILE_ATTRIBUTE_NORMAL,
|
|
||||||
None,
|
|
||||||
)?;
|
|
||||||
|
|
||||||
let mut info =
|
|
||||||
windows::Win32::Storage::FileSystem::BY_HANDLE_FILE_INFORMATION::default();
|
|
||||||
let res = GetFileInformationByHandle(handle, &mut info);
|
|
||||||
CloseHandle(handle)?;
|
|
||||||
res?;
|
|
||||||
|
|
||||||
Ok(info.nNumberOfLinks as u64)
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
#[cfg(not(any(unix, windows)))]
|
|
||||||
{
|
|
||||||
compile_error!("unsupported platform");
|
|
||||||
}
|
|
||||||
anyhow::bail!("unsupported platform")
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
struct ExtendJoinSet<T: Send + 'static>(JoinSet<T>);
|
|
||||||
|
|
||||||
impl<T: Send + 'static, F: Future<Output = T> + Send + 'static> Extend<F> for ExtendJoinSet<T> {
|
|
||||||
fn extend<I: IntoIterator<Item = F>>(&mut self, iter: I) {
|
|
||||||
for item in iter {
|
|
||||||
self.0.spawn(item);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<T: Send + 'static> Default for ExtendJoinSet<T> {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self(JoinSet::new())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn discover_cas_packages(cas_dir: &Path) -> anyhow::Result<HashMap<PathBuf, PackageFs>> {
|
|
||||||
fn read_entry(
|
|
||||||
entry: fs::DirEntry,
|
|
||||||
) -> BoxFuture<'static, anyhow::Result<HashMap<PathBuf, PackageFs>>> {
|
|
||||||
async move {
|
|
||||||
if entry
|
|
||||||
.metadata()
|
|
||||||
.await
|
|
||||||
.context("failed to read entry metadata")?
|
|
||||||
.is_dir()
|
|
||||||
{
|
|
||||||
let mut tasks = read_dir_stream(&entry.path())
|
|
||||||
.await
|
|
||||||
.context("failed to read entry directory")?
|
|
||||||
.map(|entry| async move {
|
|
||||||
read_entry(entry.context("failed to read inner cas index dir entry")?).await
|
|
||||||
})
|
|
||||||
.collect::<ExtendJoinSet<Result<_, anyhow::Error>>>()
|
|
||||||
.await
|
|
||||||
.0;
|
|
||||||
|
|
||||||
let mut res = HashMap::new();
|
|
||||||
while let Some(entry) = tasks.join_next().await {
|
|
||||||
res.extend(entry.unwrap()?);
|
|
||||||
}
|
|
||||||
|
|
||||||
return Ok(res);
|
|
||||||
};
|
|
||||||
|
|
||||||
let contents = fs::read_to_string(entry.path()).await?;
|
|
||||||
let fs = toml::from_str(&contents).context("failed to deserialize PackageFs")?;
|
|
||||||
|
|
||||||
Ok(HashMap::from([(entry.path(), fs)]))
|
|
||||||
}
|
|
||||||
.boxed()
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut tasks = ["index", "wally_index", "git_index"]
|
|
||||||
.into_iter()
|
|
||||||
.map(|index| cas_dir.join(index))
|
|
||||||
.map(|index| async move {
|
|
||||||
let mut res = HashMap::new();
|
|
||||||
|
|
||||||
let tasks = match read_dir_stream(&index).await {
|
|
||||||
Ok(tasks) => tasks,
|
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(res),
|
|
||||||
Err(e) => return Err(e).context("failed to read cas index directory"),
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut tasks = tasks
|
|
||||||
.map(|entry| async move {
|
|
||||||
read_entry(entry.context("failed to read cas index dir entry")?).await
|
|
||||||
})
|
|
||||||
.collect::<ExtendJoinSet<Result<_, anyhow::Error>>>()
|
|
||||||
.await
|
|
||||||
.0;
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
res.extend(task.unwrap()?);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(res)
|
|
||||||
})
|
|
||||||
.collect::<JoinSet<Result<_, anyhow::Error>>>();
|
|
||||||
|
|
||||||
let mut cas_entries = HashMap::new();
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
cas_entries.extend(task.unwrap()?);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(cas_entries)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn remove_hashes(cas_dir: &Path) -> anyhow::Result<HashSet<String>> {
|
|
||||||
let mut res = HashSet::new();
|
|
||||||
|
|
||||||
let tasks = match read_dir_stream(cas_dir).await {
|
|
||||||
Ok(tasks) => tasks,
|
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(res),
|
|
||||||
Err(e) => return Err(e).context("failed to read cas directory"),
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut tasks = tasks
|
|
||||||
.map(|cas_entry| async move {
|
|
||||||
let cas_entry = cas_entry.context("failed to read cas dir entry")?;
|
|
||||||
let prefix = cas_entry.file_name();
|
|
||||||
let Some(prefix) = prefix.to_str() else {
|
|
||||||
return Ok(None);
|
|
||||||
};
|
|
||||||
// we only want hash directories
|
|
||||||
if prefix.len() != 2 {
|
|
||||||
return Ok(None);
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut tasks = read_dir_stream(&cas_entry.path())
|
|
||||||
.await
|
|
||||||
.context("failed to read hash directory")?
|
|
||||||
.map(|hash_entry| {
|
|
||||||
let prefix = prefix.to_string();
|
|
||||||
async move {
|
|
||||||
let hash_entry = hash_entry.context("failed to read hash dir entry")?;
|
|
||||||
let hash = hash_entry.file_name();
|
|
||||||
let hash = hash.to_str().expect("non-UTF-8 hash").to_string();
|
|
||||||
let hash = format!("{prefix}{hash}");
|
|
||||||
|
|
||||||
let path = hash_entry.path();
|
|
||||||
let nlinks = get_nlinks(&path)
|
|
||||||
.await
|
|
||||||
.context("failed to count file usage")?;
|
|
||||||
if nlinks > 1 {
|
|
||||||
return Ok(None);
|
|
||||||
}
|
|
||||||
|
|
||||||
fs::remove_file(&path)
|
|
||||||
.await
|
|
||||||
.context("failed to remove unused file")?;
|
|
||||||
|
|
||||||
if let Some(parent) = path.parent() {
|
|
||||||
remove_empty_dir(parent).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(Some(hash))
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect::<ExtendJoinSet<Result<_, anyhow::Error>>>()
|
|
||||||
.await
|
|
||||||
.0;
|
|
||||||
|
|
||||||
let mut removed_hashes = HashSet::new();
|
|
||||||
while let Some(removed_hash) = tasks.join_next().await {
|
|
||||||
let Some(hash) = removed_hash.unwrap()? else {
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
|
|
||||||
removed_hashes.insert(hash);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(Some(removed_hashes))
|
|
||||||
})
|
|
||||||
.collect::<ExtendJoinSet<Result<_, anyhow::Error>>>()
|
|
||||||
.await
|
|
||||||
.0;
|
|
||||||
|
|
||||||
while let Some(removed_hashes) = tasks.join_next().await {
|
|
||||||
let Some(removed_hashes) = removed_hashes.unwrap()? else {
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
|
|
||||||
res.extend(removed_hashes);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(res)
|
|
||||||
}
|
|
||||||
|
|
||||||
impl PruneCommand {
|
|
||||||
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
|
||||||
// CAS structure:
|
|
||||||
// /2 first chars of hash/rest of hash
|
|
||||||
// /index/hash/name/version/target
|
|
||||||
// /wally_index/hash/name/version
|
|
||||||
// /git_index/hash/hash
|
|
||||||
// the last thing in the path is the serialized PackageFs
|
|
||||||
|
|
||||||
let (cas_entries, removed_hashes) = run_with_reporter(|_, root_progress, _| async {
|
|
||||||
let root_progress = root_progress;
|
|
||||||
root_progress.reset();
|
|
||||||
root_progress.set_message("discover packages");
|
|
||||||
let cas_entries = discover_cas_packages(project.cas_dir()).await?;
|
|
||||||
root_progress.reset();
|
|
||||||
root_progress.set_message("remove unused files");
|
|
||||||
let removed_hashes = remove_hashes(project.cas_dir()).await?;
|
|
||||||
|
|
||||||
Ok::<_, anyhow::Error>((cas_entries, removed_hashes))
|
|
||||||
})
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let mut tasks = JoinSet::new();
|
|
||||||
|
|
||||||
let mut removed_packages = 0usize;
|
|
||||||
|
|
||||||
'entry: for (path, fs) in cas_entries {
|
|
||||||
let PackageFs::Cas(entries) = fs else {
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
|
|
||||||
for entry in entries.into_values() {
|
|
||||||
let FsEntry::File(hash) = entry else {
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
|
|
||||||
if removed_hashes.contains(&hash) {
|
|
||||||
let cas_dir = project.cas_dir().to_path_buf();
|
|
||||||
tasks.spawn(async move {
|
|
||||||
fs::remove_file(&path)
|
|
||||||
.await
|
|
||||||
.context("failed to remove unused file")?;
|
|
||||||
|
|
||||||
// remove empty directories up to the cas dir
|
|
||||||
let mut path = &*path;
|
|
||||||
while let Some(parent) = path.parent() {
|
|
||||||
if parent == cas_dir {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
remove_empty_dir(parent).await?;
|
|
||||||
path = parent;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok::<_, anyhow::Error>(())
|
|
||||||
});
|
|
||||||
removed_packages += 1;
|
|
||||||
// if at least one file is removed, the package is not used
|
|
||||||
continue 'entry;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
task.unwrap()?;
|
|
||||||
}
|
|
||||||
|
|
||||||
println!(
|
|
||||||
"{} removed {} unused packages and {} individual files!",
|
|
||||||
SUCCESS_STYLE.apply_to("done!"),
|
|
||||||
INFO_STYLE.apply_to(removed_packages),
|
|
||||||
INFO_STYLE.apply_to(removed_hashes.len())
|
|
||||||
);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -3,36 +3,36 @@ use clap::Args;
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct DefaultIndexCommand {
|
pub struct DefaultIndexCommand {
|
||||||
/// The new index URL to set as default, don't pass any value to check the current default index
|
/// The new index URL to set as default, don't pass any value to check the current default index
|
||||||
#[arg(index = 1, value_parser = crate::cli::parse_gix_url)]
|
#[arg(index = 1, value_parser = crate::cli::parse_gix_url)]
|
||||||
index: Option<gix::Url>,
|
index: Option<gix::Url>,
|
||||||
|
|
||||||
/// Resets the default index to the default value
|
/// Resets the default index to the default value
|
||||||
#[arg(short, long, conflicts_with = "index")]
|
#[arg(short, long, conflicts_with = "index")]
|
||||||
reset: bool,
|
reset: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl DefaultIndexCommand {
|
impl DefaultIndexCommand {
|
||||||
pub async fn run(self) -> anyhow::Result<()> {
|
pub async fn run(self) -> anyhow::Result<()> {
|
||||||
let mut config = read_config().await?;
|
let mut config = read_config().await?;
|
||||||
|
|
||||||
let index = if self.reset {
|
let index = if self.reset {
|
||||||
Some(CliConfig::default().default_index)
|
Some(CliConfig::default().default_index)
|
||||||
} else {
|
} else {
|
||||||
self.index
|
self.index
|
||||||
};
|
};
|
||||||
|
|
||||||
match index {
|
match index {
|
||||||
Some(index) => {
|
Some(index) => {
|
||||||
config.default_index = index.clone();
|
config.default_index = index.clone();
|
||||||
write_config(&config).await?;
|
write_config(&config).await?;
|
||||||
println!("default index set to: {index}");
|
println!("default index set to: {index}");
|
||||||
}
|
}
|
||||||
None => {
|
None => {
|
||||||
println!("current default index: {}", config.default_index);
|
println!("current default index: {}", config.default_index);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -4,14 +4,14 @@ mod default_index;
|
||||||
|
|
||||||
#[derive(Debug, Subcommand)]
|
#[derive(Debug, Subcommand)]
|
||||||
pub enum ConfigCommands {
|
pub enum ConfigCommands {
|
||||||
/// Configuration for the default index
|
/// Configuration for the default index
|
||||||
DefaultIndex(default_index::DefaultIndexCommand),
|
DefaultIndex(default_index::DefaultIndexCommand),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ConfigCommands {
|
impl ConfigCommands {
|
||||||
pub async fn run(self) -> anyhow::Result<()> {
|
pub async fn run(self) -> anyhow::Result<()> {
|
||||||
match self {
|
match self {
|
||||||
ConfigCommands::DefaultIndex(default_index) => default_index.run().await,
|
ConfigCommands::DefaultIndex(default_index) => default_index.run().await,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,100 +0,0 @@
|
||||||
use crate::cli::{get_index, style::SUCCESS_STYLE};
|
|
||||||
use anyhow::Context;
|
|
||||||
use clap::Args;
|
|
||||||
use pesde::{
|
|
||||||
names::PackageName,
|
|
||||||
source::{
|
|
||||||
pesde::PesdePackageSource,
|
|
||||||
traits::{PackageSource, RefreshOptions},
|
|
||||||
},
|
|
||||||
Project,
|
|
||||||
};
|
|
||||||
use reqwest::{header::AUTHORIZATION, Method, StatusCode};
|
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
|
||||||
pub struct DeprecateCommand {
|
|
||||||
/// Whether to undeprecate the package
|
|
||||||
#[clap(long)]
|
|
||||||
undo: bool,
|
|
||||||
|
|
||||||
/// The index to deprecate the package in
|
|
||||||
#[clap(short, long)]
|
|
||||||
index: Option<String>,
|
|
||||||
|
|
||||||
/// The package to deprecate
|
|
||||||
#[clap(index = 1)]
|
|
||||||
package: PackageName,
|
|
||||||
|
|
||||||
/// The reason for deprecating the package
|
|
||||||
#[clap(index = 2, required_unless_present = "undo")]
|
|
||||||
reason: Option<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DeprecateCommand {
|
|
||||||
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
|
||||||
let index_url = get_index(&project, self.index.as_deref()).await?;
|
|
||||||
let source = PesdePackageSource::new(index_url.clone());
|
|
||||||
source
|
|
||||||
.refresh(&RefreshOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.context("failed to refresh source")?;
|
|
||||||
let config = source
|
|
||||||
.config(&project)
|
|
||||||
.await
|
|
||||||
.context("failed to get index config")?;
|
|
||||||
|
|
||||||
let mut request = reqwest.request(
|
|
||||||
if self.undo {
|
|
||||||
Method::DELETE
|
|
||||||
} else {
|
|
||||||
Method::PUT
|
|
||||||
},
|
|
||||||
format!(
|
|
||||||
"{}/v1/packages/{}/deprecate",
|
|
||||||
config.api(),
|
|
||||||
urlencoding::encode(&self.package.to_string()),
|
|
||||||
),
|
|
||||||
);
|
|
||||||
|
|
||||||
if !self.undo {
|
|
||||||
request = request.body(
|
|
||||||
self.reason
|
|
||||||
.map(|reason| reason.trim().to_string())
|
|
||||||
.filter(|reason| !reason.is_empty())
|
|
||||||
.context("deprecating must have non-empty a reason")?,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(token) = project.auth_config().tokens().get(&index_url) {
|
|
||||||
tracing::debug!("using token for {index_url}");
|
|
||||||
request = request.header(AUTHORIZATION, token);
|
|
||||||
}
|
|
||||||
|
|
||||||
let response = request.send().await.context("failed to send request")?;
|
|
||||||
|
|
||||||
let status = response.status();
|
|
||||||
let text = response
|
|
||||||
.text()
|
|
||||||
.await
|
|
||||||
.context("failed to get response text")?;
|
|
||||||
let prefix = if self.undo { "un" } else { "" };
|
|
||||||
match status {
|
|
||||||
StatusCode::CONFLICT => {
|
|
||||||
anyhow::bail!("version is already {prefix}deprecated");
|
|
||||||
}
|
|
||||||
StatusCode::FORBIDDEN => {
|
|
||||||
anyhow::bail!("unauthorized to {prefix}deprecate under this scope");
|
|
||||||
}
|
|
||||||
code if !code.is_success() => {
|
|
||||||
anyhow::bail!("failed to {prefix}deprecate package: {code} ({text})");
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
println!("{}", SUCCESS_STYLE.apply_to(text));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,242 +1,167 @@
|
||||||
use crate::cli::{
|
use crate::cli::{config::read_config, progress_bar, VersionedPackageName};
|
||||||
config::read_config,
|
|
||||||
reporters::{self, CliReporter},
|
|
||||||
VersionedPackageName,
|
|
||||||
};
|
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use console::style;
|
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use indicatif::MultiProgress;
|
|
||||||
use pesde::{
|
use pesde::{
|
||||||
download_and_link::DownloadAndLinkOptions,
|
linking::generator::generate_bin_linking_module,
|
||||||
linking::generator::generate_bin_linking_module,
|
manifest::target::TargetKind,
|
||||||
manifest::target::TargetKind,
|
names::PackageName,
|
||||||
names::{PackageName, PackageNames},
|
source::{
|
||||||
source::{
|
pesde::{specifier::PesdeDependencySpecifier, PesdePackageSource},
|
||||||
ids::PackageId,
|
traits::PackageSource,
|
||||||
pesde::{specifier::PesdeDependencySpecifier, PesdePackageSource},
|
},
|
||||||
traits::{
|
Project,
|
||||||
DownloadOptions, GetTargetOptions, PackageSource, RefreshOptions, ResolveOptions,
|
|
||||||
},
|
|
||||||
PackageSources,
|
|
||||||
},
|
|
||||||
Project, RefreshedSources,
|
|
||||||
};
|
};
|
||||||
use semver::VersionReq;
|
use semver::VersionReq;
|
||||||
use std::{
|
use std::{
|
||||||
env::current_dir,
|
collections::HashSet, env::current_dir, ffi::OsString, io::Write, process::Command, sync::Arc,
|
||||||
ffi::OsString,
|
|
||||||
io::{Stderr, Write},
|
|
||||||
process::Command,
|
|
||||||
sync::Arc,
|
|
||||||
};
|
};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct ExecuteCommand {
|
pub struct ExecuteCommand {
|
||||||
/// The package name, script name, or path to a script to run
|
/// The package name, script name, or path to a script to run
|
||||||
#[arg(index = 1)]
|
#[arg(index = 1)]
|
||||||
package: VersionedPackageName<VersionReq, PackageName>,
|
package: VersionedPackageName<VersionReq, PackageName>,
|
||||||
|
|
||||||
/// The index URL to use for the package
|
/// The index URL to use for the package
|
||||||
#[arg(short, long, value_parser = crate::cli::parse_gix_url)]
|
#[arg(short, long, value_parser = crate::cli::parse_gix_url)]
|
||||||
index: Option<gix::Url>,
|
index: Option<gix::Url>,
|
||||||
|
|
||||||
/// Arguments to pass to the script
|
/// Arguments to pass to the script
|
||||||
#[arg(index = 2, last = true)]
|
#[arg(index = 2, last = true)]
|
||||||
args: Vec<OsString>,
|
args: Vec<OsString>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ExecuteCommand {
|
impl ExecuteCommand {
|
||||||
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
||||||
let multi_progress = MultiProgress::new();
|
let index = match self.index {
|
||||||
crate::PROGRESS_BARS
|
Some(index) => Some(index),
|
||||||
.lock()
|
None => read_config().await.ok().map(|c| c.default_index),
|
||||||
.unwrap()
|
}
|
||||||
.replace(multi_progress.clone());
|
.context("no index specified")?;
|
||||||
|
let source = PesdePackageSource::new(index);
|
||||||
|
source
|
||||||
|
.refresh(&project)
|
||||||
|
.await
|
||||||
|
.context("failed to refresh source")?;
|
||||||
|
|
||||||
let refreshed_sources = RefreshedSources::new();
|
let version_req = self.package.1.unwrap_or(VersionReq::STAR);
|
||||||
|
let Some((version, pkg_ref)) = ('finder: {
|
||||||
|
let specifier = PesdeDependencySpecifier {
|
||||||
|
name: self.package.0.clone(),
|
||||||
|
version: version_req.clone(),
|
||||||
|
index: None,
|
||||||
|
target: None,
|
||||||
|
};
|
||||||
|
|
||||||
let (tempdir, bin_path) = reporters::run_with_reporter_and_writer(
|
if let Some(res) = source
|
||||||
std::io::stderr(),
|
.resolve(&specifier, &project, TargetKind::Lune, &mut HashSet::new())
|
||||||
|multi_progress, root_progress, reporter| async {
|
.await
|
||||||
let multi_progress = multi_progress;
|
.context("failed to resolve package")?
|
||||||
let root_progress = root_progress;
|
.1
|
||||||
|
.pop_last()
|
||||||
|
{
|
||||||
|
break 'finder Some(res);
|
||||||
|
}
|
||||||
|
|
||||||
root_progress.set_message("resolve");
|
source
|
||||||
|
.resolve(&specifier, &project, TargetKind::Luau, &mut HashSet::new())
|
||||||
|
.await
|
||||||
|
.context("failed to resolve package")?
|
||||||
|
.1
|
||||||
|
.pop_last()
|
||||||
|
}) else {
|
||||||
|
anyhow::bail!(
|
||||||
|
"no Lune or Luau package could be found for {}@{version_req}",
|
||||||
|
self.package.0,
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
let index = match self.index {
|
println!("using {}@{version}", pkg_ref.name);
|
||||||
Some(index) => Some(index),
|
|
||||||
None => read_config().await.ok().map(|c| c.default_index),
|
|
||||||
}
|
|
||||||
.context("no index specified")?;
|
|
||||||
let source = PesdePackageSource::new(index);
|
|
||||||
refreshed_sources
|
|
||||||
.refresh(
|
|
||||||
&PackageSources::Pesde(source.clone()),
|
|
||||||
&RefreshOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to refresh source")?;
|
|
||||||
|
|
||||||
let version_req = self.package.1.unwrap_or(VersionReq::STAR);
|
let tmp_dir = project.cas_dir().join(".tmp");
|
||||||
let Some((id, pkg_ref)) = ('finder: {
|
fs::create_dir_all(&tmp_dir)
|
||||||
let specifier = PesdeDependencySpecifier {
|
.await
|
||||||
name: self.package.0.clone(),
|
.context("failed to create temporary directory")?;
|
||||||
version: version_req.clone(),
|
let tempdir =
|
||||||
index: None,
|
tempfile::tempdir_in(tmp_dir).context("failed to create temporary directory")?;
|
||||||
target: None,
|
|
||||||
};
|
|
||||||
|
|
||||||
if let Some((v_id, pkg_ref)) = source
|
let project = Project::new(
|
||||||
.resolve(
|
tempdir.path(),
|
||||||
&specifier,
|
None::<std::path::PathBuf>,
|
||||||
&ResolveOptions {
|
project.data_dir(),
|
||||||
project: project.clone(),
|
project.cas_dir(),
|
||||||
target: TargetKind::Lune,
|
project.auth_config().clone(),
|
||||||
refreshed_sources: refreshed_sources.clone(),
|
);
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to resolve package")?
|
|
||||||
.1
|
|
||||||
.pop_last()
|
|
||||||
{
|
|
||||||
break 'finder Some((
|
|
||||||
PackageId::new(PackageNames::Pesde(self.package.0.clone()), v_id),
|
|
||||||
pkg_ref,
|
|
||||||
));
|
|
||||||
}
|
|
||||||
|
|
||||||
source
|
let (fs, target) = source
|
||||||
.resolve(
|
.download(&pkg_ref, &project, &reqwest)
|
||||||
&specifier,
|
.await
|
||||||
&ResolveOptions {
|
.context("failed to download package")?;
|
||||||
project: project.clone(),
|
let bin_path = target.bin_path().context("package has no binary export")?;
|
||||||
target: TargetKind::Luau,
|
|
||||||
refreshed_sources: refreshed_sources.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to resolve package")?
|
|
||||||
.1
|
|
||||||
.pop_last()
|
|
||||||
.map(|(v_id, pkg_ref)| {
|
|
||||||
(
|
|
||||||
PackageId::new(PackageNames::Pesde(self.package.0.clone()), v_id),
|
|
||||||
pkg_ref,
|
|
||||||
)
|
|
||||||
})
|
|
||||||
}) else {
|
|
||||||
anyhow::bail!(
|
|
||||||
"no Lune or Luau package could be found for {}@{version_req}",
|
|
||||||
self.package.0,
|
|
||||||
);
|
|
||||||
};
|
|
||||||
|
|
||||||
let tmp_dir = project.cas_dir().join(".tmp");
|
fs.write_to(tempdir.path(), project.cas_dir(), true)
|
||||||
fs::create_dir_all(&tmp_dir)
|
.await
|
||||||
.await
|
.context("failed to write package contents")?;
|
||||||
.context("failed to create temporary directory")?;
|
|
||||||
let tempdir = tempfile::tempdir_in(tmp_dir)
|
|
||||||
.context("failed to create temporary directory")?;
|
|
||||||
|
|
||||||
let project = Project::new(
|
let mut refreshed_sources = HashSet::new();
|
||||||
tempdir.path(),
|
|
||||||
None::<std::path::PathBuf>,
|
|
||||||
project.data_dir(),
|
|
||||||
project.cas_dir(),
|
|
||||||
project.auth_config().clone(),
|
|
||||||
);
|
|
||||||
|
|
||||||
let id = Arc::new(id);
|
let graph = project
|
||||||
|
.dependency_graph(None, &mut refreshed_sources, true)
|
||||||
|
.await
|
||||||
|
.context("failed to build dependency graph")?;
|
||||||
|
let graph = Arc::new(graph);
|
||||||
|
|
||||||
let fs = source
|
let (rx, downloaded_graph) = project
|
||||||
.download(
|
.download_and_link(
|
||||||
&pkg_ref,
|
&graph,
|
||||||
&DownloadOptions {
|
&Arc::new(Mutex::new(refreshed_sources)),
|
||||||
project: project.clone(),
|
&reqwest,
|
||||||
reqwest: reqwest.clone(),
|
true,
|
||||||
reporter: Arc::new(()),
|
true,
|
||||||
id: id.clone(),
|
|_| async { Ok::<_, std::io::Error>(()) },
|
||||||
},
|
)
|
||||||
)
|
.await
|
||||||
.await
|
.context("failed to download dependencies")?;
|
||||||
.context("failed to download package")?;
|
|
||||||
|
|
||||||
fs.write_to(tempdir.path(), project.cas_dir(), true)
|
progress_bar(
|
||||||
.await
|
graph.values().map(|versions| versions.len() as u64).sum(),
|
||||||
.context("failed to write package contents")?;
|
rx,
|
||||||
|
"📥 ".to_string(),
|
||||||
|
"downloading dependencies".to_string(),
|
||||||
|
"downloaded dependencies".to_string(),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
let target = source
|
downloaded_graph
|
||||||
.get_target(
|
.await
|
||||||
&pkg_ref,
|
.context("failed to download & link dependencies")?;
|
||||||
&GetTargetOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
path: Arc::from(tempdir.path()),
|
|
||||||
id: id.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to get target")?;
|
|
||||||
|
|
||||||
let bin_path = target.bin_path().context("package has no binary export")?;
|
let mut caller =
|
||||||
|
tempfile::NamedTempFile::new_in(tempdir.path()).context("failed to create tempfile")?;
|
||||||
|
caller
|
||||||
|
.write_all(
|
||||||
|
generate_bin_linking_module(
|
||||||
|
tempdir.path(),
|
||||||
|
&format!("{:?}", bin_path.to_path(tempdir.path())),
|
||||||
|
)
|
||||||
|
.as_bytes(),
|
||||||
|
)
|
||||||
|
.context("failed to write to tempfile")?;
|
||||||
|
|
||||||
let graph = project
|
let status = Command::new("lune")
|
||||||
.dependency_graph(None, refreshed_sources.clone(), true)
|
.arg("run")
|
||||||
.await
|
.arg(caller.path())
|
||||||
.context("failed to build dependency graph")?;
|
.arg("--")
|
||||||
|
.args(&self.args)
|
||||||
|
.current_dir(current_dir().context("failed to get current directory")?)
|
||||||
|
.status()
|
||||||
|
.context("failed to run script")?;
|
||||||
|
|
||||||
multi_progress.suspend(|| {
|
drop(caller);
|
||||||
eprintln!("{}", style(format!("using {}", style(id).bold())).dim());
|
drop(tempdir);
|
||||||
});
|
|
||||||
|
|
||||||
root_progress.reset();
|
std::process::exit(status.code().unwrap_or(1))
|
||||||
root_progress.set_message("download");
|
}
|
||||||
root_progress.set_style(reporters::root_progress_style_with_progress());
|
|
||||||
|
|
||||||
project
|
|
||||||
.download_and_link(
|
|
||||||
&Arc::new(graph),
|
|
||||||
DownloadAndLinkOptions::<CliReporter<Stderr>, ()>::new(reqwest)
|
|
||||||
.reporter(reporter)
|
|
||||||
.refreshed_sources(refreshed_sources)
|
|
||||||
.prod(true),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to download and link dependencies")?;
|
|
||||||
|
|
||||||
anyhow::Ok((tempdir, bin_path.to_relative_path_buf()))
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let mut caller =
|
|
||||||
tempfile::NamedTempFile::new_in(tempdir.path()).context("failed to create tempfile")?;
|
|
||||||
caller
|
|
||||||
.write_all(
|
|
||||||
generate_bin_linking_module(
|
|
||||||
tempdir.path(),
|
|
||||||
&format!("{:?}", bin_path.to_path(tempdir.path())),
|
|
||||||
)
|
|
||||||
.as_bytes(),
|
|
||||||
)
|
|
||||||
.context("failed to write to tempfile")?;
|
|
||||||
|
|
||||||
let status = Command::new("lune")
|
|
||||||
.arg("run")
|
|
||||||
.arg(caller.path())
|
|
||||||
.arg("--")
|
|
||||||
.args(&self.args)
|
|
||||||
.current_dir(current_dir().context("failed to get current directory")?)
|
|
||||||
.status()
|
|
||||||
.context("failed to run script")?;
|
|
||||||
|
|
||||||
drop(caller);
|
|
||||||
drop(tempdir);
|
|
||||||
|
|
||||||
std::process::exit(status.code().unwrap_or(1))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,289 +1,262 @@
|
||||||
use crate::cli::{
|
use crate::cli::config::read_config;
|
||||||
config::read_config,
|
|
||||||
style::{ERROR_PREFIX, INFO_STYLE, SUCCESS_STYLE},
|
|
||||||
};
|
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
|
use colored::Colorize;
|
||||||
use inquire::validator::Validation;
|
use inquire::validator::Validation;
|
||||||
use pesde::{
|
use pesde::{
|
||||||
errors::ManifestReadError,
|
errors::ManifestReadError,
|
||||||
manifest::{target::TargetKind, DependencyType},
|
manifest::{target::TargetKind, DependencyType},
|
||||||
names::{PackageName, PackageNames},
|
names::PackageName,
|
||||||
source::{
|
source::{
|
||||||
git_index::GitBasedSource,
|
git_index::GitBasedSource,
|
||||||
ids::PackageId,
|
pesde::{specifier::PesdeDependencySpecifier, PesdePackageSource},
|
||||||
pesde::{specifier::PesdeDependencySpecifier, PesdePackageSource},
|
specifiers::DependencySpecifiers,
|
||||||
specifiers::DependencySpecifiers,
|
traits::PackageSource,
|
||||||
traits::{GetTargetOptions, PackageSource, RefreshOptions, ResolveOptions},
|
},
|
||||||
PackageSources,
|
Project, DEFAULT_INDEX_NAME, SCRIPTS_LINK_FOLDER,
|
||||||
},
|
|
||||||
Project, RefreshedSources, DEFAULT_INDEX_NAME, SCRIPTS_LINK_FOLDER,
|
|
||||||
};
|
};
|
||||||
use semver::VersionReq;
|
use semver::VersionReq;
|
||||||
use std::{fmt::Display, path::Path, str::FromStr, sync::Arc};
|
use std::{collections::HashSet, fmt::Display, str::FromStr};
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct InitCommand {}
|
pub struct InitCommand {}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
enum PackageNameOrCustom {
|
enum PackageNameOrCustom {
|
||||||
PackageName(PackageName),
|
PackageName(PackageName),
|
||||||
Custom,
|
Custom,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for PackageNameOrCustom {
|
impl Display for PackageNameOrCustom {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
match self {
|
match self {
|
||||||
PackageNameOrCustom::PackageName(n) => write!(f, "{n}"),
|
PackageNameOrCustom::PackageName(n) => write!(f, "{n}"),
|
||||||
PackageNameOrCustom::Custom => write!(f, "custom"),
|
PackageNameOrCustom::Custom => write!(f, "custom"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl InitCommand {
|
impl InitCommand {
|
||||||
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
||||||
match project.read_manifest().await {
|
match project.read_manifest().await {
|
||||||
Ok(_) => {
|
Ok(_) => {
|
||||||
anyhow::bail!("project already initialized");
|
println!("{}", "project already initialized".red());
|
||||||
}
|
return Ok(());
|
||||||
Err(ManifestReadError::Io(e)) if e.kind() == std::io::ErrorKind::NotFound => {}
|
}
|
||||||
Err(e) => return Err(e.into()),
|
Err(ManifestReadError::Io(e)) if e.kind() == std::io::ErrorKind::NotFound => {}
|
||||||
};
|
Err(e) => return Err(e.into()),
|
||||||
|
};
|
||||||
|
|
||||||
let mut manifest = toml_edit::DocumentMut::new();
|
let mut manifest = toml_edit::DocumentMut::new();
|
||||||
|
|
||||||
manifest["name"] = toml_edit::value(
|
manifest["name"] = toml_edit::value(
|
||||||
inquire::Text::new("what is the name of the project?")
|
inquire::Text::new("what is the name of the project?")
|
||||||
.with_validator(|name: &str| {
|
.with_validator(|name: &str| {
|
||||||
Ok(match PackageName::from_str(name) {
|
Ok(match PackageName::from_str(name) {
|
||||||
Ok(_) => Validation::Valid,
|
Ok(_) => Validation::Valid,
|
||||||
Err(e) => Validation::Invalid(e.to_string().into()),
|
Err(e) => Validation::Invalid(e.to_string().into()),
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
.prompt()
|
.prompt()
|
||||||
.unwrap(),
|
.unwrap(),
|
||||||
);
|
);
|
||||||
manifest["version"] = toml_edit::value("0.1.0");
|
manifest["version"] = toml_edit::value("0.1.0");
|
||||||
|
|
||||||
let description = inquire::Text::new("what is the description of the project?")
|
let description = inquire::Text::new("what is the description of the project?")
|
||||||
.with_help_message("a short description of the project. leave empty for none")
|
.with_help_message("a short description of the project. leave empty for none")
|
||||||
.prompt()
|
.prompt()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
if !description.is_empty() {
|
if !description.is_empty() {
|
||||||
manifest["description"] = toml_edit::value(description);
|
manifest["description"] = toml_edit::value(description);
|
||||||
}
|
}
|
||||||
|
|
||||||
let authors = inquire::Text::new("who are the authors of this project?")
|
let authors = inquire::Text::new("who are the authors of this project?")
|
||||||
.with_help_message("comma separated list. leave empty for none")
|
.with_help_message("comma separated list. leave empty for none")
|
||||||
.prompt()
|
.prompt()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
let authors = authors
|
let authors = authors
|
||||||
.split(',')
|
.split(',')
|
||||||
.map(str::trim)
|
.map(str::trim)
|
||||||
.filter(|s| !s.is_empty())
|
.filter(|s| !s.is_empty())
|
||||||
.collect::<toml_edit::Array>();
|
.collect::<toml_edit::Array>();
|
||||||
|
|
||||||
if !authors.is_empty() {
|
if !authors.is_empty() {
|
||||||
manifest["authors"] = toml_edit::value(authors);
|
manifest["authors"] = toml_edit::value(authors);
|
||||||
}
|
}
|
||||||
|
|
||||||
let repo = inquire::Text::new("what is the repository URL of this project?")
|
let repo = inquire::Text::new("what is the repository URL of this project?")
|
||||||
.with_validator(|repo: &str| {
|
.with_validator(|repo: &str| {
|
||||||
if repo.is_empty() {
|
if repo.is_empty() {
|
||||||
return Ok(Validation::Valid);
|
return Ok(Validation::Valid);
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(match url::Url::parse(repo) {
|
Ok(match url::Url::parse(repo) {
|
||||||
Ok(_) => Validation::Valid,
|
Ok(_) => Validation::Valid,
|
||||||
Err(e) => Validation::Invalid(e.to_string().into()),
|
Err(e) => Validation::Invalid(e.to_string().into()),
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
.with_help_message("leave empty for none")
|
.with_help_message("leave empty for none")
|
||||||
.prompt()
|
.prompt()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
if !repo.is_empty() {
|
if !repo.is_empty() {
|
||||||
manifest["repository"] = toml_edit::value(repo);
|
manifest["repository"] = toml_edit::value(repo);
|
||||||
}
|
}
|
||||||
|
|
||||||
let license = inquire::Text::new("what is the license of this project?")
|
let license = inquire::Text::new("what is the license of this project?")
|
||||||
.with_initial_value("MIT")
|
.with_initial_value("MIT")
|
||||||
.with_help_message("an SPDX license identifier. leave empty for none")
|
.with_help_message("an SPDX license identifier. leave empty for none")
|
||||||
.prompt()
|
.prompt()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
if !license.is_empty() {
|
if !license.is_empty() {
|
||||||
manifest["license"] = toml_edit::value(license);
|
manifest["license"] = toml_edit::value(license);
|
||||||
}
|
}
|
||||||
|
|
||||||
let target_env = inquire::Select::new(
|
let target_env = inquire::Select::new(
|
||||||
"what environment are you targeting for your package?",
|
"what environment are you targeting for your package?",
|
||||||
TargetKind::VARIANTS.to_vec(),
|
TargetKind::VARIANTS.to_vec(),
|
||||||
)
|
)
|
||||||
.prompt()
|
.prompt()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
manifest["target"].or_insert(toml_edit::Item::Table(toml_edit::Table::new()))
|
manifest["target"].or_insert(toml_edit::Item::Table(toml_edit::Table::new()))
|
||||||
["environment"] = toml_edit::value(target_env.to_string());
|
["environment"] = toml_edit::value(target_env.to_string());
|
||||||
|
|
||||||
let source = PesdePackageSource::new(read_config().await?.default_index);
|
let source = PesdePackageSource::new(read_config().await?.default_index);
|
||||||
|
|
||||||
manifest["indices"].or_insert(toml_edit::Item::Table(toml_edit::Table::new()))
|
manifest["indices"].or_insert(toml_edit::Item::Table(toml_edit::Table::new()))
|
||||||
[DEFAULT_INDEX_NAME] = toml_edit::value(source.repo_url().to_bstring().to_string());
|
[DEFAULT_INDEX_NAME] = toml_edit::value(source.repo_url().to_bstring().to_string());
|
||||||
|
|
||||||
let refreshed_sources = RefreshedSources::new();
|
if target_env.is_roblox()
|
||||||
|
|| inquire::prompt_confirmation(
|
||||||
|
"would you like to setup default Roblox compatibility scripts?",
|
||||||
|
)
|
||||||
|
.unwrap()
|
||||||
|
{
|
||||||
|
PackageSource::refresh(&source, &project)
|
||||||
|
.await
|
||||||
|
.context("failed to refresh package source")?;
|
||||||
|
let config = source
|
||||||
|
.config(&project)
|
||||||
|
.await
|
||||||
|
.context("failed to get source config")?;
|
||||||
|
|
||||||
if target_env.is_roblox()
|
let scripts_package = if config.scripts_packages.is_empty() {
|
||||||
|| inquire::prompt_confirmation("would you like to setup Roblox compatibility scripts?")
|
PackageNameOrCustom::Custom
|
||||||
.unwrap()
|
} else {
|
||||||
{
|
inquire::Select::new(
|
||||||
refreshed_sources
|
"which scripts package do you want to use?",
|
||||||
.refresh(
|
config
|
||||||
&PackageSources::Pesde(source.clone()),
|
.scripts_packages
|
||||||
&RefreshOptions {
|
.into_iter()
|
||||||
project: project.clone(),
|
.map(PackageNameOrCustom::PackageName)
|
||||||
},
|
.chain(std::iter::once(PackageNameOrCustom::Custom))
|
||||||
)
|
.collect(),
|
||||||
.await
|
)
|
||||||
.context("failed to refresh package source")?;
|
.prompt()
|
||||||
let config = source
|
.unwrap()
|
||||||
.config(&project)
|
};
|
||||||
.await
|
|
||||||
.context("failed to get source config")?;
|
|
||||||
|
|
||||||
let scripts_package = if config.scripts_packages.is_empty() {
|
let scripts_package = match scripts_package {
|
||||||
PackageNameOrCustom::Custom
|
PackageNameOrCustom::PackageName(p) => Some(p),
|
||||||
} else {
|
PackageNameOrCustom::Custom => {
|
||||||
inquire::Select::new(
|
let name = inquire::Text::new("which scripts package to use?")
|
||||||
"which scripts package do you want to use?",
|
.with_validator(|name: &str| {
|
||||||
config
|
if name.is_empty() {
|
||||||
.scripts_packages
|
return Ok(Validation::Valid);
|
||||||
.into_iter()
|
}
|
||||||
.map(PackageNameOrCustom::PackageName)
|
|
||||||
.chain(std::iter::once(PackageNameOrCustom::Custom))
|
|
||||||
.collect(),
|
|
||||||
)
|
|
||||||
.prompt()
|
|
||||||
.unwrap()
|
|
||||||
};
|
|
||||||
|
|
||||||
let scripts_package = match scripts_package {
|
Ok(match PackageName::from_str(name) {
|
||||||
PackageNameOrCustom::PackageName(p) => Some(p),
|
Ok(_) => Validation::Valid,
|
||||||
PackageNameOrCustom::Custom => {
|
Err(e) => Validation::Invalid(e.to_string().into()),
|
||||||
let name = inquire::Text::new("which scripts package to use?")
|
})
|
||||||
.with_validator(|name: &str| {
|
})
|
||||||
if name.is_empty() {
|
.with_help_message("leave empty for none")
|
||||||
return Ok(Validation::Valid);
|
.prompt()
|
||||||
}
|
.unwrap();
|
||||||
|
|
||||||
Ok(match PackageName::from_str(name) {
|
if name.is_empty() {
|
||||||
Ok(_) => Validation::Valid,
|
None
|
||||||
Err(e) => Validation::Invalid(e.to_string().into()),
|
} else {
|
||||||
})
|
Some(PackageName::from_str(&name).unwrap())
|
||||||
})
|
}
|
||||||
.with_help_message("leave empty for none")
|
}
|
||||||
.prompt()
|
};
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
if name.is_empty() {
|
if let Some(scripts_pkg_name) = scripts_package {
|
||||||
None
|
let (v_id, pkg_ref) = source
|
||||||
} else {
|
.resolve(
|
||||||
Some(PackageName::from_str(&name).unwrap())
|
&PesdeDependencySpecifier {
|
||||||
}
|
name: scripts_pkg_name,
|
||||||
}
|
version: VersionReq::STAR,
|
||||||
};
|
index: None,
|
||||||
|
target: None,
|
||||||
|
},
|
||||||
|
&project,
|
||||||
|
TargetKind::Lune,
|
||||||
|
&mut HashSet::new(),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to resolve scripts package")?
|
||||||
|
.1
|
||||||
|
.pop_last()
|
||||||
|
.context("scripts package not found")?;
|
||||||
|
|
||||||
if let Some(scripts_pkg_name) = scripts_package {
|
let Some(scripts) = pkg_ref.target.scripts().filter(|s| !s.is_empty()) else {
|
||||||
let (v_id, pkg_ref) = source
|
anyhow::bail!("scripts package has no scripts. this is an issue with the index")
|
||||||
.resolve(
|
};
|
||||||
&PesdeDependencySpecifier {
|
|
||||||
name: scripts_pkg_name.clone(),
|
|
||||||
version: VersionReq::STAR,
|
|
||||||
index: None,
|
|
||||||
target: None,
|
|
||||||
},
|
|
||||||
&ResolveOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
target: TargetKind::Lune,
|
|
||||||
refreshed_sources,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to resolve scripts package")?
|
|
||||||
.1
|
|
||||||
.pop_last()
|
|
||||||
.context("scripts package not found")?;
|
|
||||||
|
|
||||||
let id = Arc::new(PackageId::new(PackageNames::Pesde(scripts_pkg_name), v_id));
|
let scripts_field = &mut manifest["scripts"]
|
||||||
|
.or_insert(toml_edit::Item::Table(toml_edit::Table::new()));
|
||||||
|
|
||||||
let target = source
|
for script_name in scripts.keys() {
|
||||||
.get_target(
|
scripts_field[script_name] = toml_edit::value(format!(
|
||||||
&pkg_ref,
|
"{SCRIPTS_LINK_FOLDER}/scripts/{script_name}.luau"
|
||||||
&GetTargetOptions {
|
));
|
||||||
project: project.clone(),
|
}
|
||||||
// HACK: the pesde package source doesn't use the path, so we can just use an empty one
|
|
||||||
path: Arc::from(Path::new("")),
|
|
||||||
id: id.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let Some(scripts) = target.scripts().filter(|s| !s.is_empty()) else {
|
let dev_deps = &mut manifest["dev_dependencies"]
|
||||||
anyhow::bail!("scripts package has no scripts.")
|
.or_insert(toml_edit::Item::Table(toml_edit::Table::new()));
|
||||||
};
|
|
||||||
|
|
||||||
let scripts_field = &mut manifest["scripts"]
|
let field = &mut dev_deps["scripts"];
|
||||||
.or_insert(toml_edit::Item::Table(toml_edit::Table::new()));
|
field["name"] = toml_edit::value(pkg_ref.name.to_string());
|
||||||
|
field["version"] = toml_edit::value(format!("^{}", v_id.version()));
|
||||||
|
field["target"] = toml_edit::value(v_id.target().to_string());
|
||||||
|
|
||||||
for script_name in scripts.keys() {
|
for (alias, (spec, ty)) in pkg_ref.dependencies {
|
||||||
scripts_field[script_name] = toml_edit::value(format!(
|
if ty != DependencyType::Peer {
|
||||||
"{SCRIPTS_LINK_FOLDER}/scripts/{script_name}.luau"
|
continue;
|
||||||
));
|
}
|
||||||
}
|
|
||||||
|
|
||||||
let dev_deps = &mut manifest["dev_dependencies"]
|
let DependencySpecifiers::Pesde(spec) = spec else {
|
||||||
.or_insert(toml_edit::Item::Table(toml_edit::Table::new()));
|
continue;
|
||||||
|
};
|
||||||
|
|
||||||
let field = &mut dev_deps["scripts"];
|
let field = &mut dev_deps[alias];
|
||||||
field["name"] = toml_edit::value(id.name().to_string());
|
field["name"] = toml_edit::value(spec.name.to_string());
|
||||||
field["version"] = toml_edit::value(format!("^{}", id.version_id().version()));
|
field["version"] = toml_edit::value(spec.version.to_string());
|
||||||
field["target"] = toml_edit::value(id.version_id().target().to_string());
|
field["target"] =
|
||||||
|
toml_edit::value(spec.target.unwrap_or_else(|| *v_id.target()).to_string());
|
||||||
for (alias, (spec, ty)) in pkg_ref.dependencies {
|
}
|
||||||
if ty != DependencyType::Peer {
|
} else {
|
||||||
continue;
|
println!(
|
||||||
}
|
"{}",
|
||||||
|
"no scripts package configured, this can cause issues with Roblox compatibility".red()
|
||||||
let DependencySpecifiers::Pesde(spec) = spec else {
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
|
|
||||||
let field = &mut dev_deps[alias.as_str()];
|
|
||||||
field["name"] = toml_edit::value(spec.name.to_string());
|
|
||||||
field["version"] = toml_edit::value(spec.version.to_string());
|
|
||||||
field["target"] = toml_edit::value(
|
|
||||||
spec.target
|
|
||||||
.unwrap_or_else(|| id.version_id().target())
|
|
||||||
.to_string(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
println!(
|
|
||||||
"{ERROR_PREFIX}: no scripts package configured, this can cause issues with Roblox compatibility"
|
|
||||||
);
|
);
|
||||||
if !inquire::prompt_confirmation("initialize regardless?").unwrap() {
|
if !inquire::prompt_confirmation("initialize regardless?").unwrap() {
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
project.write_manifest(manifest.to_string()).await?;
|
project.write_manifest(manifest.to_string()).await?;
|
||||||
|
|
||||||
println!(
|
println!(
|
||||||
"{}\n{}: run `install` to fully finish setup",
|
"{}\n{}: run `install` to fully finish setup",
|
||||||
SUCCESS_STYLE.apply_to("initialized project"),
|
"initialized project".green(),
|
||||||
INFO_STYLE.apply_to("tip")
|
"tip".cyan().bold()
|
||||||
);
|
);
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,55 +1,328 @@
|
||||||
use crate::cli::{
|
use crate::cli::{
|
||||||
install::{install, InstallOptions},
|
bin_dir, files::make_executable, progress_bar, run_on_workspace_members, up_to_date_lockfile,
|
||||||
run_on_workspace_members,
|
|
||||||
};
|
};
|
||||||
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use pesde::Project;
|
use colored::{ColoredString, Colorize};
|
||||||
use std::num::NonZeroUsize;
|
use fs_err::tokio as fs;
|
||||||
|
use futures::future::try_join_all;
|
||||||
|
use pesde::{
|
||||||
|
download_and_link::filter_graph, lockfile::Lockfile, manifest::target::TargetKind, Project,
|
||||||
|
MANIFEST_FILE_NAME,
|
||||||
|
};
|
||||||
|
use std::{
|
||||||
|
collections::{BTreeSet, HashMap, HashSet},
|
||||||
|
sync::Arc,
|
||||||
|
};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
#[derive(Debug, Args, Copy, Clone)]
|
#[derive(Debug, Args, Copy, Clone)]
|
||||||
pub struct InstallCommand {
|
pub struct InstallCommand {
|
||||||
/// Whether to error on changes in the lockfile
|
/// Whether to error on changes in the lockfile
|
||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
locked: bool,
|
locked: bool,
|
||||||
|
|
||||||
/// Whether to not install dev dependencies
|
/// Whether to not install dev dependencies
|
||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
prod: bool,
|
prod: bool,
|
||||||
|
}
|
||||||
|
|
||||||
/// The maximum number of concurrent network requests
|
fn bin_link_file(alias: &str) -> String {
|
||||||
#[arg(long, default_value = "16")]
|
let mut all_combinations = BTreeSet::new();
|
||||||
network_concurrency: NonZeroUsize,
|
|
||||||
|
|
||||||
/// Whether to re-install all dependencies even if they are already installed
|
for a in TargetKind::VARIANTS {
|
||||||
#[arg(long)]
|
for b in TargetKind::VARIANTS {
|
||||||
force: bool,
|
all_combinations.insert((a, b));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let all_folders = all_combinations
|
||||||
|
.into_iter()
|
||||||
|
.map(|(a, b)| format!("{:?}", a.packages_folder(b)))
|
||||||
|
.collect::<BTreeSet<_>>()
|
||||||
|
.into_iter()
|
||||||
|
.collect::<Vec<_>>()
|
||||||
|
.join(", ");
|
||||||
|
|
||||||
|
format!(
|
||||||
|
r#"local process = require("@lune/process")
|
||||||
|
local fs = require("@lune/fs")
|
||||||
|
local stdio = require("@lune/stdio")
|
||||||
|
|
||||||
|
local project_root = process.cwd
|
||||||
|
local path_components = string.split(string.gsub(project_root, "\\", "/"), "/")
|
||||||
|
|
||||||
|
for i = #path_components, 1, -1 do
|
||||||
|
local path = table.concat(path_components, "/", 1, i)
|
||||||
|
if fs.isFile(path .. "/{MANIFEST_FILE_NAME}") then
|
||||||
|
project_root = path
|
||||||
|
break
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
for _, packages_folder in {{ {all_folders} }} do
|
||||||
|
local path = `{{project_root}}/{{packages_folder}}/{alias}.bin.luau`
|
||||||
|
|
||||||
|
if fs.isFile(path) then
|
||||||
|
require(path)
|
||||||
|
return
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
stdio.ewrite(stdio.color("red") .. "binary `{alias}` not found. are you in the right directory?" .. stdio.color("reset") .. "\n")
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "patches")]
|
||||||
|
const JOBS: u8 = 5;
|
||||||
|
#[cfg(not(feature = "patches"))]
|
||||||
|
const JOBS: u8 = 4;
|
||||||
|
|
||||||
|
fn job(n: u8) -> ColoredString {
|
||||||
|
format!("[{n}/{JOBS}]").dimmed().bold()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, thiserror::Error)]
|
#[derive(Debug, thiserror::Error)]
|
||||||
#[error(transparent)]
|
#[error(transparent)]
|
||||||
struct CallbackError(#[from] anyhow::Error);
|
struct CallbackError(#[from] anyhow::Error);
|
||||||
|
|
||||||
impl InstallCommand {
|
impl InstallCommand {
|
||||||
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
||||||
let options = InstallOptions {
|
let mut refreshed_sources = HashSet::new();
|
||||||
locked: self.locked,
|
|
||||||
prod: self.prod,
|
|
||||||
write: true,
|
|
||||||
network_concurrency: self.network_concurrency,
|
|
||||||
use_lockfile: true,
|
|
||||||
force: self.force,
|
|
||||||
};
|
|
||||||
|
|
||||||
install(&options, &project, reqwest.clone(), true).await?;
|
let manifest = project
|
||||||
|
.deser_manifest()
|
||||||
|
.await
|
||||||
|
.context("failed to read manifest")?;
|
||||||
|
|
||||||
run_on_workspace_members(&project, |project| {
|
let lockfile = if self.locked {
|
||||||
let reqwest = reqwest.clone();
|
match up_to_date_lockfile(&project).await? {
|
||||||
async move {
|
None => {
|
||||||
install(&options, &project, reqwest, false).await?;
|
anyhow::bail!(
|
||||||
Ok(())
|
"lockfile is out of sync, run `{} install` to update it",
|
||||||
}
|
env!("CARGO_BIN_NAME")
|
||||||
})
|
);
|
||||||
.await?;
|
}
|
||||||
|
file => file,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
match project.deser_lockfile().await {
|
||||||
|
Ok(lockfile) => {
|
||||||
|
if lockfile.overrides != manifest.overrides {
|
||||||
|
tracing::debug!("overrides are different");
|
||||||
|
None
|
||||||
|
} else if lockfile.target != manifest.target.kind() {
|
||||||
|
tracing::debug!("target kind is different");
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(lockfile)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(pesde::errors::LockfileReadError::Io(e))
|
||||||
|
if e.kind() == std::io::ErrorKind::NotFound =>
|
||||||
|
{
|
||||||
|
None
|
||||||
|
}
|
||||||
|
Err(e) => return Err(e.into()),
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
Ok(())
|
println!(
|
||||||
}
|
"\n{}\n",
|
||||||
|
format!("[now installing {} {}]", manifest.name, manifest.target)
|
||||||
|
.bold()
|
||||||
|
.on_bright_black()
|
||||||
|
);
|
||||||
|
|
||||||
|
println!("{} ❌ removing current package folders", job(1));
|
||||||
|
|
||||||
|
{
|
||||||
|
let mut deleted_folders = HashMap::new();
|
||||||
|
|
||||||
|
for target_kind in TargetKind::VARIANTS {
|
||||||
|
let folder = manifest.target.kind().packages_folder(target_kind);
|
||||||
|
let package_dir = project.package_dir();
|
||||||
|
|
||||||
|
deleted_folders
|
||||||
|
.entry(folder.to_string())
|
||||||
|
.or_insert_with(|| async move {
|
||||||
|
tracing::debug!("deleting the {folder} folder");
|
||||||
|
|
||||||
|
if let Some(e) = fs::remove_dir_all(package_dir.join(&folder))
|
||||||
|
.await
|
||||||
|
.err()
|
||||||
|
.filter(|e| e.kind() != std::io::ErrorKind::NotFound)
|
||||||
|
{
|
||||||
|
return Err(e).context(format!("failed to remove the {folder} folder"));
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
try_join_all(deleted_folders.into_values())
|
||||||
|
.await
|
||||||
|
.context("failed to remove package folders")?;
|
||||||
|
}
|
||||||
|
|
||||||
|
let old_graph = lockfile.map(|lockfile| {
|
||||||
|
lockfile
|
||||||
|
.graph
|
||||||
|
.into_iter()
|
||||||
|
.map(|(name, versions)| {
|
||||||
|
(
|
||||||
|
name,
|
||||||
|
versions
|
||||||
|
.into_iter()
|
||||||
|
.map(|(version, node)| (version, node.node))
|
||||||
|
.collect(),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
});
|
||||||
|
|
||||||
|
println!("{} 📦 building dependency graph", job(2));
|
||||||
|
|
||||||
|
let graph = project
|
||||||
|
.dependency_graph(old_graph.as_ref(), &mut refreshed_sources, false)
|
||||||
|
.await
|
||||||
|
.context("failed to build dependency graph")?;
|
||||||
|
let graph = Arc::new(graph);
|
||||||
|
|
||||||
|
let bin_folder = bin_dir().await?;
|
||||||
|
|
||||||
|
let downloaded_graph = {
|
||||||
|
let (rx, downloaded_graph) = project
|
||||||
|
.download_and_link(
|
||||||
|
&graph,
|
||||||
|
&Arc::new(Mutex::new(refreshed_sources)),
|
||||||
|
&reqwest,
|
||||||
|
self.prod,
|
||||||
|
true,
|
||||||
|
|graph| {
|
||||||
|
let graph = graph.clone();
|
||||||
|
|
||||||
|
async move {
|
||||||
|
try_join_all(
|
||||||
|
graph
|
||||||
|
.values()
|
||||||
|
.flat_map(|versions| versions.values())
|
||||||
|
.filter(|node| node.target.bin_path().is_some())
|
||||||
|
.filter_map(|node| node.node.direct.as_ref())
|
||||||
|
.map(|(alias, _, _)| alias)
|
||||||
|
.filter(|alias| {
|
||||||
|
if *alias == env!("CARGO_BIN_NAME") {
|
||||||
|
tracing::warn!(
|
||||||
|
"package {alias} has the same name as the CLI, skipping bin link"
|
||||||
|
);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
true
|
||||||
|
})
|
||||||
|
.map(|alias| {
|
||||||
|
let bin_folder = bin_folder.clone();
|
||||||
|
async move {
|
||||||
|
let bin_exec_file = bin_folder.join(alias).with_extension(std::env::consts::EXE_EXTENSION);
|
||||||
|
|
||||||
|
let impl_folder = bin_folder.join(".impl");
|
||||||
|
fs::create_dir_all(&impl_folder).await.context("failed to create bin link folder")?;
|
||||||
|
|
||||||
|
let bin_file = impl_folder.join(alias).with_extension("luau");
|
||||||
|
fs::write(&bin_file, bin_link_file(alias))
|
||||||
|
.await
|
||||||
|
.context("failed to write bin link file")?;
|
||||||
|
|
||||||
|
|
||||||
|
#[cfg(windows)]
|
||||||
|
{
|
||||||
|
fs::copy(
|
||||||
|
std::env::current_exe()
|
||||||
|
.context("failed to get current executable path")?,
|
||||||
|
&bin_exec_file,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to copy bin link file")?;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(not(windows))]
|
||||||
|
{
|
||||||
|
fs::write(
|
||||||
|
&bin_exec_file,
|
||||||
|
format!(r#"#!/bin/sh
|
||||||
|
exec lune run "$(dirname "$0")/.impl/{alias}.luau" -- "$@""#
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to link bin link file")?;
|
||||||
|
}
|
||||||
|
|
||||||
|
make_executable(&bin_exec_file).await.context("failed to make bin link file executable")?;
|
||||||
|
|
||||||
|
Ok::<_, CallbackError>(())
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.map(|_| ())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to download dependencies")?;
|
||||||
|
|
||||||
|
progress_bar(
|
||||||
|
graph.values().map(|versions| versions.len() as u64).sum(),
|
||||||
|
rx,
|
||||||
|
format!("{} 📥 ", job(3)),
|
||||||
|
"downloading dependencies".to_string(),
|
||||||
|
"downloaded dependencies".to_string(),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
downloaded_graph
|
||||||
|
.await
|
||||||
|
.context("failed to download & link dependencies")?
|
||||||
|
};
|
||||||
|
|
||||||
|
#[cfg(feature = "patches")]
|
||||||
|
{
|
||||||
|
let rx = project
|
||||||
|
.apply_patches(&filter_graph(&downloaded_graph, self.prod))
|
||||||
|
.await
|
||||||
|
.context("failed to apply patches")?;
|
||||||
|
|
||||||
|
progress_bar(
|
||||||
|
manifest.patches.values().map(|v| v.len() as u64).sum(),
|
||||||
|
rx,
|
||||||
|
format!("{} 🩹 ", job(JOBS - 1)),
|
||||||
|
"applying patches".to_string(),
|
||||||
|
"applied patches".to_string(),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
|
println!("{} 🧹 finishing up", job(JOBS));
|
||||||
|
|
||||||
|
project
|
||||||
|
.write_lockfile(Lockfile {
|
||||||
|
name: manifest.name,
|
||||||
|
version: manifest.version,
|
||||||
|
target: manifest.target.kind(),
|
||||||
|
overrides: manifest.overrides,
|
||||||
|
|
||||||
|
graph: downloaded_graph,
|
||||||
|
|
||||||
|
workspace: run_on_workspace_members(&project, |project| {
|
||||||
|
let reqwest = reqwest.clone();
|
||||||
|
async move { Box::pin(self.run(project, reqwest)).await }
|
||||||
|
})
|
||||||
|
.await?,
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.context("failed to write lockfile")?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,51 +0,0 @@
|
||||||
use std::collections::BTreeMap;
|
|
||||||
|
|
||||||
use anyhow::Context;
|
|
||||||
use clap::Args;
|
|
||||||
|
|
||||||
use crate::cli::{
|
|
||||||
dep_type_to_key,
|
|
||||||
style::{INFO_STYLE, SUCCESS_STYLE},
|
|
||||||
};
|
|
||||||
use pesde::{
|
|
||||||
manifest::{Alias, DependencyType},
|
|
||||||
source::specifiers::DependencySpecifiers,
|
|
||||||
Project,
|
|
||||||
};
|
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
|
||||||
pub struct ListCommand {}
|
|
||||||
|
|
||||||
impl ListCommand {
|
|
||||||
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
|
||||||
let manifest = project
|
|
||||||
.deser_manifest()
|
|
||||||
.await
|
|
||||||
.context("failed to read manifest")?;
|
|
||||||
|
|
||||||
let all_deps = manifest
|
|
||||||
.all_dependencies()
|
|
||||||
.context("failed to get all dependencies")?
|
|
||||||
.into_iter()
|
|
||||||
.fold(
|
|
||||||
BTreeMap::<DependencyType, BTreeMap<Alias, DependencySpecifiers>>::new(),
|
|
||||||
|mut acc, (alias, (spec, ty))| {
|
|
||||||
acc.entry(ty).or_default().insert(alias, spec);
|
|
||||||
acc
|
|
||||||
},
|
|
||||||
);
|
|
||||||
|
|
||||||
for (dep_ty, deps) in all_deps {
|
|
||||||
let dep_key = dep_type_to_key(dep_ty);
|
|
||||||
println!("{}", INFO_STYLE.apply_to(dep_key));
|
|
||||||
|
|
||||||
for (alias, spec) in deps {
|
|
||||||
println!("{}: {spec}", SUCCESS_STYLE.apply_to(alias));
|
|
||||||
}
|
|
||||||
|
|
||||||
println!();
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -2,121 +2,95 @@ use pesde::Project;
|
||||||
|
|
||||||
mod add;
|
mod add;
|
||||||
mod auth;
|
mod auth;
|
||||||
mod cas;
|
|
||||||
mod config;
|
mod config;
|
||||||
mod deprecate;
|
|
||||||
mod execute;
|
mod execute;
|
||||||
mod init;
|
mod init;
|
||||||
mod install;
|
mod install;
|
||||||
mod list;
|
|
||||||
mod outdated;
|
mod outdated;
|
||||||
#[cfg(feature = "patches")]
|
#[cfg(feature = "patches")]
|
||||||
mod patch;
|
mod patch;
|
||||||
#[cfg(feature = "patches")]
|
#[cfg(feature = "patches")]
|
||||||
mod patch_commit;
|
mod patch_commit;
|
||||||
mod publish;
|
mod publish;
|
||||||
mod remove;
|
|
||||||
mod run;
|
mod run;
|
||||||
#[cfg(feature = "version-management")]
|
#[cfg(feature = "version-management")]
|
||||||
mod self_install;
|
mod self_install;
|
||||||
#[cfg(feature = "version-management")]
|
#[cfg(feature = "version-management")]
|
||||||
mod self_upgrade;
|
mod self_upgrade;
|
||||||
mod update;
|
mod update;
|
||||||
mod yank;
|
|
||||||
|
|
||||||
#[derive(Debug, clap::Subcommand)]
|
#[derive(Debug, clap::Subcommand)]
|
||||||
pub enum Subcommand {
|
pub enum Subcommand {
|
||||||
/// Authentication-related commands
|
/// Authentication-related commands
|
||||||
Auth(auth::AuthSubcommand),
|
Auth(auth::AuthSubcommand),
|
||||||
|
|
||||||
/// Configuration-related commands
|
/// Configuration-related commands
|
||||||
#[command(subcommand)]
|
#[command(subcommand)]
|
||||||
Config(config::ConfigCommands),
|
Config(config::ConfigCommands),
|
||||||
|
|
||||||
/// CAS-related commands
|
/// Initializes a manifest file in the current directory
|
||||||
#[command(subcommand)]
|
Init(init::InitCommand),
|
||||||
Cas(cas::CasCommands),
|
|
||||||
|
|
||||||
/// Initializes a manifest file in the current directory
|
/// Runs a script, an executable package, or a file with Lune
|
||||||
Init(init::InitCommand),
|
Run(run::RunCommand),
|
||||||
|
|
||||||
/// Adds a dependency to the project
|
/// Installs all dependencies for the project
|
||||||
Add(add::AddCommand),
|
Install(install::InstallCommand),
|
||||||
|
|
||||||
/// Removes a dependency from the project
|
/// Publishes the project to the registry
|
||||||
Remove(remove::RemoveCommand),
|
Publish(publish::PublishCommand),
|
||||||
|
|
||||||
/// Installs all dependencies for the project
|
/// Installs the pesde binary and scripts
|
||||||
Install(install::InstallCommand),
|
#[cfg(feature = "version-management")]
|
||||||
|
SelfInstall(self_install::SelfInstallCommand),
|
||||||
|
|
||||||
/// Updates the project's lockfile. Run install to apply changes
|
/// Sets up a patching environment for a package
|
||||||
Update(update::UpdateCommand),
|
#[cfg(feature = "patches")]
|
||||||
|
Patch(patch::PatchCommand),
|
||||||
|
|
||||||
/// Checks for outdated dependencies
|
/// Finalizes a patching environment for a package
|
||||||
Outdated(outdated::OutdatedCommand),
|
#[cfg(feature = "patches")]
|
||||||
|
PatchCommit(patch_commit::PatchCommitCommand),
|
||||||
|
|
||||||
/// Lists all dependencies in the project
|
/// Installs the latest version of pesde
|
||||||
List(list::ListCommand),
|
#[cfg(feature = "version-management")]
|
||||||
|
SelfUpgrade(self_upgrade::SelfUpgradeCommand),
|
||||||
|
|
||||||
/// Runs a script, an executable package, or a file with Lune
|
/// Adds a dependency to the project
|
||||||
Run(run::RunCommand),
|
Add(add::AddCommand),
|
||||||
|
|
||||||
/// Publishes the project to the registry
|
/// Updates the project's lockfile. Run install to apply changes
|
||||||
Publish(publish::PublishCommand),
|
Update(update::UpdateCommand),
|
||||||
|
|
||||||
/// Yanks a package from the registry
|
/// Checks for outdated dependencies
|
||||||
Yank(yank::YankCommand),
|
Outdated(outdated::OutdatedCommand),
|
||||||
|
|
||||||
/// Deprecates a package from the registry
|
/// Executes a binary package without needing to be run in a project directory
|
||||||
Deprecate(deprecate::DeprecateCommand),
|
#[clap(name = "x", visible_alias = "execute", visible_alias = "exec")]
|
||||||
|
Execute(execute::ExecuteCommand),
|
||||||
/// Sets up a patching environment for a package
|
|
||||||
#[cfg(feature = "patches")]
|
|
||||||
Patch(patch::PatchCommand),
|
|
||||||
|
|
||||||
/// Finalizes a patching environment for a package
|
|
||||||
#[cfg(feature = "patches")]
|
|
||||||
PatchCommit(patch_commit::PatchCommitCommand),
|
|
||||||
|
|
||||||
/// Executes a binary package without needing to be run in a project directory
|
|
||||||
#[clap(name = "x", visible_alias = "execute", visible_alias = "exec")]
|
|
||||||
Execute(execute::ExecuteCommand),
|
|
||||||
|
|
||||||
/// Installs the pesde binary and scripts
|
|
||||||
#[cfg(feature = "version-management")]
|
|
||||||
SelfInstall(self_install::SelfInstallCommand),
|
|
||||||
|
|
||||||
/// Installs the latest version of pesde
|
|
||||||
#[cfg(feature = "version-management")]
|
|
||||||
SelfUpgrade(self_upgrade::SelfUpgradeCommand),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Subcommand {
|
impl Subcommand {
|
||||||
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
||||||
match self {
|
match self {
|
||||||
Subcommand::Auth(auth) => auth.run(project, reqwest).await,
|
Subcommand::Auth(auth) => auth.run(project, reqwest).await,
|
||||||
Subcommand::Config(config) => config.run().await,
|
Subcommand::Config(config) => config.run().await,
|
||||||
Subcommand::Cas(cas) => cas.run(project).await,
|
Subcommand::Init(init) => init.run(project).await,
|
||||||
Subcommand::Init(init) => init.run(project).await,
|
Subcommand::Run(run) => run.run(project).await,
|
||||||
Subcommand::Add(add) => add.run(project).await,
|
Subcommand::Install(install) => install.run(project, reqwest).await,
|
||||||
Subcommand::Remove(remove) => remove.run(project).await,
|
Subcommand::Publish(publish) => publish.run(project, reqwest).await,
|
||||||
Subcommand::Install(install) => install.run(project, reqwest).await,
|
#[cfg(feature = "version-management")]
|
||||||
Subcommand::Update(update) => update.run(project, reqwest).await,
|
Subcommand::SelfInstall(self_install) => self_install.run().await,
|
||||||
Subcommand::Outdated(outdated) => outdated.run(project).await,
|
#[cfg(feature = "patches")]
|
||||||
Subcommand::List(list) => list.run(project).await,
|
Subcommand::Patch(patch) => patch.run(project, reqwest).await,
|
||||||
Subcommand::Run(run) => run.run(project).await,
|
#[cfg(feature = "patches")]
|
||||||
Subcommand::Publish(publish) => publish.run(project, reqwest).await,
|
Subcommand::PatchCommit(patch_commit) => patch_commit.run(project).await,
|
||||||
Subcommand::Yank(yank) => yank.run(project, reqwest).await,
|
#[cfg(feature = "version-management")]
|
||||||
Subcommand::Deprecate(deprecate) => deprecate.run(project, reqwest).await,
|
Subcommand::SelfUpgrade(self_upgrade) => self_upgrade.run(reqwest).await,
|
||||||
#[cfg(feature = "patches")]
|
Subcommand::Add(add) => add.run(project).await,
|
||||||
Subcommand::Patch(patch) => patch.run(project, reqwest).await,
|
Subcommand::Update(update) => update.run(project, reqwest).await,
|
||||||
#[cfg(feature = "patches")]
|
Subcommand::Outdated(outdated) => outdated.run(project).await,
|
||||||
Subcommand::PatchCommit(patch_commit) => patch_commit.run(project).await,
|
Subcommand::Execute(execute) => execute.run(project, reqwest).await,
|
||||||
Subcommand::Execute(execute) => execute.run(project, reqwest).await,
|
}
|
||||||
#[cfg(feature = "version-management")]
|
}
|
||||||
Subcommand::SelfInstall(self_install) => self_install.run().await,
|
|
||||||
#[cfg(feature = "version-management")]
|
|
||||||
Subcommand::SelfUpgrade(self_upgrade) => self_upgrade.run(reqwest).await,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,130 +1,136 @@
|
||||||
use crate::cli::up_to_date_lockfile;
|
use crate::cli::up_to_date_lockfile;
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
|
use futures::future::try_join_all;
|
||||||
use pesde::{
|
use pesde::{
|
||||||
source::{
|
refresh_sources,
|
||||||
specifiers::DependencySpecifiers,
|
source::{
|
||||||
traits::{PackageRef, PackageSource, RefreshOptions, ResolveOptions},
|
refs::PackageRefs,
|
||||||
},
|
specifiers::DependencySpecifiers,
|
||||||
Project, RefreshedSources,
|
traits::{PackageRef, PackageSource},
|
||||||
|
},
|
||||||
|
Project,
|
||||||
};
|
};
|
||||||
use semver::VersionReq;
|
use semver::VersionReq;
|
||||||
use tokio::task::JoinSet;
|
use std::{collections::HashSet, sync::Arc};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct OutdatedCommand {
|
pub struct OutdatedCommand {
|
||||||
/// Whether to check within version requirements
|
/// Whether to check within version requirements
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
strict: bool,
|
strict: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl OutdatedCommand {
|
impl OutdatedCommand {
|
||||||
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
||||||
let graph = match up_to_date_lockfile(&project).await? {
|
let graph = match up_to_date_lockfile(&project).await? {
|
||||||
Some(file) => file.graph,
|
Some(file) => file.graph,
|
||||||
None => {
|
None => {
|
||||||
anyhow::bail!(
|
anyhow::bail!(
|
||||||
"lockfile is out of sync, run `{} install` to update it",
|
"lockfile is out of sync, run `{} install` to update it",
|
||||||
env!("CARGO_BIN_NAME")
|
env!("CARGO_BIN_NAME")
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let manifest = project
|
let manifest = project
|
||||||
.deser_manifest()
|
.deser_manifest()
|
||||||
.await
|
.await
|
||||||
.context("failed to read manifest")?;
|
.context("failed to read manifest")?;
|
||||||
let manifest_target_kind = manifest.target.kind();
|
let manifest_target_kind = manifest.target.kind();
|
||||||
|
|
||||||
let refreshed_sources = RefreshedSources::new();
|
let mut refreshed_sources = HashSet::new();
|
||||||
|
|
||||||
let mut tasks = graph
|
refresh_sources(
|
||||||
.into_iter()
|
&project,
|
||||||
.map(|(current_id, node)| {
|
graph
|
||||||
let project = project.clone();
|
.iter()
|
||||||
let refreshed_sources = refreshed_sources.clone();
|
.flat_map(|(_, versions)| versions.iter())
|
||||||
async move {
|
.map(|(_, node)| node.node.pkg_ref.source()),
|
||||||
let Some((alias, mut specifier, _)) = node.direct else {
|
&mut refreshed_sources,
|
||||||
return Ok::<bool, anyhow::Error>(true);
|
)
|
||||||
};
|
.await?;
|
||||||
|
|
||||||
if matches!(
|
let refreshed_sources = Arc::new(Mutex::new(refreshed_sources));
|
||||||
specifier,
|
|
||||||
DependencySpecifiers::Git(_)
|
|
||||||
| DependencySpecifiers::Workspace(_)
|
|
||||||
| DependencySpecifiers::Path(_)
|
|
||||||
) {
|
|
||||||
return Ok(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
let source = node.pkg_ref.source();
|
if try_join_all(
|
||||||
refreshed_sources
|
graph
|
||||||
.refresh(
|
.into_iter()
|
||||||
&source,
|
.flat_map(|(_, versions)| versions.into_iter())
|
||||||
&RefreshOptions {
|
.map(|(current_version_id, node)| {
|
||||||
project: project.clone(),
|
let project = project.clone();
|
||||||
},
|
let refreshed_sources = refreshed_sources.clone();
|
||||||
)
|
async move {
|
||||||
.await?;
|
let Some((alias, mut specifier, _)) = node.node.direct else {
|
||||||
|
return Ok::<bool, anyhow::Error>(true);
|
||||||
|
};
|
||||||
|
|
||||||
if !self.strict {
|
if matches!(
|
||||||
match &mut specifier {
|
specifier,
|
||||||
DependencySpecifiers::Pesde(spec) => {
|
DependencySpecifiers::Git(_) | DependencySpecifiers::Workspace(_)
|
||||||
spec.version = VersionReq::STAR;
|
) {
|
||||||
}
|
return Ok(true);
|
||||||
#[cfg(feature = "wally-compat")]
|
}
|
||||||
DependencySpecifiers::Wally(spec) => {
|
|
||||||
spec.version = VersionReq::STAR;
|
|
||||||
}
|
|
||||||
DependencySpecifiers::Git(_) => {}
|
|
||||||
DependencySpecifiers::Workspace(_) => {}
|
|
||||||
DependencySpecifiers::Path(_) => {}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
let version_id = source
|
let source = node.node.pkg_ref.source();
|
||||||
.resolve(
|
|
||||||
&specifier,
|
|
||||||
&ResolveOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
target: manifest_target_kind,
|
|
||||||
refreshed_sources: refreshed_sources.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to resolve package versions")?
|
|
||||||
.1
|
|
||||||
.pop_last()
|
|
||||||
.map(|(v_id, _)| v_id)
|
|
||||||
.with_context(|| format!("no versions of {specifier} found"))?;
|
|
||||||
|
|
||||||
if version_id != *current_id.version_id() {
|
if !self.strict {
|
||||||
println!(
|
match specifier {
|
||||||
"{} ({alias}) {} -> {version_id}",
|
DependencySpecifiers::Pesde(ref mut spec) => {
|
||||||
current_id.name(),
|
spec.version = VersionReq::STAR;
|
||||||
current_id.version_id(),
|
}
|
||||||
);
|
#[cfg(feature = "wally-compat")]
|
||||||
|
DependencySpecifiers::Wally(ref mut spec) => {
|
||||||
|
spec.version = VersionReq::STAR;
|
||||||
|
}
|
||||||
|
DependencySpecifiers::Git(_) => {}
|
||||||
|
DependencySpecifiers::Workspace(_) => {}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
return Ok(false);
|
let version_id = source
|
||||||
}
|
.resolve(
|
||||||
|
&specifier,
|
||||||
|
&project,
|
||||||
|
manifest_target_kind,
|
||||||
|
&mut *refreshed_sources.lock().await,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to resolve package versions")?
|
||||||
|
.1
|
||||||
|
.pop_last()
|
||||||
|
.map(|(v_id, _)| v_id)
|
||||||
|
.context(format!("no versions of {specifier} found"))?;
|
||||||
|
|
||||||
Ok(true)
|
if version_id != current_version_id {
|
||||||
}
|
println!(
|
||||||
})
|
"{} {} ({alias}) {} -> {}",
|
||||||
.collect::<JoinSet<_>>();
|
match node.node.pkg_ref {
|
||||||
|
PackageRefs::Pesde(pkg_ref) => pkg_ref.name.to_string(),
|
||||||
|
#[cfg(feature = "wally-compat")]
|
||||||
|
PackageRefs::Wally(pkg_ref) => pkg_ref.name.to_string(),
|
||||||
|
_ => unreachable!(),
|
||||||
|
},
|
||||||
|
current_version_id.target(),
|
||||||
|
current_version_id.version(),
|
||||||
|
version_id.version()
|
||||||
|
);
|
||||||
|
|
||||||
let mut all_up_to_date = true;
|
return Ok(false);
|
||||||
|
}
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
Ok(true)
|
||||||
if !task.unwrap()? {
|
}
|
||||||
all_up_to_date = false;
|
}),
|
||||||
}
|
)
|
||||||
}
|
.await?
|
||||||
|
.into_iter()
|
||||||
|
.all(|b| b)
|
||||||
|
{
|
||||||
|
println!("all packages are up to date");
|
||||||
|
}
|
||||||
|
|
||||||
if all_up_to_date {
|
Ok(())
|
||||||
println!("all packages are up to date");
|
}
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,87 +1,79 @@
|
||||||
use std::sync::Arc;
|
use crate::cli::{up_to_date_lockfile, VersionedPackageName};
|
||||||
|
|
||||||
use crate::cli::{
|
|
||||||
style::{CLI_STYLE, INFO_STYLE, WARN_PREFIX},
|
|
||||||
up_to_date_lockfile, VersionedPackageName,
|
|
||||||
};
|
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use console::style;
|
use colored::Colorize;
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use pesde::{
|
use pesde::{
|
||||||
patches::setup_patches_repo,
|
patches::setup_patches_repo,
|
||||||
source::{
|
source::{
|
||||||
refs::PackageRefs,
|
refs::PackageRefs,
|
||||||
traits::{DownloadOptions, PackageRef, PackageSource},
|
traits::{PackageRef, PackageSource},
|
||||||
},
|
},
|
||||||
Project, MANIFEST_FILE_NAME,
|
Project, MANIFEST_FILE_NAME,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct PatchCommand {
|
pub struct PatchCommand {
|
||||||
/// The package name to patch
|
/// The package name to patch
|
||||||
#[arg(index = 1)]
|
#[arg(index = 1)]
|
||||||
package: VersionedPackageName,
|
package: VersionedPackageName,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl PatchCommand {
|
impl PatchCommand {
|
||||||
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
||||||
let graph = if let Some(lockfile) = up_to_date_lockfile(&project).await? {
|
let graph = if let Some(lockfile) = up_to_date_lockfile(&project).await? {
|
||||||
lockfile.graph
|
lockfile.graph
|
||||||
} else {
|
} else {
|
||||||
anyhow::bail!("outdated lockfile, please run the install command first")
|
anyhow::bail!("outdated lockfile, please run the install command first")
|
||||||
};
|
};
|
||||||
|
|
||||||
let id = self.package.get(&graph)?;
|
let (name, version_id) = self.package.get(&graph)?;
|
||||||
|
|
||||||
let node = graph.get(&id).context("package not found in graph")?;
|
let node = graph
|
||||||
|
.get(&name)
|
||||||
|
.and_then(|versions| versions.get(&version_id))
|
||||||
|
.context("package not found in graph")?;
|
||||||
|
|
||||||
if matches!(
|
if matches!(node.node.pkg_ref, PackageRefs::Workspace(_)) {
|
||||||
node.pkg_ref,
|
anyhow::bail!("cannot patch a workspace package")
|
||||||
PackageRefs::Workspace(_) | PackageRefs::Path(_)
|
}
|
||||||
) {
|
|
||||||
anyhow::bail!("cannot patch a workspace or a path package")
|
|
||||||
}
|
|
||||||
|
|
||||||
let source = node.pkg_ref.source();
|
let source = node.node.pkg_ref.source();
|
||||||
|
|
||||||
let directory = project
|
let directory = project
|
||||||
.data_dir()
|
.data_dir()
|
||||||
.join("patches")
|
.join("patches")
|
||||||
.join(id.name().escaped())
|
.join(name.escaped())
|
||||||
.join(id.version_id().escaped())
|
.join(version_id.escaped())
|
||||||
.join(jiff::Timestamp::now().as_second().to_string());
|
.join(chrono::Utc::now().timestamp().to_string());
|
||||||
fs::create_dir_all(&directory).await?;
|
fs::create_dir_all(&directory).await?;
|
||||||
|
|
||||||
source
|
source
|
||||||
.download(
|
.download(&node.node.pkg_ref, &project, &reqwest)
|
||||||
&node.pkg_ref,
|
.await?
|
||||||
&DownloadOptions {
|
.0
|
||||||
project: project.clone(),
|
.write_to(&directory, project.cas_dir(), false)
|
||||||
reqwest,
|
.await
|
||||||
reporter: Arc::new(()),
|
.context("failed to write package contents")?;
|
||||||
id: Arc::new(id),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await?
|
|
||||||
.write_to(&directory, project.cas_dir(), false)
|
|
||||||
.await
|
|
||||||
.context("failed to write package contents")?;
|
|
||||||
|
|
||||||
setup_patches_repo(&directory)?;
|
setup_patches_repo(&directory)?;
|
||||||
|
|
||||||
println!(
|
println!(
|
||||||
r#"done! modify the files in the directory, then run {} {}{} to apply.
|
concat!(
|
||||||
{WARN_PREFIX}: do not commit these changes
|
"done! modify the files in the directory, then run `",
|
||||||
{}: the {MANIFEST_FILE_NAME} file will be ignored when patching"#,
|
env!("CARGO_BIN_NAME"),
|
||||||
CLI_STYLE.apply_to(concat!("`", env!("CARGO_BIN_NAME"), " patch-commit")),
|
r#" patch-commit {}` to apply.
|
||||||
style(format!("'{}'", directory.display())).cyan().bold(),
|
{}: do not commit these changes
|
||||||
CLI_STYLE.apply_to("`"),
|
{}: the {} file will be ignored when patching"#
|
||||||
INFO_STYLE.apply_to("note")
|
),
|
||||||
);
|
directory.display().to_string().bold().cyan(),
|
||||||
|
"warning".yellow(),
|
||||||
|
"note".blue(),
|
||||||
|
MANIFEST_FILE_NAME
|
||||||
|
);
|
||||||
|
|
||||||
open::that(directory)?;
|
open::that(directory)?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,102 +2,96 @@ use crate::cli::up_to_date_lockfile;
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use pesde::{
|
use pesde::{names::PackageNames, patches::create_patch, source::version_id::VersionId, Project};
|
||||||
names::PackageNames,
|
|
||||||
patches::create_patch,
|
|
||||||
source::ids::{PackageId, VersionId},
|
|
||||||
Project,
|
|
||||||
};
|
|
||||||
use std::{path::PathBuf, str::FromStr};
|
use std::{path::PathBuf, str::FromStr};
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct PatchCommitCommand {
|
pub struct PatchCommitCommand {
|
||||||
/// The directory containing the patch to commit
|
/// The directory containing the patch to commit
|
||||||
#[arg(index = 1)]
|
#[arg(index = 1)]
|
||||||
directory: PathBuf,
|
directory: PathBuf,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl PatchCommitCommand {
|
impl PatchCommitCommand {
|
||||||
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
||||||
let graph = if let Some(lockfile) = up_to_date_lockfile(&project).await? {
|
let graph = if let Some(lockfile) = up_to_date_lockfile(&project).await? {
|
||||||
lockfile.graph
|
lockfile.graph
|
||||||
} else {
|
} else {
|
||||||
anyhow::bail!("outdated lockfile, please run the install command first")
|
anyhow::bail!("outdated lockfile, please run the install command first")
|
||||||
};
|
};
|
||||||
|
|
||||||
let id = PackageId::new(
|
let (name, version_id) = (
|
||||||
PackageNames::from_escaped(
|
PackageNames::from_escaped(
|
||||||
self.directory
|
self.directory
|
||||||
.parent()
|
.parent()
|
||||||
.context("directory has no parent")?
|
.context("directory has no parent")?
|
||||||
.parent()
|
.parent()
|
||||||
.context("directory has no grandparent")?
|
.context("directory has no grandparent")?
|
||||||
.file_name()
|
.file_name()
|
||||||
.context("directory grandparent has no name")?
|
.context("directory grandparent has no name")?
|
||||||
.to_str()
|
.to_str()
|
||||||
.context("directory grandparent name is not valid")?,
|
.context("directory grandparent name is not valid")?,
|
||||||
)?,
|
)?,
|
||||||
VersionId::from_escaped(
|
VersionId::from_escaped(
|
||||||
self.directory
|
self.directory
|
||||||
.parent()
|
.parent()
|
||||||
.context("directory has no parent")?
|
.context("directory has no parent")?
|
||||||
.file_name()
|
.file_name()
|
||||||
.context("directory parent has no name")?
|
.context("directory parent has no name")?
|
||||||
.to_str()
|
.to_str()
|
||||||
.context("directory parent name is not valid")?,
|
.context("directory parent name is not valid")?,
|
||||||
)?,
|
)?,
|
||||||
);
|
);
|
||||||
|
|
||||||
graph.get(&id).context("package not found in graph")?;
|
graph
|
||||||
|
.get(&name)
|
||||||
|
.and_then(|versions| versions.get(&version_id))
|
||||||
|
.context("package not found in graph")?;
|
||||||
|
|
||||||
let mut manifest = toml_edit::DocumentMut::from_str(
|
let mut manifest = toml_edit::DocumentMut::from_str(
|
||||||
&project
|
&project
|
||||||
.read_manifest()
|
.read_manifest()
|
||||||
.await
|
.await
|
||||||
.context("failed to read manifest")?,
|
.context("failed to read manifest")?,
|
||||||
)
|
)
|
||||||
.context("failed to parse manifest")?;
|
.context("failed to parse manifest")?;
|
||||||
|
|
||||||
let patch = create_patch(&self.directory).context("failed to create patch")?;
|
let patch = create_patch(&self.directory).context("failed to create patch")?;
|
||||||
fs::remove_dir_all(self.directory)
|
fs::remove_dir_all(self.directory)
|
||||||
.await
|
.await
|
||||||
.context("failed to remove patch directory")?;
|
.context("failed to remove patch directory")?;
|
||||||
|
|
||||||
let patches_dir = project.package_dir().join("patches");
|
let patches_dir = project.package_dir().join("patches");
|
||||||
fs::create_dir_all(&patches_dir)
|
fs::create_dir_all(&patches_dir)
|
||||||
.await
|
.await
|
||||||
.context("failed to create patches directory")?;
|
.context("failed to create patches directory")?;
|
||||||
|
|
||||||
let patch_file_name = format!(
|
let patch_file_name = format!("{}-{}.patch", name.escaped(), version_id.escaped());
|
||||||
"{}-{}.patch",
|
|
||||||
id.name().escaped(),
|
|
||||||
id.version_id().escaped()
|
|
||||||
);
|
|
||||||
|
|
||||||
let patch_file = patches_dir.join(&patch_file_name);
|
let patch_file = patches_dir.join(&patch_file_name);
|
||||||
if patch_file.exists() {
|
if patch_file.exists() {
|
||||||
anyhow::bail!("patch file already exists: {}", patch_file.display());
|
anyhow::bail!("patch file already exists: {}", patch_file.display());
|
||||||
}
|
}
|
||||||
|
|
||||||
fs::write(&patch_file, patch)
|
fs::write(&patch_file, patch)
|
||||||
.await
|
.await
|
||||||
.context("failed to write patch file")?;
|
.context("failed to write patch file")?;
|
||||||
|
|
||||||
manifest["patches"].or_insert(toml_edit::Item::Table(toml_edit::Table::new()))
|
manifest["patches"].or_insert(toml_edit::Item::Table(toml_edit::Table::new()))
|
||||||
[&id.name().to_string()][&id.version_id().to_string()] =
|
[&name.to_string()][&version_id.to_string()] =
|
||||||
toml_edit::value(format!("patches/{patch_file_name}"));
|
toml_edit::value(format!("patches/{patch_file_name}"));
|
||||||
|
|
||||||
project
|
project
|
||||||
.write_manifest(manifest.to_string())
|
.write_manifest(manifest.to_string())
|
||||||
.await
|
.await
|
||||||
.context("failed to write manifest")?;
|
.context("failed to write manifest")?;
|
||||||
|
|
||||||
println!(concat!(
|
println!(concat!(
|
||||||
"done! run `",
|
"done! run `",
|
||||||
env!("CARGO_BIN_NAME"),
|
env!("CARGO_BIN_NAME"),
|
||||||
" install` to apply the patch"
|
" install` to apply the patch"
|
||||||
));
|
));
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
File diff suppressed because it is too large
Load diff
|
@ -1,59 +0,0 @@
|
||||||
use std::str::FromStr;
|
|
||||||
|
|
||||||
use anyhow::Context;
|
|
||||||
use clap::Args;
|
|
||||||
|
|
||||||
use crate::cli::{
|
|
||||||
dep_type_to_key,
|
|
||||||
style::{INFO_STYLE, SUCCESS_STYLE},
|
|
||||||
};
|
|
||||||
use pesde::{
|
|
||||||
manifest::{Alias, DependencyType},
|
|
||||||
Project,
|
|
||||||
};
|
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
|
||||||
pub struct RemoveCommand {
|
|
||||||
/// The alias of the package to remove
|
|
||||||
#[arg(index = 1)]
|
|
||||||
alias: Alias,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl RemoveCommand {
|
|
||||||
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
|
||||||
let mut manifest = toml_edit::DocumentMut::from_str(
|
|
||||||
&project
|
|
||||||
.read_manifest()
|
|
||||||
.await
|
|
||||||
.context("failed to read manifest")?,
|
|
||||||
)
|
|
||||||
.context("failed to parse manifest")?;
|
|
||||||
|
|
||||||
let Some(dep_key) = DependencyType::VARIANTS
|
|
||||||
.iter()
|
|
||||||
.copied()
|
|
||||||
.map(dep_type_to_key)
|
|
||||||
.find(|dependency_key| {
|
|
||||||
manifest[dependency_key]
|
|
||||||
.as_table_mut()
|
|
||||||
.is_some_and(|table| table.remove(self.alias.as_str()).is_some())
|
|
||||||
})
|
|
||||||
else {
|
|
||||||
anyhow::bail!("package under alias `{}` not found in manifest", self.alias)
|
|
||||||
};
|
|
||||||
|
|
||||||
project
|
|
||||||
.write_manifest(manifest.to_string())
|
|
||||||
.await
|
|
||||||
.context("failed to write manifest")?;
|
|
||||||
|
|
||||||
println!(
|
|
||||||
"{} removed {} from {}!",
|
|
||||||
SUCCESS_STYLE.apply_to("success!"),
|
|
||||||
INFO_STYLE.apply_to(self.alias),
|
|
||||||
INFO_STYLE.apply_to(dep_key)
|
|
||||||
);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -3,196 +3,175 @@ use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use futures::{StreamExt, TryStreamExt};
|
use futures::{StreamExt, TryStreamExt};
|
||||||
use pesde::{
|
use pesde::{
|
||||||
errors::{ManifestReadError, WorkspaceMembersError},
|
linking::generator::generate_bin_linking_module,
|
||||||
linking::generator::generate_bin_linking_module,
|
names::{PackageName, PackageNames},
|
||||||
names::{PackageName, PackageNames},
|
Project, MANIFEST_FILE_NAME, PACKAGES_CONTAINER_NAME,
|
||||||
source::traits::{GetTargetOptions, PackageRef, PackageSource, RefreshOptions},
|
|
||||||
Project, MANIFEST_FILE_NAME,
|
|
||||||
};
|
};
|
||||||
use relative_path::RelativePathBuf;
|
use relative_path::RelativePathBuf;
|
||||||
use std::{
|
use std::{
|
||||||
collections::HashSet, env::current_dir, ffi::OsString, io::Write, path::Path, process::Command,
|
collections::HashSet, env::current_dir, ffi::OsString, io::Write, path::PathBuf,
|
||||||
sync::Arc,
|
process::Command,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct RunCommand {
|
pub struct RunCommand {
|
||||||
/// The package name, script name, or path to a script to run
|
/// The package name, script name, or path to a script to run
|
||||||
#[arg(index = 1)]
|
#[arg(index = 1)]
|
||||||
package_or_script: Option<String>,
|
package_or_script: Option<String>,
|
||||||
|
|
||||||
/// Arguments to pass to the script
|
/// Arguments to pass to the script
|
||||||
#[arg(index = 2, last = true)]
|
#[arg(index = 2, last = true)]
|
||||||
args: Vec<OsString>,
|
args: Vec<OsString>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl RunCommand {
|
impl RunCommand {
|
||||||
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project) -> anyhow::Result<()> {
|
||||||
let run = |root: &Path, file_path: &Path| {
|
let run = |root: PathBuf, file_path: PathBuf| {
|
||||||
let mut caller = tempfile::NamedTempFile::new().expect("failed to create tempfile");
|
let mut caller = tempfile::NamedTempFile::new().expect("failed to create tempfile");
|
||||||
caller
|
caller
|
||||||
.write_all(
|
.write_all(
|
||||||
generate_bin_linking_module(
|
generate_bin_linking_module(
|
||||||
root,
|
root,
|
||||||
&format!("{:?}", file_path.to_string_lossy()),
|
&format!("{:?}", file_path.to_string_lossy()),
|
||||||
)
|
)
|
||||||
.as_bytes(),
|
.as_bytes(),
|
||||||
)
|
)
|
||||||
.expect("failed to write to tempfile");
|
.expect("failed to write to tempfile");
|
||||||
|
|
||||||
let status = Command::new("lune")
|
let status = Command::new("lune")
|
||||||
.arg("run")
|
.arg("run")
|
||||||
.arg(caller.path())
|
.arg(caller.path())
|
||||||
.arg("--")
|
.arg("--")
|
||||||
.args(&self.args)
|
.args(&self.args)
|
||||||
.current_dir(current_dir().expect("failed to get current directory"))
|
.current_dir(current_dir().expect("failed to get current directory"))
|
||||||
.status()
|
.status()
|
||||||
.expect("failed to run script");
|
.expect("failed to run script");
|
||||||
|
|
||||||
drop(caller);
|
drop(caller);
|
||||||
|
|
||||||
std::process::exit(status.code().unwrap_or(1))
|
std::process::exit(status.code().unwrap_or(1))
|
||||||
};
|
};
|
||||||
|
|
||||||
let Some(package_or_script) = self.package_or_script else {
|
let Some(package_or_script) = self.package_or_script else {
|
||||||
if let Some(script_path) = project.deser_manifest().await?.target.bin_path() {
|
if let Some(script_path) = project.deser_manifest().await?.target.bin_path() {
|
||||||
run(
|
run(
|
||||||
project.package_dir(),
|
project.package_dir().to_owned(),
|
||||||
&script_path.to_path(project.package_dir()),
|
script_path.to_path(project.package_dir()),
|
||||||
);
|
);
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
anyhow::bail!("no package or script specified, and no bin path found in manifest")
|
anyhow::bail!("no package or script specified, and no bin path found in manifest")
|
||||||
};
|
};
|
||||||
|
|
||||||
if let Ok(pkg_name) = package_or_script.parse::<PackageName>() {
|
if let Ok(pkg_name) = package_or_script.parse::<PackageName>() {
|
||||||
let graph = if let Some(lockfile) = up_to_date_lockfile(&project).await? {
|
let graph = if let Some(lockfile) = up_to_date_lockfile(&project).await? {
|
||||||
lockfile.graph
|
lockfile.graph
|
||||||
} else {
|
} else {
|
||||||
anyhow::bail!("outdated lockfile, please run the install command first")
|
anyhow::bail!("outdated lockfile, please run the install command first")
|
||||||
};
|
};
|
||||||
|
|
||||||
let pkg_name = PackageNames::Pesde(pkg_name);
|
let pkg_name = PackageNames::Pesde(pkg_name);
|
||||||
|
|
||||||
let mut versions = graph
|
for (version_id, node) in graph.get(&pkg_name).context("package not found in graph")? {
|
||||||
.into_iter()
|
if node.node.direct.is_none() {
|
||||||
.filter(|(id, node)| *id.name() == pkg_name && node.direct.is_some())
|
continue;
|
||||||
.collect::<Vec<_>>();
|
}
|
||||||
|
|
||||||
let (id, node) = match versions.len() {
|
let Some(bin_path) = node.target.bin_path() else {
|
||||||
0 => anyhow::bail!("package not found"),
|
anyhow::bail!("package has no bin path");
|
||||||
1 => versions.pop().unwrap(),
|
};
|
||||||
_ => anyhow::bail!("multiple versions found. use the package's alias instead."),
|
|
||||||
};
|
|
||||||
|
|
||||||
let container_folder = node.container_folder_from_project(
|
let base_folder = project
|
||||||
&id,
|
.deser_manifest()
|
||||||
&project,
|
.await?
|
||||||
project
|
.target
|
||||||
.deser_manifest()
|
.kind()
|
||||||
.await
|
.packages_folder(version_id.target());
|
||||||
.context("failed to deserialize manifest")?
|
let container_folder = node.node.container_folder(
|
||||||
.target
|
&project
|
||||||
.kind(),
|
.package_dir()
|
||||||
);
|
.join(base_folder)
|
||||||
|
.join(PACKAGES_CONTAINER_NAME),
|
||||||
|
&pkg_name,
|
||||||
|
version_id.version(),
|
||||||
|
);
|
||||||
|
|
||||||
let source = node.pkg_ref.source();
|
let path = bin_path.to_path(&container_folder);
|
||||||
source
|
|
||||||
.refresh(&RefreshOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.context("failed to refresh source")?;
|
|
||||||
let target = source
|
|
||||||
.get_target(
|
|
||||||
&node.pkg_ref,
|
|
||||||
&GetTargetOptions {
|
|
||||||
project,
|
|
||||||
path: Arc::from(container_folder.as_path()),
|
|
||||||
id: Arc::new(id),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let Some(bin_path) = target.bin_path() else {
|
run(path.clone(), path);
|
||||||
anyhow::bail!("package has no bin path");
|
return Ok(());
|
||||||
};
|
}
|
||||||
|
}
|
||||||
|
|
||||||
let path = bin_path.to_path(&container_folder);
|
if let Ok(manifest) = project.deser_manifest().await {
|
||||||
|
if let Some(script_path) = manifest.scripts.get(&package_or_script) {
|
||||||
|
run(
|
||||||
|
project.package_dir().to_path_buf(),
|
||||||
|
script_path.to_path(project.package_dir()),
|
||||||
|
);
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
run(&path, &path);
|
let relative_path = RelativePathBuf::from(package_or_script);
|
||||||
return Ok(());
|
let path = relative_path.to_path(project.package_dir());
|
||||||
}
|
|
||||||
|
|
||||||
if let Ok(manifest) = project.deser_manifest().await {
|
if !path.exists() {
|
||||||
if let Some(script_path) = manifest.scripts.get(&package_or_script) {
|
anyhow::bail!("path `{}` does not exist", path.display());
|
||||||
run(
|
}
|
||||||
project.package_dir(),
|
|
||||||
&script_path.to_path(project.package_dir()),
|
|
||||||
);
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let relative_path = RelativePathBuf::from(package_or_script);
|
let workspace_dir = project
|
||||||
let path = relative_path.to_path(project.package_dir());
|
.workspace_dir()
|
||||||
|
.unwrap_or_else(|| project.package_dir());
|
||||||
|
|
||||||
if !path.exists() {
|
let members = match project.workspace_members(workspace_dir, false).await {
|
||||||
anyhow::bail!("path `{}` does not exist", path.display());
|
Ok(members) => members.boxed(),
|
||||||
}
|
Err(pesde::errors::WorkspaceMembersError::ManifestMissing(e))
|
||||||
|
if e.kind() == std::io::ErrorKind::NotFound =>
|
||||||
|
{
|
||||||
|
futures::stream::empty().boxed()
|
||||||
|
}
|
||||||
|
Err(e) => Err(e).context("failed to get workspace members")?,
|
||||||
|
};
|
||||||
|
|
||||||
let workspace_dir = project
|
let members = members
|
||||||
.workspace_dir()
|
.map(|res| {
|
||||||
.unwrap_or_else(|| project.package_dir());
|
res.map_err(anyhow::Error::from)
|
||||||
|
.and_then(|(path, _)| path.canonicalize().map_err(Into::into))
|
||||||
|
})
|
||||||
|
.chain(futures::stream::once(async {
|
||||||
|
workspace_dir.canonicalize().map_err(Into::into)
|
||||||
|
}))
|
||||||
|
.try_collect::<HashSet<_>>()
|
||||||
|
.await
|
||||||
|
.context("failed to collect workspace members")?;
|
||||||
|
|
||||||
let members = match project.workspace_members(false).await {
|
let root = 'finder: {
|
||||||
Ok(members) => members.boxed(),
|
let mut current_path = path.to_path_buf();
|
||||||
Err(WorkspaceMembersError::ManifestParse(ManifestReadError::Io(e)))
|
loop {
|
||||||
if e.kind() == std::io::ErrorKind::NotFound =>
|
let canonical_path = current_path
|
||||||
{
|
.canonicalize()
|
||||||
futures::stream::empty().boxed()
|
.context("failed to canonicalize parent")?;
|
||||||
}
|
|
||||||
Err(e) => Err(e).context("failed to get workspace members")?,
|
|
||||||
};
|
|
||||||
|
|
||||||
let members = members
|
if members.contains(&canonical_path)
|
||||||
.map(|res| {
|
&& canonical_path.join(MANIFEST_FILE_NAME).exists()
|
||||||
res.map_err(anyhow::Error::from)
|
{
|
||||||
.and_then(|(path, _)| path.canonicalize().map_err(Into::into))
|
break 'finder canonical_path;
|
||||||
})
|
}
|
||||||
.chain(futures::stream::once(async {
|
|
||||||
workspace_dir.canonicalize().map_err(Into::into)
|
|
||||||
}))
|
|
||||||
.try_collect::<HashSet<_>>()
|
|
||||||
.await
|
|
||||||
.context("failed to collect workspace members")?;
|
|
||||||
|
|
||||||
let root = 'finder: {
|
if let Some(parent) = current_path.parent() {
|
||||||
let mut current_path = path.to_path_buf();
|
current_path = parent.to_path_buf();
|
||||||
loop {
|
} else {
|
||||||
let canonical_path = current_path
|
break;
|
||||||
.canonicalize()
|
}
|
||||||
.context("failed to canonicalize parent")?;
|
}
|
||||||
|
|
||||||
if members.contains(&canonical_path)
|
project.package_dir().to_path_buf()
|
||||||
&& canonical_path.join(MANIFEST_FILE_NAME).exists()
|
};
|
||||||
{
|
|
||||||
break 'finder canonical_path;
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(parent) = current_path.parent() {
|
run(root, path);
|
||||||
current_path = parent.to_path_buf();
|
|
||||||
} else {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
project.package_dir().to_path_buf()
|
Ok(())
|
||||||
};
|
}
|
||||||
|
|
||||||
run(&root, &path);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,77 +1,77 @@
|
||||||
use crate::cli::{
|
use crate::cli::{version::update_bin_exe, HOME_DIR};
|
||||||
style::{ADDED_STYLE, CLI_STYLE},
|
|
||||||
version::replace_pesde_bin_exe,
|
|
||||||
HOME_DIR,
|
|
||||||
};
|
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use console::style;
|
use colored::Colorize;
|
||||||
use std::env::current_exe;
|
use std::env::current_exe;
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct SelfInstallCommand {
|
pub struct SelfInstallCommand {
|
||||||
/// Skip adding the bin directory to the PATH
|
/// Skip adding the bin directory to the PATH
|
||||||
#[cfg(windows)]
|
#[cfg(windows)]
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
skip_add_to_path: bool,
|
skip_add_to_path: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl SelfInstallCommand {
|
impl SelfInstallCommand {
|
||||||
pub async fn run(self) -> anyhow::Result<()> {
|
pub async fn run(self) -> anyhow::Result<()> {
|
||||||
#[cfg(windows)]
|
#[cfg(windows)]
|
||||||
{
|
{
|
||||||
if !self.skip_add_to_path {
|
if !self.skip_add_to_path {
|
||||||
use crate::cli::style::WARN_STYLE;
|
use anyhow::Context;
|
||||||
use anyhow::Context;
|
use winreg::{enums::HKEY_CURRENT_USER, RegKey};
|
||||||
use windows_registry::CURRENT_USER;
|
|
||||||
|
|
||||||
let env = CURRENT_USER
|
let current_user = RegKey::predef(HKEY_CURRENT_USER);
|
||||||
.create("Environment")
|
let env = current_user
|
||||||
.context("failed to open Environment key")?;
|
.create_subkey("Environment")
|
||||||
let path = env.get_string("Path").context("failed to get Path value")?;
|
.context("failed to open Environment key")?
|
||||||
|
.0;
|
||||||
|
let path: String = env.get_value("Path").context("failed to get Path value")?;
|
||||||
|
|
||||||
let bin_dir = crate::cli::bin_dir().await?;
|
let bin_dir = crate::cli::bin_dir().await?;
|
||||||
let bin_dir = bin_dir.to_string_lossy();
|
let bin_dir = bin_dir.to_string_lossy();
|
||||||
|
|
||||||
let exists = path.split(';').any(|part| *part == bin_dir);
|
let exists = path.split(';').any(|part| *part == bin_dir);
|
||||||
|
|
||||||
if !exists {
|
if !exists {
|
||||||
let new_path = format!("{path};{bin_dir}");
|
let new_path = format!("{path};{bin_dir}");
|
||||||
env.set_string("Path", &new_path)
|
env.set_value("Path", &new_path)
|
||||||
.context("failed to set Path value")?;
|
.context("failed to set Path value")?;
|
||||||
|
|
||||||
println!(
|
println!(
|
||||||
"\nin order to allow proper functionality {} was added to PATH.\n\n{}",
|
"\nin order to allow binary exports as executables {}.\n\n{}",
|
||||||
style(format!("`~/{HOME_DIR}/bin`")).green(),
|
format!("`~/{HOME_DIR}/bin` was added to PATH").green(),
|
||||||
WARN_STYLE.apply_to("please restart your shell for this to take effect")
|
"please restart your shell for this to take effect"
|
||||||
);
|
.yellow()
|
||||||
}
|
.bold()
|
||||||
}
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
println!(
|
println!(
|
||||||
"installed {} {}!",
|
"installed {} {}!",
|
||||||
CLI_STYLE.apply_to(env!("CARGO_BIN_NAME")),
|
env!("CARGO_BIN_NAME").cyan(),
|
||||||
ADDED_STYLE.apply_to(env!("CARGO_PKG_VERSION")),
|
env!("CARGO_PKG_VERSION").yellow(),
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(unix)]
|
#[cfg(unix)]
|
||||||
{
|
{
|
||||||
println!(
|
println!(
|
||||||
r#"installed {} {}! add the following line to your shell profile in order to get the binary and binary exports as executables usable from anywhere:
|
r#"installed {} {}! add the following line to your shell profile in order to get the binary and binary exports as executables usable from anywhere:
|
||||||
|
|
||||||
{}
|
{}
|
||||||
|
|
||||||
and then restart your shell.
|
and then restart your shell.
|
||||||
"#,
|
"#,
|
||||||
CLI_STYLE.apply_to(env!("CARGO_BIN_NAME")),
|
env!("CARGO_BIN_NAME").cyan(),
|
||||||
ADDED_STYLE.apply_to(env!("CARGO_PKG_VERSION")),
|
env!("CARGO_PKG_VERSION").yellow(),
|
||||||
style(format!(r#"export PATH="$PATH:$HOME/{HOME_DIR}/bin""#)).green(),
|
format!(r#"export PATH="$PATH:~/{HOME_DIR}/bin""#)
|
||||||
);
|
.bold()
|
||||||
}
|
.green()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
replace_pesde_bin_exe(¤t_exe().context("failed to get current exe path")?).await?;
|
update_bin_exe(¤t_exe().context("failed to get current exe path")?).await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,66 +1,58 @@
|
||||||
use crate::{
|
use crate::cli::{
|
||||||
cli::{
|
config::read_config,
|
||||||
config::read_config,
|
version::{
|
||||||
style::{ADDED_STYLE, CLI_STYLE, REMOVED_STYLE},
|
current_version, get_or_download_version, get_remote_version, no_build_metadata,
|
||||||
version::{
|
update_bin_exe, TagInfo, VersionType,
|
||||||
current_version, find_latest_version, get_or_download_engine, replace_pesde_bin_exe,
|
},
|
||||||
},
|
|
||||||
},
|
|
||||||
util::no_build_metadata,
|
|
||||||
};
|
};
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use pesde::engine::EngineKind;
|
use colored::Colorize;
|
||||||
use semver::VersionReq;
|
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
#[derive(Debug, Args)]
|
||||||
pub struct SelfUpgradeCommand {
|
pub struct SelfUpgradeCommand {
|
||||||
/// Whether to use the version from the "upgrades available" message
|
/// Whether to use the version from the "upgrades available" message
|
||||||
#[clap(long, default_value_t = false)]
|
#[clap(long, default_value_t = false)]
|
||||||
use_cached: bool,
|
use_cached: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl SelfUpgradeCommand {
|
impl SelfUpgradeCommand {
|
||||||
pub async fn run(self, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
pub async fn run(self, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
||||||
let latest_version = if self.use_cached {
|
let latest_version = if self.use_cached {
|
||||||
read_config()
|
read_config()
|
||||||
.await?
|
.await?
|
||||||
.last_checked_updates
|
.last_checked_updates
|
||||||
.context("no cached version found")?
|
.context("no cached version found")?
|
||||||
.1
|
.1
|
||||||
} else {
|
} else {
|
||||||
find_latest_version(&reqwest).await?
|
get_remote_version(&reqwest, VersionType::Latest).await?
|
||||||
};
|
};
|
||||||
|
|
||||||
let latest_version_no_metadata = no_build_metadata(&latest_version);
|
let latest_version_no_metadata = no_build_metadata(&latest_version);
|
||||||
|
|
||||||
if latest_version_no_metadata <= current_version() {
|
if latest_version_no_metadata <= current_version() {
|
||||||
println!("already up to date");
|
println!("already up to date");
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
let display_latest_version = ADDED_STYLE.apply_to(latest_version_no_metadata);
|
let display_latest_version = latest_version_no_metadata.to_string().yellow().bold();
|
||||||
|
|
||||||
let confirmed = inquire::prompt_confirmation(format!(
|
if !inquire::prompt_confirmation(format!(
|
||||||
"are you sure you want to upgrade {} from {} to {display_latest_version}?",
|
"are you sure you want to upgrade {} from {} to {display_latest_version}?",
|
||||||
CLI_STYLE.apply_to(env!("CARGO_BIN_NAME")),
|
env!("CARGO_BIN_NAME").cyan(),
|
||||||
REMOVED_STYLE.apply_to(env!("CARGO_PKG_VERSION"))
|
env!("CARGO_PKG_VERSION").yellow().bold()
|
||||||
))?;
|
))? {
|
||||||
if !confirmed {
|
println!("cancelled upgrade");
|
||||||
println!("cancelled upgrade");
|
return Ok(());
|
||||||
return Ok(());
|
}
|
||||||
}
|
|
||||||
|
|
||||||
let path = get_or_download_engine(
|
let path = get_or_download_version(&reqwest, &TagInfo::Complete(latest_version), true)
|
||||||
&reqwest,
|
.await?
|
||||||
EngineKind::Pesde,
|
.unwrap();
|
||||||
VersionReq::parse(&format!("={latest_version}")).unwrap(),
|
update_bin_exe(&path).await?;
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
replace_pesde_bin_exe(&path).await?;
|
|
||||||
|
|
||||||
println!("upgraded to version {display_latest_version}!");
|
println!("upgraded to version {display_latest_version}!");
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,48 +1,85 @@
|
||||||
use crate::cli::{
|
use crate::cli::{progress_bar, run_on_workspace_members};
|
||||||
install::{install, InstallOptions},
|
use anyhow::Context;
|
||||||
run_on_workspace_members,
|
|
||||||
};
|
|
||||||
use clap::Args;
|
use clap::Args;
|
||||||
use pesde::Project;
|
use colored::Colorize;
|
||||||
use std::num::NonZeroUsize;
|
use pesde::{lockfile::Lockfile, Project};
|
||||||
|
use std::{collections::HashSet, sync::Arc};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
#[derive(Debug, Args, Copy, Clone)]
|
#[derive(Debug, Args, Copy, Clone)]
|
||||||
pub struct UpdateCommand {
|
pub struct UpdateCommand {}
|
||||||
/// Update the dependencies but don't install them
|
|
||||||
#[arg(long)]
|
|
||||||
no_install: bool,
|
|
||||||
|
|
||||||
/// The maximum number of concurrent network requests
|
|
||||||
#[arg(long, default_value = "16")]
|
|
||||||
network_concurrency: NonZeroUsize,
|
|
||||||
|
|
||||||
/// Whether to re-install all dependencies even if they are already installed
|
|
||||||
#[arg(long)]
|
|
||||||
force: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl UpdateCommand {
|
impl UpdateCommand {
|
||||||
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
||||||
let options = InstallOptions {
|
let mut refreshed_sources = HashSet::new();
|
||||||
locked: false,
|
|
||||||
prod: false,
|
|
||||||
write: !self.no_install,
|
|
||||||
network_concurrency: self.network_concurrency,
|
|
||||||
use_lockfile: false,
|
|
||||||
force: self.force,
|
|
||||||
};
|
|
||||||
|
|
||||||
install(&options, &project, reqwest.clone(), true).await?;
|
let manifest = project
|
||||||
|
.deser_manifest()
|
||||||
|
.await
|
||||||
|
.context("failed to read manifest")?;
|
||||||
|
|
||||||
run_on_workspace_members(&project, |project| {
|
println!(
|
||||||
let reqwest = reqwest.clone();
|
"\n{}\n",
|
||||||
async move {
|
format!("[now updating {} {}]", manifest.name, manifest.target)
|
||||||
install(&options, &project, reqwest, false).await?;
|
.bold()
|
||||||
Ok(())
|
.on_bright_black()
|
||||||
}
|
);
|
||||||
})
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(())
|
let graph = project
|
||||||
}
|
.dependency_graph(None, &mut refreshed_sources, false)
|
||||||
|
.await
|
||||||
|
.context("failed to build dependency graph")?;
|
||||||
|
let graph = Arc::new(graph);
|
||||||
|
|
||||||
|
project
|
||||||
|
.write_lockfile(Lockfile {
|
||||||
|
name: manifest.name,
|
||||||
|
version: manifest.version,
|
||||||
|
target: manifest.target.kind(),
|
||||||
|
overrides: manifest.overrides,
|
||||||
|
|
||||||
|
graph: {
|
||||||
|
let (rx, downloaded_graph) = project
|
||||||
|
.download_and_link(
|
||||||
|
&graph,
|
||||||
|
&Arc::new(Mutex::new(refreshed_sources)),
|
||||||
|
&reqwest,
|
||||||
|
false,
|
||||||
|
false,
|
||||||
|
|_| async { Ok::<_, std::io::Error>(()) },
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to download dependencies")?;
|
||||||
|
|
||||||
|
progress_bar(
|
||||||
|
graph.values().map(|versions| versions.len() as u64).sum(),
|
||||||
|
rx,
|
||||||
|
"📥 ".to_string(),
|
||||||
|
"downloading dependencies".to_string(),
|
||||||
|
"downloaded dependencies".to_string(),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
downloaded_graph
|
||||||
|
.await
|
||||||
|
.context("failed to download dependencies")?
|
||||||
|
},
|
||||||
|
|
||||||
|
workspace: run_on_workspace_members(&project, |project| {
|
||||||
|
let reqwest = reqwest.clone();
|
||||||
|
async move { Box::pin(self.run(project, reqwest)).await }
|
||||||
|
})
|
||||||
|
.await?,
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.context("failed to write lockfile")?;
|
||||||
|
|
||||||
|
println!(
|
||||||
|
"\n\n{}. run `{} install` in order to install the new dependencies",
|
||||||
|
"✅ done".green(),
|
||||||
|
env!("CARGO_BIN_NAME")
|
||||||
|
);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,148 +0,0 @@
|
||||||
use crate::cli::{get_index, style::SUCCESS_STYLE};
|
|
||||||
use anyhow::Context;
|
|
||||||
use clap::Args;
|
|
||||||
use pesde::{
|
|
||||||
manifest::target::TargetKind,
|
|
||||||
names::PackageName,
|
|
||||||
source::{
|
|
||||||
pesde::PesdePackageSource,
|
|
||||||
traits::{PackageSource, RefreshOptions},
|
|
||||||
},
|
|
||||||
Project,
|
|
||||||
};
|
|
||||||
use reqwest::{header::AUTHORIZATION, Method, StatusCode};
|
|
||||||
use semver::Version;
|
|
||||||
use std::{fmt::Display, str::FromStr};
|
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
|
||||||
enum TargetKindOrAll {
|
|
||||||
All,
|
|
||||||
Specific(TargetKind),
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Display for TargetKindOrAll {
|
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
||||||
match self {
|
|
||||||
TargetKindOrAll::All => write!(f, "all"),
|
|
||||||
TargetKindOrAll::Specific(kind) => write!(f, "{kind}"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl FromStr for TargetKindOrAll {
|
|
||||||
type Err = anyhow::Error;
|
|
||||||
|
|
||||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
||||||
if s.eq_ignore_ascii_case("all") {
|
|
||||||
return Ok(TargetKindOrAll::All);
|
|
||||||
}
|
|
||||||
|
|
||||||
s.parse()
|
|
||||||
.map(TargetKindOrAll::Specific)
|
|
||||||
.context("failed to parse target kind")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
|
||||||
struct YankId(PackageName, Version, TargetKindOrAll);
|
|
||||||
|
|
||||||
impl FromStr for YankId {
|
|
||||||
type Err = anyhow::Error;
|
|
||||||
|
|
||||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
||||||
let (package, version) = s
|
|
||||||
.split_once('@')
|
|
||||||
.context("package is not in format of `scope/name@version target`")?;
|
|
||||||
let target = match version.split(' ').nth(1) {
|
|
||||||
Some(target) => target
|
|
||||||
.parse()
|
|
||||||
.context("package is not in format of `scope/name@version target`")?,
|
|
||||||
None => TargetKindOrAll::All,
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(YankId(
|
|
||||||
package.parse().context("failed to parse package name")?,
|
|
||||||
version.parse().context("failed to parse version")?,
|
|
||||||
target,
|
|
||||||
))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Args)]
|
|
||||||
pub struct YankCommand {
|
|
||||||
/// Whether to unyank the package
|
|
||||||
#[clap(long)]
|
|
||||||
undo: bool,
|
|
||||||
|
|
||||||
/// The index to yank the package from
|
|
||||||
#[clap(short, long)]
|
|
||||||
index: Option<String>,
|
|
||||||
|
|
||||||
/// The package to yank
|
|
||||||
#[clap(index = 1)]
|
|
||||||
package: YankId,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl YankCommand {
|
|
||||||
pub async fn run(self, project: Project, reqwest: reqwest::Client) -> anyhow::Result<()> {
|
|
||||||
let YankId(package, version, target) = self.package;
|
|
||||||
|
|
||||||
let index_url = get_index(&project, self.index.as_deref()).await?;
|
|
||||||
let source = PesdePackageSource::new(index_url.clone());
|
|
||||||
source
|
|
||||||
.refresh(&RefreshOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.context("failed to refresh source")?;
|
|
||||||
let config = source
|
|
||||||
.config(&project)
|
|
||||||
.await
|
|
||||||
.context("failed to get index config")?;
|
|
||||||
|
|
||||||
let mut request = reqwest.request(
|
|
||||||
if self.undo {
|
|
||||||
Method::DELETE
|
|
||||||
} else {
|
|
||||||
Method::PUT
|
|
||||||
},
|
|
||||||
format!(
|
|
||||||
"{}/v1/packages/{}/{}/{}/yank",
|
|
||||||
config.api(),
|
|
||||||
urlencoding::encode(&package.to_string()),
|
|
||||||
urlencoding::encode(&version.to_string()),
|
|
||||||
urlencoding::encode(&target.to_string()),
|
|
||||||
),
|
|
||||||
);
|
|
||||||
|
|
||||||
if let Some(token) = project.auth_config().tokens().get(&index_url) {
|
|
||||||
tracing::debug!("using token for {index_url}");
|
|
||||||
request = request.header(AUTHORIZATION, token);
|
|
||||||
}
|
|
||||||
|
|
||||||
let response = request.send().await.context("failed to send request")?;
|
|
||||||
|
|
||||||
let status = response.status();
|
|
||||||
let text = response
|
|
||||||
.text()
|
|
||||||
.await
|
|
||||||
.context("failed to get response text")?;
|
|
||||||
let prefix = if self.undo { "un" } else { "" };
|
|
||||||
match status {
|
|
||||||
StatusCode::CONFLICT => {
|
|
||||||
anyhow::bail!("version is already {prefix}yanked");
|
|
||||||
}
|
|
||||||
StatusCode::FORBIDDEN => {
|
|
||||||
anyhow::bail!("unauthorized to {prefix}yank under this scope");
|
|
||||||
}
|
|
||||||
code if !code.is_success() => {
|
|
||||||
anyhow::bail!("failed to {prefix}yank package: {code} ({text})");
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
println!("{}", SUCCESS_STYLE.apply_to(text));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -7,51 +7,51 @@ use tracing::instrument;
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub struct CliConfig {
|
pub struct CliConfig {
|
||||||
#[serde(
|
#[serde(
|
||||||
serialize_with = "crate::util::serialize_gix_url",
|
serialize_with = "crate::util::serialize_gix_url",
|
||||||
deserialize_with = "crate::util::deserialize_gix_url"
|
deserialize_with = "crate::util::deserialize_gix_url"
|
||||||
)]
|
)]
|
||||||
pub default_index: gix::Url,
|
pub default_index: gix::Url,
|
||||||
|
|
||||||
pub tokens: Tokens,
|
pub tokens: Tokens,
|
||||||
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
pub last_checked_updates: Option<(jiff::Timestamp, semver::Version)>,
|
pub last_checked_updates: Option<(chrono::DateTime<chrono::Utc>, semver::Version)>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for CliConfig {
|
impl Default for CliConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
default_index: "https://github.com/pesde-pkg/index".try_into().unwrap(),
|
default_index: "https://github.com/pesde-pkg/index".try_into().unwrap(),
|
||||||
|
|
||||||
tokens: Tokens::default(),
|
tokens: Tokens(Default::default()),
|
||||||
|
|
||||||
last_checked_updates: None,
|
last_checked_updates: None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(level = "trace")]
|
#[instrument(level = "trace")]
|
||||||
pub async fn read_config() -> anyhow::Result<CliConfig> {
|
pub async fn read_config() -> anyhow::Result<CliConfig> {
|
||||||
let config_string = match fs::read_to_string(home_dir()?.join("config.toml")).await {
|
let config_string = match fs::read_to_string(home_dir()?.join("config.toml")).await {
|
||||||
Ok(config_string) => config_string,
|
Ok(config_string) => config_string,
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => {
|
Err(e) if e.kind() == std::io::ErrorKind::NotFound => {
|
||||||
return Ok(CliConfig::default());
|
return Ok(CliConfig::default());
|
||||||
}
|
}
|
||||||
Err(e) => return Err(e).context("failed to read config file"),
|
Err(e) => return Err(e).context("failed to read config file"),
|
||||||
};
|
};
|
||||||
|
|
||||||
let config = toml::from_str(&config_string).context("failed to parse config file")?;
|
let config = toml::from_str(&config_string).context("failed to parse config file")?;
|
||||||
|
|
||||||
Ok(config)
|
Ok(config)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(level = "trace")]
|
#[instrument(level = "trace")]
|
||||||
pub async fn write_config(config: &CliConfig) -> anyhow::Result<()> {
|
pub async fn write_config(config: &CliConfig) -> anyhow::Result<()> {
|
||||||
let config_string = toml::to_string(config).context("failed to serialize config")?;
|
let config_string = toml::to_string(config).context("failed to serialize config")?;
|
||||||
fs::write(home_dir()?.join("config.toml"), config_string)
|
fs::write(home_dir()?.join("config.toml"), config_string)
|
||||||
.await
|
.await
|
||||||
.context("failed to write config file")?;
|
.context("failed to write config file")?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,21 +1,21 @@
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
|
|
||||||
pub async fn make_executable<P: AsRef<Path>>(_path: P) -> anyhow::Result<()> {
|
pub async fn make_executable<P: AsRef<Path>>(_path: P) -> anyhow::Result<()> {
|
||||||
#[cfg(unix)]
|
#[cfg(unix)]
|
||||||
{
|
{
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use std::os::unix::fs::PermissionsExt;
|
use std::os::unix::fs::PermissionsExt;
|
||||||
|
|
||||||
let mut perms = fs::metadata(&_path)
|
let mut perms = fs::metadata(&_path)
|
||||||
.await
|
.await
|
||||||
.context("failed to get bin link file metadata")?
|
.context("failed to get bin link file metadata")?
|
||||||
.permissions();
|
.permissions();
|
||||||
perms.set_mode(perms.mode() | 0o111);
|
perms.set_mode(perms.mode() | 0o111);
|
||||||
fs::set_permissions(&_path, perms)
|
fs::set_permissions(&_path, perms)
|
||||||
.await
|
.await
|
||||||
.context("failed to set bin link file permissions")?;
|
.context("failed to set bin link file permissions")?;
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,574 +0,0 @@
|
||||||
use super::files::make_executable;
|
|
||||||
use crate::cli::{
|
|
||||||
bin_dir, dep_type_to_key,
|
|
||||||
reporters::{self, CliReporter},
|
|
||||||
resolve_overrides, run_on_workspace_members,
|
|
||||||
style::{ADDED_STYLE, REMOVED_STYLE, WARN_PREFIX},
|
|
||||||
up_to_date_lockfile,
|
|
||||||
};
|
|
||||||
use anyhow::Context;
|
|
||||||
use console::style;
|
|
||||||
use fs_err::tokio as fs;
|
|
||||||
use pesde::{
|
|
||||||
download_and_link::{DownloadAndLinkHooks, DownloadAndLinkOptions},
|
|
||||||
engine::EngineKind,
|
|
||||||
graph::{DependencyGraph, DependencyGraphWithTarget},
|
|
||||||
lockfile::Lockfile,
|
|
||||||
manifest::{target::TargetKind, Alias, DependencyType, Manifest},
|
|
||||||
names::PackageNames,
|
|
||||||
source::{pesde::PesdePackageSource, refs::PackageRefs, traits::PackageRef, PackageSources},
|
|
||||||
version_matches, Project, RefreshedSources, LOCKFILE_FILE_NAME, MANIFEST_FILE_NAME,
|
|
||||||
};
|
|
||||||
use std::{
|
|
||||||
collections::{BTreeMap, BTreeSet, HashMap},
|
|
||||||
num::NonZeroUsize,
|
|
||||||
sync::Arc,
|
|
||||||
time::Instant,
|
|
||||||
};
|
|
||||||
use tokio::task::JoinSet;
|
|
||||||
|
|
||||||
fn bin_link_file(alias: &Alias) -> String {
|
|
||||||
let mut all_combinations = BTreeSet::new();
|
|
||||||
|
|
||||||
for a in TargetKind::VARIANTS {
|
|
||||||
for b in TargetKind::VARIANTS {
|
|
||||||
all_combinations.insert((*a, *b));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let all_folders = all_combinations
|
|
||||||
.into_iter()
|
|
||||||
.map(|(a, b)| format!("{:?}", a.packages_folder(b)))
|
|
||||||
.collect::<BTreeSet<_>>()
|
|
||||||
.into_iter()
|
|
||||||
.collect::<Vec<_>>()
|
|
||||||
.join(", ");
|
|
||||||
|
|
||||||
format!(
|
|
||||||
include_str!("bin_link.luau"),
|
|
||||||
alias = alias,
|
|
||||||
all_folders = all_folders,
|
|
||||||
MANIFEST_FILE_NAME = MANIFEST_FILE_NAME,
|
|
||||||
LOCKFILE_FILE_NAME = LOCKFILE_FILE_NAME
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct InstallHooks {
|
|
||||||
pub bin_folder: std::path::PathBuf,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, thiserror::Error)]
|
|
||||||
#[error(transparent)]
|
|
||||||
pub struct InstallHooksError(#[from] anyhow::Error);
|
|
||||||
|
|
||||||
impl DownloadAndLinkHooks for InstallHooks {
|
|
||||||
type Error = InstallHooksError;
|
|
||||||
|
|
||||||
async fn on_bins_downloaded(
|
|
||||||
&self,
|
|
||||||
graph: &DependencyGraphWithTarget,
|
|
||||||
) -> Result<(), Self::Error> {
|
|
||||||
let mut tasks = graph
|
|
||||||
.values()
|
|
||||||
.filter(|node| node.target.bin_path().is_some())
|
|
||||||
.filter_map(|node| node.node.direct.as_ref())
|
|
||||||
.map(|(alias, _, _)| {
|
|
||||||
let bin_folder = self.bin_folder.clone();
|
|
||||||
let alias = alias.clone();
|
|
||||||
|
|
||||||
async move {
|
|
||||||
let bin_exec_file = bin_folder
|
|
||||||
.join(alias.as_str())
|
|
||||||
.with_extension(std::env::consts::EXE_EXTENSION);
|
|
||||||
|
|
||||||
let impl_folder = bin_folder.join(".impl");
|
|
||||||
fs::create_dir_all(&impl_folder)
|
|
||||||
.await
|
|
||||||
.context("failed to create bin link folder")?;
|
|
||||||
|
|
||||||
let bin_file = impl_folder.join(alias.as_str()).with_extension("luau");
|
|
||||||
fs::write(&bin_file, bin_link_file(&alias))
|
|
||||||
.await
|
|
||||||
.context("failed to write bin link file")?;
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
match fs::symlink_file(
|
|
||||||
std::env::current_exe().context("failed to get current executable path")?,
|
|
||||||
&bin_exec_file,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(_) => {}
|
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::AlreadyExists => {}
|
|
||||||
e => e.context("failed to copy bin link file")?,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(not(windows))]
|
|
||||||
fs::write(
|
|
||||||
&bin_exec_file,
|
|
||||||
format!(
|
|
||||||
r#"#!/bin/sh
|
|
||||||
exec lune run "$(dirname "$0")/.impl/{alias}.luau" -- "$@""#
|
|
||||||
),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to link bin link file")?;
|
|
||||||
|
|
||||||
make_executable(&bin_exec_file)
|
|
||||||
.await
|
|
||||||
.context("failed to make bin link file executable")?;
|
|
||||||
|
|
||||||
Ok::<_, anyhow::Error>(())
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect::<JoinSet<_>>();
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
task.unwrap()?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy)]
|
|
||||||
pub struct InstallOptions {
|
|
||||||
pub locked: bool,
|
|
||||||
pub prod: bool,
|
|
||||||
pub write: bool,
|
|
||||||
pub use_lockfile: bool,
|
|
||||||
pub network_concurrency: NonZeroUsize,
|
|
||||||
pub force: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn install(
|
|
||||||
options: &InstallOptions,
|
|
||||||
project: &Project,
|
|
||||||
reqwest: reqwest::Client,
|
|
||||||
is_root: bool,
|
|
||||||
) -> anyhow::Result<()> {
|
|
||||||
let start = Instant::now();
|
|
||||||
|
|
||||||
let refreshed_sources = RefreshedSources::new();
|
|
||||||
|
|
||||||
let manifest = project
|
|
||||||
.deser_manifest()
|
|
||||||
.await
|
|
||||||
.context("failed to read manifest")?;
|
|
||||||
|
|
||||||
let mut has_irrecoverable_changes = false;
|
|
||||||
|
|
||||||
let lockfile = if options.locked {
|
|
||||||
match up_to_date_lockfile(project).await? {
|
|
||||||
None => {
|
|
||||||
anyhow::bail!(
|
|
||||||
"lockfile is out of sync, run `{} install` to update it",
|
|
||||||
env!("CARGO_BIN_NAME")
|
|
||||||
);
|
|
||||||
}
|
|
||||||
file => file,
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
match project.deser_lockfile().await {
|
|
||||||
Ok(lockfile) => {
|
|
||||||
if lockfile.overrides != resolve_overrides(&manifest)? {
|
|
||||||
tracing::debug!("overrides are different");
|
|
||||||
has_irrecoverable_changes = true;
|
|
||||||
None
|
|
||||||
} else if lockfile.target != manifest.target.kind() {
|
|
||||||
tracing::debug!("target kind is different");
|
|
||||||
has_irrecoverable_changes = true;
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
Some(lockfile)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Err(pesde::errors::LockfileReadError::Io(e))
|
|
||||||
if e.kind() == std::io::ErrorKind::NotFound =>
|
|
||||||
{
|
|
||||||
None
|
|
||||||
}
|
|
||||||
Err(e) => return Err(e.into()),
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let overrides = resolve_overrides(&manifest)?;
|
|
||||||
|
|
||||||
let (new_lockfile, old_graph) =
|
|
||||||
reporters::run_with_reporter(|multi, root_progress, reporter| async {
|
|
||||||
let multi = multi;
|
|
||||||
let root_progress = root_progress;
|
|
||||||
|
|
||||||
root_progress.set_prefix(format!("{} {}: ", manifest.name, manifest.target));
|
|
||||||
#[cfg(feature = "version-management")]
|
|
||||||
{
|
|
||||||
root_progress.reset();
|
|
||||||
root_progress.set_message("update engine linkers");
|
|
||||||
|
|
||||||
let mut tasks = manifest
|
|
||||||
.engines
|
|
||||||
.keys()
|
|
||||||
.map(|engine| crate::cli::version::make_linker_if_needed(*engine))
|
|
||||||
.collect::<JoinSet<_>>();
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
task.unwrap()?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
root_progress.reset();
|
|
||||||
root_progress.set_message("resolve");
|
|
||||||
|
|
||||||
let old_graph = lockfile.map(|lockfile| lockfile.graph);
|
|
||||||
|
|
||||||
let graph = project
|
|
||||||
.dependency_graph(
|
|
||||||
old_graph.as_ref().filter(|_| options.use_lockfile),
|
|
||||||
refreshed_sources.clone(),
|
|
||||||
false,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to build dependency graph")?;
|
|
||||||
|
|
||||||
let mut tasks = graph
|
|
||||||
.iter()
|
|
||||||
.filter_map(|(id, node)| {
|
|
||||||
let PackageSources::Pesde(source) = node.pkg_ref.source() else {
|
|
||||||
return None;
|
|
||||||
};
|
|
||||||
#[allow(irrefutable_let_patterns)]
|
|
||||||
let PackageNames::Pesde(name) = id.name().clone() else {
|
|
||||||
panic!("unexpected package name");
|
|
||||||
};
|
|
||||||
let project = project.clone();
|
|
||||||
|
|
||||||
Some(async move {
|
|
||||||
let file = source.read_index_file(&name, &project).await.context("failed to read package index file")?.context("package not found in index")?;
|
|
||||||
|
|
||||||
Ok::<_, anyhow::Error>(if file.meta.deprecated.is_empty() {
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
Some((name, file.meta.deprecated))
|
|
||||||
})
|
|
||||||
})
|
|
||||||
})
|
|
||||||
.collect::<JoinSet<_>>();
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
let Some((name, reason)) = task.unwrap()? else {
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
|
|
||||||
multi.suspend(|| {
|
|
||||||
println!("{WARN_PREFIX}: package {name} is deprecated: {reason}");
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
let graph = Arc::new(graph);
|
|
||||||
|
|
||||||
if options.write {
|
|
||||||
root_progress.reset();
|
|
||||||
root_progress.set_length(0);
|
|
||||||
root_progress.set_message("download");
|
|
||||||
root_progress.set_style(reporters::root_progress_style_with_progress());
|
|
||||||
|
|
||||||
let hooks = InstallHooks {
|
|
||||||
bin_folder: bin_dir().await?,
|
|
||||||
};
|
|
||||||
|
|
||||||
#[allow(unused_variables)]
|
|
||||||
let downloaded_graph = project
|
|
||||||
.download_and_link(
|
|
||||||
&graph,
|
|
||||||
DownloadAndLinkOptions::<CliReporter, InstallHooks>::new(reqwest.clone())
|
|
||||||
.reporter(
|
|
||||||
#[cfg(feature = "patches")]
|
|
||||||
reporter.clone(),
|
|
||||||
#[cfg(not(feature = "patches"))]
|
|
||||||
reporter,
|
|
||||||
)
|
|
||||||
.hooks(hooks)
|
|
||||||
.refreshed_sources(refreshed_sources)
|
|
||||||
.prod(options.prod)
|
|
||||||
.network_concurrency(options.network_concurrency)
|
|
||||||
.force(options.force || has_irrecoverable_changes),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to download and link dependencies")?;
|
|
||||||
|
|
||||||
#[cfg(feature = "patches")]
|
|
||||||
{
|
|
||||||
use pesde::graph::ConvertableGraph;
|
|
||||||
root_progress.reset();
|
|
||||||
root_progress.set_length(0);
|
|
||||||
root_progress.set_message("patch");
|
|
||||||
|
|
||||||
project
|
|
||||||
.apply_patches(&downloaded_graph.clone().convert(), reporter)
|
|
||||||
.await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(feature = "version-management")]
|
|
||||||
{
|
|
||||||
let mut tasks = manifest
|
|
||||||
.engines
|
|
||||||
.into_iter()
|
|
||||||
.map(|(engine, req)| async move {
|
|
||||||
Ok::<_, anyhow::Error>(
|
|
||||||
crate::cli::version::get_installed_versions(engine)
|
|
||||||
.await?
|
|
||||||
.into_iter()
|
|
||||||
.filter(|version| version_matches(&req, version))
|
|
||||||
.next_back()
|
|
||||||
.map(|version| (engine, version)),
|
|
||||||
)
|
|
||||||
})
|
|
||||||
.collect::<JoinSet<_>>();
|
|
||||||
|
|
||||||
let mut resolved_engine_versions = HashMap::new();
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
let Some((engine, version)) = task.unwrap()? else {
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
resolved_engine_versions.insert(engine, version);
|
|
||||||
}
|
|
||||||
|
|
||||||
let manifest_target_kind = manifest.target.kind();
|
|
||||||
let mut tasks = downloaded_graph.iter()
|
|
||||||
.map(|(id, node)| {
|
|
||||||
let id = id.clone();
|
|
||||||
let node = node.clone();
|
|
||||||
let project = project.clone();
|
|
||||||
|
|
||||||
async move {
|
|
||||||
let engines = match &node.node.pkg_ref {
|
|
||||||
PackageRefs::Pesde(pkg_ref) => {
|
|
||||||
let source = PesdePackageSource::new(pkg_ref.index_url.clone());
|
|
||||||
#[allow(irrefutable_let_patterns)]
|
|
||||||
let PackageNames::Pesde(name) = id.name() else {
|
|
||||||
panic!("unexpected package name");
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut file = source.read_index_file(name, &project).await.context("failed to read package index file")?.context("package not found in index")?;
|
|
||||||
file
|
|
||||||
.entries
|
|
||||||
.remove(id.version_id())
|
|
||||||
.context("package version not found in index")?
|
|
||||||
.engines
|
|
||||||
}
|
|
||||||
#[cfg(feature = "wally-compat")]
|
|
||||||
PackageRefs::Wally(_) => Default::default(),
|
|
||||||
_ => {
|
|
||||||
let path = node.node.container_folder_from_project(
|
|
||||||
&id,
|
|
||||||
&project,
|
|
||||||
manifest_target_kind,
|
|
||||||
);
|
|
||||||
|
|
||||||
match fs::read_to_string(path.join(MANIFEST_FILE_NAME)).await {
|
|
||||||
Ok(manifest) => match toml::from_str::<Manifest>(&manifest) {
|
|
||||||
Ok(manifest) => manifest.engines,
|
|
||||||
Err(e) => return Err(e).context("failed to read package manifest"),
|
|
||||||
},
|
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => Default::default(),
|
|
||||||
Err(e) => return Err(e).context("failed to read package manifest"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok((id, engines))
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect::<JoinSet<_>>();
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
let (id, required_engines) = task.unwrap()?;
|
|
||||||
|
|
||||||
for (engine, req) in required_engines {
|
|
||||||
if engine == EngineKind::Pesde {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
let Some(version) = resolved_engine_versions.get(&engine) else {
|
|
||||||
tracing::debug!("package {id} requires {engine} {req}, but it is not installed");
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
|
|
||||||
if !version_matches(&req, version) {
|
|
||||||
multi.suspend(|| {
|
|
||||||
println!("{WARN_PREFIX}: package {id} requires {engine} {req}, but {version} is installed");
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
root_progress.reset();
|
|
||||||
root_progress.set_message("finish");
|
|
||||||
|
|
||||||
let new_lockfile = Lockfile {
|
|
||||||
name: manifest.name.clone(),
|
|
||||||
version: manifest.version,
|
|
||||||
target: manifest.target.kind(),
|
|
||||||
overrides,
|
|
||||||
|
|
||||||
graph: Arc::into_inner(graph).unwrap(),
|
|
||||||
|
|
||||||
workspace: run_on_workspace_members(project, |_| async { Ok(()) }).await?,
|
|
||||||
};
|
|
||||||
|
|
||||||
project
|
|
||||||
.write_lockfile(&new_lockfile)
|
|
||||||
.await
|
|
||||||
.context("failed to write lockfile")?;
|
|
||||||
|
|
||||||
anyhow::Ok((new_lockfile, old_graph.unwrap_or_default()))
|
|
||||||
})
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let elapsed = start.elapsed();
|
|
||||||
|
|
||||||
if is_root {
|
|
||||||
println!();
|
|
||||||
}
|
|
||||||
|
|
||||||
print_package_diff(
|
|
||||||
&format!("{} {}:", manifest.name, manifest.target),
|
|
||||||
old_graph,
|
|
||||||
new_lockfile.graph,
|
|
||||||
);
|
|
||||||
|
|
||||||
println!("done in {:.2}s", elapsed.as_secs_f64());
|
|
||||||
println!();
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Prints the difference between two graphs.
|
|
||||||
pub fn print_package_diff(prefix: &str, old_graph: DependencyGraph, new_graph: DependencyGraph) {
|
|
||||||
let mut old_pkg_map = BTreeMap::new();
|
|
||||||
let mut old_direct_pkg_map = BTreeMap::new();
|
|
||||||
let mut new_pkg_map = BTreeMap::new();
|
|
||||||
let mut new_direct_pkg_map = BTreeMap::new();
|
|
||||||
|
|
||||||
for (id, node) in &old_graph {
|
|
||||||
old_pkg_map.insert(id, node);
|
|
||||||
if node.direct.is_some() {
|
|
||||||
old_direct_pkg_map.insert(id, node);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (id, node) in &new_graph {
|
|
||||||
new_pkg_map.insert(id, node);
|
|
||||||
if node.direct.is_some() {
|
|
||||||
new_direct_pkg_map.insert(id, node);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let added_pkgs = new_pkg_map
|
|
||||||
.iter()
|
|
||||||
.filter(|(key, _)| !old_pkg_map.contains_key(*key))
|
|
||||||
.map(|(key, &node)| (key, node))
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
let removed_pkgs = old_pkg_map
|
|
||||||
.iter()
|
|
||||||
.filter(|(key, _)| !new_pkg_map.contains_key(*key))
|
|
||||||
.map(|(key, &node)| (key, node))
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
let added_direct_pkgs = new_direct_pkg_map
|
|
||||||
.iter()
|
|
||||||
.filter(|(key, _)| !old_direct_pkg_map.contains_key(*key))
|
|
||||||
.map(|(key, &node)| (key, node))
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
let removed_direct_pkgs = old_direct_pkg_map
|
|
||||||
.iter()
|
|
||||||
.filter(|(key, _)| !new_direct_pkg_map.contains_key(*key))
|
|
||||||
.map(|(key, &node)| (key, node))
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
|
|
||||||
let prefix = style(prefix).bold();
|
|
||||||
|
|
||||||
let no_changes = added_pkgs.is_empty()
|
|
||||||
&& removed_pkgs.is_empty()
|
|
||||||
&& added_direct_pkgs.is_empty()
|
|
||||||
&& removed_direct_pkgs.is_empty();
|
|
||||||
|
|
||||||
if no_changes {
|
|
||||||
println!("{prefix} already up to date");
|
|
||||||
} else {
|
|
||||||
let mut change_signs = [
|
|
||||||
(!added_pkgs.is_empty()).then(|| {
|
|
||||||
ADDED_STYLE
|
|
||||||
.apply_to(format!("+{}", added_pkgs.len()))
|
|
||||||
.to_string()
|
|
||||||
}),
|
|
||||||
(!removed_pkgs.is_empty()).then(|| {
|
|
||||||
REMOVED_STYLE
|
|
||||||
.apply_to(format!("-{}", removed_pkgs.len()))
|
|
||||||
.to_string()
|
|
||||||
}),
|
|
||||||
]
|
|
||||||
.into_iter()
|
|
||||||
.flatten()
|
|
||||||
.collect::<Vec<_>>()
|
|
||||||
.join(" ");
|
|
||||||
|
|
||||||
let changes_empty = change_signs.is_empty();
|
|
||||||
if changes_empty {
|
|
||||||
change_signs = style("(no changes)").dim().to_string();
|
|
||||||
}
|
|
||||||
|
|
||||||
println!("{prefix} {change_signs}");
|
|
||||||
|
|
||||||
if !changes_empty {
|
|
||||||
println!(
|
|
||||||
"{}{}",
|
|
||||||
ADDED_STYLE.apply_to("+".repeat(added_pkgs.len())),
|
|
||||||
REMOVED_STYLE.apply_to("-".repeat(removed_pkgs.len()))
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
let dependency_groups = added_direct_pkgs
|
|
||||||
.iter()
|
|
||||||
.map(|(key, node)| (true, key, node))
|
|
||||||
.chain(
|
|
||||||
removed_direct_pkgs
|
|
||||||
.iter()
|
|
||||||
.map(|(key, node)| (false, key, node)),
|
|
||||||
)
|
|
||||||
.filter_map(|(added, key, node)| {
|
|
||||||
node.direct.as_ref().map(|(_, _, ty)| (added, key, ty))
|
|
||||||
})
|
|
||||||
.fold(
|
|
||||||
BTreeMap::<DependencyType, BTreeSet<_>>::new(),
|
|
||||||
|mut map, (added, key, &ty)| {
|
|
||||||
map.entry(ty).or_default().insert((key, added));
|
|
||||||
map
|
|
||||||
},
|
|
||||||
);
|
|
||||||
|
|
||||||
for (ty, set) in dependency_groups {
|
|
||||||
println!();
|
|
||||||
println!(
|
|
||||||
"{}",
|
|
||||||
style(format!("{}:", dep_type_to_key(ty))).yellow().bold()
|
|
||||||
);
|
|
||||||
|
|
||||||
for (id, added) in set {
|
|
||||||
println!(
|
|
||||||
"{} {} {}",
|
|
||||||
if added {
|
|
||||||
ADDED_STYLE.apply_to("+")
|
|
||||||
} else {
|
|
||||||
REMOVED_STYLE.apply_to("-")
|
|
||||||
},
|
|
||||||
id.name(),
|
|
||||||
style(id.version_id()).dim()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
println!();
|
|
||||||
}
|
|
||||||
}
|
|
498
src/cli/mod.rs
498
src/cli/mod.rs
|
@ -1,30 +1,21 @@
|
||||||
use crate::cli::{
|
|
||||||
config::read_config,
|
|
||||||
style::{ERROR_STYLE, INFO_STYLE, WARN_STYLE},
|
|
||||||
};
|
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
|
use colored::Colorize;
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
use pesde::{
|
use pesde::{
|
||||||
errors::ManifestReadError,
|
lockfile::Lockfile,
|
||||||
lockfile::Lockfile,
|
manifest::target::TargetKind,
|
||||||
manifest::{
|
names::{PackageName, PackageNames},
|
||||||
overrides::{OverrideKey, OverrideSpecifier},
|
source::{version_id::VersionId, workspace::specifier::VersionTypeOrReq},
|
||||||
target::TargetKind,
|
Project,
|
||||||
DependencyType, Manifest,
|
|
||||||
},
|
|
||||||
names::{PackageName, PackageNames},
|
|
||||||
source::{
|
|
||||||
ids::VersionId, specifiers::DependencySpecifiers, workspace::specifier::VersionTypeOrReq,
|
|
||||||
},
|
|
||||||
Project, DEFAULT_INDEX_NAME,
|
|
||||||
};
|
};
|
||||||
use relative_path::RelativePathBuf;
|
use relative_path::RelativePathBuf;
|
||||||
use std::{
|
use std::{
|
||||||
collections::{BTreeMap, HashSet},
|
collections::{BTreeMap, HashSet},
|
||||||
future::Future,
|
future::Future,
|
||||||
path::PathBuf,
|
path::PathBuf,
|
||||||
str::FromStr,
|
str::FromStr,
|
||||||
|
time::Duration,
|
||||||
};
|
};
|
||||||
use tokio::pin;
|
use tokio::pin;
|
||||||
use tracing::instrument;
|
use tracing::instrument;
|
||||||
|
@ -33,327 +24,280 @@ pub mod auth;
|
||||||
pub mod commands;
|
pub mod commands;
|
||||||
pub mod config;
|
pub mod config;
|
||||||
pub mod files;
|
pub mod files;
|
||||||
pub mod install;
|
|
||||||
pub mod reporters;
|
|
||||||
pub mod style;
|
|
||||||
#[cfg(feature = "version-management")]
|
#[cfg(feature = "version-management")]
|
||||||
pub mod version;
|
pub mod version;
|
||||||
|
|
||||||
pub const HOME_DIR: &str = concat!(".", env!("CARGO_PKG_NAME"));
|
pub const HOME_DIR: &str = concat!(".", env!("CARGO_PKG_NAME"));
|
||||||
|
|
||||||
pub fn home_dir() -> anyhow::Result<PathBuf> {
|
pub fn home_dir() -> anyhow::Result<PathBuf> {
|
||||||
Ok(dirs::home_dir()
|
Ok(dirs::home_dir()
|
||||||
.context("failed to get home directory")?
|
.context("failed to get home directory")?
|
||||||
.join(HOME_DIR))
|
.join(HOME_DIR))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn bin_dir() -> anyhow::Result<PathBuf> {
|
pub async fn bin_dir() -> anyhow::Result<PathBuf> {
|
||||||
let bin_dir = home_dir()?.join("bin");
|
let bin_dir = home_dir()?.join("bin");
|
||||||
fs::create_dir_all(&bin_dir)
|
fs::create_dir_all(&bin_dir)
|
||||||
.await
|
.await
|
||||||
.context("failed to create bin folder")?;
|
.context("failed to create bin folder")?;
|
||||||
Ok(bin_dir)
|
Ok(bin_dir)
|
||||||
}
|
|
||||||
|
|
||||||
pub fn resolve_overrides(
|
|
||||||
manifest: &Manifest,
|
|
||||||
) -> anyhow::Result<BTreeMap<OverrideKey, DependencySpecifiers>> {
|
|
||||||
let mut dependencies = None;
|
|
||||||
let mut overrides = BTreeMap::new();
|
|
||||||
|
|
||||||
for (key, spec) in &manifest.overrides {
|
|
||||||
overrides.insert(
|
|
||||||
key.clone(),
|
|
||||||
match spec {
|
|
||||||
OverrideSpecifier::Specifier(spec) => spec,
|
|
||||||
OverrideSpecifier::Alias(alias) => {
|
|
||||||
if dependencies.is_none() {
|
|
||||||
dependencies = Some(
|
|
||||||
manifest
|
|
||||||
.all_dependencies()
|
|
||||||
.context("failed to get all dependencies")?,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
&dependencies
|
|
||||||
.as_ref()
|
|
||||||
.and_then(|deps| deps.get(alias))
|
|
||||||
.with_context(|| format!("alias `{alias}` not found in manifest"))?
|
|
||||||
.0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
.clone(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(overrides)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(skip(project), ret(level = "trace"), level = "debug")]
|
#[instrument(skip(project), ret(level = "trace"), level = "debug")]
|
||||||
pub async fn up_to_date_lockfile(project: &Project) -> anyhow::Result<Option<Lockfile>> {
|
pub async fn up_to_date_lockfile(project: &Project) -> anyhow::Result<Option<Lockfile>> {
|
||||||
let manifest = project.deser_manifest().await?;
|
let manifest = project.deser_manifest().await?;
|
||||||
let lockfile = match project.deser_lockfile().await {
|
let lockfile = match project.deser_lockfile().await {
|
||||||
Ok(lockfile) => lockfile,
|
Ok(lockfile) => lockfile,
|
||||||
Err(pesde::errors::LockfileReadError::Io(e))
|
Err(pesde::errors::LockfileReadError::Io(e))
|
||||||
if e.kind() == std::io::ErrorKind::NotFound =>
|
if e.kind() == std::io::ErrorKind::NotFound =>
|
||||||
{
|
{
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
Err(e) => return Err(e.into()),
|
Err(e) => return Err(e.into()),
|
||||||
};
|
};
|
||||||
|
|
||||||
if resolve_overrides(&manifest)? != lockfile.overrides {
|
if manifest.overrides != lockfile.overrides {
|
||||||
tracing::debug!("overrides are different");
|
tracing::debug!("overrides are different");
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
|
|
||||||
if manifest.target.kind() != lockfile.target {
|
if manifest.target.kind() != lockfile.target {
|
||||||
tracing::debug!("target kind is different");
|
tracing::debug!("target kind is different");
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
|
|
||||||
if manifest.name != lockfile.name || manifest.version != lockfile.version {
|
if manifest.name != lockfile.name || manifest.version != lockfile.version {
|
||||||
tracing::debug!("name or version is different");
|
tracing::debug!("name or version is different");
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
|
|
||||||
let specs = lockfile
|
let specs = lockfile
|
||||||
.graph
|
.graph
|
||||||
.iter()
|
.iter()
|
||||||
.filter_map(|(_, node)| {
|
.flat_map(|(_, versions)| versions)
|
||||||
node.direct
|
.filter_map(|(_, node)| {
|
||||||
.as_ref()
|
node.node
|
||||||
.map(|(_, spec, source_ty)| (spec, source_ty))
|
.direct
|
||||||
})
|
.as_ref()
|
||||||
.collect::<HashSet<_>>();
|
.map(|(_, spec, source_ty)| (spec, source_ty))
|
||||||
|
})
|
||||||
|
.collect::<HashSet<_>>();
|
||||||
|
|
||||||
let same_dependencies = manifest
|
let same_dependencies = manifest
|
||||||
.all_dependencies()
|
.all_dependencies()
|
||||||
.context("failed to get all dependencies")?
|
.context("failed to get all dependencies")?
|
||||||
.iter()
|
.iter()
|
||||||
.all(|(_, (spec, ty))| specs.contains(&(spec, ty)));
|
.all(|(_, (spec, ty))| specs.contains(&(spec, ty)));
|
||||||
|
|
||||||
tracing::debug!("dependencies are the same: {same_dependencies}");
|
tracing::debug!("dependencies are the same: {same_dependencies}");
|
||||||
|
|
||||||
Ok(if same_dependencies {
|
Ok(if same_dependencies {
|
||||||
Some(lockfile)
|
Some(lockfile)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
struct VersionedPackageName<V: FromStr = VersionId, N: FromStr = PackageNames>(N, Option<V>);
|
struct VersionedPackageName<V: FromStr = VersionId, N: FromStr = PackageNames>(N, Option<V>);
|
||||||
|
|
||||||
impl<V: FromStr<Err = E>, E: Into<anyhow::Error>, N: FromStr<Err = F>, F: Into<anyhow::Error>>
|
impl<V: FromStr<Err = E>, E: Into<anyhow::Error>, N: FromStr<Err = F>, F: Into<anyhow::Error>>
|
||||||
FromStr for VersionedPackageName<V, N>
|
FromStr for VersionedPackageName<V, N>
|
||||||
{
|
{
|
||||||
type Err = anyhow::Error;
|
type Err = anyhow::Error;
|
||||||
|
|
||||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||||
let mut parts = s.splitn(2, '@');
|
let mut parts = s.splitn(2, '@');
|
||||||
let name = parts.next().unwrap();
|
let name = parts.next().unwrap();
|
||||||
let version = parts
|
let version = parts
|
||||||
.next()
|
.next()
|
||||||
.map(FromStr::from_str)
|
.map(FromStr::from_str)
|
||||||
.transpose()
|
.transpose()
|
||||||
.map_err(Into::into)?;
|
.map_err(Into::into)?;
|
||||||
|
|
||||||
Ok(VersionedPackageName(
|
Ok(VersionedPackageName(
|
||||||
name.parse().map_err(Into::into)?,
|
name.parse().map_err(Into::into)?,
|
||||||
version,
|
version,
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl VersionedPackageName {
|
impl VersionedPackageName {
|
||||||
#[cfg(feature = "patches")]
|
#[cfg(feature = "patches")]
|
||||||
fn get(
|
fn get(
|
||||||
self,
|
self,
|
||||||
graph: &pesde::graph::DependencyGraph,
|
graph: &pesde::lockfile::DownloadedGraph,
|
||||||
) -> anyhow::Result<pesde::source::ids::PackageId> {
|
) -> anyhow::Result<(PackageNames, VersionId)> {
|
||||||
let version_id = match self.1 {
|
let version_id = match self.1 {
|
||||||
Some(version) => version,
|
Some(version) => version,
|
||||||
None => {
|
None => {
|
||||||
let versions = graph
|
let versions = graph.get(&self.0).context("package not found in graph")?;
|
||||||
.keys()
|
if versions.len() == 1 {
|
||||||
.filter(|id| *id.name() == self.0)
|
let version = versions.keys().next().unwrap().clone();
|
||||||
.collect::<Vec<_>>();
|
tracing::debug!("only one version found, using {version}");
|
||||||
|
version
|
||||||
|
} else {
|
||||||
|
anyhow::bail!(
|
||||||
|
"multiple versions found, please specify one of: {}",
|
||||||
|
versions
|
||||||
|
.keys()
|
||||||
|
.map(|v| v.to_string())
|
||||||
|
.collect::<Vec<_>>()
|
||||||
|
.join(", ")
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
match versions.len() {
|
Ok((self.0, version_id))
|
||||||
0 => anyhow::bail!("package not found"),
|
}
|
||||||
1 => versions[0].version_id().clone(),
|
|
||||||
_ => anyhow::bail!(
|
|
||||||
"multiple versions found, please specify one of: {}",
|
|
||||||
versions
|
|
||||||
.iter()
|
|
||||||
.map(|v| v.to_string())
|
|
||||||
.collect::<Vec<_>>()
|
|
||||||
.join(", ")
|
|
||||||
),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(pesde::source::ids::PackageId::new(self.0, version_id))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
enum AnyPackageIdentifier<V: FromStr = VersionId, N: FromStr = PackageNames> {
|
enum AnyPackageIdentifier<V: FromStr = VersionId, N: FromStr = PackageNames> {
|
||||||
PackageName(VersionedPackageName<V, N>),
|
PackageName(VersionedPackageName<V, N>),
|
||||||
Url((gix::Url, String)),
|
Url((gix::Url, String)),
|
||||||
Workspace(VersionedPackageName<VersionTypeOrReq, PackageName>),
|
Workspace(VersionedPackageName<VersionTypeOrReq, PackageName>),
|
||||||
Path(PathBuf),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<V: FromStr<Err = E>, E: Into<anyhow::Error>, N: FromStr<Err = F>, F: Into<anyhow::Error>>
|
impl<V: FromStr<Err = E>, E: Into<anyhow::Error>, N: FromStr<Err = F>, F: Into<anyhow::Error>>
|
||||||
FromStr for AnyPackageIdentifier<V, N>
|
FromStr for AnyPackageIdentifier<V, N>
|
||||||
{
|
{
|
||||||
type Err = anyhow::Error;
|
type Err = anyhow::Error;
|
||||||
|
|
||||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||||
if let Some(s) = s.strip_prefix("gh#") {
|
if let Some(s) = s.strip_prefix("gh#") {
|
||||||
let s = format!("https://github.com/{s}");
|
let s = format!("https://github.com/{s}");
|
||||||
let (repo, rev) = s.split_once('#').context("missing revision")?;
|
let (repo, rev) = s.split_once('#').context("missing revision")?;
|
||||||
|
|
||||||
Ok(AnyPackageIdentifier::Url((
|
Ok(AnyPackageIdentifier::Url((
|
||||||
repo.try_into()?,
|
repo.try_into()?,
|
||||||
rev.to_string(),
|
rev.to_string(),
|
||||||
)))
|
)))
|
||||||
} else if let Some(rest) = s.strip_prefix("workspace:") {
|
} else if let Some(rest) = s.strip_prefix("workspace:") {
|
||||||
Ok(AnyPackageIdentifier::Workspace(rest.parse()?))
|
Ok(AnyPackageIdentifier::Workspace(rest.parse()?))
|
||||||
} else if let Some(rest) = s.strip_prefix("path:") {
|
} else if s.contains(':') {
|
||||||
Ok(AnyPackageIdentifier::Path(rest.into()))
|
let (url, rev) = s.split_once('#').context("missing revision")?;
|
||||||
} else if s.contains(':') {
|
|
||||||
let (url, rev) = s.split_once('#').context("missing revision")?;
|
|
||||||
|
|
||||||
Ok(AnyPackageIdentifier::Url((
|
Ok(AnyPackageIdentifier::Url((
|
||||||
url.try_into()?,
|
url.try_into()?,
|
||||||
rev.to_string(),
|
rev.to_string(),
|
||||||
)))
|
)))
|
||||||
} else {
|
} else {
|
||||||
Ok(AnyPackageIdentifier::PackageName(s.parse()?))
|
Ok(AnyPackageIdentifier::PackageName(s.parse()?))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn parse_gix_url(s: &str) -> Result<gix::Url, gix::url::parse::Error> {
|
pub fn parse_gix_url(s: &str) -> Result<gix::Url, gix::url::parse::Error> {
|
||||||
s.try_into()
|
s.try_into()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn progress_bar<E: std::error::Error + Into<anyhow::Error>>(
|
||||||
|
len: u64,
|
||||||
|
mut rx: tokio::sync::mpsc::Receiver<Result<String, E>>,
|
||||||
|
prefix: String,
|
||||||
|
progress_msg: String,
|
||||||
|
finish_msg: String,
|
||||||
|
) -> anyhow::Result<()> {
|
||||||
|
let bar = indicatif::ProgressBar::new(len)
|
||||||
|
.with_style(
|
||||||
|
indicatif::ProgressStyle::default_bar()
|
||||||
|
.template("{prefix}[{elapsed_precise}] {bar:40.208/166} {pos}/{len} {msg}")?
|
||||||
|
.progress_chars("█▓▒░ "),
|
||||||
|
)
|
||||||
|
.with_prefix(prefix)
|
||||||
|
.with_message(progress_msg);
|
||||||
|
bar.enable_steady_tick(Duration::from_millis(100));
|
||||||
|
|
||||||
|
while let Some(result) = rx.recv().await {
|
||||||
|
bar.inc(1);
|
||||||
|
|
||||||
|
match result {
|
||||||
|
Ok(text) => {
|
||||||
|
bar.set_message(text);
|
||||||
|
}
|
||||||
|
Err(e) => return Err(e.into()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bar.finish_with_message(finish_msg);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn shift_project_dir(project: &Project, pkg_dir: PathBuf) -> Project {
|
pub fn shift_project_dir(project: &Project, pkg_dir: PathBuf) -> Project {
|
||||||
Project::new(
|
Project::new(
|
||||||
pkg_dir,
|
pkg_dir,
|
||||||
Some(project.package_dir()),
|
Some(project.package_dir()),
|
||||||
project.data_dir(),
|
project.data_dir(),
|
||||||
project.cas_dir(),
|
project.cas_dir(),
|
||||||
project.auth_config().clone(),
|
project.auth_config().clone(),
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn run_on_workspace_members<F: Future<Output = anyhow::Result<()>>>(
|
pub async fn run_on_workspace_members<F: Future<Output = anyhow::Result<()>>>(
|
||||||
project: &Project,
|
project: &Project,
|
||||||
f: impl Fn(Project) -> F,
|
f: impl Fn(Project) -> F,
|
||||||
) -> anyhow::Result<BTreeMap<PackageName, BTreeMap<TargetKind, RelativePathBuf>>> {
|
) -> anyhow::Result<BTreeMap<PackageName, BTreeMap<TargetKind, RelativePathBuf>>> {
|
||||||
// this might seem counterintuitive, but remember that
|
// this might seem counterintuitive, but remember that
|
||||||
// the presence of a workspace dir means that this project is a member of one
|
// the presence of a workspace dir means that this project is a member of one
|
||||||
if project.workspace_dir().is_some() {
|
if project.workspace_dir().is_some() {
|
||||||
return Ok(Default::default());
|
return Ok(Default::default());
|
||||||
}
|
}
|
||||||
|
|
||||||
let members_future = project.workspace_members(true).await?;
|
let members_future = project
|
||||||
pin!(members_future);
|
.workspace_members(project.package_dir(), true)
|
||||||
|
.await?;
|
||||||
|
pin!(members_future);
|
||||||
|
|
||||||
let mut results = BTreeMap::<PackageName, BTreeMap<TargetKind, RelativePathBuf>>::new();
|
let mut results = BTreeMap::<PackageName, BTreeMap<TargetKind, RelativePathBuf>>::new();
|
||||||
|
|
||||||
while let Some((path, manifest)) = members_future.next().await.transpose()? {
|
while let Some((path, manifest)) = members_future.next().await.transpose()? {
|
||||||
let relative_path =
|
let relative_path =
|
||||||
RelativePathBuf::from_path(path.strip_prefix(project.package_dir()).unwrap()).unwrap();
|
RelativePathBuf::from_path(path.strip_prefix(project.package_dir()).unwrap()).unwrap();
|
||||||
|
|
||||||
// don't run on the current workspace root
|
// don't run on the current workspace root
|
||||||
if relative_path != "" {
|
if relative_path != "" {
|
||||||
f(shift_project_dir(project, path)).await?;
|
f(shift_project_dir(project, path)).await?;
|
||||||
}
|
}
|
||||||
|
|
||||||
results
|
results
|
||||||
.entry(manifest.name)
|
.entry(manifest.name)
|
||||||
.or_default()
|
.or_default()
|
||||||
.insert(manifest.target.kind(), relative_path);
|
.insert(manifest.target.kind(), relative_path);
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(results)
|
Ok(results)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn display_err(result: anyhow::Result<()>, prefix: &str) {
|
pub fn display_err(result: anyhow::Result<()>, prefix: &str) {
|
||||||
if let Err(err) = result {
|
if let Err(err) = result {
|
||||||
eprintln!(
|
eprintln!("{}: {err}\n", format!("error{prefix}").red().bold());
|
||||||
"{}: {err}\n",
|
|
||||||
ERROR_STYLE.apply_to(format!("error{prefix}"))
|
|
||||||
);
|
|
||||||
|
|
||||||
let cause = err.chain().skip(1).collect::<Vec<_>>();
|
let cause = err.chain().skip(1).collect::<Vec<_>>();
|
||||||
|
|
||||||
if !cause.is_empty() {
|
if !cause.is_empty() {
|
||||||
eprintln!("{}:", ERROR_STYLE.apply_to("caused by"));
|
eprintln!("{}:", "caused by".red().bold());
|
||||||
for err in cause {
|
for err in cause {
|
||||||
eprintln!("\t- {err}");
|
eprintln!(" - {err}");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let backtrace = err.backtrace();
|
let backtrace = err.backtrace();
|
||||||
match backtrace.status() {
|
match backtrace.status() {
|
||||||
std::backtrace::BacktraceStatus::Disabled => {
|
std::backtrace::BacktraceStatus::Disabled => {
|
||||||
eprintln!(
|
eprintln!(
|
||||||
"\n{}: set RUST_BACKTRACE=1 for a backtrace",
|
"\n{}: set RUST_BACKTRACE=1 for a backtrace",
|
||||||
INFO_STYLE.apply_to("help")
|
"help".yellow().bold()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
std::backtrace::BacktraceStatus::Captured => {
|
std::backtrace::BacktraceStatus::Captured => {
|
||||||
eprintln!("\n{}:\n{backtrace}", WARN_STYLE.apply_to("backtrace"));
|
eprintln!("\n{}:\n{backtrace}", "backtrace".yellow().bold());
|
||||||
}
|
}
|
||||||
_ => {
|
_ => {
|
||||||
eprintln!("\n{}: not captured", WARN_STYLE.apply_to("backtrace"));
|
eprintln!("\n{}: not captured", "backtrace".yellow().bold());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_index(project: &Project, index: Option<&str>) -> anyhow::Result<gix::Url> {
|
|
||||||
let manifest = match project.deser_manifest().await {
|
|
||||||
Ok(manifest) => Some(manifest),
|
|
||||||
Err(e) => match e {
|
|
||||||
ManifestReadError::Io(e) if e.kind() == std::io::ErrorKind::NotFound => None,
|
|
||||||
e => return Err(e.into()),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
let index_url = match index {
|
|
||||||
Some(index) => index.try_into().ok(),
|
|
||||||
None => match manifest {
|
|
||||||
Some(_) => None,
|
|
||||||
None => Some(read_config().await?.default_index),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
match index_url {
|
|
||||||
Some(url) => Ok(url),
|
|
||||||
None => {
|
|
||||||
let index_name = index.unwrap_or(DEFAULT_INDEX_NAME);
|
|
||||||
|
|
||||||
manifest
|
|
||||||
.unwrap()
|
|
||||||
.indices
|
|
||||||
.remove(index_name)
|
|
||||||
.with_context(|| format!("index {index_name} not found in manifest"))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn dep_type_to_key(dep_type: DependencyType) -> &'static str {
|
|
||||||
match dep_type {
|
|
||||||
DependencyType::Standard => "dependencies",
|
|
||||||
DependencyType::Dev => "dev_dependencies",
|
|
||||||
DependencyType::Peer => "peer_dependencies",
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,211 +0,0 @@
|
||||||
//! Progress reporters for the CLI
|
|
||||||
|
|
||||||
use std::{
|
|
||||||
future::Future,
|
|
||||||
io::{Stdout, Write},
|
|
||||||
sync::{Arc, Mutex, Once, OnceLock},
|
|
||||||
time::Duration,
|
|
||||||
};
|
|
||||||
|
|
||||||
use indicatif::{MultiProgress, ProgressBar, ProgressStyle};
|
|
||||||
use pesde::reporters::{
|
|
||||||
DownloadProgressReporter, DownloadsReporter, PatchProgressReporter, PatchesReporter,
|
|
||||||
};
|
|
||||||
|
|
||||||
pub const TICK_CHARS: &str = "⣷⣯⣟⡿⢿⣻⣽⣾";
|
|
||||||
|
|
||||||
pub fn root_progress_style() -> ProgressStyle {
|
|
||||||
ProgressStyle::with_template("{prefix:.dim}{msg:>8.214/yellow} {spinner} [{elapsed_precise}]")
|
|
||||||
.unwrap()
|
|
||||||
.tick_chars(TICK_CHARS)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn root_progress_style_with_progress() -> ProgressStyle {
|
|
||||||
ProgressStyle::with_template(
|
|
||||||
"{prefix:.dim}{msg:>8.214/yellow} {spinner} [{elapsed_precise}] {bar:20} {pos}/{len}",
|
|
||||||
)
|
|
||||||
.unwrap()
|
|
||||||
.tick_chars(TICK_CHARS)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn run_with_reporter_and_writer<W, F, R, Fut>(writer: W, f: F) -> R
|
|
||||||
where
|
|
||||||
W: Write + Send + Sync + 'static,
|
|
||||||
F: FnOnce(MultiProgress, ProgressBar, Arc<CliReporter<W>>) -> Fut,
|
|
||||||
Fut: Future<Output = R>,
|
|
||||||
{
|
|
||||||
let multi_progress = MultiProgress::new();
|
|
||||||
crate::PROGRESS_BARS
|
|
||||||
.lock()
|
|
||||||
.unwrap()
|
|
||||||
.replace(multi_progress.clone());
|
|
||||||
|
|
||||||
let root_progress = multi_progress.add(ProgressBar::new(0));
|
|
||||||
root_progress.set_style(root_progress_style());
|
|
||||||
root_progress.enable_steady_tick(Duration::from_millis(100));
|
|
||||||
|
|
||||||
let reporter = Arc::new(CliReporter::with_writer(
|
|
||||||
writer,
|
|
||||||
multi_progress.clone(),
|
|
||||||
root_progress.clone(),
|
|
||||||
));
|
|
||||||
let result = f(multi_progress.clone(), root_progress.clone(), reporter).await;
|
|
||||||
|
|
||||||
root_progress.finish();
|
|
||||||
multi_progress.clear().unwrap();
|
|
||||||
crate::PROGRESS_BARS.lock().unwrap().take();
|
|
||||||
|
|
||||||
result
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn run_with_reporter<F, R, Fut>(f: F) -> R
|
|
||||||
where
|
|
||||||
F: FnOnce(MultiProgress, ProgressBar, Arc<CliReporter<Stdout>>) -> Fut,
|
|
||||||
Fut: Future<Output = R>,
|
|
||||||
{
|
|
||||||
run_with_reporter_and_writer(std::io::stdout(), f).await
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct CliReporter<W = Stdout> {
|
|
||||||
writer: Mutex<W>,
|
|
||||||
child_style: ProgressStyle,
|
|
||||||
child_style_with_bytes: ProgressStyle,
|
|
||||||
child_style_with_bytes_without_total: ProgressStyle,
|
|
||||||
multi_progress: MultiProgress,
|
|
||||||
root_progress: ProgressBar,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<W> CliReporter<W> {
|
|
||||||
#[allow(unknown_lints, clippy::literal_string_with_formatting_args)]
|
|
||||||
pub fn with_writer(
|
|
||||||
writer: W,
|
|
||||||
multi_progress: MultiProgress,
|
|
||||||
root_progress: ProgressBar,
|
|
||||||
) -> Self {
|
|
||||||
Self {
|
|
||||||
writer: Mutex::new(writer),
|
|
||||||
child_style: ProgressStyle::with_template("{msg:.dim}").unwrap(),
|
|
||||||
child_style_with_bytes: ProgressStyle::with_template(
|
|
||||||
"{msg:.dim} {bytes:.dim}/{total_bytes:.dim}",
|
|
||||||
)
|
|
||||||
.unwrap(),
|
|
||||||
child_style_with_bytes_without_total: ProgressStyle::with_template(
|
|
||||||
"{msg:.dim} {bytes:.dim}",
|
|
||||||
)
|
|
||||||
.unwrap(),
|
|
||||||
multi_progress,
|
|
||||||
root_progress,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct CliDownloadProgressReporter<W> {
|
|
||||||
root_reporter: Arc<CliReporter<W>>,
|
|
||||||
name: String,
|
|
||||||
progress: OnceLock<ProgressBar>,
|
|
||||||
set_progress: Once,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<W: Write + Send + Sync + 'static> DownloadsReporter for CliReporter<W> {
|
|
||||||
type DownloadProgressReporter = CliDownloadProgressReporter<W>;
|
|
||||||
|
|
||||||
fn report_download(self: Arc<Self>, name: String) -> Self::DownloadProgressReporter {
|
|
||||||
self.root_progress.inc_length(1);
|
|
||||||
|
|
||||||
CliDownloadProgressReporter {
|
|
||||||
root_reporter: self,
|
|
||||||
name,
|
|
||||||
progress: OnceLock::new(),
|
|
||||||
set_progress: Once::new(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<W: Write + Send + Sync + 'static> DownloadProgressReporter for CliDownloadProgressReporter<W> {
|
|
||||||
fn report_start(&self) {
|
|
||||||
let progress = self.root_reporter.multi_progress.add(ProgressBar::new(0));
|
|
||||||
progress.set_style(self.root_reporter.child_style.clone());
|
|
||||||
progress.set_message(format!("- {}", self.name));
|
|
||||||
|
|
||||||
self.progress
|
|
||||||
.set(progress)
|
|
||||||
.expect("report_start called more than once");
|
|
||||||
}
|
|
||||||
|
|
||||||
fn report_progress(&self, total: u64, len: u64) {
|
|
||||||
if let Some(progress) = self.progress.get() {
|
|
||||||
progress.set_length(total);
|
|
||||||
progress.set_position(len);
|
|
||||||
|
|
||||||
self.set_progress.call_once(|| {
|
|
||||||
if total > 0 {
|
|
||||||
progress.set_style(self.root_reporter.child_style_with_bytes.clone());
|
|
||||||
} else {
|
|
||||||
progress.set_style(
|
|
||||||
self.root_reporter
|
|
||||||
.child_style_with_bytes_without_total
|
|
||||||
.clone(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn report_done(&self) {
|
|
||||||
if let Some(progress) = self.progress.get() {
|
|
||||||
if progress.is_hidden() {
|
|
||||||
writeln!(
|
|
||||||
self.root_reporter.writer.lock().unwrap(),
|
|
||||||
"downloaded {}",
|
|
||||||
self.name
|
|
||||||
)
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
progress.finish();
|
|
||||||
self.root_reporter.multi_progress.remove(progress);
|
|
||||||
self.root_reporter.root_progress.inc(1);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct CliPatchProgressReporter<W> {
|
|
||||||
root_reporter: Arc<CliReporter<W>>,
|
|
||||||
name: String,
|
|
||||||
progress: ProgressBar,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<W: Write + Send + Sync + 'static> PatchesReporter for CliReporter<W> {
|
|
||||||
type PatchProgressReporter = CliPatchProgressReporter<W>;
|
|
||||||
|
|
||||||
fn report_patch(self: Arc<Self>, name: String) -> Self::PatchProgressReporter {
|
|
||||||
let progress = self.multi_progress.add(ProgressBar::new(0));
|
|
||||||
progress.set_style(self.child_style.clone());
|
|
||||||
progress.set_message(format!("- {name}"));
|
|
||||||
|
|
||||||
self.root_progress.inc_length(1);
|
|
||||||
|
|
||||||
CliPatchProgressReporter {
|
|
||||||
root_reporter: self,
|
|
||||||
name: name.to_string(),
|
|
||||||
progress,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<W: Write + Send + Sync + 'static> PatchProgressReporter for CliPatchProgressReporter<W> {
|
|
||||||
fn report_done(&self) {
|
|
||||||
if self.progress.is_hidden() {
|
|
||||||
writeln!(
|
|
||||||
self.root_reporter.writer.lock().unwrap(),
|
|
||||||
"patched {}",
|
|
||||||
self.name
|
|
||||||
)
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
self.progress.finish();
|
|
||||||
self.root_reporter.multi_progress.remove(&self.progress);
|
|
||||||
self.root_reporter.root_progress.inc(1);
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,54 +0,0 @@
|
||||||
use console::{Style, StyledObject};
|
|
||||||
use paste::paste;
|
|
||||||
use std::{fmt::Display, sync::LazyLock};
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
pub struct LazyStyle<T>(LazyLock<T>);
|
|
||||||
|
|
||||||
impl LazyStyle<Style> {
|
|
||||||
pub fn apply_to<D>(&self, text: D) -> StyledObject<D> {
|
|
||||||
LazyLock::force(&self.0).apply_to(text)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<T: Display> Display for LazyStyle<T> {
|
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
||||||
write!(f, "{}", LazyLock::force(&self.0))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
macro_rules! make_style {
|
|
||||||
($name:ident, $color:ident) => {
|
|
||||||
make_style!($name, $color());
|
|
||||||
};
|
|
||||||
($name:ident, $($color:tt)+) => {
|
|
||||||
paste! {
|
|
||||||
pub static [<$name _STYLE>]: LazyStyle<Style> = LazyStyle(LazyLock::new(||
|
|
||||||
Style::new().$($color)+.bold()
|
|
||||||
));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
macro_rules! make_prefix {
|
|
||||||
($name:ident) => {
|
|
||||||
paste! {
|
|
||||||
pub static [<$name:upper _PREFIX>]: LazyStyle<StyledObject<&'static str>> = LazyStyle(LazyLock::new(||
|
|
||||||
[<$name:upper _STYLE>].apply_to(stringify!($name))
|
|
||||||
));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
pub const CLI_COLOR_256: u8 = 214;
|
|
||||||
|
|
||||||
make_style!(INFO, cyan);
|
|
||||||
make_style!(WARN, yellow);
|
|
||||||
make_prefix!(warn);
|
|
||||||
make_style!(ERROR, red);
|
|
||||||
make_prefix!(error);
|
|
||||||
make_style!(SUCCESS, green);
|
|
||||||
make_style!(CLI, color256(CLI_COLOR_256));
|
|
||||||
make_style!(ADDED, green);
|
|
||||||
make_style!(REMOVED, red);
|
|
||||||
make_style!(URL, blue().underlined());
|
|
|
@ -1,294 +1,361 @@
|
||||||
use crate::{
|
use crate::cli::{
|
||||||
cli::{
|
bin_dir,
|
||||||
bin_dir,
|
config::{read_config, write_config, CliConfig},
|
||||||
config::{read_config, write_config, CliConfig},
|
files::make_executable,
|
||||||
files::make_executable,
|
home_dir,
|
||||||
home_dir,
|
|
||||||
reporters::run_with_reporter,
|
|
||||||
style::{ADDED_STYLE, CLI_STYLE, REMOVED_STYLE, URL_STYLE},
|
|
||||||
},
|
|
||||||
util::no_build_metadata,
|
|
||||||
};
|
};
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use console::Style;
|
use colored::Colorize;
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use jiff::SignedDuration;
|
use futures::StreamExt;
|
||||||
use pesde::{
|
use reqwest::header::ACCEPT;
|
||||||
engine::{
|
use semver::Version;
|
||||||
source::{
|
use serde::Deserialize;
|
||||||
traits::{DownloadOptions, EngineSource, ResolveOptions},
|
|
||||||
EngineSources,
|
|
||||||
},
|
|
||||||
EngineKind,
|
|
||||||
},
|
|
||||||
reporters::DownloadsReporter,
|
|
||||||
version_matches,
|
|
||||||
};
|
|
||||||
use semver::{Version, VersionReq};
|
|
||||||
use std::{
|
use std::{
|
||||||
collections::BTreeSet,
|
env::current_exe,
|
||||||
env::current_exe,
|
path::{Path, PathBuf},
|
||||||
path::{Path, PathBuf},
|
|
||||||
sync::Arc,
|
|
||||||
};
|
};
|
||||||
|
use tokio::io::AsyncWrite;
|
||||||
use tracing::instrument;
|
use tracing::instrument;
|
||||||
|
|
||||||
pub fn current_version() -> Version {
|
pub fn current_version() -> Version {
|
||||||
Version::parse(env!("CARGO_PKG_VERSION")).unwrap()
|
Version::parse(env!("CARGO_PKG_VERSION")).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
const CHECK_INTERVAL: SignedDuration = SignedDuration::from_hours(6);
|
#[derive(Debug, Deserialize)]
|
||||||
|
struct Release {
|
||||||
pub async fn find_latest_version(reqwest: &reqwest::Client) -> anyhow::Result<Version> {
|
tag_name: String,
|
||||||
let version = EngineSources::pesde()
|
assets: Vec<Asset>,
|
||||||
.resolve(
|
|
||||||
&VersionReq::STAR,
|
|
||||||
&ResolveOptions {
|
|
||||||
reqwest: reqwest.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to resolve version")?
|
|
||||||
.pop_last()
|
|
||||||
.context("no versions found")?
|
|
||||||
.0;
|
|
||||||
|
|
||||||
Ok(version)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
struct Asset {
|
||||||
|
name: String,
|
||||||
|
url: url::Url,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument(level = "trace")]
|
||||||
|
fn get_repo() -> (String, String) {
|
||||||
|
let mut parts = env!("CARGO_PKG_REPOSITORY").split('/').skip(3);
|
||||||
|
let (owner, repo) = (
|
||||||
|
parts.next().unwrap().to_string(),
|
||||||
|
parts.next().unwrap().to_string(),
|
||||||
|
);
|
||||||
|
|
||||||
|
tracing::trace!("repository for updates: {owner}/{repo}");
|
||||||
|
|
||||||
|
(owner, repo)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum VersionType {
|
||||||
|
Latest,
|
||||||
|
Specific(Version),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[instrument(skip(reqwest), level = "trace")]
|
||||||
|
pub async fn get_remote_version(
|
||||||
|
reqwest: &reqwest::Client,
|
||||||
|
ty: VersionType,
|
||||||
|
) -> anyhow::Result<Version> {
|
||||||
|
let (owner, repo) = get_repo();
|
||||||
|
|
||||||
|
let mut releases = reqwest
|
||||||
|
.get(format!(
|
||||||
|
"https://api.github.com/repos/{owner}/{repo}/releases",
|
||||||
|
))
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.context("failed to send request to GitHub API")?
|
||||||
|
.error_for_status()
|
||||||
|
.context("failed to get GitHub API response")?
|
||||||
|
.json::<Vec<Release>>()
|
||||||
|
.await
|
||||||
|
.context("failed to parse GitHub API response")?
|
||||||
|
.into_iter()
|
||||||
|
.filter_map(|release| Version::parse(release.tag_name.trim_start_matches('v')).ok());
|
||||||
|
|
||||||
|
match ty {
|
||||||
|
VersionType::Latest => releases.max(),
|
||||||
|
VersionType::Specific(version) => {
|
||||||
|
releases.find(|v| no_build_metadata(v) == no_build_metadata(&version))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.context("failed to find latest version")
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn no_build_metadata(version: &Version) -> Version {
|
||||||
|
let mut version = version.clone();
|
||||||
|
version.build = semver::BuildMetadata::EMPTY;
|
||||||
|
version
|
||||||
|
}
|
||||||
|
|
||||||
|
const CHECK_INTERVAL: chrono::Duration = chrono::Duration::hours(6);
|
||||||
|
|
||||||
#[instrument(skip(reqwest), level = "trace")]
|
#[instrument(skip(reqwest), level = "trace")]
|
||||||
pub async fn check_for_updates(reqwest: &reqwest::Client) -> anyhow::Result<()> {
|
pub async fn check_for_updates(reqwest: &reqwest::Client) -> anyhow::Result<()> {
|
||||||
let config = read_config().await?;
|
let config = read_config().await?;
|
||||||
|
|
||||||
let version = if let Some((_, version)) = config
|
let version = if let Some((_, version)) = config
|
||||||
.last_checked_updates
|
.last_checked_updates
|
||||||
.filter(|(time, _)| jiff::Timestamp::now().duration_since(*time) < CHECK_INTERVAL)
|
.filter(|(time, _)| chrono::Utc::now() - *time < CHECK_INTERVAL)
|
||||||
{
|
{
|
||||||
tracing::debug!("using cached version");
|
tracing::debug!("using cached version");
|
||||||
version
|
version
|
||||||
} else {
|
} else {
|
||||||
tracing::debug!("checking for updates");
|
tracing::debug!("checking for updates");
|
||||||
let version = find_latest_version(reqwest).await?;
|
let version = get_remote_version(reqwest, VersionType::Latest).await?;
|
||||||
|
|
||||||
write_config(&CliConfig {
|
write_config(&CliConfig {
|
||||||
last_checked_updates: Some((jiff::Timestamp::now(), version.clone())),
|
last_checked_updates: Some((chrono::Utc::now(), version.clone())),
|
||||||
..config
|
..config
|
||||||
})
|
})
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
version
|
version
|
||||||
};
|
};
|
||||||
let current_version = current_version();
|
let current_version = current_version();
|
||||||
let version_no_metadata = no_build_metadata(&version);
|
let version_no_metadata = no_build_metadata(&version);
|
||||||
|
|
||||||
if version_no_metadata <= current_version {
|
if version_no_metadata <= current_version {
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
let alert_style = Style::new().yellow();
|
let name = env!("CARGO_BIN_NAME");
|
||||||
let changelog = format!("{}/releases/tag/v{version}", env!("CARGO_PKG_REPOSITORY"));
|
let changelog = format!("{}/releases/tag/v{version}", env!("CARGO_PKG_REPOSITORY"));
|
||||||
|
|
||||||
let messages = [
|
let unformatted_messages = [
|
||||||
format!(
|
"".to_string(),
|
||||||
"{} {} → {}",
|
format!("update available! {current_version} → {version_no_metadata}"),
|
||||||
alert_style.apply_to("update available!").bold(),
|
format!("changelog: {changelog}"),
|
||||||
REMOVED_STYLE.apply_to(current_version),
|
format!("run `{name} self-upgrade` to upgrade"),
|
||||||
ADDED_STYLE.apply_to(version_no_metadata)
|
"".to_string(),
|
||||||
),
|
];
|
||||||
format!(
|
|
||||||
"run {} to upgrade",
|
|
||||||
CLI_STYLE.apply_to(concat!("`", env!("CARGO_BIN_NAME"), " self-upgrade`")),
|
|
||||||
),
|
|
||||||
"".to_string(),
|
|
||||||
format!("changelog: {}", URL_STYLE.apply_to(changelog)),
|
|
||||||
];
|
|
||||||
|
|
||||||
let column = alert_style.apply_to("┃");
|
let width = unformatted_messages
|
||||||
|
.iter()
|
||||||
|
.map(|s| s.chars().count())
|
||||||
|
.max()
|
||||||
|
.unwrap()
|
||||||
|
+ 4;
|
||||||
|
|
||||||
let message = messages
|
let column = "│".bright_magenta();
|
||||||
.into_iter()
|
|
||||||
.map(|s| format!("{column} {s}"))
|
|
||||||
.collect::<Vec<_>>()
|
|
||||||
.join("\n");
|
|
||||||
|
|
||||||
println!("\n{message}\n");
|
let message = [
|
||||||
|
"".to_string(),
|
||||||
|
format!(
|
||||||
|
"update available! {} → {}",
|
||||||
|
current_version.to_string().red(),
|
||||||
|
version_no_metadata.to_string().green()
|
||||||
|
),
|
||||||
|
format!("changelog: {}", changelog.blue()),
|
||||||
|
format!(
|
||||||
|
"run `{} {}` to upgrade",
|
||||||
|
name.blue(),
|
||||||
|
"self-upgrade".yellow()
|
||||||
|
),
|
||||||
|
"".to_string(),
|
||||||
|
]
|
||||||
|
.into_iter()
|
||||||
|
.enumerate()
|
||||||
|
.map(|(i, s)| {
|
||||||
|
let text_length = unformatted_messages[i].chars().count();
|
||||||
|
let padding = (width as f32 - text_length as f32) / 2f32;
|
||||||
|
let padding_l = " ".repeat(padding.floor() as usize);
|
||||||
|
let padding_r = " ".repeat(padding.ceil() as usize);
|
||||||
|
format!("{column}{padding_l}{s}{padding_r}{column}")
|
||||||
|
})
|
||||||
|
.collect::<Vec<_>>()
|
||||||
|
.join("\n");
|
||||||
|
|
||||||
Ok(())
|
let lines = "─".repeat(width).bright_magenta();
|
||||||
|
|
||||||
|
let tl = "╭".bright_magenta();
|
||||||
|
let tr = "╮".bright_magenta();
|
||||||
|
let bl = "╰".bright_magenta();
|
||||||
|
let br = "╯".bright_magenta();
|
||||||
|
|
||||||
|
println!("\n{tl}{lines}{tr}\n{message}\n{bl}{lines}{br}\n");
|
||||||
|
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
const ENGINES_DIR: &str = "engines";
|
#[instrument(skip(reqwest, writer), level = "trace")]
|
||||||
|
pub async fn download_github_release<W: AsyncWrite + Unpin>(
|
||||||
|
reqwest: &reqwest::Client,
|
||||||
|
version: &Version,
|
||||||
|
mut writer: W,
|
||||||
|
) -> anyhow::Result<()> {
|
||||||
|
let (owner, repo) = get_repo();
|
||||||
|
|
||||||
#[instrument(level = "trace")]
|
let release = reqwest
|
||||||
pub async fn get_installed_versions(engine: EngineKind) -> anyhow::Result<BTreeSet<Version>> {
|
.get(format!(
|
||||||
let source = engine.source();
|
"https://api.github.com/repos/{owner}/{repo}/releases/tags/v{version}",
|
||||||
let path = home_dir()?.join(ENGINES_DIR).join(source.directory());
|
))
|
||||||
let mut installed_versions = BTreeSet::new();
|
.send()
|
||||||
|
.await
|
||||||
|
.context("failed to send request to GitHub API")?
|
||||||
|
.error_for_status()
|
||||||
|
.context("failed to get GitHub API response")?
|
||||||
|
.json::<Release>()
|
||||||
|
.await
|
||||||
|
.context("failed to parse GitHub API response")?;
|
||||||
|
|
||||||
let mut read_dir = match fs::read_dir(&path).await {
|
let asset = release
|
||||||
Ok(read_dir) => read_dir,
|
.assets
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(installed_versions),
|
.into_iter()
|
||||||
Err(e) => return Err(e).context("failed to read engines directory"),
|
.find(|asset| {
|
||||||
};
|
asset.name.ends_with(&format!(
|
||||||
|
"-{}-{}.tar.gz",
|
||||||
|
std::env::consts::OS,
|
||||||
|
std::env::consts::ARCH
|
||||||
|
))
|
||||||
|
})
|
||||||
|
.context("failed to find asset for current platform")?;
|
||||||
|
|
||||||
while let Some(entry) = read_dir.next_entry().await? {
|
let bytes = reqwest
|
||||||
let path = entry.path();
|
.get(asset.url)
|
||||||
|
.header(ACCEPT, "application/octet-stream")
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.context("failed to send request to download asset")?
|
||||||
|
.error_for_status()
|
||||||
|
.context("failed to download asset")?
|
||||||
|
.bytes()
|
||||||
|
.await
|
||||||
|
.context("failed to download asset")?;
|
||||||
|
|
||||||
let Some(version) = path.file_name().and_then(|s| s.to_str()) else {
|
let mut decoder = async_compression::tokio::bufread::GzipDecoder::new(bytes.as_ref());
|
||||||
continue;
|
let mut archive = tokio_tar::Archive::new(&mut decoder);
|
||||||
};
|
|
||||||
|
|
||||||
if let Ok(version) = Version::parse(version) {
|
let mut entry = archive
|
||||||
installed_versions.insert(version);
|
.entries()
|
||||||
}
|
.context("failed to read archive entries")?
|
||||||
}
|
.next()
|
||||||
|
.await
|
||||||
|
.context("archive has no entry")?
|
||||||
|
.context("failed to get first archive entry")?;
|
||||||
|
|
||||||
Ok(installed_versions)
|
tokio::io::copy(&mut entry, &mut writer)
|
||||||
|
.await
|
||||||
|
.context("failed to write archive entry to file")
|
||||||
|
.map(|_| ())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum TagInfo {
|
||||||
|
Complete(Version),
|
||||||
|
Incomplete(Version),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(skip(reqwest), level = "trace")]
|
#[instrument(skip(reqwest), level = "trace")]
|
||||||
pub async fn get_or_download_engine(
|
pub async fn get_or_download_version(
|
||||||
reqwest: &reqwest::Client,
|
reqwest: &reqwest::Client,
|
||||||
engine: EngineKind,
|
tag: &TagInfo,
|
||||||
req: VersionReq,
|
always_give_path: bool,
|
||||||
) -> anyhow::Result<PathBuf> {
|
) -> anyhow::Result<Option<PathBuf>> {
|
||||||
let source = engine.source();
|
let path = home_dir()?.join("versions");
|
||||||
let path = home_dir()?.join(ENGINES_DIR).join(source.directory());
|
fs::create_dir_all(&path)
|
||||||
|
.await
|
||||||
|
.context("failed to create versions directory")?;
|
||||||
|
|
||||||
let installed_versions = get_installed_versions(engine).await?;
|
let version = match tag {
|
||||||
|
TagInfo::Complete(version) => version,
|
||||||
|
// don't fetch the version since it could be cached
|
||||||
|
TagInfo::Incomplete(version) => version,
|
||||||
|
};
|
||||||
|
|
||||||
let max_matching = installed_versions
|
let path = path.join(format!(
|
||||||
.iter()
|
"{}{}",
|
||||||
.filter(|v| version_matches(&req, v))
|
no_build_metadata(version),
|
||||||
.next_back();
|
std::env::consts::EXE_SUFFIX
|
||||||
if let Some(version) = max_matching {
|
));
|
||||||
return Ok(path
|
|
||||||
.join(version.to_string())
|
|
||||||
.join(source.expected_file_name())
|
|
||||||
.with_extension(std::env::consts::EXE_EXTENSION));
|
|
||||||
}
|
|
||||||
|
|
||||||
let path = run_with_reporter(|_, root_progress, reporter| async {
|
let is_requested_version = !always_give_path && *version == current_version();
|
||||||
let root_progress = root_progress;
|
|
||||||
let reporter = reporter;
|
|
||||||
|
|
||||||
root_progress.set_message("resolve version");
|
if path.exists() {
|
||||||
let mut versions = source
|
tracing::debug!("version already exists");
|
||||||
.resolve(
|
|
||||||
&req,
|
|
||||||
&ResolveOptions {
|
|
||||||
reqwest: reqwest.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to resolve versions")?;
|
|
||||||
let (version, engine_ref) = versions.pop_last().context("no matching versions found")?;
|
|
||||||
|
|
||||||
root_progress.set_message("download");
|
return Ok(if is_requested_version {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(path)
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
let reporter = reporter.report_download(format!("{engine} v{version}"));
|
if is_requested_version {
|
||||||
|
tracing::debug!("copying current executable to version directory");
|
||||||
|
fs::copy(current_exe()?, &path)
|
||||||
|
.await
|
||||||
|
.context("failed to copy current executable to version directory")?;
|
||||||
|
} else {
|
||||||
|
let version = match tag {
|
||||||
|
TagInfo::Complete(version) => version.clone(),
|
||||||
|
TagInfo::Incomplete(version) => {
|
||||||
|
get_remote_version(reqwest, VersionType::Specific(version.clone()))
|
||||||
|
.await
|
||||||
|
.context("failed to get remote version")?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
let archive = source
|
tracing::debug!("downloading version");
|
||||||
.download(
|
download_github_release(
|
||||||
&engine_ref,
|
reqwest,
|
||||||
&DownloadOptions {
|
&version,
|
||||||
reqwest: reqwest.clone(),
|
fs::File::create(&path)
|
||||||
reporter: Arc::new(reporter),
|
.await
|
||||||
version: version.clone(),
|
.context("failed to create version file")?,
|
||||||
},
|
)
|
||||||
)
|
.await?;
|
||||||
.await
|
}
|
||||||
.context("failed to download engine")?;
|
|
||||||
|
|
||||||
let path = path.join(version.to_string());
|
make_executable(&path)
|
||||||
fs::create_dir_all(&path)
|
.await
|
||||||
.await
|
.context("failed to make downloaded version executable")?;
|
||||||
.context("failed to create engine container folder")?;
|
|
||||||
let path = path
|
|
||||||
.join(source.expected_file_name())
|
|
||||||
.with_extension(std::env::consts::EXE_EXTENSION);
|
|
||||||
|
|
||||||
let mut file = fs::File::create(&path)
|
Ok(if is_requested_version {
|
||||||
.await
|
None
|
||||||
.context("failed to create new file")?;
|
} else {
|
||||||
|
Some(path)
|
||||||
tokio::io::copy(
|
})
|
||||||
&mut archive
|
|
||||||
.find_executable(source.expected_file_name())
|
|
||||||
.await
|
|
||||||
.context("failed to find executable")?,
|
|
||||||
&mut file,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.context("failed to write to file")?;
|
|
||||||
|
|
||||||
Ok::<_, anyhow::Error>(path)
|
|
||||||
})
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
make_executable(&path)
|
|
||||||
.await
|
|
||||||
.context("failed to make downloaded version executable")?;
|
|
||||||
|
|
||||||
if engine != EngineKind::Pesde {
|
|
||||||
make_linker_if_needed(engine).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(path)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(level = "trace")]
|
#[instrument(level = "trace")]
|
||||||
pub async fn replace_pesde_bin_exe(with: &Path) -> anyhow::Result<()> {
|
pub async fn update_bin_exe(downloaded_file: &Path) -> anyhow::Result<()> {
|
||||||
let bin_exe_path = bin_dir()
|
let bin_exe_path = bin_dir().await?.join(format!(
|
||||||
.await?
|
"{}{}",
|
||||||
.join(EngineKind::Pesde.to_string())
|
env!("CARGO_BIN_NAME"),
|
||||||
.with_extension(std::env::consts::EXE_EXTENSION);
|
std::env::consts::EXE_SUFFIX
|
||||||
|
));
|
||||||
|
let mut downloaded_file = downloaded_file.to_path_buf();
|
||||||
|
|
||||||
let exists = bin_exe_path.exists();
|
let exists = bin_exe_path.exists();
|
||||||
|
|
||||||
if cfg!(target_os = "linux") && exists {
|
if cfg!(target_os = "linux") && exists {
|
||||||
fs::remove_file(&bin_exe_path)
|
fs::remove_file(&bin_exe_path)
|
||||||
.await
|
.await
|
||||||
.context("failed to remove existing executable")?;
|
.context("failed to remove existing executable")?;
|
||||||
} else if exists {
|
} else if exists {
|
||||||
let tempfile = tempfile::Builder::new()
|
let tempfile = tempfile::Builder::new()
|
||||||
.make(|_| Ok(()))
|
.make(|_| Ok(()))
|
||||||
.context("failed to create temporary file")?;
|
.context("failed to create temporary file")?;
|
||||||
let temp_path = tempfile.into_temp_path().to_path_buf();
|
let path = tempfile.into_temp_path().to_path_buf();
|
||||||
#[cfg(windows)]
|
#[cfg(windows)]
|
||||||
let temp_path = temp_path.with_extension("exe");
|
let path = path.with_extension("exe");
|
||||||
|
|
||||||
match fs::rename(&bin_exe_path, &temp_path).await {
|
let current_exe = current_exe().context("failed to get current exe path")?;
|
||||||
Ok(_) => {}
|
if current_exe == downloaded_file {
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => {}
|
downloaded_file = path.to_path_buf();
|
||||||
Err(e) => return Err(e).context("failed to rename existing executable"),
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fs::copy(with, &bin_exe_path)
|
fs::rename(&bin_exe_path, &path)
|
||||||
.await
|
.await
|
||||||
.context("failed to copy executable to bin folder")?;
|
.context("failed to rename current executable")?;
|
||||||
|
}
|
||||||
|
|
||||||
make_executable(&bin_exe_path).await
|
fs::copy(downloaded_file, &bin_exe_path)
|
||||||
}
|
.await
|
||||||
|
.context("failed to copy executable to bin folder")?;
|
||||||
#[instrument(level = "trace")]
|
|
||||||
pub async fn make_linker_if_needed(engine: EngineKind) -> anyhow::Result<()> {
|
make_executable(&bin_exe_path).await
|
||||||
let bin_dir = bin_dir().await?;
|
|
||||||
let linker = bin_dir
|
|
||||||
.join(engine.to_string())
|
|
||||||
.with_extension(std::env::consts::EXE_EXTENSION);
|
|
||||||
let exists = linker.exists();
|
|
||||||
|
|
||||||
if !exists {
|
|
||||||
let exe = current_exe().context("failed to get current exe path")?;
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
|
||||||
let result = fs::symlink_file(exe, linker);
|
|
||||||
#[cfg(not(windows))]
|
|
||||||
let result = fs::symlink(exe, linker);
|
|
||||||
|
|
||||||
result.await.context("failed to create symlink")?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
}
|
||||||
|
|
337
src/download.rs
337
src/download.rs
|
@ -1,203 +1,190 @@
|
||||||
use crate::{
|
use crate::{
|
||||||
graph::{DependencyGraph, DependencyGraphNode},
|
lockfile::{DependencyGraph, DownloadedDependencyGraphNode, DownloadedGraph},
|
||||||
reporters::{DownloadProgressReporter, DownloadsReporter},
|
manifest::DependencyType,
|
||||||
source::{
|
refresh_sources,
|
||||||
fs::PackageFs,
|
source::{
|
||||||
ids::PackageId,
|
traits::{PackageRef, PackageSource},
|
||||||
traits::{DownloadOptions, PackageRef, PackageSource, RefreshOptions},
|
PackageSources,
|
||||||
},
|
},
|
||||||
Project, RefreshedSources,
|
Project, PACKAGES_CONTAINER_NAME,
|
||||||
|
};
|
||||||
|
use fs_err::tokio as fs;
|
||||||
|
use std::{
|
||||||
|
collections::HashSet,
|
||||||
|
sync::{Arc, Mutex},
|
||||||
};
|
};
|
||||||
use async_stream::try_stream;
|
|
||||||
use futures::Stream;
|
|
||||||
use std::{num::NonZeroUsize, sync::Arc};
|
|
||||||
use tokio::{sync::Semaphore, task::JoinSet};
|
|
||||||
use tracing::{instrument, Instrument};
|
use tracing::{instrument, Instrument};
|
||||||
|
|
||||||
/// Options for downloading.
|
type MultithreadedGraph = Arc<Mutex<DownloadedGraph>>;
|
||||||
#[derive(Debug)]
|
|
||||||
pub(crate) struct DownloadGraphOptions<Reporter> {
|
|
||||||
/// The reqwest client.
|
|
||||||
pub reqwest: reqwest::Client,
|
|
||||||
/// The downloads reporter.
|
|
||||||
pub reporter: Option<Arc<Reporter>>,
|
|
||||||
/// The refreshed sources.
|
|
||||||
pub refreshed_sources: RefreshedSources,
|
|
||||||
/// The max number of concurrent network requests.
|
|
||||||
pub network_concurrency: NonZeroUsize,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<Reporter> DownloadGraphOptions<Reporter>
|
pub(crate) type MultithreadDownloadJob = (
|
||||||
where
|
tokio::sync::mpsc::Receiver<Result<String, errors::DownloadGraphError>>,
|
||||||
Reporter: DownloadsReporter + Send + Sync + 'static,
|
MultithreadedGraph,
|
||||||
{
|
);
|
||||||
/// Creates a new download options with the given reqwest client and reporter.
|
|
||||||
pub(crate) fn new(reqwest: reqwest::Client) -> Self {
|
|
||||||
Self {
|
|
||||||
reqwest,
|
|
||||||
reporter: None,
|
|
||||||
refreshed_sources: Default::default(),
|
|
||||||
network_concurrency: NonZeroUsize::new(16).unwrap(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sets the downloads reporter.
|
|
||||||
pub(crate) fn reporter(mut self, reporter: impl Into<Arc<Reporter>>) -> Self {
|
|
||||||
self.reporter.replace(reporter.into());
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sets the refreshed sources.
|
|
||||||
pub(crate) fn refreshed_sources(mut self, refreshed_sources: RefreshedSources) -> Self {
|
|
||||||
self.refreshed_sources = refreshed_sources;
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sets the max number of concurrent network requests.
|
|
||||||
pub(crate) fn network_concurrency(mut self, network_concurrency: NonZeroUsize) -> Self {
|
|
||||||
self.network_concurrency = network_concurrency;
|
|
||||||
self
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<Reporter> Clone for DownloadGraphOptions<Reporter> {
|
|
||||||
fn clone(&self) -> Self {
|
|
||||||
Self {
|
|
||||||
reqwest: self.reqwest.clone(),
|
|
||||||
reporter: self.reporter.clone(),
|
|
||||||
refreshed_sources: self.refreshed_sources.clone(),
|
|
||||||
network_concurrency: self.network_concurrency,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Project {
|
impl Project {
|
||||||
/// Downloads a graph of dependencies.
|
/// Downloads a graph of dependencies
|
||||||
#[instrument(skip_all, level = "debug")]
|
#[instrument(skip(self, graph, refreshed_sources, reqwest), level = "debug")]
|
||||||
pub(crate) async fn download_graph<Reporter>(
|
pub async fn download_graph(
|
||||||
&self,
|
&self,
|
||||||
graph: &DependencyGraph,
|
graph: &DependencyGraph,
|
||||||
options: DownloadGraphOptions<Reporter>,
|
refreshed_sources: &mut HashSet<PackageSources>,
|
||||||
) -> Result<
|
reqwest: &reqwest::Client,
|
||||||
impl Stream<
|
prod: bool,
|
||||||
Item = Result<(PackageId, DependencyGraphNode, PackageFs), errors::DownloadGraphError>,
|
write: bool,
|
||||||
>,
|
wally: bool,
|
||||||
errors::DownloadGraphError,
|
) -> Result<MultithreadDownloadJob, errors::DownloadGraphError> {
|
||||||
>
|
let manifest = self.deser_manifest().await?;
|
||||||
where
|
let manifest_target_kind = manifest.target.kind();
|
||||||
Reporter: DownloadsReporter + Send + Sync + 'static,
|
let downloaded_graph: MultithreadedGraph = Arc::new(Mutex::new(Default::default()));
|
||||||
{
|
|
||||||
let DownloadGraphOptions {
|
|
||||||
reqwest,
|
|
||||||
reporter,
|
|
||||||
refreshed_sources,
|
|
||||||
network_concurrency,
|
|
||||||
} = options;
|
|
||||||
|
|
||||||
let semaphore = Arc::new(Semaphore::new(network_concurrency.get()));
|
let (tx, rx) = tokio::sync::mpsc::channel(
|
||||||
|
graph
|
||||||
|
.iter()
|
||||||
|
.map(|(_, versions)| versions.len())
|
||||||
|
.sum::<usize>()
|
||||||
|
.max(1),
|
||||||
|
);
|
||||||
|
|
||||||
let mut tasks = graph
|
refresh_sources(
|
||||||
.iter()
|
self,
|
||||||
.map(|(package_id, node)| {
|
graph
|
||||||
let span = tracing::info_span!("download", package_id = package_id.to_string());
|
.iter()
|
||||||
|
.flat_map(|(_, versions)| versions.iter())
|
||||||
|
.map(|(_, node)| node.pkg_ref.source()),
|
||||||
|
refreshed_sources,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
let project = self.clone();
|
let project = Arc::new(self.clone());
|
||||||
let reqwest = reqwest.clone();
|
|
||||||
let reporter = reporter.clone();
|
|
||||||
let refreshed_sources = refreshed_sources.clone();
|
|
||||||
let semaphore = semaphore.clone();
|
|
||||||
let package_id = Arc::new(package_id.clone());
|
|
||||||
let node = node.clone();
|
|
||||||
|
|
||||||
async move {
|
for (name, versions) in graph {
|
||||||
let progress_reporter = reporter
|
for (version_id, node) in versions {
|
||||||
.clone()
|
// we need to download pesde packages first, since scripts (for target finding for example) can depend on them
|
||||||
.map(|reporter| reporter.report_download(package_id.to_string()));
|
if node.pkg_ref.like_wally() != wally {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
let _permit = semaphore.acquire().await;
|
let tx = tx.clone();
|
||||||
|
|
||||||
if let Some(ref progress_reporter) = progress_reporter {
|
let name = name.clone();
|
||||||
progress_reporter.report_start();
|
let version_id = version_id.clone();
|
||||||
}
|
let node = node.clone();
|
||||||
|
|
||||||
let source = node.pkg_ref.source();
|
let span = tracing::info_span!(
|
||||||
refreshed_sources
|
"download",
|
||||||
.refresh(
|
name = name.to_string(),
|
||||||
&source,
|
version_id = version_id.to_string()
|
||||||
&RefreshOptions {
|
);
|
||||||
project: project.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
tracing::debug!("downloading");
|
let project = project.clone();
|
||||||
|
let reqwest = reqwest.clone();
|
||||||
|
let downloaded_graph = downloaded_graph.clone();
|
||||||
|
|
||||||
let fs = match progress_reporter {
|
let package_dir = self.package_dir().to_path_buf();
|
||||||
Some(progress_reporter) => {
|
|
||||||
source
|
|
||||||
.download(
|
|
||||||
&node.pkg_ref,
|
|
||||||
&DownloadOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
reqwest,
|
|
||||||
id: package_id.clone(),
|
|
||||||
reporter: Arc::new(progress_reporter),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
None => {
|
|
||||||
source
|
|
||||||
.download(
|
|
||||||
&node.pkg_ref,
|
|
||||||
&DownloadOptions {
|
|
||||||
project: project.clone(),
|
|
||||||
reqwest,
|
|
||||||
id: package_id.clone(),
|
|
||||||
reporter: Arc::new(()),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
}
|
|
||||||
.map_err(Box::new)?;
|
|
||||||
|
|
||||||
tracing::debug!("downloaded");
|
tokio::spawn(
|
||||||
|
async move {
|
||||||
|
let source = node.pkg_ref.source();
|
||||||
|
|
||||||
Ok((Arc::into_inner(package_id).unwrap(), node, fs))
|
let container_folder = node.container_folder(
|
||||||
}
|
&package_dir
|
||||||
.instrument(span)
|
.join(manifest_target_kind.packages_folder(version_id.target()))
|
||||||
})
|
.join(PACKAGES_CONTAINER_NAME),
|
||||||
.collect::<JoinSet<Result<_, errors::DownloadGraphError>>>();
|
&name,
|
||||||
|
version_id.version(),
|
||||||
|
);
|
||||||
|
|
||||||
let stream = try_stream! {
|
match fs::create_dir_all(&container_folder).await {
|
||||||
while let Some(res) = tasks.join_next().await {
|
Ok(_) => {}
|
||||||
yield res.unwrap()?;
|
Err(e) => {
|
||||||
}
|
tx.send(Err(errors::DownloadGraphError::Io(e)))
|
||||||
};
|
.await
|
||||||
|
.unwrap();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Ok(stream)
|
let project = project.clone();
|
||||||
}
|
|
||||||
|
tracing::debug!("downloading");
|
||||||
|
|
||||||
|
let (fs, target) =
|
||||||
|
match source.download(&node.pkg_ref, &project, &reqwest).await {
|
||||||
|
Ok(target) => target,
|
||||||
|
Err(e) => {
|
||||||
|
tx.send(Err(Box::new(e).into())).await.unwrap();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
tracing::debug!("downloaded");
|
||||||
|
|
||||||
|
if write {
|
||||||
|
if !prod || node.resolved_ty != DependencyType::Dev {
|
||||||
|
match fs.write_to(container_folder, project.cas_dir(), true).await {
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
tx.send(Err(errors::DownloadGraphError::WriteFailed(e)))
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
tracing::debug!(
|
||||||
|
"skipping write to disk, dev dependency in prod mode"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let display_name = format!("{name}@{version_id}");
|
||||||
|
|
||||||
|
{
|
||||||
|
let mut downloaded_graph = downloaded_graph.lock().unwrap();
|
||||||
|
downloaded_graph
|
||||||
|
.entry(name)
|
||||||
|
.or_default()
|
||||||
|
.insert(version_id, DownloadedDependencyGraphNode { node, target });
|
||||||
|
}
|
||||||
|
|
||||||
|
tx.send(Ok(display_name)).await.unwrap();
|
||||||
|
}
|
||||||
|
.instrument(span),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok((rx, downloaded_graph))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Errors that can occur when downloading a graph
|
/// Errors that can occur when downloading a graph
|
||||||
pub mod errors {
|
pub mod errors {
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
||||||
|
|
||||||
/// Errors that can occur when downloading a graph
|
/// Errors that can occur when downloading a graph
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub enum DownloadGraphError {
|
pub enum DownloadGraphError {
|
||||||
/// An error occurred refreshing a package source
|
/// An error occurred deserializing the project manifest
|
||||||
#[error("failed to refresh package source")]
|
#[error("error deserializing project manifest")]
|
||||||
RefreshFailed(#[from] crate::source::errors::RefreshError),
|
ManifestDeserializationFailed(#[from] crate::errors::ManifestReadError),
|
||||||
|
|
||||||
/// Error interacting with the filesystem
|
/// An error occurred refreshing a package source
|
||||||
#[error("error interacting with the filesystem")]
|
#[error("failed to refresh package source")]
|
||||||
Io(#[from] std::io::Error),
|
RefreshFailed(#[from] Box<crate::source::errors::RefreshError>),
|
||||||
|
|
||||||
/// Error downloading a package
|
/// Error interacting with the filesystem
|
||||||
#[error("failed to download package")]
|
#[error("error interacting with the filesystem")]
|
||||||
DownloadFailed(#[from] Box<crate::source::errors::DownloadError>),
|
Io(#[from] std::io::Error),
|
||||||
}
|
|
||||||
|
/// Error downloading a package
|
||||||
|
#[error("failed to download package")]
|
||||||
|
DownloadFailed(#[from] Box<crate::source::errors::DownloadError>),
|
||||||
|
|
||||||
|
/// Error writing package contents
|
||||||
|
#[error("failed to write package contents")]
|
||||||
|
WriteFailed(#[source] std::io::Error),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,425 +1,176 @@
|
||||||
use crate::{
|
use crate::{
|
||||||
all_packages_dirs,
|
lockfile::{DependencyGraph, DownloadedGraph},
|
||||||
download::DownloadGraphOptions,
|
manifest::DependencyType,
|
||||||
graph::{
|
source::PackageSources,
|
||||||
DependencyGraph, DependencyGraphNode, DependencyGraphNodeWithTarget,
|
Project,
|
||||||
DependencyGraphWithTarget,
|
|
||||||
},
|
|
||||||
manifest::{target::TargetKind, DependencyType},
|
|
||||||
reporters::DownloadsReporter,
|
|
||||||
source::{
|
|
||||||
ids::PackageId,
|
|
||||||
traits::{GetTargetOptions, PackageRef, PackageSource},
|
|
||||||
},
|
|
||||||
Project, RefreshedSources, SCRIPTS_LINK_FOLDER,
|
|
||||||
};
|
};
|
||||||
use fs_err::tokio as fs;
|
use futures::FutureExt;
|
||||||
use futures::TryStreamExt;
|
|
||||||
use std::{
|
use std::{
|
||||||
collections::HashMap,
|
collections::HashSet,
|
||||||
convert::Infallible,
|
future::Future,
|
||||||
future::{self, Future},
|
sync::{Arc, Mutex as StdMutex},
|
||||||
num::NonZeroUsize,
|
|
||||||
path::PathBuf,
|
|
||||||
sync::Arc,
|
|
||||||
};
|
};
|
||||||
use tokio::{pin, task::JoinSet};
|
use tokio::sync::Mutex;
|
||||||
use tracing::{instrument, Instrument};
|
use tracing::{instrument, Instrument};
|
||||||
|
|
||||||
/// Hooks to perform actions after certain events during download and linking.
|
/// Filters a graph to only include production dependencies, if `prod` is `true`
|
||||||
#[allow(unused_variables)]
|
pub fn filter_graph(graph: &DownloadedGraph, prod: bool) -> DownloadedGraph {
|
||||||
pub trait DownloadAndLinkHooks {
|
if !prod {
|
||||||
/// The error type for the hooks.
|
return graph.clone();
|
||||||
type Error: std::error::Error + Send + Sync + 'static;
|
}
|
||||||
|
|
||||||
/// Called after scripts have been downloaded. The `downloaded_graph`
|
graph
|
||||||
/// contains all downloaded packages.
|
.iter()
|
||||||
fn on_scripts_downloaded(
|
.map(|(name, versions)| {
|
||||||
&self,
|
(
|
||||||
graph: &DependencyGraphWithTarget,
|
name.clone(),
|
||||||
) -> impl Future<Output = Result<(), Self::Error>> + Send {
|
versions
|
||||||
future::ready(Ok(()))
|
.iter()
|
||||||
}
|
.filter(|(_, node)| node.node.resolved_ty != DependencyType::Dev)
|
||||||
|
.map(|(v_id, node)| (v_id.clone(), node.clone()))
|
||||||
/// Called after binary dependencies have been downloaded. The
|
.collect(),
|
||||||
/// `downloaded_graph` contains all downloaded packages.
|
)
|
||||||
fn on_bins_downloaded(
|
})
|
||||||
&self,
|
.collect()
|
||||||
graph: &DependencyGraphWithTarget,
|
|
||||||
) -> impl Future<Output = Result<(), Self::Error>> + Send {
|
|
||||||
future::ready(Ok(()))
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Called after all dependencies have been downloaded. The
|
|
||||||
/// `downloaded_graph` contains all downloaded packages.
|
|
||||||
fn on_all_downloaded(
|
|
||||||
&self,
|
|
||||||
graph: &DependencyGraphWithTarget,
|
|
||||||
) -> impl Future<Output = Result<(), Self::Error>> + Send {
|
|
||||||
future::ready(Ok(()))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl DownloadAndLinkHooks for () {
|
/// Receiver for dependencies downloaded and linked
|
||||||
type Error = Infallible;
|
pub type DownloadAndLinkReceiver =
|
||||||
}
|
tokio::sync::mpsc::Receiver<Result<String, crate::download::errors::DownloadGraphError>>;
|
||||||
|
|
||||||
/// Options for downloading and linking.
|
|
||||||
#[derive(Debug)]
|
|
||||||
pub struct DownloadAndLinkOptions<Reporter = (), Hooks = ()> {
|
|
||||||
/// The reqwest client.
|
|
||||||
pub reqwest: reqwest::Client,
|
|
||||||
/// The downloads reporter.
|
|
||||||
pub reporter: Option<Arc<Reporter>>,
|
|
||||||
/// The download and link hooks.
|
|
||||||
pub hooks: Option<Arc<Hooks>>,
|
|
||||||
/// The refreshed sources.
|
|
||||||
pub refreshed_sources: RefreshedSources,
|
|
||||||
/// Whether to skip dev dependencies.
|
|
||||||
pub prod: bool,
|
|
||||||
/// The max number of concurrent network requests.
|
|
||||||
pub network_concurrency: NonZeroUsize,
|
|
||||||
/// Whether to re-install all dependencies even if they are already installed
|
|
||||||
pub force: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<Reporter, Hooks> DownloadAndLinkOptions<Reporter, Hooks>
|
|
||||||
where
|
|
||||||
Reporter: DownloadsReporter + Send + Sync + 'static,
|
|
||||||
Hooks: DownloadAndLinkHooks + Send + Sync + 'static,
|
|
||||||
{
|
|
||||||
/// Creates a new download options with the given reqwest client and reporter.
|
|
||||||
pub fn new(reqwest: reqwest::Client) -> Self {
|
|
||||||
Self {
|
|
||||||
reqwest,
|
|
||||||
reporter: None,
|
|
||||||
hooks: None,
|
|
||||||
refreshed_sources: Default::default(),
|
|
||||||
prod: false,
|
|
||||||
network_concurrency: NonZeroUsize::new(16).unwrap(),
|
|
||||||
force: false,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sets the downloads reporter.
|
|
||||||
pub fn reporter(mut self, reporter: impl Into<Arc<Reporter>>) -> Self {
|
|
||||||
self.reporter.replace(reporter.into());
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sets the download and link hooks.
|
|
||||||
pub fn hooks(mut self, hooks: impl Into<Arc<Hooks>>) -> Self {
|
|
||||||
self.hooks.replace(hooks.into());
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sets the refreshed sources.
|
|
||||||
pub fn refreshed_sources(mut self, refreshed_sources: RefreshedSources) -> Self {
|
|
||||||
self.refreshed_sources = refreshed_sources;
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sets whether to skip dev dependencies.
|
|
||||||
pub fn prod(mut self, prod: bool) -> Self {
|
|
||||||
self.prod = prod;
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sets the max number of concurrent network requests.
|
|
||||||
pub fn network_concurrency(mut self, network_concurrency: NonZeroUsize) -> Self {
|
|
||||||
self.network_concurrency = network_concurrency;
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sets whether to re-install all dependencies even if they are already installed
|
|
||||||
pub fn force(mut self, force: bool) -> Self {
|
|
||||||
self.force = force;
|
|
||||||
self
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Clone for DownloadAndLinkOptions {
|
|
||||||
fn clone(&self) -> Self {
|
|
||||||
Self {
|
|
||||||
reqwest: self.reqwest.clone(),
|
|
||||||
reporter: self.reporter.clone(),
|
|
||||||
hooks: self.hooks.clone(),
|
|
||||||
refreshed_sources: self.refreshed_sources.clone(),
|
|
||||||
prod: self.prod,
|
|
||||||
network_concurrency: self.network_concurrency,
|
|
||||||
force: self.force,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Project {
|
impl Project {
|
||||||
/// Downloads a graph of dependencies and links them in the correct order
|
/// Downloads a graph of dependencies and links them in the correct order
|
||||||
#[instrument(skip_all, fields(prod = options.prod), level = "debug")]
|
#[instrument(
|
||||||
pub async fn download_and_link<Reporter, Hooks>(
|
skip(self, graph, refreshed_sources, reqwest, pesde_cb),
|
||||||
&self,
|
level = "debug"
|
||||||
graph: &Arc<DependencyGraph>,
|
)]
|
||||||
options: DownloadAndLinkOptions<Reporter, Hooks>,
|
pub async fn download_and_link<
|
||||||
) -> Result<DependencyGraphWithTarget, errors::DownloadAndLinkError<Hooks::Error>>
|
F: FnOnce(&Arc<DownloadedGraph>) -> R + Send + 'static,
|
||||||
where
|
R: Future<Output = Result<(), E>> + Send,
|
||||||
Reporter: DownloadsReporter + 'static,
|
E: Send + Sync + 'static,
|
||||||
Hooks: DownloadAndLinkHooks + 'static,
|
>(
|
||||||
{
|
&self,
|
||||||
let DownloadAndLinkOptions {
|
graph: &Arc<DependencyGraph>,
|
||||||
reqwest,
|
refreshed_sources: &Arc<Mutex<HashSet<PackageSources>>>,
|
||||||
reporter,
|
reqwest: &reqwest::Client,
|
||||||
hooks,
|
prod: bool,
|
||||||
refreshed_sources,
|
write: bool,
|
||||||
prod,
|
pesde_cb: F,
|
||||||
network_concurrency,
|
) -> Result<
|
||||||
force,
|
(
|
||||||
} = options;
|
DownloadAndLinkReceiver,
|
||||||
|
impl Future<Output = Result<DownloadedGraph, errors::DownloadAndLinkError<E>>>,
|
||||||
|
),
|
||||||
|
errors::DownloadAndLinkError<E>,
|
||||||
|
> {
|
||||||
|
let (tx, rx) = tokio::sync::mpsc::channel(
|
||||||
|
graph
|
||||||
|
.iter()
|
||||||
|
.map(|(_, versions)| versions.len())
|
||||||
|
.sum::<usize>()
|
||||||
|
.max(1),
|
||||||
|
);
|
||||||
|
let downloaded_graph = Arc::new(StdMutex::new(DownloadedGraph::default()));
|
||||||
|
|
||||||
let graph = graph.clone();
|
let this = self.clone();
|
||||||
let reqwest = reqwest.clone();
|
let graph = graph.clone();
|
||||||
let manifest = self.deser_manifest().await?;
|
let reqwest = reqwest.clone();
|
||||||
|
let refreshed_sources = refreshed_sources.clone();
|
||||||
|
|
||||||
if force {
|
Ok((
|
||||||
async fn remove_dir(dir: PathBuf) -> std::io::Result<()> {
|
rx,
|
||||||
tracing::debug!("force deleting the `{}` folder", dir.display());
|
tokio::spawn(async move {
|
||||||
|
let mut refreshed_sources = refreshed_sources.lock().await;
|
||||||
|
|
||||||
match fs::remove_dir_all(dir).await {
|
// step 1. download pesde dependencies
|
||||||
Ok(()) => Ok(()),
|
let (mut pesde_rx, pesde_graph) = this
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => Ok(()),
|
.download_graph(&graph, &mut refreshed_sources, &reqwest, prod, write, false)
|
||||||
Err(e) => Err(e),
|
.instrument(tracing::debug_span!("download (pesde)"))
|
||||||
}
|
.await?;
|
||||||
}
|
|
||||||
|
|
||||||
let mut tasks = all_packages_dirs()
|
while let Some(result) = pesde_rx.recv().await {
|
||||||
.into_iter()
|
tx.send(result).await.unwrap();
|
||||||
.map(|folder| remove_dir(self.package_dir().join(&folder)))
|
}
|
||||||
.chain(std::iter::once(remove_dir(
|
|
||||||
self.package_dir().join(SCRIPTS_LINK_FOLDER),
|
|
||||||
)))
|
|
||||||
.collect::<JoinSet<_>>();
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
let pesde_graph = Arc::into_inner(pesde_graph).unwrap().into_inner().unwrap();
|
||||||
task.unwrap()?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// step 1. download dependencies
|
// step 2. link pesde dependencies. do so without types
|
||||||
let downloaded_graph = {
|
if write {
|
||||||
let mut download_graph_options = DownloadGraphOptions::<Reporter>::new(reqwest.clone())
|
this.link_dependencies(&filter_graph(&pesde_graph, prod), false)
|
||||||
.refreshed_sources(refreshed_sources.clone())
|
.instrument(tracing::debug_span!("link (pesde)"))
|
||||||
.network_concurrency(network_concurrency);
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
if let Some(reporter) = reporter {
|
let pesde_graph = Arc::new(pesde_graph);
|
||||||
download_graph_options = download_graph_options.reporter(reporter.clone());
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut downloaded_graph = DependencyGraph::new();
|
pesde_cb(&pesde_graph)
|
||||||
|
.await
|
||||||
|
.map_err(errors::DownloadAndLinkError::PesdeCallback)?;
|
||||||
|
|
||||||
let graph_to_download = if force {
|
let pesde_graph = Arc::into_inner(pesde_graph).unwrap();
|
||||||
graph.clone()
|
|
||||||
} else {
|
|
||||||
let mut tasks = graph
|
|
||||||
.iter()
|
|
||||||
.map(|(id, node)| {
|
|
||||||
let id = id.clone();
|
|
||||||
let node = node.clone();
|
|
||||||
let container_folder =
|
|
||||||
node.container_folder_from_project(&id, self, manifest.target.kind());
|
|
||||||
|
|
||||||
async move {
|
// step 3. download wally dependencies
|
||||||
return (id, node, fs::metadata(&container_folder).await.is_ok());
|
let (mut wally_rx, wally_graph) = this
|
||||||
}
|
.download_graph(&graph, &mut refreshed_sources, &reqwest, prod, write, true)
|
||||||
})
|
.instrument(tracing::debug_span!("download (wally)"))
|
||||||
.collect::<JoinSet<_>>();
|
.await?;
|
||||||
|
|
||||||
let mut graph_to_download = DependencyGraph::new();
|
while let Some(result) = wally_rx.recv().await {
|
||||||
while let Some(task) = tasks.join_next().await {
|
tx.send(result).await.unwrap();
|
||||||
let (id, node, installed) = task.unwrap();
|
}
|
||||||
if installed {
|
|
||||||
downloaded_graph.insert(id, node);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
graph_to_download.insert(id, node);
|
let wally_graph = Arc::into_inner(wally_graph).unwrap().into_inner().unwrap();
|
||||||
}
|
|
||||||
|
|
||||||
Arc::new(graph_to_download)
|
{
|
||||||
};
|
let mut downloaded_graph = downloaded_graph.lock().unwrap();
|
||||||
|
downloaded_graph.extend(pesde_graph);
|
||||||
|
for (name, versions) in wally_graph {
|
||||||
|
for (version_id, node) in versions {
|
||||||
|
downloaded_graph
|
||||||
|
.entry(name.clone())
|
||||||
|
.or_default()
|
||||||
|
.insert(version_id, node);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
let downloaded = self
|
let graph = Arc::into_inner(downloaded_graph)
|
||||||
.download_graph(&graph_to_download, download_graph_options.clone())
|
.unwrap()
|
||||||
.instrument(tracing::debug_span!("download"))
|
.into_inner()
|
||||||
.await?;
|
.unwrap();
|
||||||
pin!(downloaded);
|
|
||||||
|
|
||||||
let mut tasks = JoinSet::new();
|
// step 4. link ALL dependencies. do so with types
|
||||||
|
if write {
|
||||||
|
this.link_dependencies(&filter_graph(&graph, prod), true)
|
||||||
|
.instrument(tracing::debug_span!("link (all)"))
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
while let Some((id, node, fs)) = downloaded.try_next().await? {
|
Ok(graph)
|
||||||
let container_folder =
|
})
|
||||||
node.container_folder_from_project(&id, self, manifest.target.kind());
|
.map(|r| r.unwrap()),
|
||||||
|
))
|
||||||
downloaded_graph.insert(id, node);
|
}
|
||||||
|
|
||||||
let cas_dir = self.cas_dir().to_path_buf();
|
|
||||||
tasks.spawn(async move {
|
|
||||||
fs::create_dir_all(&container_folder).await?;
|
|
||||||
fs.write_to(container_folder, cas_dir, true).await
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
task.unwrap()?;
|
|
||||||
}
|
|
||||||
|
|
||||||
downloaded_graph
|
|
||||||
};
|
|
||||||
|
|
||||||
let (downloaded_wally_graph, downloaded_other_graph) = downloaded_graph
|
|
||||||
.into_iter()
|
|
||||||
.partition::<HashMap<_, _>, _>(|(_, node)| node.pkg_ref.is_wally_package());
|
|
||||||
|
|
||||||
let mut graph = Arc::new(DependencyGraphWithTarget::new());
|
|
||||||
|
|
||||||
async fn get_graph_targets<Hooks: DownloadAndLinkHooks>(
|
|
||||||
graph: &mut Arc<DependencyGraphWithTarget>,
|
|
||||||
project: &Project,
|
|
||||||
manifest_target_kind: TargetKind,
|
|
||||||
downloaded_graph: HashMap<PackageId, DependencyGraphNode>,
|
|
||||||
) -> Result<(), errors::DownloadAndLinkError<Hooks::Error>> {
|
|
||||||
let mut tasks = downloaded_graph
|
|
||||||
.into_iter()
|
|
||||||
.map(|(id, node)| {
|
|
||||||
let source = node.pkg_ref.source();
|
|
||||||
let path = Arc::from(
|
|
||||||
node.container_folder_from_project(&id, project, manifest_target_kind)
|
|
||||||
.as_path(),
|
|
||||||
);
|
|
||||||
let id = Arc::new(id.clone());
|
|
||||||
let project = project.clone();
|
|
||||||
|
|
||||||
async move {
|
|
||||||
let target = source
|
|
||||||
.get_target(
|
|
||||||
&node.pkg_ref,
|
|
||||||
&GetTargetOptions {
|
|
||||||
project,
|
|
||||||
path,
|
|
||||||
id: id.clone(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok::<_, errors::DownloadAndLinkError<Hooks::Error>>((
|
|
||||||
Arc::into_inner(id).unwrap(),
|
|
||||||
DependencyGraphNodeWithTarget { node, target },
|
|
||||||
))
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect::<JoinSet<_>>();
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
let (id, node) = task.unwrap()?;
|
|
||||||
Arc::get_mut(graph).unwrap().insert(id, node);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
// step 2. get targets for non Wally packages (Wally packages require the scripts packages to be downloaded first)
|
|
||||||
get_graph_targets::<Hooks>(
|
|
||||||
&mut graph,
|
|
||||||
self,
|
|
||||||
manifest.target.kind(),
|
|
||||||
downloaded_other_graph,
|
|
||||||
)
|
|
||||||
.instrument(tracing::debug_span!("get targets (non-wally)"))
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
self.link_dependencies(graph.clone(), false)
|
|
||||||
.instrument(tracing::debug_span!("link (non-wally)"))
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
if let Some(hooks) = &hooks {
|
|
||||||
hooks
|
|
||||||
.on_scripts_downloaded(&graph)
|
|
||||||
.await
|
|
||||||
.map_err(errors::DownloadAndLinkError::Hook)?;
|
|
||||||
|
|
||||||
hooks
|
|
||||||
.on_bins_downloaded(&graph)
|
|
||||||
.await
|
|
||||||
.map_err(errors::DownloadAndLinkError::Hook)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
// step 3. get targets for Wally packages
|
|
||||||
get_graph_targets::<Hooks>(
|
|
||||||
&mut graph,
|
|
||||||
self,
|
|
||||||
manifest.target.kind(),
|
|
||||||
downloaded_wally_graph,
|
|
||||||
)
|
|
||||||
.instrument(tracing::debug_span!("get targets (wally)"))
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
// step 4. link ALL dependencies. do so with types
|
|
||||||
self.link_dependencies(graph.clone(), true)
|
|
||||||
.instrument(tracing::debug_span!("link (all)"))
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
if let Some(hooks) = &hooks {
|
|
||||||
hooks
|
|
||||||
.on_all_downloaded(&graph)
|
|
||||||
.await
|
|
||||||
.map_err(errors::DownloadAndLinkError::Hook)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut graph = Arc::into_inner(graph).unwrap();
|
|
||||||
|
|
||||||
if prod {
|
|
||||||
graph.retain(|_, node| node.node.resolved_ty != DependencyType::Dev);
|
|
||||||
}
|
|
||||||
|
|
||||||
if prod || !force {
|
|
||||||
self.remove_unused(&graph).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(graph)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Errors that can occur when downloading and linking dependencies
|
/// Errors that can occur when downloading and linking dependencies
|
||||||
pub mod errors {
|
pub mod errors {
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
||||||
|
|
||||||
/// An error that can occur when downloading and linking dependencies
|
/// An error that can occur when downloading and linking dependencies
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub enum DownloadAndLinkError<E> {
|
pub enum DownloadAndLinkError<E> {
|
||||||
/// Reading the manifest failed
|
/// An error occurred while downloading the graph
|
||||||
#[error("error reading manifest")]
|
#[error("error downloading graph")]
|
||||||
ManifestRead(#[from] crate::errors::ManifestReadError),
|
DownloadGraph(#[from] crate::download::errors::DownloadGraphError),
|
||||||
|
|
||||||
/// An error occurred while downloading the graph
|
/// An error occurred while linking dependencies
|
||||||
#[error("error downloading graph")]
|
#[error("error linking dependencies")]
|
||||||
DownloadGraph(#[from] crate::download::errors::DownloadGraphError),
|
Linking(#[from] crate::linking::errors::LinkingError),
|
||||||
|
|
||||||
/// An error occurred while linking dependencies
|
/// An error occurred while executing the pesde callback
|
||||||
#[error("error linking dependencies")]
|
#[error("error executing pesde callback")]
|
||||||
Linking(#[from] crate::linking::errors::LinkingError),
|
PesdeCallback(#[source] E),
|
||||||
|
}
|
||||||
/// An error occurred while executing the pesde callback
|
|
||||||
#[error("error executing hook")]
|
|
||||||
Hook(#[source] E),
|
|
||||||
|
|
||||||
/// IO error
|
|
||||||
#[error("io error")]
|
|
||||||
Io(#[from] std::io::Error),
|
|
||||||
|
|
||||||
/// Error getting a target
|
|
||||||
#[error("error getting target")]
|
|
||||||
GetTarget(#[from] crate::source::errors::GetTargetError),
|
|
||||||
|
|
||||||
/// Removing unused dependencies failed
|
|
||||||
#[error("error removing unused dependencies")]
|
|
||||||
RemoveUnused(#[from] crate::linking::incremental::errors::RemoveUnusedError),
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,61 +0,0 @@
|
||||||
/// Sources of engines
|
|
||||||
pub mod source;
|
|
||||||
|
|
||||||
use crate::{engine::source::EngineSources, ser_display_deser_fromstr};
|
|
||||||
use std::{fmt::Display, str::FromStr};
|
|
||||||
|
|
||||||
/// All supported engines
|
|
||||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)]
|
|
||||||
#[cfg_attr(test, derive(schemars::JsonSchema))]
|
|
||||||
#[cfg_attr(test, schemars(rename_all = "snake_case"))]
|
|
||||||
pub enum EngineKind {
|
|
||||||
/// The pesde package manager
|
|
||||||
Pesde,
|
|
||||||
/// The Lune runtime
|
|
||||||
Lune,
|
|
||||||
}
|
|
||||||
ser_display_deser_fromstr!(EngineKind);
|
|
||||||
|
|
||||||
impl Display for EngineKind {
|
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
||||||
match self {
|
|
||||||
EngineKind::Pesde => write!(f, "pesde"),
|
|
||||||
EngineKind::Lune => write!(f, "lune"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl FromStr for EngineKind {
|
|
||||||
type Err = errors::EngineKindFromStrError;
|
|
||||||
|
|
||||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
||||||
match s.to_lowercase().as_str() {
|
|
||||||
"pesde" => Ok(EngineKind::Pesde),
|
|
||||||
"lune" => Ok(EngineKind::Lune),
|
|
||||||
_ => Err(errors::EngineKindFromStrError::Unknown(s.to_string())),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl EngineKind {
|
|
||||||
/// Returns the source to get this engine from
|
|
||||||
pub fn source(&self) -> EngineSources {
|
|
||||||
match self {
|
|
||||||
EngineKind::Pesde => EngineSources::pesde(),
|
|
||||||
EngineKind::Lune => EngineSources::lune(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Errors related to engine kinds
|
|
||||||
pub mod errors {
|
|
||||||
use thiserror::Error;
|
|
||||||
|
|
||||||
/// Errors which can occur while using the FromStr implementation of EngineKind
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
pub enum EngineKindFromStrError {
|
|
||||||
/// The string isn't a recognized EngineKind
|
|
||||||
#[error("unknown engine kind {0}")]
|
|
||||||
Unknown(String),
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,320 +0,0 @@
|
||||||
use futures::StreamExt;
|
|
||||||
use std::{
|
|
||||||
collections::BTreeSet,
|
|
||||||
mem::ManuallyDrop,
|
|
||||||
path::{Path, PathBuf},
|
|
||||||
pin::Pin,
|
|
||||||
str::FromStr,
|
|
||||||
task::{Context, Poll},
|
|
||||||
};
|
|
||||||
use tokio::{
|
|
||||||
io::{AsyncBufRead, AsyncRead, AsyncReadExt, ReadBuf},
|
|
||||||
pin,
|
|
||||||
};
|
|
||||||
use tokio_util::compat::{Compat, FuturesAsyncReadCompatExt};
|
|
||||||
|
|
||||||
/// The kind of encoding used for the archive
|
|
||||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
|
||||||
pub enum EncodingKind {
|
|
||||||
/// Gzip
|
|
||||||
Gzip,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// The kind of archive
|
|
||||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
|
||||||
pub enum ArchiveKind {
|
|
||||||
/// Tar
|
|
||||||
Tar,
|
|
||||||
/// Zip
|
|
||||||
Zip,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
|
||||||
pub(crate) struct ArchiveInfo(ArchiveKind, Option<EncodingKind>);
|
|
||||||
|
|
||||||
impl FromStr for ArchiveInfo {
|
|
||||||
type Err = errors::ArchiveInfoFromStrError;
|
|
||||||
|
|
||||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
||||||
let parts = s.split('.').collect::<Vec<_>>();
|
|
||||||
|
|
||||||
Ok(match &*parts {
|
|
||||||
[.., "tar", "gz"] => ArchiveInfo(ArchiveKind::Tar, Some(EncodingKind::Gzip)),
|
|
||||||
[.., "tar"] => ArchiveInfo(ArchiveKind::Tar, None),
|
|
||||||
[.., "zip", "gz"] => {
|
|
||||||
return Err(errors::ArchiveInfoFromStrError::Unsupported(
|
|
||||||
ArchiveKind::Zip,
|
|
||||||
Some(EncodingKind::Gzip),
|
|
||||||
))
|
|
||||||
}
|
|
||||||
[.., "zip"] => ArchiveInfo(ArchiveKind::Zip, None),
|
|
||||||
_ => return Err(errors::ArchiveInfoFromStrError::Invalid(s.to_string())),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) type ArchiveReader = Pin<Box<dyn AsyncBufRead>>;
|
|
||||||
|
|
||||||
/// An archive
|
|
||||||
pub struct Archive {
|
|
||||||
pub(crate) info: ArchiveInfo,
|
|
||||||
pub(crate) reader: ArchiveReader,
|
|
||||||
}
|
|
||||||
|
|
||||||
enum TarReader {
|
|
||||||
Gzip(async_compression::tokio::bufread::GzipDecoder<ArchiveReader>),
|
|
||||||
Plain(ArchiveReader),
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO: try to see if we can avoid the unsafe blocks
|
|
||||||
|
|
||||||
impl AsyncRead for TarReader {
|
|
||||||
fn poll_read(
|
|
||||||
self: Pin<&mut Self>,
|
|
||||||
cx: &mut Context<'_>,
|
|
||||||
buf: &mut ReadBuf<'_>,
|
|
||||||
) -> Poll<std::io::Result<()>> {
|
|
||||||
unsafe {
|
|
||||||
match self.get_unchecked_mut() {
|
|
||||||
Self::Gzip(r) => Pin::new_unchecked(r).poll_read(cx, buf),
|
|
||||||
Self::Plain(r) => Pin::new_unchecked(r).poll_read(cx, buf),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
enum ArchiveEntryInner {
|
|
||||||
Tar(tokio_tar::Entry<tokio_tar::Archive<TarReader>>),
|
|
||||||
Zip {
|
|
||||||
archive: *mut async_zip::tokio::read::seek::ZipFileReader<std::io::Cursor<Vec<u8>>>,
|
|
||||||
reader: ManuallyDrop<
|
|
||||||
Compat<
|
|
||||||
async_zip::tokio::read::ZipEntryReader<
|
|
||||||
'static,
|
|
||||||
std::io::Cursor<Vec<u8>>,
|
|
||||||
async_zip::base::read::WithoutEntry,
|
|
||||||
>,
|
|
||||||
>,
|
|
||||||
>,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Drop for ArchiveEntryInner {
|
|
||||||
fn drop(&mut self) {
|
|
||||||
match self {
|
|
||||||
Self::Tar(_) => {}
|
|
||||||
Self::Zip { archive, reader } => unsafe {
|
|
||||||
ManuallyDrop::drop(reader);
|
|
||||||
drop(Box::from_raw(*archive));
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// An entry in an archive. Usually the executable
|
|
||||||
pub struct ArchiveEntry(ArchiveEntryInner);
|
|
||||||
|
|
||||||
impl AsyncRead for ArchiveEntry {
|
|
||||||
fn poll_read(
|
|
||||||
self: Pin<&mut Self>,
|
|
||||||
cx: &mut Context<'_>,
|
|
||||||
buf: &mut ReadBuf<'_>,
|
|
||||||
) -> Poll<std::io::Result<()>> {
|
|
||||||
unsafe {
|
|
||||||
match &mut self.get_unchecked_mut().0 {
|
|
||||||
ArchiveEntryInner::Tar(r) => Pin::new_unchecked(r).poll_read(cx, buf),
|
|
||||||
ArchiveEntryInner::Zip { reader, .. } => {
|
|
||||||
Pin::new_unchecked(&mut **reader).poll_read(cx, buf)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Archive {
|
|
||||||
/// Finds the executable in the archive and returns it as an [`ArchiveEntry`]
|
|
||||||
pub async fn find_executable(
|
|
||||||
self,
|
|
||||||
expected_file_name: &str,
|
|
||||||
) -> Result<ArchiveEntry, errors::FindExecutableError> {
|
|
||||||
#[derive(Debug, PartialEq, Eq)]
|
|
||||||
struct Candidate {
|
|
||||||
path: PathBuf,
|
|
||||||
file_name_matches: bool,
|
|
||||||
extension_matches: bool,
|
|
||||||
has_permissions: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Candidate {
|
|
||||||
fn new(path: PathBuf, perms: u32, expected_file_name: &str) -> Self {
|
|
||||||
Self {
|
|
||||||
file_name_matches: path
|
|
||||||
.file_name()
|
|
||||||
.is_some_and(|name| name == expected_file_name),
|
|
||||||
extension_matches: match path.extension() {
|
|
||||||
Some(ext) if ext == std::env::consts::EXE_EXTENSION => true,
|
|
||||||
None if std::env::consts::EXE_EXTENSION.is_empty() => true,
|
|
||||||
_ => false,
|
|
||||||
},
|
|
||||||
path,
|
|
||||||
has_permissions: perms & 0o111 != 0,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn should_be_considered(&self) -> bool {
|
|
||||||
// if nothing matches, we should not consider this candidate as it is most likely not
|
|
||||||
self.file_name_matches || self.extension_matches || self.has_permissions
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Ord for Candidate {
|
|
||||||
fn cmp(&self, other: &Self) -> std::cmp::Ordering {
|
|
||||||
self.file_name_matches
|
|
||||||
.cmp(&other.file_name_matches)
|
|
||||||
.then(self.extension_matches.cmp(&other.extension_matches))
|
|
||||||
.then(self.has_permissions.cmp(&other.has_permissions))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl PartialOrd for Candidate {
|
|
||||||
fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
|
|
||||||
Some(self.cmp(other))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut candidates = BTreeSet::new();
|
|
||||||
|
|
||||||
match self.info {
|
|
||||||
ArchiveInfo(ArchiveKind::Tar, encoding) => {
|
|
||||||
use async_compression::tokio::bufread as decoders;
|
|
||||||
|
|
||||||
let reader = match encoding {
|
|
||||||
Some(EncodingKind::Gzip) => {
|
|
||||||
TarReader::Gzip(decoders::GzipDecoder::new(self.reader))
|
|
||||||
}
|
|
||||||
None => TarReader::Plain(self.reader),
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut archive = tokio_tar::Archive::new(reader);
|
|
||||||
let mut entries = archive.entries()?;
|
|
||||||
|
|
||||||
while let Some(entry) = entries.next().await.transpose()? {
|
|
||||||
if entry.header().entry_type().is_dir() {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
let candidate = Candidate::new(
|
|
||||||
entry.path()?.to_path_buf(),
|
|
||||||
entry.header().mode()?,
|
|
||||||
expected_file_name,
|
|
||||||
);
|
|
||||||
if candidate.should_be_considered() {
|
|
||||||
candidates.insert(candidate);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let Some(candidate) = candidates.pop_last() else {
|
|
||||||
return Err(errors::FindExecutableError::ExecutableNotFound);
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut entries = archive.entries()?;
|
|
||||||
|
|
||||||
while let Some(entry) = entries.next().await.transpose()? {
|
|
||||||
if entry.header().entry_type().is_dir() {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
let path = entry.path()?;
|
|
||||||
if path == candidate.path {
|
|
||||||
return Ok(ArchiveEntry(ArchiveEntryInner::Tar(entry)));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
ArchiveInfo(ArchiveKind::Zip, _) => {
|
|
||||||
let reader = self.reader;
|
|
||||||
pin!(reader);
|
|
||||||
|
|
||||||
// TODO: would be lovely to not have to read the whole archive into memory
|
|
||||||
let mut buf = vec![];
|
|
||||||
reader.read_to_end(&mut buf).await?;
|
|
||||||
|
|
||||||
let archive = async_zip::base::read::seek::ZipFileReader::with_tokio(
|
|
||||||
std::io::Cursor::new(buf),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
for entry in archive.file().entries() {
|
|
||||||
if entry.dir()? {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
let path: &Path = entry.filename().as_str()?.as_ref();
|
|
||||||
let candidate = Candidate::new(
|
|
||||||
path.to_path_buf(),
|
|
||||||
entry.unix_permissions().unwrap_or(0) as u32,
|
|
||||||
expected_file_name,
|
|
||||||
);
|
|
||||||
if candidate.should_be_considered() {
|
|
||||||
candidates.insert(candidate);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let Some(candidate) = candidates.pop_last() else {
|
|
||||||
return Err(errors::FindExecutableError::ExecutableNotFound);
|
|
||||||
};
|
|
||||||
|
|
||||||
for (i, entry) in archive.file().entries().iter().enumerate() {
|
|
||||||
if entry.dir()? {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
let path: &Path = entry.filename().as_str()?.as_ref();
|
|
||||||
if candidate.path == path {
|
|
||||||
let ptr = Box::into_raw(Box::new(archive));
|
|
||||||
let reader = (unsafe { &mut *ptr }).reader_without_entry(i).await?;
|
|
||||||
return Ok(ArchiveEntry(ArchiveEntryInner::Zip {
|
|
||||||
archive: ptr,
|
|
||||||
reader: ManuallyDrop::new(reader.compat()),
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Err(errors::FindExecutableError::ExecutableNotFound)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Errors that can occur when working with archives
|
|
||||||
pub mod errors {
|
|
||||||
use thiserror::Error;
|
|
||||||
|
|
||||||
/// Errors that can occur when parsing archive info
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
#[non_exhaustive]
|
|
||||||
pub enum ArchiveInfoFromStrError {
|
|
||||||
/// The string is not a valid archive descriptor. E.g. `{name}.tar.gz`
|
|
||||||
#[error("string `{0}` is not a valid archive descriptor")]
|
|
||||||
Invalid(String),
|
|
||||||
|
|
||||||
/// The archive type is not supported. E.g. `{name}.zip.gz`
|
|
||||||
#[error("archive type {0:?} with encoding {1:?} is not supported")]
|
|
||||||
Unsupported(super::ArchiveKind, Option<super::EncodingKind>),
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Errors that can occur when finding an executable in an archive
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
#[non_exhaustive]
|
|
||||||
pub enum FindExecutableError {
|
|
||||||
/// The executable was not found in the archive
|
|
||||||
#[error("failed to find executable in archive")]
|
|
||||||
ExecutableNotFound,
|
|
||||||
|
|
||||||
/// An IO error occurred
|
|
||||||
#[error("IO error")]
|
|
||||||
Io(#[from] std::io::Error),
|
|
||||||
|
|
||||||
/// An error occurred reading the zip archive
|
|
||||||
#[error("failed to read zip archive")]
|
|
||||||
Zip(#[from] async_zip::error::ZipError),
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,19 +0,0 @@
|
||||||
use serde::Deserialize;
|
|
||||||
|
|
||||||
/// A GitHub release
|
|
||||||
#[derive(Debug, Eq, PartialEq, Hash, Clone, Deserialize)]
|
|
||||||
pub struct Release {
|
|
||||||
/// The tag name of the release
|
|
||||||
pub tag_name: String,
|
|
||||||
/// The assets of the release
|
|
||||||
pub assets: Vec<Asset>,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// An asset of a GitHub release
|
|
||||||
#[derive(Debug, Eq, PartialEq, Hash, Clone, Deserialize)]
|
|
||||||
pub struct Asset {
|
|
||||||
/// The name of the asset
|
|
||||||
pub name: String,
|
|
||||||
/// The download URL of the asset
|
|
||||||
pub url: url::Url,
|
|
||||||
}
|
|
|
@ -1,146 +0,0 @@
|
||||||
/// The GitHub engine reference
|
|
||||||
pub mod engine_ref;
|
|
||||||
|
|
||||||
use crate::{
|
|
||||||
engine::source::{
|
|
||||||
archive::Archive,
|
|
||||||
github::engine_ref::Release,
|
|
||||||
traits::{DownloadOptions, EngineSource, ResolveOptions},
|
|
||||||
},
|
|
||||||
reporters::{response_to_async_read, DownloadProgressReporter},
|
|
||||||
util::no_build_metadata,
|
|
||||||
version_matches,
|
|
||||||
};
|
|
||||||
use reqwest::header::ACCEPT;
|
|
||||||
use semver::{Version, VersionReq};
|
|
||||||
use std::{collections::BTreeMap, path::PathBuf};
|
|
||||||
|
|
||||||
/// The GitHub engine source
|
|
||||||
#[derive(Debug, Eq, PartialEq, Hash, Clone)]
|
|
||||||
pub struct GitHubEngineSource {
|
|
||||||
/// The owner of the repository to download from
|
|
||||||
pub owner: String,
|
|
||||||
/// The repository of which to download releases from
|
|
||||||
pub repo: String,
|
|
||||||
/// The template for the asset name. `{VERSION}` will be replaced with the version
|
|
||||||
pub asset_template: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl EngineSource for GitHubEngineSource {
|
|
||||||
type Ref = Release;
|
|
||||||
type ResolveError = errors::ResolveError;
|
|
||||||
type DownloadError = errors::DownloadError;
|
|
||||||
|
|
||||||
fn directory(&self) -> PathBuf {
|
|
||||||
PathBuf::from("github").join(&self.owner).join(&self.repo)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn expected_file_name(&self) -> &str {
|
|
||||||
&self.repo
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn resolve(
|
|
||||||
&self,
|
|
||||||
requirement: &VersionReq,
|
|
||||||
options: &ResolveOptions,
|
|
||||||
) -> Result<BTreeMap<Version, Self::Ref>, Self::ResolveError> {
|
|
||||||
let ResolveOptions { reqwest, .. } = options;
|
|
||||||
|
|
||||||
Ok(reqwest
|
|
||||||
.get(format!(
|
|
||||||
"https://api.github.com/repos/{}/{}/releases",
|
|
||||||
urlencoding::encode(&self.owner),
|
|
||||||
urlencoding::encode(&self.repo),
|
|
||||||
))
|
|
||||||
.send()
|
|
||||||
.await?
|
|
||||||
.error_for_status()?
|
|
||||||
.json::<Vec<Release>>()
|
|
||||||
.await?
|
|
||||||
.into_iter()
|
|
||||||
.filter_map(
|
|
||||||
|release| match release.tag_name.trim_start_matches('v').parse() {
|
|
||||||
Ok(version) if version_matches(requirement, &version) => {
|
|
||||||
Some((version, release))
|
|
||||||
}
|
|
||||||
_ => None,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn download<R: DownloadProgressReporter + 'static>(
|
|
||||||
&self,
|
|
||||||
engine_ref: &Self::Ref,
|
|
||||||
options: &DownloadOptions<R>,
|
|
||||||
) -> Result<Archive, Self::DownloadError> {
|
|
||||||
let DownloadOptions {
|
|
||||||
reqwest,
|
|
||||||
reporter,
|
|
||||||
version,
|
|
||||||
..
|
|
||||||
} = options;
|
|
||||||
|
|
||||||
let desired_asset_names = [
|
|
||||||
self.asset_template
|
|
||||||
.replace("{VERSION}", &version.to_string()),
|
|
||||||
self.asset_template
|
|
||||||
.replace("{VERSION}", &no_build_metadata(version).to_string()),
|
|
||||||
];
|
|
||||||
|
|
||||||
let asset = engine_ref
|
|
||||||
.assets
|
|
||||||
.iter()
|
|
||||||
.find(|asset| {
|
|
||||||
desired_asset_names
|
|
||||||
.iter()
|
|
||||||
.any(|name| asset.name.eq_ignore_ascii_case(name))
|
|
||||||
})
|
|
||||||
.ok_or(errors::DownloadError::AssetNotFound)?;
|
|
||||||
|
|
||||||
reporter.report_start();
|
|
||||||
|
|
||||||
let response = reqwest
|
|
||||||
.get(asset.url.clone())
|
|
||||||
.header(ACCEPT, "application/octet-stream")
|
|
||||||
.send()
|
|
||||||
.await?
|
|
||||||
.error_for_status()?;
|
|
||||||
|
|
||||||
Ok(Archive {
|
|
||||||
info: asset.name.parse()?,
|
|
||||||
reader: Box::pin(response_to_async_read(response, reporter.clone())),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Errors that can occur when working with the GitHub engine source
|
|
||||||
pub mod errors {
|
|
||||||
use thiserror::Error;
|
|
||||||
|
|
||||||
/// Errors that can occur when resolving a GitHub engine
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
#[non_exhaustive]
|
|
||||||
pub enum ResolveError {
|
|
||||||
/// Handling the request failed
|
|
||||||
#[error("failed to handle GitHub API request")]
|
|
||||||
Request(#[from] reqwest::Error),
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Errors that can occur when downloading a GitHub engine
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
#[non_exhaustive]
|
|
||||||
pub enum DownloadError {
|
|
||||||
/// An asset for the current platform could not be found
|
|
||||||
#[error("failed to find asset for current platform")]
|
|
||||||
AssetNotFound,
|
|
||||||
|
|
||||||
/// Handling the request failed
|
|
||||||
#[error("failed to handle GitHub API request")]
|
|
||||||
Request(#[from] reqwest::Error),
|
|
||||||
|
|
||||||
/// The asset's name could not be parsed
|
|
||||||
#[error("failed to parse asset name")]
|
|
||||||
ParseAssetName(#[from] crate::engine::source::archive::errors::ArchiveInfoFromStrError),
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,143 +0,0 @@
|
||||||
use crate::{
|
|
||||||
engine::source::{
|
|
||||||
archive::Archive,
|
|
||||||
traits::{DownloadOptions, EngineSource, ResolveOptions},
|
|
||||||
},
|
|
||||||
reporters::DownloadProgressReporter,
|
|
||||||
};
|
|
||||||
use semver::{Version, VersionReq};
|
|
||||||
use std::{collections::BTreeMap, path::PathBuf};
|
|
||||||
|
|
||||||
/// Archives
|
|
||||||
pub mod archive;
|
|
||||||
/// The GitHub engine source
|
|
||||||
pub mod github;
|
|
||||||
/// Traits for engine sources
|
|
||||||
pub mod traits;
|
|
||||||
|
|
||||||
/// Engine references
|
|
||||||
#[derive(Debug, Eq, PartialEq, Hash, Clone)]
|
|
||||||
pub enum EngineRefs {
|
|
||||||
/// A GitHub engine reference
|
|
||||||
GitHub(github::engine_ref::Release),
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Engine sources
|
|
||||||
#[derive(Debug, Eq, PartialEq, Hash, Clone)]
|
|
||||||
pub enum EngineSources {
|
|
||||||
/// A GitHub engine source
|
|
||||||
GitHub(github::GitHubEngineSource),
|
|
||||||
}
|
|
||||||
|
|
||||||
impl EngineSource for EngineSources {
|
|
||||||
type Ref = EngineRefs;
|
|
||||||
type ResolveError = errors::ResolveError;
|
|
||||||
type DownloadError = errors::DownloadError;
|
|
||||||
|
|
||||||
fn directory(&self) -> PathBuf {
|
|
||||||
match self {
|
|
||||||
EngineSources::GitHub(source) => source.directory(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn expected_file_name(&self) -> &str {
|
|
||||||
match self {
|
|
||||||
EngineSources::GitHub(source) => source.expected_file_name(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn resolve(
|
|
||||||
&self,
|
|
||||||
requirement: &VersionReq,
|
|
||||||
options: &ResolveOptions,
|
|
||||||
) -> Result<BTreeMap<Version, Self::Ref>, Self::ResolveError> {
|
|
||||||
match self {
|
|
||||||
EngineSources::GitHub(source) => source
|
|
||||||
.resolve(requirement, options)
|
|
||||||
.await
|
|
||||||
.map(|map| {
|
|
||||||
map.into_iter()
|
|
||||||
.map(|(version, release)| (version, EngineRefs::GitHub(release)))
|
|
||||||
.collect()
|
|
||||||
})
|
|
||||||
.map_err(Into::into),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn download<R: DownloadProgressReporter + 'static>(
|
|
||||||
&self,
|
|
||||||
engine_ref: &Self::Ref,
|
|
||||||
options: &DownloadOptions<R>,
|
|
||||||
) -> Result<Archive, Self::DownloadError> {
|
|
||||||
match (self, engine_ref) {
|
|
||||||
(EngineSources::GitHub(source), EngineRefs::GitHub(release)) => {
|
|
||||||
source.download(release, options).await.map_err(Into::into)
|
|
||||||
}
|
|
||||||
|
|
||||||
// for the future
|
|
||||||
#[allow(unreachable_patterns)]
|
|
||||||
_ => Err(errors::DownloadError::Mismatch),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl EngineSources {
|
|
||||||
/// Returns the source for the pesde engine
|
|
||||||
pub fn pesde() -> Self {
|
|
||||||
let mut parts = env!("CARGO_PKG_REPOSITORY").split('/').skip(3);
|
|
||||||
let (owner, repo) = (
|
|
||||||
parts.next().unwrap().to_string(),
|
|
||||||
parts.next().unwrap().to_string(),
|
|
||||||
);
|
|
||||||
|
|
||||||
EngineSources::GitHub(github::GitHubEngineSource {
|
|
||||||
owner,
|
|
||||||
repo,
|
|
||||||
asset_template: format!(
|
|
||||||
"pesde-{{VERSION}}-{}-{}.zip",
|
|
||||||
std::env::consts::OS,
|
|
||||||
std::env::consts::ARCH
|
|
||||||
),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns the source for the lune engine
|
|
||||||
pub fn lune() -> Self {
|
|
||||||
EngineSources::GitHub(github::GitHubEngineSource {
|
|
||||||
owner: "lune-org".into(),
|
|
||||||
repo: "lune".into(),
|
|
||||||
asset_template: format!(
|
|
||||||
"lune-{{VERSION}}-{}-{}.zip",
|
|
||||||
std::env::consts::OS,
|
|
||||||
std::env::consts::ARCH
|
|
||||||
),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Errors that can occur when working with engine sources
|
|
||||||
pub mod errors {
|
|
||||||
use thiserror::Error;
|
|
||||||
|
|
||||||
/// Errors that can occur when resolving an engine
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
#[non_exhaustive]
|
|
||||||
pub enum ResolveError {
|
|
||||||
/// Failed to resolve the GitHub engine
|
|
||||||
#[error("failed to resolve github engine")]
|
|
||||||
GitHub(#[from] super::github::errors::ResolveError),
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Errors that can occur when downloading an engine
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
#[non_exhaustive]
|
|
||||||
pub enum DownloadError {
|
|
||||||
/// Failed to download the GitHub engine
|
|
||||||
#[error("failed to download github engine")]
|
|
||||||
GitHub(#[from] super::github::errors::DownloadError),
|
|
||||||
|
|
||||||
/// Mismatched engine reference
|
|
||||||
#[error("mismatched engine reference")]
|
|
||||||
Mismatch,
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,51 +0,0 @@
|
||||||
use crate::{engine::source::archive::Archive, reporters::DownloadProgressReporter};
|
|
||||||
use semver::{Version, VersionReq};
|
|
||||||
use std::{collections::BTreeMap, fmt::Debug, future::Future, path::PathBuf, sync::Arc};
|
|
||||||
|
|
||||||
/// Options for resolving an engine
|
|
||||||
#[derive(Debug, Clone)]
|
|
||||||
pub struct ResolveOptions {
|
|
||||||
/// The reqwest client to use
|
|
||||||
pub reqwest: reqwest::Client,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Options for downloading an engine
|
|
||||||
#[derive(Debug, Clone)]
|
|
||||||
pub struct DownloadOptions<R: DownloadProgressReporter> {
|
|
||||||
/// The reqwest client to use
|
|
||||||
pub reqwest: reqwest::Client,
|
|
||||||
/// The reporter to use
|
|
||||||
pub reporter: Arc<R>,
|
|
||||||
/// The version of the engine to be downloaded
|
|
||||||
pub version: Version,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// A source of engines
|
|
||||||
pub trait EngineSource: Debug {
|
|
||||||
/// The reference type for this source
|
|
||||||
type Ref;
|
|
||||||
/// The error type for resolving an engine from this source
|
|
||||||
type ResolveError: std::error::Error + Send + Sync + 'static;
|
|
||||||
/// The error type for downloading an engine from this source
|
|
||||||
type DownloadError: std::error::Error + Send + Sync + 'static;
|
|
||||||
|
|
||||||
/// Returns the folder to store the engine's versions in
|
|
||||||
fn directory(&self) -> PathBuf;
|
|
||||||
|
|
||||||
/// Returns the expected file name of the engine in the archive
|
|
||||||
fn expected_file_name(&self) -> &str;
|
|
||||||
|
|
||||||
/// Resolves a requirement to a reference
|
|
||||||
fn resolve(
|
|
||||||
&self,
|
|
||||||
requirement: &VersionReq,
|
|
||||||
options: &ResolveOptions,
|
|
||||||
) -> impl Future<Output = Result<BTreeMap<Version, Self::Ref>, Self::ResolveError>> + Send + Sync;
|
|
||||||
|
|
||||||
/// Downloads an engine
|
|
||||||
fn download<R: DownloadProgressReporter + 'static>(
|
|
||||||
&self,
|
|
||||||
engine_ref: &Self::Ref,
|
|
||||||
options: &DownloadOptions<R>,
|
|
||||||
) -> impl Future<Output = Result<Archive, Self::DownloadError>> + Send + Sync;
|
|
||||||
}
|
|
110
src/graph.rs
110
src/graph.rs
|
@ -1,110 +0,0 @@
|
||||||
use crate::{
|
|
||||||
manifest::{
|
|
||||||
target::{Target, TargetKind},
|
|
||||||
Alias, DependencyType,
|
|
||||||
},
|
|
||||||
source::{
|
|
||||||
ids::{PackageId, VersionId},
|
|
||||||
refs::PackageRefs,
|
|
||||||
specifiers::DependencySpecifiers,
|
|
||||||
traits::PackageRef,
|
|
||||||
},
|
|
||||||
Project, PACKAGES_CONTAINER_NAME,
|
|
||||||
};
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::{collections::BTreeMap, path::PathBuf};
|
|
||||||
|
|
||||||
/// A graph of dependencies
|
|
||||||
pub type Graph<Node> = BTreeMap<PackageId, Node>;
|
|
||||||
|
|
||||||
/// A dependency graph node
|
|
||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
||||||
pub struct DependencyGraphNode {
|
|
||||||
/// The alias, specifier, and original (as in the manifest) type for the dependency, if it is a direct dependency (i.e. used by the current project)
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
pub direct: Option<(Alias, DependencySpecifiers, DependencyType)>,
|
|
||||||
/// The dependencies of the package
|
|
||||||
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
|
||||||
pub dependencies: BTreeMap<PackageId, Alias>,
|
|
||||||
/// The resolved (transformed, for example Peer -> Standard) type of the dependency
|
|
||||||
pub resolved_ty: DependencyType,
|
|
||||||
/// Whether the resolved type should be Peer if this isn't depended on
|
|
||||||
#[serde(default, skip_serializing_if = "std::ops::Not::not")]
|
|
||||||
pub is_peer: bool,
|
|
||||||
/// The package reference
|
|
||||||
pub pkg_ref: PackageRefs,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DependencyGraphNode {
|
|
||||||
pub(crate) fn dependencies_dir(
|
|
||||||
&self,
|
|
||||||
version_id: &VersionId,
|
|
||||||
project_target: TargetKind,
|
|
||||||
) -> String {
|
|
||||||
if self.pkg_ref.use_new_structure() {
|
|
||||||
version_id.target().packages_folder(project_target)
|
|
||||||
} else {
|
|
||||||
"..".to_string()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns the folder to store the contents of the package in
|
|
||||||
pub fn container_folder(&self, package_id: &PackageId) -> PathBuf {
|
|
||||||
let (name, v_id) = package_id.parts();
|
|
||||||
|
|
||||||
if self.pkg_ref.is_wally_package() {
|
|
||||||
return PathBuf::from(format!(
|
|
||||||
"{}_{}@{}",
|
|
||||||
name.scope(),
|
|
||||||
name.name(),
|
|
||||||
v_id.version()
|
|
||||||
))
|
|
||||||
.join(name.name());
|
|
||||||
}
|
|
||||||
|
|
||||||
PathBuf::from(name.escaped())
|
|
||||||
.join(v_id.version().to_string())
|
|
||||||
.join(name.name())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns the folder to store the contents of the package in starting from the project's package directory
|
|
||||||
pub fn container_folder_from_project(
|
|
||||||
&self,
|
|
||||||
package_id: &PackageId,
|
|
||||||
project: &Project,
|
|
||||||
manifest_target_kind: TargetKind,
|
|
||||||
) -> PathBuf {
|
|
||||||
project
|
|
||||||
.package_dir()
|
|
||||||
.join(manifest_target_kind.packages_folder(package_id.version_id().target()))
|
|
||||||
.join(PACKAGES_CONTAINER_NAME)
|
|
||||||
.join(self.container_folder(package_id))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// A graph of `DependencyGraphNode`s
|
|
||||||
pub type DependencyGraph = Graph<DependencyGraphNode>;
|
|
||||||
|
|
||||||
/// A dependency graph node with a `Target`
|
|
||||||
#[derive(Debug, Clone)]
|
|
||||||
pub struct DependencyGraphNodeWithTarget {
|
|
||||||
/// The target of the package
|
|
||||||
pub target: Target,
|
|
||||||
/// The node
|
|
||||||
pub node: DependencyGraphNode,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// A graph of `DownloadedDependencyGraphNode`s
|
|
||||||
pub type DependencyGraphWithTarget = Graph<DependencyGraphNodeWithTarget>;
|
|
||||||
|
|
||||||
/// A trait for converting a graph to a different type of graph
|
|
||||||
pub trait ConvertableGraph<Node> {
|
|
||||||
/// Converts the graph to a different type of graph
|
|
||||||
fn convert(self) -> Graph<Node>;
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ConvertableGraph<DependencyGraphNode> for DependencyGraphWithTarget {
|
|
||||||
fn convert(self) -> Graph<DependencyGraphNode> {
|
|
||||||
self.into_iter().map(|(id, node)| (id, node.node)).collect()
|
|
||||||
}
|
|
||||||
}
|
|
774
src/lib.rs
774
src/lib.rs
|
@ -1,27 +1,21 @@
|
||||||
#![warn(missing_docs, clippy::redundant_closure_for_method_calls)]
|
#![deny(missing_docs)]
|
||||||
//! A package manager for the Luau programming language, supporting multiple runtimes including Roblox and Lune.
|
//! A package manager for the Luau programming language, supporting multiple runtimes including Roblox and Lune.
|
||||||
//! pesde has its own registry, however it can also use Wally, and Git repositories as package sources.
|
//! pesde has its own registry, however it can also use Wally, and Git repositories as package sources.
|
||||||
//! It has been designed with multiple targets in mind, namely Roblox, Lune, and Luau.
|
//! It has been designed with multiple targets in mind, namely Roblox, Lune, and Luau.
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
lockfile::Lockfile,
|
lockfile::Lockfile,
|
||||||
manifest::{target::TargetKind, Manifest},
|
manifest::Manifest,
|
||||||
source::{
|
source::{traits::PackageSource, PackageSources},
|
||||||
traits::{PackageSource, RefreshOptions},
|
|
||||||
PackageSources,
|
|
||||||
},
|
|
||||||
};
|
};
|
||||||
use async_stream::try_stream;
|
use async_stream::stream;
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use futures::Stream;
|
use futures::{future::try_join_all, Stream};
|
||||||
use gix::sec::identity::Account;
|
use gix::sec::identity::Account;
|
||||||
use semver::{Version, VersionReq};
|
|
||||||
use std::{
|
use std::{
|
||||||
collections::{HashMap, HashSet},
|
collections::{HashMap, HashSet},
|
||||||
fmt::Debug,
|
fmt::Debug,
|
||||||
hash::{Hash, Hasher},
|
path::{Path, PathBuf},
|
||||||
path::{Path, PathBuf},
|
|
||||||
sync::Arc,
|
|
||||||
};
|
};
|
||||||
use tracing::instrument;
|
use tracing::instrument;
|
||||||
use wax::Pattern;
|
use wax::Pattern;
|
||||||
|
@ -30,10 +24,6 @@ use wax::Pattern;
|
||||||
pub mod download;
|
pub mod download;
|
||||||
/// Utility for downloading and linking in the correct order
|
/// Utility for downloading and linking in the correct order
|
||||||
pub mod download_and_link;
|
pub mod download_and_link;
|
||||||
/// Handling of engines
|
|
||||||
pub mod engine;
|
|
||||||
/// Graphs
|
|
||||||
pub mod graph;
|
|
||||||
/// Linking packages
|
/// Linking packages
|
||||||
pub mod linking;
|
pub mod linking;
|
||||||
/// Lockfile
|
/// Lockfile
|
||||||
|
@ -45,7 +35,6 @@ pub mod names;
|
||||||
/// Patching packages
|
/// Patching packages
|
||||||
#[cfg(feature = "patches")]
|
#[cfg(feature = "patches")]
|
||||||
pub mod patches;
|
pub mod patches;
|
||||||
pub mod reporters;
|
|
||||||
/// Resolving packages
|
/// Resolving packages
|
||||||
pub mod resolver;
|
pub mod resolver;
|
||||||
/// Running scripts
|
/// Running scripts
|
||||||
|
@ -66,468 +55,415 @@ pub(crate) const LINK_LIB_NO_FILE_FOUND: &str = "____pesde_no_export_file_found"
|
||||||
/// The folder in which scripts are linked
|
/// The folder in which scripts are linked
|
||||||
pub const SCRIPTS_LINK_FOLDER: &str = ".pesde";
|
pub const SCRIPTS_LINK_FOLDER: &str = ".pesde";
|
||||||
|
|
||||||
#[derive(Debug, Default)]
|
|
||||||
struct AuthConfigShared {
|
|
||||||
tokens: HashMap<gix::Url, String>,
|
|
||||||
git_credentials: Option<Account>,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Struct containing the authentication configuration
|
/// Struct containing the authentication configuration
|
||||||
#[derive(Debug, Clone, Default)]
|
#[derive(Debug, Default, Clone)]
|
||||||
pub struct AuthConfig {
|
pub struct AuthConfig {
|
||||||
shared: Arc<AuthConfigShared>,
|
tokens: HashMap<gix::Url, String>,
|
||||||
|
git_credentials: Option<Account>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AuthConfig {
|
impl AuthConfig {
|
||||||
/// Create a new `AuthConfig`
|
/// Create a new `AuthConfig`
|
||||||
pub fn new() -> Self {
|
pub fn new() -> Self {
|
||||||
AuthConfig::default()
|
AuthConfig::default()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set the tokens
|
/// Set the tokens
|
||||||
/// Panics if the `AuthConfig` is shared
|
pub fn with_tokens<I: IntoIterator<Item = (gix::Url, S)>, S: AsRef<str>>(
|
||||||
pub fn with_tokens<I: IntoIterator<Item = (gix::Url, S)>, S: AsRef<str>>(
|
mut self,
|
||||||
mut self,
|
tokens: I,
|
||||||
tokens: I,
|
) -> Self {
|
||||||
) -> Self {
|
self.tokens = tokens
|
||||||
Arc::get_mut(&mut self.shared).unwrap().tokens = tokens
|
.into_iter()
|
||||||
.into_iter()
|
.map(|(url, s)| (url, s.as_ref().to_string()))
|
||||||
.map(|(url, s)| (url, s.as_ref().to_string()))
|
.collect();
|
||||||
.collect();
|
self
|
||||||
self
|
}
|
||||||
}
|
|
||||||
|
|
||||||
/// Set the git credentials
|
/// Set the git credentials
|
||||||
/// Panics if the `AuthConfig` is shared
|
pub fn with_git_credentials(mut self, git_credentials: Option<Account>) -> Self {
|
||||||
pub fn with_git_credentials(mut self, git_credentials: Option<Account>) -> Self {
|
self.git_credentials = git_credentials;
|
||||||
Arc::get_mut(&mut self.shared).unwrap().git_credentials = git_credentials;
|
self
|
||||||
self
|
}
|
||||||
}
|
|
||||||
|
|
||||||
/// Get the tokens
|
/// Get the tokens
|
||||||
pub fn tokens(&self) -> &HashMap<gix::Url, String> {
|
pub fn tokens(&self) -> &HashMap<gix::Url, String> {
|
||||||
&self.shared.tokens
|
&self.tokens
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the git credentials
|
/// Get the git credentials
|
||||||
pub fn git_credentials(&self) -> Option<&Account> {
|
pub fn git_credentials(&self) -> Option<&Account> {
|
||||||
self.shared.git_credentials.as_ref()
|
self.git_credentials.as_ref()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
struct ProjectShared {
|
|
||||||
package_dir: PathBuf,
|
|
||||||
workspace_dir: Option<PathBuf>,
|
|
||||||
data_dir: PathBuf,
|
|
||||||
cas_dir: PathBuf,
|
|
||||||
auth_config: AuthConfig,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The main struct of the pesde library, representing a project
|
/// The main struct of the pesde library, representing a project
|
||||||
/// Unlike `ProjectShared`, this struct is `Send` and `Sync` and is cheap to clone because it is `Arc`-backed
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct Project {
|
pub struct Project {
|
||||||
shared: Arc<ProjectShared>,
|
package_dir: PathBuf,
|
||||||
|
workspace_dir: Option<PathBuf>,
|
||||||
|
data_dir: PathBuf,
|
||||||
|
auth_config: AuthConfig,
|
||||||
|
cas_dir: PathBuf,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Project {
|
impl Project {
|
||||||
/// Create a new `Project`
|
/// Create a new `Project`
|
||||||
pub fn new(
|
pub fn new<P: AsRef<Path>, Q: AsRef<Path>, R: AsRef<Path>, S: AsRef<Path>>(
|
||||||
package_dir: impl AsRef<Path>,
|
package_dir: P,
|
||||||
workspace_dir: Option<impl AsRef<Path>>,
|
workspace_dir: Option<Q>,
|
||||||
data_dir: impl AsRef<Path>,
|
data_dir: R,
|
||||||
cas_dir: impl AsRef<Path>,
|
cas_dir: S,
|
||||||
auth_config: AuthConfig,
|
auth_config: AuthConfig,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
Project {
|
Project {
|
||||||
shared: Arc::new(ProjectShared {
|
package_dir: package_dir.as_ref().to_path_buf(),
|
||||||
package_dir: package_dir.as_ref().to_path_buf(),
|
workspace_dir: workspace_dir.map(|d| d.as_ref().to_path_buf()),
|
||||||
workspace_dir: workspace_dir.map(|d| d.as_ref().to_path_buf()),
|
data_dir: data_dir.as_ref().to_path_buf(),
|
||||||
data_dir: data_dir.as_ref().to_path_buf(),
|
auth_config,
|
||||||
cas_dir: cas_dir.as_ref().to_path_buf(),
|
cas_dir: cas_dir.as_ref().to_path_buf(),
|
||||||
auth_config,
|
}
|
||||||
}),
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// The directory of the package
|
/// The directory of the package
|
||||||
pub fn package_dir(&self) -> &Path {
|
pub fn package_dir(&self) -> &Path {
|
||||||
&self.shared.package_dir
|
&self.package_dir
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The directory of the workspace this package belongs to, if any
|
/// The directory of the workspace this package belongs to, if any
|
||||||
pub fn workspace_dir(&self) -> Option<&Path> {
|
pub fn workspace_dir(&self) -> Option<&Path> {
|
||||||
self.shared.workspace_dir.as_deref()
|
self.workspace_dir.as_deref()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The directory to store general-purpose data
|
/// The directory to store general-purpose data
|
||||||
pub fn data_dir(&self) -> &Path {
|
pub fn data_dir(&self) -> &Path {
|
||||||
&self.shared.data_dir
|
&self.data_dir
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The CAS (content-addressable storage) directory
|
/// The authentication configuration
|
||||||
pub fn cas_dir(&self) -> &Path {
|
pub fn auth_config(&self) -> &AuthConfig {
|
||||||
&self.shared.cas_dir
|
&self.auth_config
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The authentication configuration
|
/// The CAS (content-addressable storage) directory
|
||||||
pub fn auth_config(&self) -> &AuthConfig {
|
pub fn cas_dir(&self) -> &Path {
|
||||||
&self.shared.auth_config
|
&self.cas_dir
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Read the manifest file
|
/// Read the manifest file
|
||||||
#[instrument(skip(self), ret(level = "trace"), level = "debug")]
|
#[instrument(skip(self), ret(level = "trace"), level = "debug")]
|
||||||
pub async fn read_manifest(&self) -> Result<String, errors::ManifestReadError> {
|
pub async fn read_manifest(&self) -> Result<String, errors::ManifestReadError> {
|
||||||
let string = fs::read_to_string(self.package_dir().join(MANIFEST_FILE_NAME)).await?;
|
let string = fs::read_to_string(self.package_dir.join(MANIFEST_FILE_NAME)).await?;
|
||||||
Ok(string)
|
Ok(string)
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: cache the manifest
|
// TODO: cache the manifest
|
||||||
/// Deserialize the manifest file
|
/// Deserialize the manifest file
|
||||||
#[instrument(skip(self), ret(level = "trace"), level = "debug")]
|
#[instrument(skip(self), ret(level = "trace"), level = "debug")]
|
||||||
pub async fn deser_manifest(&self) -> Result<Manifest, errors::ManifestReadError> {
|
pub async fn deser_manifest(&self) -> Result<Manifest, errors::ManifestReadError> {
|
||||||
deser_manifest(self.package_dir()).await
|
let string = fs::read_to_string(self.package_dir.join(MANIFEST_FILE_NAME)).await?;
|
||||||
}
|
Ok(toml::from_str(&string)?)
|
||||||
|
}
|
||||||
|
|
||||||
/// Deserialize the manifest file of the workspace root
|
/// Write the manifest file
|
||||||
#[instrument(skip(self), ret(level = "trace"), level = "debug")]
|
#[instrument(skip(self, manifest), level = "debug")]
|
||||||
pub async fn deser_workspace_manifest(
|
pub async fn write_manifest<S: AsRef<[u8]>>(&self, manifest: S) -> Result<(), std::io::Error> {
|
||||||
&self,
|
fs::write(self.package_dir.join(MANIFEST_FILE_NAME), manifest.as_ref()).await
|
||||||
) -> Result<Option<Manifest>, errors::ManifestReadError> {
|
}
|
||||||
let Some(workspace_dir) = self.workspace_dir() else {
|
|
||||||
return Ok(None);
|
|
||||||
};
|
|
||||||
|
|
||||||
deser_manifest(workspace_dir).await.map(Some)
|
/// Deserialize the lockfile
|
||||||
}
|
#[instrument(skip(self), ret(level = "trace"), level = "debug")]
|
||||||
|
pub async fn deser_lockfile(&self) -> Result<Lockfile, errors::LockfileReadError> {
|
||||||
|
let string = fs::read_to_string(self.package_dir.join(LOCKFILE_FILE_NAME)).await?;
|
||||||
|
Ok(toml::from_str(&string)?)
|
||||||
|
}
|
||||||
|
|
||||||
/// Write the manifest file
|
/// Write the lockfile
|
||||||
#[instrument(skip(self, manifest), level = "debug")]
|
#[instrument(skip(self, lockfile), level = "debug")]
|
||||||
pub async fn write_manifest<S: AsRef<[u8]>>(&self, manifest: S) -> Result<(), std::io::Error> {
|
pub async fn write_lockfile(
|
||||||
fs::write(
|
&self,
|
||||||
self.package_dir().join(MANIFEST_FILE_NAME),
|
lockfile: Lockfile,
|
||||||
manifest.as_ref(),
|
) -> Result<(), errors::LockfileWriteError> {
|
||||||
)
|
let string = toml::to_string(&lockfile)?;
|
||||||
.await
|
fs::write(self.package_dir.join(LOCKFILE_FILE_NAME), string).await?;
|
||||||
}
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
/// Deserialize the lockfile
|
/// Get the workspace members
|
||||||
#[instrument(skip(self), ret(level = "trace"), level = "debug")]
|
#[instrument(skip(self), level = "debug")]
|
||||||
pub async fn deser_lockfile(&self) -> Result<Lockfile, errors::LockfileReadError> {
|
pub async fn workspace_members<P: AsRef<Path> + Debug>(
|
||||||
let string = fs::read_to_string(self.package_dir().join(LOCKFILE_FILE_NAME)).await?;
|
&self,
|
||||||
Ok(match toml::from_str(&string) {
|
dir: P,
|
||||||
Ok(lockfile) => lockfile,
|
can_ref_self: bool,
|
||||||
Err(e) => {
|
) -> Result<
|
||||||
#[allow(deprecated)]
|
impl Stream<Item = Result<(PathBuf, Manifest), errors::WorkspaceMembersError>>,
|
||||||
let Ok(old_lockfile) = toml::from_str::<lockfile::old::LockfileOld>(&string) else {
|
errors::WorkspaceMembersError,
|
||||||
return Err(errors::LockfileReadError::Serde(e));
|
> {
|
||||||
};
|
let dir = dir.as_ref().to_path_buf();
|
||||||
|
let manifest = fs::read_to_string(dir.join(MANIFEST_FILE_NAME))
|
||||||
|
.await
|
||||||
|
.map_err(errors::WorkspaceMembersError::ManifestMissing)?;
|
||||||
|
let manifest = toml::from_str::<Manifest>(&manifest).map_err(|e| {
|
||||||
|
errors::WorkspaceMembersError::ManifestDeser(dir.to_path_buf(), Box::new(e))
|
||||||
|
})?;
|
||||||
|
|
||||||
#[allow(deprecated)]
|
let members = matching_globs(
|
||||||
old_lockfile.to_new()
|
dir,
|
||||||
}
|
manifest.workspace_members.iter().map(|s| s.as_str()),
|
||||||
})
|
false,
|
||||||
}
|
can_ref_self,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
/// Write the lockfile
|
Ok(stream! {
|
||||||
#[instrument(skip(self, lockfile), level = "debug")]
|
for path in members {
|
||||||
pub async fn write_lockfile(
|
let manifest = fs::read_to_string(path.join(MANIFEST_FILE_NAME))
|
||||||
&self,
|
.await
|
||||||
lockfile: &Lockfile,
|
.map_err(errors::WorkspaceMembersError::ManifestMissing)?;
|
||||||
) -> Result<(), errors::LockfileWriteError> {
|
let manifest = toml::from_str::<Manifest>(&manifest).map_err(|e| {
|
||||||
let string = toml::to_string(lockfile)?;
|
errors::WorkspaceMembersError::ManifestDeser(path.clone(), Box::new(e))
|
||||||
fs::write(self.package_dir().join(LOCKFILE_FILE_NAME), string).await?;
|
})?;
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Get the workspace members
|
yield Ok((path, manifest));
|
||||||
#[instrument(skip(self), level = "debug")]
|
}
|
||||||
pub async fn workspace_members(
|
})
|
||||||
&self,
|
}
|
||||||
can_ref_self: bool,
|
}
|
||||||
) -> Result<
|
|
||||||
impl Stream<Item = Result<(PathBuf, Manifest), errors::WorkspaceMembersError>>,
|
|
||||||
errors::WorkspaceMembersError,
|
|
||||||
> {
|
|
||||||
let dir = self.workspace_dir().unwrap_or(self.package_dir());
|
|
||||||
let manifest = deser_manifest(dir).await?;
|
|
||||||
|
|
||||||
let members = matching_globs(
|
/// Gets all matching paths in a directory
|
||||||
dir,
|
#[deprecated(
|
||||||
manifest.workspace_members.iter().map(String::as_str),
|
since = "0.5.0-rc.13",
|
||||||
false,
|
note = "use `matching_globs` instead, which does not have the old behaviour of including whole directories by their name (`src` instead of `src/**`)"
|
||||||
can_ref_self,
|
)]
|
||||||
)
|
#[instrument(ret, level = "trace")]
|
||||||
.await?;
|
pub async fn matching_globs_old_behaviour<
|
||||||
|
'a,
|
||||||
|
P: AsRef<Path> + Debug,
|
||||||
|
I: IntoIterator<Item = &'a str> + Debug,
|
||||||
|
>(
|
||||||
|
dir: P,
|
||||||
|
globs: I,
|
||||||
|
relative: bool,
|
||||||
|
) -> Result<HashSet<PathBuf>, errors::MatchingGlobsError> {
|
||||||
|
let (negative_globs, positive_globs) = globs
|
||||||
|
.into_iter()
|
||||||
|
.partition::<Vec<_>, _>(|glob| glob.starts_with('!'));
|
||||||
|
|
||||||
Ok(try_stream! {
|
let negative_globs = wax::any(
|
||||||
for path in members {
|
negative_globs
|
||||||
let manifest = deser_manifest(&path).await?;
|
.into_iter()
|
||||||
yield (path, manifest);
|
.map(|glob| wax::Glob::new(&glob[1..]))
|
||||||
}
|
.collect::<Result<Vec<_>, _>>()?,
|
||||||
})
|
)?;
|
||||||
}
|
|
||||||
|
let (positive_globs, file_names) = positive_globs
|
||||||
|
.into_iter()
|
||||||
|
// only globs we can be sure of (maintaining compatibility with old "only file/dir name" system)
|
||||||
|
.partition::<Vec<_>, _>(|glob| glob.contains('/'));
|
||||||
|
let file_names = file_names.into_iter().collect::<HashSet<_>>();
|
||||||
|
|
||||||
|
let positive_globs = wax::any(
|
||||||
|
positive_globs
|
||||||
|
.into_iter()
|
||||||
|
.map(wax::Glob::new)
|
||||||
|
.collect::<Result<Vec<_>, _>>()?,
|
||||||
|
)?;
|
||||||
|
|
||||||
|
let mut read_dirs = vec![(fs::read_dir(dir.as_ref().to_path_buf()).await?, false)];
|
||||||
|
let mut paths = HashSet::new();
|
||||||
|
|
||||||
|
let mut is_root = true;
|
||||||
|
|
||||||
|
while let Some((mut read_dir, is_entire_dir_included)) = read_dirs.pop() {
|
||||||
|
while let Some(entry) = read_dir.next_entry().await? {
|
||||||
|
let path = entry.path();
|
||||||
|
let relative_path = path.strip_prefix(dir.as_ref()).unwrap();
|
||||||
|
let file_name = path.file_name().unwrap();
|
||||||
|
let is_filename_match =
|
||||||
|
is_root && file_name.to_str().is_some_and(|s| file_names.contains(s));
|
||||||
|
|
||||||
|
if entry.file_type().await?.is_dir() {
|
||||||
|
read_dirs.push((
|
||||||
|
fs::read_dir(&path).await?,
|
||||||
|
is_entire_dir_included || is_filename_match,
|
||||||
|
));
|
||||||
|
if is_filename_match {
|
||||||
|
tracing::warn!("directory name usage found for {}. this is deprecated and will be removed in the future", path.display());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (is_entire_dir_included || is_filename_match)
|
||||||
|
|| (positive_globs.is_match(relative_path)
|
||||||
|
&& !negative_globs.is_match(relative_path))
|
||||||
|
{
|
||||||
|
paths.insert(if relative {
|
||||||
|
relative_path.to_path_buf()
|
||||||
|
} else {
|
||||||
|
path.to_path_buf()
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
is_root = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(paths)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Gets all matching paths in a directory
|
/// Gets all matching paths in a directory
|
||||||
#[instrument(ret, level = "trace")]
|
#[instrument(ret, level = "trace")]
|
||||||
pub async fn matching_globs<'a, P: AsRef<Path> + Debug, I: IntoIterator<Item = &'a str> + Debug>(
|
pub async fn matching_globs<'a, P: AsRef<Path> + Debug, I: IntoIterator<Item = &'a str> + Debug>(
|
||||||
dir: P,
|
dir: P,
|
||||||
globs: I,
|
globs: I,
|
||||||
relative: bool,
|
relative: bool,
|
||||||
can_ref_self: bool,
|
can_ref_self: bool,
|
||||||
) -> Result<HashSet<PathBuf>, errors::MatchingGlobsError> {
|
) -> Result<HashSet<PathBuf>, errors::MatchingGlobsError> {
|
||||||
let (negative_globs, mut positive_globs): (HashSet<&str>, _) =
|
let (negative_globs, mut positive_globs): (HashSet<&str>, _) =
|
||||||
globs.into_iter().partition(|glob| glob.starts_with('!'));
|
globs.into_iter().partition(|glob| glob.starts_with('!'));
|
||||||
|
|
||||||
let include_self = positive_globs.remove(".") && can_ref_self;
|
let include_self = positive_globs.remove(".") && can_ref_self;
|
||||||
|
|
||||||
let negative_globs = wax::any(
|
let negative_globs = wax::any(
|
||||||
negative_globs
|
negative_globs
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|glob| wax::Glob::new(&glob[1..]))
|
.map(|glob| wax::Glob::new(&glob[1..]))
|
||||||
.collect::<Result<Vec<_>, _>>()?,
|
.collect::<Result<Vec<_>, _>>()?,
|
||||||
)?;
|
)?;
|
||||||
let positive_globs = wax::any(
|
let positive_globs = wax::any(
|
||||||
positive_globs
|
positive_globs
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(wax::Glob::new)
|
.map(wax::Glob::new)
|
||||||
.collect::<Result<Vec<_>, _>>()?,
|
.collect::<Result<Vec<_>, _>>()?,
|
||||||
)?;
|
)?;
|
||||||
|
|
||||||
let mut read_dirs = vec![fs::read_dir(dir.as_ref().to_path_buf()).await?];
|
let mut read_dirs = vec![fs::read_dir(dir.as_ref().to_path_buf()).await?];
|
||||||
let mut paths = HashSet::new();
|
let mut paths = HashSet::new();
|
||||||
|
|
||||||
if include_self {
|
if include_self {
|
||||||
paths.insert(if relative {
|
paths.insert(if relative {
|
||||||
PathBuf::new()
|
PathBuf::new()
|
||||||
} else {
|
} else {
|
||||||
dir.as_ref().to_path_buf()
|
dir.as_ref().to_path_buf()
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
while let Some(mut read_dir) = read_dirs.pop() {
|
while let Some(mut read_dir) = read_dirs.pop() {
|
||||||
while let Some(entry) = read_dir.next_entry().await? {
|
while let Some(entry) = read_dir.next_entry().await? {
|
||||||
let path = entry.path();
|
let path = entry.path();
|
||||||
if entry.file_type().await?.is_dir() {
|
if entry.file_type().await?.is_dir() {
|
||||||
read_dirs.push(fs::read_dir(&path).await?);
|
read_dirs.push(fs::read_dir(&path).await?);
|
||||||
}
|
}
|
||||||
|
|
||||||
let relative_path = path.strip_prefix(dir.as_ref()).unwrap();
|
let relative_path = path.strip_prefix(dir.as_ref()).unwrap();
|
||||||
|
|
||||||
if positive_globs.is_match(relative_path) && !negative_globs.is_match(relative_path) {
|
if positive_globs.is_match(relative_path) && !negative_globs.is_match(relative_path) {
|
||||||
paths.insert(if relative {
|
paths.insert(if relative {
|
||||||
relative_path.to_path_buf()
|
relative_path.to_path_buf()
|
||||||
} else {
|
} else {
|
||||||
path.to_path_buf()
|
path.to_path_buf()
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(paths)
|
Ok(paths)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// A struct containing sources already having been refreshed
|
/// Refreshes the sources asynchronously
|
||||||
#[derive(Debug, Clone, Default)]
|
pub async fn refresh_sources<I: Iterator<Item = PackageSources>>(
|
||||||
pub struct RefreshedSources(Arc<tokio::sync::Mutex<HashSet<u64>>>);
|
project: &Project,
|
||||||
|
sources: I,
|
||||||
impl RefreshedSources {
|
refreshed_sources: &mut HashSet<PackageSources>,
|
||||||
/// Create a new empty `RefreshedSources`
|
) -> Result<(), Box<source::errors::RefreshError>> {
|
||||||
pub fn new() -> Self {
|
try_join_all(sources.map(|source| {
|
||||||
RefreshedSources::default()
|
let needs_refresh = refreshed_sources.insert(source.clone());
|
||||||
}
|
async move {
|
||||||
|
if needs_refresh {
|
||||||
/// Refreshes the source asynchronously if it has not already been refreshed.
|
source.refresh(project).await.map_err(Box::new)
|
||||||
/// Will prevent more refreshes of the same source.
|
} else {
|
||||||
pub async fn refresh(
|
Ok(())
|
||||||
&self,
|
}
|
||||||
source: &PackageSources,
|
}
|
||||||
options: &RefreshOptions,
|
}))
|
||||||
) -> Result<(), source::errors::RefreshError> {
|
.await
|
||||||
let mut hasher = std::hash::DefaultHasher::new();
|
.map(|_| ())
|
||||||
source.hash(&mut hasher);
|
|
||||||
let hash = hasher.finish();
|
|
||||||
|
|
||||||
let mut refreshed_sources = self.0.lock().await;
|
|
||||||
|
|
||||||
if refreshed_sources.insert(hash) {
|
|
||||||
source.refresh(options).await
|
|
||||||
} else {
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn deser_manifest(path: &Path) -> Result<Manifest, errors::ManifestReadError> {
|
|
||||||
let string = fs::read_to_string(path.join(MANIFEST_FILE_NAME)).await?;
|
|
||||||
toml::from_str(&string).map_err(|e| errors::ManifestReadError::Serde(path.to_path_buf(), e))
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Find the project & workspace directory roots
|
|
||||||
pub async fn find_roots(
|
|
||||||
cwd: PathBuf,
|
|
||||||
) -> Result<(PathBuf, Option<PathBuf>), errors::FindRootsError> {
|
|
||||||
let mut current_path = Some(cwd.clone());
|
|
||||||
let mut project_root = None::<PathBuf>;
|
|
||||||
let mut workspace_dir = None::<PathBuf>;
|
|
||||||
|
|
||||||
async fn get_workspace_members(
|
|
||||||
path: &Path,
|
|
||||||
) -> Result<HashSet<PathBuf>, errors::FindRootsError> {
|
|
||||||
let manifest = deser_manifest(path).await?;
|
|
||||||
|
|
||||||
if manifest.workspace_members.is_empty() {
|
|
||||||
return Ok(HashSet::new());
|
|
||||||
}
|
|
||||||
|
|
||||||
matching_globs(
|
|
||||||
path,
|
|
||||||
manifest.workspace_members.iter().map(String::as_str),
|
|
||||||
false,
|
|
||||||
false,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.map_err(errors::FindRootsError::Globbing)
|
|
||||||
}
|
|
||||||
|
|
||||||
while let Some(path) = current_path {
|
|
||||||
current_path = path.parent().map(Path::to_path_buf);
|
|
||||||
|
|
||||||
if !path.join(MANIFEST_FILE_NAME).exists() {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
match (project_root.as_ref(), workspace_dir.as_ref()) {
|
|
||||||
(Some(project_root), Some(workspace_dir)) => {
|
|
||||||
return Ok((project_root.clone(), Some(workspace_dir.clone())));
|
|
||||||
}
|
|
||||||
|
|
||||||
(Some(project_root), None) => {
|
|
||||||
if get_workspace_members(&path).await?.contains(project_root) {
|
|
||||||
workspace_dir = Some(path);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
(None, None) => {
|
|
||||||
if get_workspace_members(&path).await?.contains(&cwd) {
|
|
||||||
// initializing a new member of a workspace
|
|
||||||
return Ok((cwd, Some(path)));
|
|
||||||
} else {
|
|
||||||
project_root = Some(path);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
(None, Some(_)) => unreachable!(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// we mustn't expect the project root to be found, as that would
|
|
||||||
// disable the ability to run pesde in a non-project directory (for example to init it)
|
|
||||||
Ok((project_root.unwrap_or(cwd), workspace_dir))
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns whether a version matches a version requirement
|
|
||||||
/// Differs from `VersionReq::matches` in that EVERY version matches `*`
|
|
||||||
pub fn version_matches(req: &VersionReq, version: &Version) -> bool {
|
|
||||||
*req == VersionReq::STAR || req.matches(version)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn all_packages_dirs() -> HashSet<String> {
|
|
||||||
let mut dirs = HashSet::new();
|
|
||||||
for target_kind_a in TargetKind::VARIANTS {
|
|
||||||
for target_kind_b in TargetKind::VARIANTS {
|
|
||||||
dirs.insert(target_kind_a.packages_folder(*target_kind_b));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
dirs
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Errors that can occur when using the pesde library
|
/// Errors that can occur when using the pesde library
|
||||||
pub mod errors {
|
pub mod errors {
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
||||||
|
|
||||||
/// Errors that can occur when reading the manifest file
|
/// Errors that can occur when reading the manifest file
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub enum ManifestReadError {
|
pub enum ManifestReadError {
|
||||||
/// An IO error occurred
|
/// An IO error occurred
|
||||||
#[error("io error reading manifest file")]
|
#[error("io error reading manifest file")]
|
||||||
Io(#[from] std::io::Error),
|
Io(#[from] std::io::Error),
|
||||||
|
|
||||||
/// An error occurred while deserializing the manifest file
|
/// An error occurred while deserializing the manifest file
|
||||||
#[error("error deserializing manifest file at {0}")]
|
#[error("error deserializing manifest file")]
|
||||||
Serde(PathBuf, #[source] toml::de::Error),
|
Serde(#[from] toml::de::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Errors that can occur when reading the lockfile
|
/// Errors that can occur when reading the lockfile
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub enum LockfileReadError {
|
pub enum LockfileReadError {
|
||||||
/// An IO error occurred
|
/// An IO error occurred
|
||||||
#[error("io error reading lockfile")]
|
#[error("io error reading lockfile")]
|
||||||
Io(#[from] std::io::Error),
|
Io(#[from] std::io::Error),
|
||||||
|
|
||||||
/// An error occurred while deserializing the lockfile
|
/// An error occurred while deserializing the lockfile
|
||||||
#[error("error deserializing lockfile")]
|
#[error("error deserializing lockfile")]
|
||||||
Serde(#[from] toml::de::Error),
|
Serde(#[from] toml::de::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Errors that can occur when writing the lockfile
|
/// Errors that can occur when writing the lockfile
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub enum LockfileWriteError {
|
pub enum LockfileWriteError {
|
||||||
/// An IO error occurred
|
/// An IO error occurred
|
||||||
#[error("io error writing lockfile")]
|
#[error("io error writing lockfile")]
|
||||||
Io(#[from] std::io::Error),
|
Io(#[from] std::io::Error),
|
||||||
|
|
||||||
/// An error occurred while serializing the lockfile
|
/// An error occurred while serializing the lockfile
|
||||||
#[error("error serializing lockfile")]
|
#[error("error serializing lockfile")]
|
||||||
Serde(#[from] toml::ser::Error),
|
Serde(#[from] toml::ser::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Errors that can occur when finding workspace members
|
/// Errors that can occur when finding workspace members
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub enum WorkspaceMembersError {
|
pub enum WorkspaceMembersError {
|
||||||
/// An error occurred parsing the manifest file
|
/// The manifest file could not be found
|
||||||
#[error("error parsing manifest file")]
|
#[error("missing manifest file")]
|
||||||
ManifestParse(#[from] ManifestReadError),
|
ManifestMissing(#[source] std::io::Error),
|
||||||
|
|
||||||
/// An error occurred interacting with the filesystem
|
/// An error occurred deserializing the manifest file
|
||||||
#[error("error interacting with the filesystem")]
|
#[error("error deserializing manifest file at {0}")]
|
||||||
Io(#[from] std::io::Error),
|
ManifestDeser(PathBuf, #[source] Box<toml::de::Error>),
|
||||||
|
|
||||||
/// An error occurred while globbing
|
/// An error occurred interacting with the filesystem
|
||||||
#[error("error globbing")]
|
#[error("error interacting with the filesystem")]
|
||||||
Globbing(#[from] MatchingGlobsError),
|
Io(#[from] std::io::Error),
|
||||||
}
|
|
||||||
|
|
||||||
/// Errors that can occur when finding matching globs
|
/// An error occurred while globbing
|
||||||
#[derive(Debug, Error)]
|
#[error("error globbing")]
|
||||||
#[non_exhaustive]
|
Globbing(#[from] MatchingGlobsError),
|
||||||
pub enum MatchingGlobsError {
|
}
|
||||||
/// An error occurred interacting with the filesystem
|
|
||||||
#[error("error interacting with the filesystem")]
|
|
||||||
Io(#[from] std::io::Error),
|
|
||||||
|
|
||||||
/// An error occurred while building a glob
|
/// Errors that can occur when finding matching globs
|
||||||
#[error("error building glob")]
|
#[derive(Debug, Error)]
|
||||||
BuildGlob(#[from] wax::BuildError),
|
#[non_exhaustive]
|
||||||
}
|
pub enum MatchingGlobsError {
|
||||||
|
/// An error occurred interacting with the filesystem
|
||||||
|
#[error("error interacting with the filesystem")]
|
||||||
|
Io(#[from] std::io::Error),
|
||||||
|
|
||||||
/// Errors that can occur when finding project roots
|
/// An error occurred while building a glob
|
||||||
#[derive(Debug, Error)]
|
#[error("error building glob")]
|
||||||
#[non_exhaustive]
|
BuildGlob(#[from] wax::BuildError),
|
||||||
pub enum FindRootsError {
|
}
|
||||||
/// Reading the manifest failed
|
|
||||||
#[error("error reading manifest")]
|
|
||||||
ManifestRead(#[from] ManifestReadError),
|
|
||||||
|
|
||||||
/// Globbing failed
|
|
||||||
#[error("error globbing")]
|
|
||||||
Globbing(#[from] MatchingGlobsError),
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,247 +2,230 @@ use std::path::{Component, Path};
|
||||||
|
|
||||||
use crate::manifest::{target::TargetKind, Manifest};
|
use crate::manifest::{target::TargetKind, Manifest};
|
||||||
use full_moon::{ast::luau::ExportedTypeDeclaration, visitors::Visitor};
|
use full_moon::{ast::luau::ExportedTypeDeclaration, visitors::Visitor};
|
||||||
use relative_path::RelativePath;
|
use relative_path::RelativePathBuf;
|
||||||
use tracing::instrument;
|
|
||||||
|
|
||||||
struct TypeVisitor {
|
struct TypeVisitor {
|
||||||
types: Vec<String>,
|
types: Vec<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Visitor for TypeVisitor {
|
impl Visitor for TypeVisitor {
|
||||||
fn visit_exported_type_declaration(&mut self, node: &ExportedTypeDeclaration) {
|
fn visit_exported_type_declaration(&mut self, node: &ExportedTypeDeclaration) {
|
||||||
let name = node.type_declaration().type_name().to_string();
|
let name = node.type_declaration().type_name().to_string();
|
||||||
|
|
||||||
let (declaration_generics, generics) =
|
let (declaration_generics, generics) =
|
||||||
if let Some(declaration) = node.type_declaration().generics() {
|
if let Some(declaration) = node.type_declaration().generics() {
|
||||||
let mut declaration_generics = vec![];
|
let mut declaration_generics = vec![];
|
||||||
let mut generics = vec![];
|
let mut generics = vec![];
|
||||||
|
|
||||||
for generic in declaration.generics().iter() {
|
for generic in declaration.generics().iter() {
|
||||||
declaration_generics.push(generic.to_string());
|
declaration_generics.push(generic.to_string());
|
||||||
|
|
||||||
if generic.default_type().is_some() {
|
if generic.default_type().is_some() {
|
||||||
generics.push(generic.parameter().to_string())
|
generics.push(generic.parameter().to_string())
|
||||||
} else {
|
} else {
|
||||||
generics.push(generic.to_string())
|
generics.push(generic.to_string())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
(
|
(
|
||||||
format!("<{}>", declaration_generics.join(", ")),
|
format!("<{}>", declaration_generics.join(", ")),
|
||||||
format!("<{}>", generics.join(", ")),
|
format!("<{}>", generics.join(", ")),
|
||||||
)
|
)
|
||||||
} else {
|
} else {
|
||||||
("".to_string(), "".to_string())
|
("".to_string(), "".to_string())
|
||||||
};
|
};
|
||||||
|
|
||||||
self.types.push(format!(
|
self.types.push(format!(
|
||||||
"export type {name}{declaration_generics} = module.{name}{generics}\n"
|
"export type {name}{declaration_generics} = module.{name}{generics}\n"
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn get_file_types(file: &str) -> Vec<String> {
|
/// Get the types exported by a file
|
||||||
let ast = match full_moon::parse(file) {
|
pub fn get_file_types(file: &str) -> Result<Vec<String>, Vec<full_moon::Error>> {
|
||||||
Ok(ast) => ast,
|
let ast = full_moon::parse(file)?;
|
||||||
Err(err) => {
|
let mut visitor = TypeVisitor { types: vec![] };
|
||||||
tracing::error!(
|
visitor.visit_ast(&ast);
|
||||||
"failed to parse file to extract types:\n{}",
|
|
||||||
err.into_iter()
|
|
||||||
.map(|err| format!("\t- {err}"))
|
|
||||||
.collect::<Vec<_>>()
|
|
||||||
.join("\n")
|
|
||||||
);
|
|
||||||
|
|
||||||
return vec![];
|
Ok(visitor.types)
|
||||||
}
|
|
||||||
};
|
|
||||||
let mut visitor = TypeVisitor { types: vec![] };
|
|
||||||
visitor.visit_ast(&ast);
|
|
||||||
|
|
||||||
visitor.types
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Generate a linking module for a library
|
/// Generate a linking module for a library
|
||||||
pub fn generate_lib_linking_module<I: IntoIterator<Item = S>, S: AsRef<str>>(
|
pub fn generate_lib_linking_module<I: IntoIterator<Item = S>, S: AsRef<str>>(
|
||||||
path: &str,
|
path: &str,
|
||||||
types: I,
|
types: I,
|
||||||
) -> String {
|
) -> String {
|
||||||
let mut output = format!("local module = require({path})\n");
|
let mut output = format!("local module = require({path})\n");
|
||||||
|
|
||||||
for ty in types {
|
for ty in types {
|
||||||
output.push_str(ty.as_ref());
|
output.push_str(ty.as_ref());
|
||||||
}
|
}
|
||||||
|
|
||||||
output.push_str("return module");
|
output.push_str("return module");
|
||||||
|
|
||||||
output
|
output
|
||||||
}
|
}
|
||||||
|
|
||||||
fn luau_style_path(path: &Path) -> String {
|
fn luau_style_path(path: &Path) -> String {
|
||||||
let path = path
|
let path = path
|
||||||
.components()
|
.components()
|
||||||
.zip(
|
.zip(
|
||||||
path.components()
|
path.components()
|
||||||
.skip(1)
|
.skip(1)
|
||||||
.map(Some)
|
.map(Some)
|
||||||
.chain(std::iter::repeat(None)),
|
.chain(std::iter::repeat(None)),
|
||||||
)
|
)
|
||||||
.filter_map(|(ct, next_ct)| match ct {
|
.filter_map(|(ct, next_ct)| match ct {
|
||||||
Component::CurDir => Some(".".to_string()),
|
Component::CurDir => Some(".".to_string()),
|
||||||
Component::ParentDir => Some("..".to_string()),
|
Component::ParentDir => Some("..".to_string()),
|
||||||
Component::Normal(part) => {
|
Component::Normal(part) => {
|
||||||
let str = part.to_string_lossy();
|
let str = part.to_string_lossy();
|
||||||
|
|
||||||
Some(
|
Some(
|
||||||
(if next_ct.is_some() {
|
(if next_ct.is_some() {
|
||||||
&str
|
&str
|
||||||
} else {
|
} else {
|
||||||
str.strip_suffix(".luau")
|
str.strip_suffix(".luau")
|
||||||
.or_else(|| str.strip_suffix(".lua"))
|
.or_else(|| str.strip_suffix(".lua"))
|
||||||
.unwrap_or(&str)
|
.unwrap_or(&str)
|
||||||
})
|
})
|
||||||
.to_string(),
|
.to_string(),
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
_ => None,
|
_ => None,
|
||||||
})
|
})
|
||||||
.collect::<Vec<_>>()
|
.collect::<Vec<_>>()
|
||||||
.join("/");
|
.join("/");
|
||||||
|
|
||||||
let require = format!("./{path}");
|
let require = format!("./{path}");
|
||||||
format!("{require:?}")
|
format!("{require:?}")
|
||||||
}
|
}
|
||||||
|
|
||||||
// This function should be simplified (especially to reduce the number of arguments),
|
// This function should be simplified (especially to reduce the number of arguments),
|
||||||
// but it's not clear how to do that while maintaining the current functionality.
|
// but it's not clear how to do that while maintaining the current functionality.
|
||||||
/// Get the require path for a library
|
/// Get the require path for a library
|
||||||
#[instrument(skip(project_manifest), level = "trace")]
|
|
||||||
#[allow(clippy::too_many_arguments)]
|
#[allow(clippy::too_many_arguments)]
|
||||||
pub fn get_lib_require_path(
|
pub fn get_lib_require_path(
|
||||||
target: TargetKind,
|
target: &TargetKind,
|
||||||
base_dir: &Path,
|
base_dir: &Path,
|
||||||
lib_file: &RelativePath,
|
lib_file: &RelativePathBuf,
|
||||||
destination_dir: &Path,
|
destination_dir: &Path,
|
||||||
use_new_structure: bool,
|
use_new_structure: bool,
|
||||||
root_container_dir: &Path,
|
root_container_dir: &Path,
|
||||||
container_dir: &Path,
|
container_dir: &Path,
|
||||||
project_manifest: &Manifest,
|
project_manifest: &Manifest,
|
||||||
) -> Result<String, errors::GetLibRequirePath> {
|
) -> Result<String, errors::GetLibRequirePath> {
|
||||||
let path = pathdiff::diff_paths(destination_dir, base_dir).unwrap();
|
let path = pathdiff::diff_paths(destination_dir, base_dir).unwrap();
|
||||||
tracing::debug!("diffed path: {}", path.display());
|
let path = if use_new_structure {
|
||||||
let path = if use_new_structure {
|
tracing::debug!("using new structure for require path with {lib_file:?}");
|
||||||
lib_file.to_path(path)
|
lib_file.to_path(path)
|
||||||
} else {
|
} else {
|
||||||
path
|
tracing::debug!("using old structure for require path with {lib_file:?}");
|
||||||
};
|
path
|
||||||
|
};
|
||||||
|
|
||||||
if matches!(target, TargetKind::Roblox | TargetKind::RobloxServer) {
|
if matches!(target, TargetKind::Roblox | TargetKind::RobloxServer) {
|
||||||
let (prefix, path) = match target.try_into() {
|
let (prefix, path) = match target.try_into() {
|
||||||
Ok(place_kind) if !destination_dir.starts_with(root_container_dir) => (
|
Ok(place_kind) if !destination_dir.starts_with(root_container_dir) => (
|
||||||
project_manifest
|
project_manifest
|
||||||
.place
|
.place
|
||||||
.get(&place_kind)
|
.get(&place_kind)
|
||||||
.ok_or(errors::GetLibRequirePath::RobloxPlaceKindPathNotFound(
|
.ok_or(errors::GetLibRequirePath::RobloxPlaceKindPathNotFound(
|
||||||
place_kind,
|
place_kind,
|
||||||
))?
|
))?
|
||||||
.as_str(),
|
.as_str(),
|
||||||
if use_new_structure {
|
if use_new_structure {
|
||||||
lib_file.to_path(container_dir)
|
lib_file.to_path(container_dir)
|
||||||
} else {
|
} else {
|
||||||
container_dir.to_path_buf()
|
container_dir.to_path_buf()
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
_ => ("script.Parent", path),
|
_ => ("script.Parent", path),
|
||||||
};
|
};
|
||||||
|
|
||||||
let path = path
|
let path = path
|
||||||
.components()
|
.components()
|
||||||
.zip(
|
.zip(
|
||||||
path.components()
|
path.components()
|
||||||
.skip(1)
|
.skip(1)
|
||||||
.map(Some)
|
.map(Some)
|
||||||
.chain(std::iter::repeat(None)),
|
.chain(std::iter::repeat(None)),
|
||||||
)
|
)
|
||||||
.filter_map(|(component, next_comp)| match component {
|
.filter_map(|(component, next_comp)| match component {
|
||||||
Component::ParentDir => Some(".Parent".to_string()),
|
Component::ParentDir => Some(".Parent".to_string()),
|
||||||
Component::Normal(part) if part != "init.lua" && part != "init.luau" => {
|
Component::Normal(part) if part != "init.lua" && part != "init.luau" => {
|
||||||
let str = part.to_string_lossy();
|
let str = part.to_string_lossy();
|
||||||
|
|
||||||
Some(format!(
|
Some(format!(
|
||||||
"[{:?}]",
|
"[{:?}]",
|
||||||
if next_comp.is_some() {
|
if next_comp.is_some() {
|
||||||
&str
|
&str
|
||||||
} else {
|
} else {
|
||||||
str.strip_suffix(".luau")
|
str.strip_suffix(".luau")
|
||||||
.or_else(|| str.strip_suffix(".lua"))
|
.or_else(|| str.strip_suffix(".lua"))
|
||||||
.unwrap_or(&str)
|
.unwrap_or(&str)
|
||||||
}
|
}
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
_ => None,
|
_ => None,
|
||||||
})
|
})
|
||||||
.collect::<Vec<_>>()
|
.collect::<Vec<_>>()
|
||||||
.join("");
|
.join("");
|
||||||
|
|
||||||
return Ok(format!("{prefix}{path}"));
|
return Ok(format!("{prefix}{path}"));
|
||||||
};
|
};
|
||||||
|
|
||||||
Ok(luau_style_path(&path))
|
Ok(luau_style_path(&path))
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Generate a linking module for a binary
|
/// Generate a linking module for a binary
|
||||||
pub fn generate_bin_linking_module<P: AsRef<Path>>(package_root: P, require_path: &str) -> String {
|
pub fn generate_bin_linking_module<P: AsRef<Path>>(package_root: P, require_path: &str) -> String {
|
||||||
format!(
|
format!(
|
||||||
r#"_G.PESDE_ROOT = {:?}
|
r#"_G.PESDE_ROOT = {:?}
|
||||||
return require({require_path})"#,
|
return require({require_path})"#,
|
||||||
package_root.as_ref().to_string_lossy()
|
package_root.as_ref().to_string_lossy()
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the require path for a binary
|
/// Get the require path for a binary
|
||||||
#[instrument(level = "trace")]
|
|
||||||
pub fn get_bin_require_path(
|
pub fn get_bin_require_path(
|
||||||
base_dir: &Path,
|
base_dir: &Path,
|
||||||
bin_file: &RelativePath,
|
bin_file: &RelativePathBuf,
|
||||||
destination_dir: &Path,
|
destination_dir: &Path,
|
||||||
) -> String {
|
) -> String {
|
||||||
let path = pathdiff::diff_paths(destination_dir, base_dir).unwrap();
|
let path = pathdiff::diff_paths(destination_dir, base_dir).unwrap();
|
||||||
tracing::debug!("diffed path: {}", path.display());
|
let path = bin_file.to_path(path);
|
||||||
let path = bin_file.to_path(path);
|
|
||||||
|
|
||||||
luau_style_path(&path)
|
luau_style_path(&path)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Generate a linking module for a script
|
/// Generate a linking module for a script
|
||||||
pub fn generate_script_linking_module(require_path: &str) -> String {
|
pub fn generate_script_linking_module(require_path: &str) -> String {
|
||||||
format!(r#"return require({require_path})"#)
|
format!(r#"return require({require_path})"#)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the require path for a script
|
/// Get the require path for a script
|
||||||
#[instrument(level = "trace")]
|
|
||||||
pub fn get_script_require_path(
|
pub fn get_script_require_path(
|
||||||
base_dir: &Path,
|
base_dir: &Path,
|
||||||
script_file: &RelativePath,
|
script_file: &RelativePathBuf,
|
||||||
destination_dir: &Path,
|
destination_dir: &Path,
|
||||||
) -> String {
|
) -> String {
|
||||||
let path = pathdiff::diff_paths(destination_dir, base_dir).unwrap();
|
let path = pathdiff::diff_paths(destination_dir, base_dir).unwrap();
|
||||||
tracing::debug!("diffed path: {}", path.display());
|
let path = script_file.to_path(path);
|
||||||
let path = script_file.to_path(path);
|
|
||||||
|
|
||||||
luau_style_path(&path)
|
luau_style_path(&path)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Errors for the linking module utilities
|
/// Errors for the linking module utilities
|
||||||
pub mod errors {
|
pub mod errors {
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
||||||
|
|
||||||
/// An error occurred while getting the require path for a library
|
/// An error occurred while getting the require path for a library
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub enum GetLibRequirePath {
|
pub enum GetLibRequirePath {
|
||||||
/// The path for the RobloxPlaceKind could not be found
|
/// The path for the RobloxPlaceKind could not be found
|
||||||
#[error("could not find the path for the RobloxPlaceKind {0}")]
|
#[error("could not find the path for the RobloxPlaceKind {0}")]
|
||||||
RobloxPlaceKindPathNotFound(crate::manifest::target::RobloxPlaceKind),
|
RobloxPlaceKindPathNotFound(crate::manifest::target::RobloxPlaceKind),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,271 +0,0 @@
|
||||||
use crate::{
|
|
||||||
all_packages_dirs, graph::DependencyGraphWithTarget, manifest::Alias, util::remove_empty_dir,
|
|
||||||
Project, PACKAGES_CONTAINER_NAME, SCRIPTS_LINK_FOLDER,
|
|
||||||
};
|
|
||||||
use fs_err::tokio as fs;
|
|
||||||
use futures::FutureExt;
|
|
||||||
use std::{
|
|
||||||
collections::HashSet,
|
|
||||||
path::{Path, PathBuf},
|
|
||||||
sync::Arc,
|
|
||||||
};
|
|
||||||
use tokio::task::JoinSet;
|
|
||||||
|
|
||||||
fn index_entry(
|
|
||||||
entry: fs::DirEntry,
|
|
||||||
packages_index_dir: &Path,
|
|
||||||
tasks: &mut JoinSet<std::io::Result<()>>,
|
|
||||||
used_paths: &Arc<HashSet<PathBuf>>,
|
|
||||||
) {
|
|
||||||
let path = entry.path();
|
|
||||||
let path_relative = path.strip_prefix(packages_index_dir).unwrap().to_path_buf();
|
|
||||||
|
|
||||||
let is_wally = entry
|
|
||||||
.file_name()
|
|
||||||
.to_str()
|
|
||||||
.expect("non UTF-8 folder name in packages index")
|
|
||||||
.contains("@");
|
|
||||||
|
|
||||||
let used_paths = used_paths.clone();
|
|
||||||
tasks.spawn(async move {
|
|
||||||
if is_wally {
|
|
||||||
#[cfg(not(feature = "wally-compat"))]
|
|
||||||
{
|
|
||||||
tracing::error!(
|
|
||||||
"found Wally package in index despite feature being disabled at `{}`",
|
|
||||||
path.display()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
#[cfg(feature = "wally-compat")]
|
|
||||||
{
|
|
||||||
if !used_paths.contains(&path_relative) {
|
|
||||||
fs::remove_dir_all(path).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut tasks = JoinSet::new();
|
|
||||||
|
|
||||||
let mut entries = fs::read_dir(&path).await?;
|
|
||||||
while let Some(entry) = entries.next_entry().await? {
|
|
||||||
let version = entry.file_name();
|
|
||||||
let path_relative = path_relative.join(&version);
|
|
||||||
|
|
||||||
if used_paths.contains(&path_relative) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
let path = entry.path();
|
|
||||||
tasks.spawn(async { fs::remove_dir_all(path).await });
|
|
||||||
}
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
task.unwrap()?;
|
|
||||||
}
|
|
||||||
|
|
||||||
remove_empty_dir(&path).await
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
fn packages_entry(
|
|
||||||
entry: fs::DirEntry,
|
|
||||||
tasks: &mut JoinSet<std::io::Result<()>>,
|
|
||||||
expected_aliases: &Arc<HashSet<Alias>>,
|
|
||||||
) {
|
|
||||||
let expected_aliases = expected_aliases.clone();
|
|
||||||
tasks.spawn(async move {
|
|
||||||
if !entry.file_type().await?.is_file() {
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
|
|
||||||
let path = entry.path();
|
|
||||||
let name = path
|
|
||||||
.file_stem()
|
|
||||||
.unwrap()
|
|
||||||
.to_str()
|
|
||||||
.expect("non UTF-8 file name in packages folder");
|
|
||||||
let name = name.strip_suffix(".bin").unwrap_or(name);
|
|
||||||
let name = match name.parse::<Alias>() {
|
|
||||||
Ok(name) => name,
|
|
||||||
Err(e) => {
|
|
||||||
tracing::error!("invalid alias in packages folder: {e}");
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
if !expected_aliases.contains(&name) {
|
|
||||||
fs::remove_file(path).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
fn scripts_entry(
|
|
||||||
entry: fs::DirEntry,
|
|
||||||
tasks: &mut JoinSet<std::io::Result<()>>,
|
|
||||||
expected_aliases: &Arc<HashSet<Alias>>,
|
|
||||||
) {
|
|
||||||
let expected_aliases = expected_aliases.clone();
|
|
||||||
tasks.spawn(async move {
|
|
||||||
if !entry.file_type().await?.is_dir() {
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
|
|
||||||
let path = entry.path();
|
|
||||||
let name = path
|
|
||||||
.file_name()
|
|
||||||
.unwrap()
|
|
||||||
.to_str()
|
|
||||||
.expect("non UTF-8 file name in scripts folder");
|
|
||||||
let name = match name.parse::<Alias>() {
|
|
||||||
Ok(name) => name,
|
|
||||||
Err(e) => {
|
|
||||||
tracing::error!("invalid alias in scripts folder: {e}");
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
if !expected_aliases.contains(&name) {
|
|
||||||
fs::remove_dir_all(&path).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Project {
|
|
||||||
/// Removes unused packages from the project
|
|
||||||
pub async fn remove_unused(
|
|
||||||
&self,
|
|
||||||
graph: &DependencyGraphWithTarget,
|
|
||||||
) -> Result<(), errors::RemoveUnusedError> {
|
|
||||||
let manifest = self.deser_manifest().await?;
|
|
||||||
let used_paths = graph
|
|
||||||
.iter()
|
|
||||||
.map(|(id, node)| {
|
|
||||||
node.node
|
|
||||||
.container_folder(id)
|
|
||||||
.parent()
|
|
||||||
.unwrap()
|
|
||||||
.to_path_buf()
|
|
||||||
})
|
|
||||||
.collect::<HashSet<_>>();
|
|
||||||
let used_paths = Arc::new(used_paths);
|
|
||||||
|
|
||||||
let mut tasks = all_packages_dirs()
|
|
||||||
.into_iter()
|
|
||||||
.map(|folder| {
|
|
||||||
let packages_dir = self.package_dir().join(&folder);
|
|
||||||
let packages_index_dir = packages_dir.join(PACKAGES_CONTAINER_NAME);
|
|
||||||
let used_paths = used_paths.clone();
|
|
||||||
|
|
||||||
let expected_aliases = graph
|
|
||||||
.iter()
|
|
||||||
.filter(|(id, _)| {
|
|
||||||
manifest
|
|
||||||
.target
|
|
||||||
.kind()
|
|
||||||
.packages_folder(id.version_id().target())
|
|
||||||
== folder
|
|
||||||
})
|
|
||||||
.filter_map(|(_, node)| {
|
|
||||||
node.node.direct.as_ref().map(|(alias, _, _)| alias.clone())
|
|
||||||
})
|
|
||||||
.collect::<HashSet<_>>();
|
|
||||||
let expected_aliases = Arc::new(expected_aliases);
|
|
||||||
|
|
||||||
async move {
|
|
||||||
let mut index_entries = match fs::read_dir(&packages_index_dir).await {
|
|
||||||
Ok(entries) => entries,
|
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(()),
|
|
||||||
Err(e) => return Err(e),
|
|
||||||
};
|
|
||||||
// we don't handle NotFound here because the upper level will handle it
|
|
||||||
let mut packages_entries = fs::read_dir(&packages_dir).await?;
|
|
||||||
let mut tasks = JoinSet::new();
|
|
||||||
|
|
||||||
loop {
|
|
||||||
tokio::select! {
|
|
||||||
Some(entry) = index_entries.next_entry().map(Result::transpose) => {
|
|
||||||
index_entry(
|
|
||||||
entry?,
|
|
||||||
&packages_index_dir,
|
|
||||||
&mut tasks,
|
|
||||||
&used_paths,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
Some(entry) = packages_entries.next_entry().map(Result::transpose) => {
|
|
||||||
packages_entry(
|
|
||||||
entry?,
|
|
||||||
&mut tasks,
|
|
||||||
&expected_aliases,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
else => break,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
task.unwrap()?;
|
|
||||||
}
|
|
||||||
|
|
||||||
remove_empty_dir(&packages_index_dir).await?;
|
|
||||||
remove_empty_dir(&packages_dir).await?;
|
|
||||||
|
|
||||||
Ok::<_, std::io::Error>(())
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect::<JoinSet<_>>();
|
|
||||||
|
|
||||||
let scripts_dir = self.package_dir().join(SCRIPTS_LINK_FOLDER);
|
|
||||||
match fs::read_dir(&scripts_dir).await {
|
|
||||||
Ok(mut entries) => {
|
|
||||||
let expected_aliases = graph
|
|
||||||
.iter()
|
|
||||||
.filter_map(|(_, node)| {
|
|
||||||
node.node
|
|
||||||
.direct
|
|
||||||
.as_ref()
|
|
||||||
.map(|(alias, _, _)| alias.clone())
|
|
||||||
.filter(|_| node.target.scripts().is_some_and(|s| !s.is_empty()))
|
|
||||||
})
|
|
||||||
.collect::<HashSet<_>>();
|
|
||||||
let expected_aliases = Arc::new(expected_aliases);
|
|
||||||
|
|
||||||
while let Some(entry) = entries.next_entry().await? {
|
|
||||||
scripts_entry(entry, &mut tasks, &expected_aliases);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => {}
|
|
||||||
Err(e) => return Err(e.into()),
|
|
||||||
}
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
task.unwrap()?;
|
|
||||||
}
|
|
||||||
|
|
||||||
remove_empty_dir(&scripts_dir).await?;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Errors that can occur when using incremental installs
|
|
||||||
pub mod errors {
|
|
||||||
use thiserror::Error;
|
|
||||||
|
|
||||||
/// Errors that can occur when removing unused packages
|
|
||||||
#[derive(Debug, Error)]
|
|
||||||
pub enum RemoveUnusedError {
|
|
||||||
/// Reading the manifest failed
|
|
||||||
#[error("error reading manifest")]
|
|
||||||
ManifestRead(#[from] crate::errors::ManifestReadError),
|
|
||||||
|
|
||||||
/// IO error
|
|
||||||
#[error("IO error")]
|
|
||||||
Io(#[from] std::io::Error),
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,424 +1,399 @@
|
||||||
use crate::{
|
use crate::{
|
||||||
graph::{DependencyGraphNodeWithTarget, DependencyGraphWithTarget},
|
linking::generator::get_file_types,
|
||||||
linking::generator::get_file_types,
|
lockfile::{DownloadedDependencyGraphNode, DownloadedGraph},
|
||||||
manifest::{Alias, Manifest},
|
manifest::Manifest,
|
||||||
scripts::{execute_script, ExecuteScriptHooks, ScriptName},
|
names::PackageNames,
|
||||||
source::{
|
scripts::{execute_script, ScriptName},
|
||||||
fs::{cas_path, store_in_cas},
|
source::{
|
||||||
ids::PackageId,
|
fs::{cas_path, store_in_cas},
|
||||||
traits::PackageRef,
|
traits::PackageRef,
|
||||||
},
|
version_id::VersionId,
|
||||||
Project, LINK_LIB_NO_FILE_FOUND, PACKAGES_CONTAINER_NAME, SCRIPTS_LINK_FOLDER,
|
},
|
||||||
|
Project, LINK_LIB_NO_FILE_FOUND, PACKAGES_CONTAINER_NAME, SCRIPTS_LINK_FOLDER,
|
||||||
};
|
};
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
|
use futures::future::try_join_all;
|
||||||
use std::{
|
use std::{
|
||||||
collections::HashMap,
|
collections::HashMap,
|
||||||
ffi::OsStr,
|
ffi::OsStr,
|
||||||
path::{Path, PathBuf},
|
path::{Path, PathBuf},
|
||||||
sync::Arc,
|
sync::Arc,
|
||||||
};
|
};
|
||||||
use tokio::task::{spawn_blocking, JoinSet};
|
use tokio::task::spawn_blocking;
|
||||||
use tracing::{instrument, Instrument};
|
use tracing::{instrument, Instrument};
|
||||||
|
|
||||||
/// Generates linking modules for a project
|
/// Generates linking modules for a project
|
||||||
pub mod generator;
|
pub mod generator;
|
||||||
/// Incremental installs
|
|
||||||
pub mod incremental;
|
|
||||||
|
|
||||||
async fn create_and_canonicalize<P: AsRef<Path>>(path: P) -> std::io::Result<PathBuf> {
|
async fn create_and_canonicalize<P: AsRef<Path>>(path: P) -> std::io::Result<PathBuf> {
|
||||||
let p = path.as_ref();
|
let p = path.as_ref();
|
||||||
fs::create_dir_all(p).await?;
|
fs::create_dir_all(p).await?;
|
||||||
p.canonicalize()
|
p.canonicalize()
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn write_cas(destination: PathBuf, cas_dir: &Path, contents: &str) -> std::io::Result<()> {
|
async fn write_cas(destination: PathBuf, cas_dir: &Path, contents: &str) -> std::io::Result<()> {
|
||||||
let hash = store_in_cas(cas_dir, contents.as_bytes()).await?;
|
let hash = store_in_cas(cas_dir, contents.as_bytes(), |_| async { Ok(()) }).await?;
|
||||||
|
|
||||||
match fs::remove_file(&destination).await {
|
match fs::remove_file(&destination).await {
|
||||||
Ok(_) => {}
|
Ok(_) => {}
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => {}
|
Err(e) if e.kind() == std::io::ErrorKind::NotFound => {}
|
||||||
// TODO: investigate why this happens and whether we can avoid it without ignoring all PermissionDenied errors
|
Err(e) => return Err(e),
|
||||||
#[cfg(windows)]
|
};
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::PermissionDenied => {}
|
|
||||||
Err(e) => return Err(e),
|
|
||||||
};
|
|
||||||
|
|
||||||
fs::hard_link(cas_path(&hash, cas_dir), destination).await
|
fs::hard_link(cas_path(&hash, cas_dir), destination).await
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy)]
|
|
||||||
struct LinkingExecuteScriptHooks;
|
|
||||||
|
|
||||||
impl ExecuteScriptHooks for LinkingExecuteScriptHooks {
|
|
||||||
fn not_found(&self, script: ScriptName) {
|
|
||||||
tracing::warn!(
|
|
||||||
"not having a `{script}` script in the manifest might cause issues with linking"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type PackageTypes = HashMap<PackageId, Vec<String>>;
|
|
||||||
|
|
||||||
impl Project {
|
impl Project {
|
||||||
/// Links the dependencies of the project
|
/// Links the dependencies of the project
|
||||||
#[instrument(skip(self, graph), level = "debug")]
|
#[instrument(skip(self, graph), level = "debug")]
|
||||||
pub(crate) async fn link_dependencies(
|
pub async fn link_dependencies(
|
||||||
&self,
|
&self,
|
||||||
graph: Arc<DependencyGraphWithTarget>,
|
graph: &DownloadedGraph,
|
||||||
with_types: bool,
|
with_types: bool,
|
||||||
) -> Result<(), errors::LinkingError> {
|
) -> Result<(), errors::LinkingError> {
|
||||||
let manifest = self.deser_manifest().await?;
|
let manifest = self.deser_manifest().await?;
|
||||||
let manifest_target_kind = manifest.target.kind();
|
let manifest_target_kind = manifest.target.kind();
|
||||||
let manifest = Arc::new(manifest);
|
let manifest = Arc::new(manifest);
|
||||||
|
|
||||||
// step 1. link all non-wally packages (and their dependencies) temporarily without types
|
// step 1. link all non-wally packages (and their dependencies) temporarily without types
|
||||||
// we do this separately to allow the required tools for the scripts to be installed
|
// we do this separately to allow the required tools for the scripts to be installed
|
||||||
self.link(&graph, &manifest, &Arc::new(PackageTypes::default()), false)
|
self.link(graph, &manifest, &Arc::new(Default::default()), false)
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if !with_types {
|
if !with_types {
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
// step 2. extract the types from libraries, prepare Roblox packages for syncing
|
// step 2. extract the types from libraries, prepare Roblox packages for syncing
|
||||||
let mut tasks = graph
|
let roblox_sync_config_gen_script = manifest
|
||||||
.iter()
|
.scripts
|
||||||
.map(|(package_id, node)| {
|
.get(&ScriptName::RobloxSyncConfigGenerator.to_string());
|
||||||
let span =
|
|
||||||
tracing::info_span!("extract types", package_id = package_id.to_string());
|
|
||||||
|
|
||||||
let package_id = package_id.clone();
|
let package_types = try_join_all(graph.iter().map(|(name, versions)| async move {
|
||||||
let node = node.clone();
|
Ok::<_, errors::LinkingError>((
|
||||||
let project = self.clone();
|
name,
|
||||||
|
try_join_all(versions.iter().map(|(version_id, node)| async move {
|
||||||
|
let Some(lib_file) = node.target.lib_path() else {
|
||||||
|
return Ok((version_id, vec![]));
|
||||||
|
};
|
||||||
|
|
||||||
async move {
|
let container_folder = node.node.container_folder(
|
||||||
let Some(lib_file) = node.target.lib_path() else {
|
&self
|
||||||
return Ok((package_id, vec![]));
|
.package_dir()
|
||||||
};
|
.join(manifest_target_kind.packages_folder(version_id.target()))
|
||||||
|
.join(PACKAGES_CONTAINER_NAME),
|
||||||
|
name,
|
||||||
|
version_id.version(),
|
||||||
|
);
|
||||||
|
|
||||||
let container_folder = node.node.container_folder_from_project(
|
let types = if lib_file.as_str() != LINK_LIB_NO_FILE_FOUND {
|
||||||
&package_id,
|
let lib_file = lib_file.to_path(&container_folder);
|
||||||
&project,
|
|
||||||
manifest_target_kind,
|
|
||||||
);
|
|
||||||
|
|
||||||
let types = if lib_file.as_str() != LINK_LIB_NO_FILE_FOUND {
|
let contents = match fs::read_to_string(&lib_file).await {
|
||||||
let lib_file = lib_file.to_path(&container_folder);
|
Ok(contents) => contents,
|
||||||
|
Err(e) if e.kind() == std::io::ErrorKind::NotFound => {
|
||||||
|
return Err(errors::LinkingError::LibFileNotFound(
|
||||||
|
lib_file.display().to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
Err(e) => return Err(e.into()),
|
||||||
|
};
|
||||||
|
|
||||||
let contents = match fs::read_to_string(&lib_file).await {
|
let types = match spawn_blocking(move || get_file_types(&contents))
|
||||||
Ok(contents) => contents,
|
.await
|
||||||
Err(e) if e.kind() == std::io::ErrorKind::NotFound => {
|
.unwrap()
|
||||||
return Err(errors::LinkingError::LibFileNotFound(
|
{
|
||||||
lib_file.display().to_string(),
|
Ok(types) => types,
|
||||||
));
|
Err(e) => {
|
||||||
}
|
return Err(errors::LinkingError::FullMoon(
|
||||||
Err(e) => return Err(e.into()),
|
lib_file.display().to_string(),
|
||||||
};
|
e,
|
||||||
|
))
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
let types = spawn_blocking(move || get_file_types(&contents))
|
tracing::debug!("contains {} exported types", types.len());
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
tracing::debug!("contains {} exported types", types.len());
|
types
|
||||||
|
} else {
|
||||||
|
vec![]
|
||||||
|
};
|
||||||
|
|
||||||
types
|
if let Some(build_files) = Some(&node.target)
|
||||||
} else {
|
.filter(|_| !node.node.pkg_ref.like_wally())
|
||||||
vec![]
|
.and_then(|t| t.build_files())
|
||||||
};
|
{
|
||||||
|
let Some(script_path) = roblox_sync_config_gen_script else {
|
||||||
|
tracing::warn!("not having a `{}` script in the manifest might cause issues with Roblox linking", ScriptName::RobloxSyncConfigGenerator);
|
||||||
|
return Ok((version_id, types));
|
||||||
|
};
|
||||||
|
|
||||||
if let Some(build_files) = Some(&node.target)
|
execute_script(
|
||||||
.filter(|_| !node.node.pkg_ref.is_wally_package())
|
ScriptName::RobloxSyncConfigGenerator,
|
||||||
.and_then(|t| t.build_files())
|
&script_path.to_path(self.package_dir()),
|
||||||
{
|
std::iter::once(container_folder.as_os_str())
|
||||||
execute_script(
|
.chain(build_files.iter().map(OsStr::new)),
|
||||||
ScriptName::RobloxSyncConfigGenerator,
|
self,
|
||||||
&project,
|
false,
|
||||||
LinkingExecuteScriptHooks,
|
).await
|
||||||
std::iter::once(container_folder.as_os_str())
|
.map_err(|e| {
|
||||||
.chain(build_files.iter().map(OsStr::new)),
|
errors::LinkingError::GenerateRobloxSyncConfig(
|
||||||
false,
|
container_folder.display().to_string(),
|
||||||
)
|
e,
|
||||||
.await
|
)
|
||||||
.map_err(errors::LinkingError::ExecuteScript)?;
|
})?;
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok((package_id, types))
|
Ok((version_id, types))
|
||||||
}
|
}.instrument(tracing::debug_span!("extract types", name = name.to_string(), version_id = version_id.to_string()))))
|
||||||
.instrument(span)
|
.await?
|
||||||
})
|
.into_iter()
|
||||||
.collect::<JoinSet<_>>();
|
.collect::<HashMap<_, _>>(),
|
||||||
|
))
|
||||||
|
}))
|
||||||
|
.await?
|
||||||
|
.into_iter()
|
||||||
|
.collect::<HashMap<_, _>>();
|
||||||
|
|
||||||
let mut package_types = PackageTypes::new();
|
// step 3. link all packages (and their dependencies), this time with types
|
||||||
while let Some(task) = tasks.join_next().await {
|
self.link(graph, &manifest, &Arc::new(package_types), true)
|
||||||
let (package_id, types) = task.unwrap()?;
|
.await
|
||||||
package_types.insert(package_id, types);
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// step 3. link all packages (and their dependencies), this time with types
|
#[allow(clippy::too_many_arguments)]
|
||||||
self.link(&graph, &manifest, &Arc::new(package_types), true)
|
async fn link_files(
|
||||||
.await
|
&self,
|
||||||
}
|
base_folder: &Path,
|
||||||
|
container_folder: &Path,
|
||||||
|
root_container_folder: &Path,
|
||||||
|
relative_container_folder: &Path,
|
||||||
|
node: &DownloadedDependencyGraphNode,
|
||||||
|
name: &PackageNames,
|
||||||
|
version_id: &VersionId,
|
||||||
|
alias: &str,
|
||||||
|
package_types: &HashMap<&PackageNames, HashMap<&VersionId, Vec<String>>>,
|
||||||
|
manifest: &Manifest,
|
||||||
|
) -> Result<(), errors::LinkingError> {
|
||||||
|
static NO_TYPES: Vec<String> = Vec::new();
|
||||||
|
|
||||||
async fn link(
|
if let Some(lib_file) = node.target.lib_path() {
|
||||||
&self,
|
let lib_module = generator::generate_lib_linking_module(
|
||||||
graph: &Arc<DependencyGraphWithTarget>,
|
&generator::get_lib_require_path(
|
||||||
manifest: &Arc<Manifest>,
|
&node.target.kind(),
|
||||||
package_types: &Arc<PackageTypes>,
|
base_folder,
|
||||||
is_complete: bool,
|
lib_file,
|
||||||
) -> Result<(), errors::LinkingError> {
|
container_folder,
|
||||||
let mut tasks = JoinSet::<Result<_, errors::LinkingError>>::new();
|
node.node.pkg_ref.use_new_structure(),
|
||||||
let mut link_files = |base_folder: &Path,
|
root_container_folder,
|
||||||
container_folder: &Path,
|
relative_container_folder,
|
||||||
root_container_folder: &Path,
|
manifest,
|
||||||
relative_container_folder: &Path,
|
)?,
|
||||||
node: &DependencyGraphNodeWithTarget,
|
package_types
|
||||||
package_id: &PackageId,
|
.get(name)
|
||||||
alias: &Alias,
|
.and_then(|v| v.get(version_id))
|
||||||
is_root: bool|
|
.unwrap_or(&NO_TYPES),
|
||||||
-> Result<(), errors::LinkingError> {
|
);
|
||||||
static NO_TYPES: Vec<String> = Vec::new();
|
|
||||||
|
|
||||||
if let Some(lib_file) = node.target.lib_path() {
|
write_cas(
|
||||||
let destination = base_folder.join(format!("{alias}.luau"));
|
base_folder.join(format!("{alias}.luau")),
|
||||||
|
self.cas_dir(),
|
||||||
|
&lib_module,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
let lib_module = generator::generate_lib_linking_module(
|
if let Some(bin_file) = node.target.bin_path() {
|
||||||
&generator::get_lib_require_path(
|
let bin_module = generator::generate_bin_linking_module(
|
||||||
node.target.kind(),
|
container_folder,
|
||||||
base_folder,
|
&generator::get_bin_require_path(base_folder, bin_file, container_folder),
|
||||||
lib_file,
|
);
|
||||||
container_folder,
|
|
||||||
node.node.pkg_ref.use_new_structure(),
|
|
||||||
root_container_folder,
|
|
||||||
relative_container_folder,
|
|
||||||
manifest,
|
|
||||||
)?,
|
|
||||||
package_types.get(package_id).unwrap_or(&NO_TYPES),
|
|
||||||
);
|
|
||||||
let cas_dir = self.cas_dir().to_path_buf();
|
|
||||||
|
|
||||||
tasks.spawn(async move {
|
write_cas(
|
||||||
write_cas(destination, &cas_dir, &lib_module)
|
base_folder.join(format!("{alias}.bin.luau")),
|
||||||
.await
|
self.cas_dir(),
|
||||||
.map_err(Into::into)
|
&bin_module,
|
||||||
});
|
)
|
||||||
}
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
if let Some(bin_file) = node.target.bin_path() {
|
if let Some(scripts) = node.target.scripts().filter(|s| !s.is_empty()) {
|
||||||
let destination = base_folder.join(format!("{alias}.bin.luau"));
|
let scripts_base =
|
||||||
|
create_and_canonicalize(self.package_dir().join(SCRIPTS_LINK_FOLDER).join(alias))
|
||||||
|
.await?;
|
||||||
|
|
||||||
let bin_module = generator::generate_bin_linking_module(
|
for (script_name, script_path) in scripts {
|
||||||
container_folder,
|
let script_module =
|
||||||
&generator::get_bin_require_path(base_folder, bin_file, container_folder),
|
generator::generate_script_linking_module(&generator::get_script_require_path(
|
||||||
);
|
&scripts_base,
|
||||||
let cas_dir = self.cas_dir().to_path_buf();
|
script_path,
|
||||||
|
container_folder,
|
||||||
|
));
|
||||||
|
|
||||||
tasks.spawn(async move {
|
write_cas(
|
||||||
write_cas(destination, &cas_dir, &bin_module)
|
scripts_base.join(format!("{script_name}.luau")),
|
||||||
.await
|
self.cas_dir(),
|
||||||
.map_err(Into::into)
|
&script_module,
|
||||||
});
|
)
|
||||||
}
|
.await?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if let Some(scripts) = node
|
Ok(())
|
||||||
.target
|
}
|
||||||
.scripts()
|
|
||||||
.filter(|s| !s.is_empty() && node.node.direct.is_some() && is_root)
|
|
||||||
{
|
|
||||||
let scripts_base = self
|
|
||||||
.package_dir()
|
|
||||||
.join(SCRIPTS_LINK_FOLDER)
|
|
||||||
.join(alias.as_str());
|
|
||||||
|
|
||||||
for (script_name, script_path) in scripts {
|
async fn link(
|
||||||
let destination = scripts_base.join(format!("{script_name}.luau"));
|
&self,
|
||||||
let script_module = generator::generate_script_linking_module(
|
graph: &DownloadedGraph,
|
||||||
&generator::get_script_require_path(
|
manifest: &Arc<Manifest>,
|
||||||
&scripts_base,
|
package_types: &Arc<HashMap<&PackageNames, HashMap<&VersionId, Vec<String>>>>,
|
||||||
script_path,
|
is_complete: bool,
|
||||||
container_folder,
|
) -> Result<(), errors::LinkingError> {
|
||||||
),
|
try_join_all(graph.iter().flat_map(|(name, versions)| {
|
||||||
);
|
versions.iter().map(|(version_id, node)| {
|
||||||
let cas_dir = self.cas_dir().to_path_buf();
|
let name = name.clone();
|
||||||
|
let manifest = manifest.clone();
|
||||||
|
let package_types = package_types.clone();
|
||||||
|
|
||||||
tasks.spawn(async move {
|
let span = tracing::info_span!(
|
||||||
fs::create_dir_all(destination.parent().unwrap()).await?;
|
"link",
|
||||||
|
name = name.to_string(),
|
||||||
|
version_id = version_id.to_string()
|
||||||
|
);
|
||||||
|
|
||||||
write_cas(destination, &cas_dir, &script_module)
|
async move {
|
||||||
.await
|
let (node_container_folder, node_packages_folder) = {
|
||||||
.map_err(Into::into)
|
let base_folder = create_and_canonicalize(
|
||||||
});
|
self.package_dir()
|
||||||
}
|
.join(manifest.target.kind().packages_folder(version_id.target())),
|
||||||
}
|
)
|
||||||
|
.await?;
|
||||||
|
let packages_container_folder = base_folder.join(PACKAGES_CONTAINER_NAME);
|
||||||
|
|
||||||
Ok(())
|
let container_folder = node.node.container_folder(
|
||||||
};
|
&packages_container_folder,
|
||||||
|
&name,
|
||||||
|
version_id.version(),
|
||||||
|
);
|
||||||
|
|
||||||
let mut node_tasks = graph
|
if let Some((alias, _, _)) = &node.node.direct {
|
||||||
.iter()
|
self.link_files(
|
||||||
.map(|(id, node)| {
|
&base_folder,
|
||||||
let base_folder = self.package_dir().join(
|
&container_folder,
|
||||||
manifest
|
&base_folder,
|
||||||
.target
|
container_folder.strip_prefix(&base_folder).unwrap(),
|
||||||
.kind()
|
node,
|
||||||
.packages_folder(id.version_id().target()),
|
&name,
|
||||||
);
|
version_id,
|
||||||
|
alias,
|
||||||
|
&package_types,
|
||||||
|
&manifest,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
let id = id.clone();
|
(container_folder, base_folder)
|
||||||
let node = node.clone();
|
};
|
||||||
|
|
||||||
async move {
|
for (dependency_name, (dependency_version_id, dependency_alias)) in
|
||||||
Ok::<_, errors::LinkingError>((
|
&node.node.dependencies
|
||||||
id,
|
{
|
||||||
node,
|
let Some(dependency_node) = graph
|
||||||
create_and_canonicalize(base_folder).await?,
|
.get(dependency_name)
|
||||||
))
|
.and_then(|v| v.get(dependency_version_id))
|
||||||
}
|
else {
|
||||||
})
|
if is_complete {
|
||||||
.collect::<JoinSet<_>>();
|
return Err(errors::LinkingError::DependencyNotFound(
|
||||||
|
format!("{dependency_name}@{dependency_version_id}"),
|
||||||
|
format!("{name}@{version_id}"),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
let mut dependency_tasks = JoinSet::<Result<_, errors::LinkingError>>::new();
|
continue;
|
||||||
|
};
|
||||||
|
|
||||||
loop {
|
let base_folder = create_and_canonicalize(
|
||||||
tokio::select! {
|
self.package_dir().join(
|
||||||
Some(res) = node_tasks.join_next() => {
|
version_id
|
||||||
let (package_id, node, base_folder) = res.unwrap()?;
|
.target()
|
||||||
let (node_container_folder, node_packages_folder) = {
|
.packages_folder(dependency_version_id.target()),
|
||||||
let packages_container_folder = base_folder.join(PACKAGES_CONTAINER_NAME);
|
),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
let packages_container_folder = base_folder.join(PACKAGES_CONTAINER_NAME);
|
||||||
|
|
||||||
let container_folder =
|
let container_folder = dependency_node.node.container_folder(
|
||||||
packages_container_folder.join(node.node.container_folder(&package_id));
|
&packages_container_folder,
|
||||||
|
dependency_name,
|
||||||
|
dependency_version_id.version(),
|
||||||
|
);
|
||||||
|
|
||||||
if let Some((alias, _, _)) = &node.node.direct {
|
let linker_folder = create_and_canonicalize(
|
||||||
link_files(
|
node_container_folder.join(
|
||||||
&base_folder,
|
node.node
|
||||||
&container_folder,
|
.base_folder(version_id, dependency_node.target.kind()),
|
||||||
&base_folder,
|
),
|
||||||
container_folder.strip_prefix(&base_folder).unwrap(),
|
)
|
||||||
&node,
|
.await?;
|
||||||
&package_id,
|
|
||||||
alias,
|
|
||||||
true,
|
|
||||||
)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
(container_folder, base_folder)
|
self.link_files(
|
||||||
};
|
&linker_folder,
|
||||||
|
&container_folder,
|
||||||
|
&node_packages_folder,
|
||||||
|
container_folder.strip_prefix(&base_folder).unwrap(),
|
||||||
|
dependency_node,
|
||||||
|
dependency_name,
|
||||||
|
dependency_version_id,
|
||||||
|
dependency_alias,
|
||||||
|
&package_types,
|
||||||
|
&manifest,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
for (dep_id, dep_alias) in &node.node.dependencies {
|
Ok(())
|
||||||
let dep_id = dep_id.clone();
|
}
|
||||||
let dep_alias = dep_alias.clone();
|
.instrument(span)
|
||||||
let graph = graph.clone();
|
})
|
||||||
let node = node.clone();
|
}))
|
||||||
let package_id = package_id.clone();
|
.await
|
||||||
let node_container_folder = node_container_folder.clone();
|
.map(|_| ())
|
||||||
let node_packages_folder = node_packages_folder.clone();
|
}
|
||||||
let package_dir = self.package_dir().to_path_buf();
|
|
||||||
|
|
||||||
dependency_tasks.spawn(async move {
|
|
||||||
let Some(dep_node) = graph.get(&dep_id) else {
|
|
||||||
return if is_complete {
|
|
||||||
Err(errors::LinkingError::DependencyNotFound(
|
|
||||||
dep_id.to_string(),
|
|
||||||
package_id.to_string(),
|
|
||||||
))
|
|
||||||
} else {
|
|
||||||
Ok(None)
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
let base_folder = package_dir.join(
|
|
||||||
package_id
|
|
||||||
.version_id()
|
|
||||||
.target()
|
|
||||||
.packages_folder(dep_id.version_id().target()),
|
|
||||||
);
|
|
||||||
let linker_folder = node_container_folder.join(node.node.dependencies_dir(
|
|
||||||
package_id.version_id(),
|
|
||||||
dep_id.version_id().target(),
|
|
||||||
));
|
|
||||||
|
|
||||||
Ok(Some((
|
|
||||||
dep_node.clone(),
|
|
||||||
dep_id,
|
|
||||||
dep_alias,
|
|
||||||
create_and_canonicalize(base_folder).await?,
|
|
||||||
create_and_canonicalize(linker_folder).await?,
|
|
||||||
node_packages_folder,
|
|
||||||
)))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
},
|
|
||||||
Some(res) = dependency_tasks.join_next() => {
|
|
||||||
let Some((
|
|
||||||
dependency_node,
|
|
||||||
dependency_id,
|
|
||||||
dependency_alias,
|
|
||||||
base_folder,
|
|
||||||
linker_folder,
|
|
||||||
node_packages_folder,
|
|
||||||
)) = res.unwrap()?
|
|
||||||
else {
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
|
|
||||||
let packages_container_folder = base_folder.join(PACKAGES_CONTAINER_NAME);
|
|
||||||
|
|
||||||
let container_folder = packages_container_folder
|
|
||||||
.join(dependency_node.node.container_folder(&dependency_id));
|
|
||||||
|
|
||||||
link_files(
|
|
||||||
&linker_folder,
|
|
||||||
&container_folder,
|
|
||||||
&node_packages_folder,
|
|
||||||
container_folder.strip_prefix(&base_folder).unwrap(),
|
|
||||||
&dependency_node,
|
|
||||||
&dependency_id,
|
|
||||||
&dependency_alias,
|
|
||||||
false,
|
|
||||||
)?;
|
|
||||||
},
|
|
||||||
else => break,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
while let Some(task) = tasks.join_next().await {
|
|
||||||
task.unwrap()?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Errors that can occur while linking dependencies
|
/// Errors that can occur while linking dependencies
|
||||||
pub mod errors {
|
pub mod errors {
|
||||||
use thiserror::Error;
|
use thiserror::Error;
|
||||||
|
|
||||||
/// Errors that can occur while linking dependencies
|
/// Errors that can occur while linking dependencies
|
||||||
#[derive(Debug, Error)]
|
#[derive(Debug, Error)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub enum LinkingError {
|
pub enum LinkingError {
|
||||||
/// An error occurred while deserializing the project manifest
|
/// An error occurred while deserializing the project manifest
|
||||||
#[error("error deserializing project manifest")]
|
#[error("error deserializing project manifest")]
|
||||||
Manifest(#[from] crate::errors::ManifestReadError),
|
Manifest(#[from] crate::errors::ManifestReadError),
|
||||||
|
|
||||||
/// An error occurred while interacting with the filesystem
|
/// An error occurred while interacting with the filesystem
|
||||||
#[error("error interacting with filesystem")]
|
#[error("error interacting with filesystem")]
|
||||||
Io(#[from] std::io::Error),
|
Io(#[from] std::io::Error),
|
||||||
|
|
||||||
/// A dependency was not found
|
/// A dependency was not found
|
||||||
#[error("dependency `{0}` of `{1}` not found")]
|
#[error("dependency `{0}` of `{1}` not found")]
|
||||||
DependencyNotFound(String, String),
|
DependencyNotFound(String, String),
|
||||||
|
|
||||||
/// The library file was not found
|
/// The library file was not found
|
||||||
#[error("library file at {0} not found")]
|
#[error("library file at {0} not found")]
|
||||||
LibFileNotFound(String),
|
LibFileNotFound(String),
|
||||||
|
|
||||||
/// Executing a script failed
|
/// An error occurred while parsing a Luau script
|
||||||
#[error("error executing script")]
|
#[error("error parsing Luau script at {0}")]
|
||||||
ExecuteScript(#[from] crate::scripts::errors::ExecuteScriptError),
|
FullMoon(String, Vec<full_moon::Error>),
|
||||||
|
|
||||||
/// An error occurred while getting the require path for a library
|
/// An error occurred while generating a Roblox sync config
|
||||||
#[error("error getting require path for library")]
|
#[error("error generating roblox sync config for {0}")]
|
||||||
GetLibRequirePath(#[from] super::generator::errors::GetLibRequirePath),
|
GenerateRobloxSyncConfig(String, #[source] std::io::Error),
|
||||||
}
|
|
||||||
|
/// An error occurred while getting the require path for a library
|
||||||
|
#[error("error getting require path for library")]
|
||||||
|
GetLibRequirePath(#[from] super::generator::errors::GetLibRequirePath),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
234
src/lockfile.rs
234
src/lockfile.rs
|
@ -1,145 +1,113 @@
|
||||||
#![allow(deprecated)]
|
|
||||||
use crate::{
|
use crate::{
|
||||||
graph::DependencyGraph,
|
manifest::{
|
||||||
manifest::{overrides::OverrideKey, target::TargetKind},
|
overrides::OverrideKey,
|
||||||
names::PackageName,
|
target::{Target, TargetKind},
|
||||||
source::specifiers::DependencySpecifiers,
|
DependencyType,
|
||||||
|
},
|
||||||
|
names::{PackageName, PackageNames},
|
||||||
|
source::{
|
||||||
|
refs::PackageRefs, specifiers::DependencySpecifiers, traits::PackageRef,
|
||||||
|
version_id::VersionId,
|
||||||
|
},
|
||||||
};
|
};
|
||||||
use relative_path::RelativePathBuf;
|
use relative_path::RelativePathBuf;
|
||||||
use semver::Version;
|
use semver::Version;
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use std::collections::BTreeMap;
|
use std::{
|
||||||
|
collections::BTreeMap,
|
||||||
|
path::{Path, PathBuf},
|
||||||
|
};
|
||||||
|
|
||||||
|
/// A graph of dependencies
|
||||||
|
pub type Graph<Node> = BTreeMap<PackageNames, BTreeMap<VersionId, Node>>;
|
||||||
|
|
||||||
|
/// A dependency graph node
|
||||||
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||||
|
pub struct DependencyGraphNode {
|
||||||
|
/// The alias, specifier, and original (as in the manifest) type for the dependency, if it is a direct dependency (i.e. used by the current project)
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
pub direct: Option<(String, DependencySpecifiers, DependencyType)>,
|
||||||
|
/// The dependencies of the package
|
||||||
|
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
||||||
|
pub dependencies: BTreeMap<PackageNames, (VersionId, String)>,
|
||||||
|
/// The resolved (transformed, for example Peer -> Standard) type of the dependency
|
||||||
|
pub resolved_ty: DependencyType,
|
||||||
|
/// Whether the resolved type should be Peer if this isn't depended on
|
||||||
|
#[serde(default, skip_serializing_if = "std::ops::Not::not")]
|
||||||
|
pub is_peer: bool,
|
||||||
|
/// The package reference
|
||||||
|
pub pkg_ref: PackageRefs,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl DependencyGraphNode {
|
||||||
|
pub(crate) fn base_folder(&self, version_id: &VersionId, project_target: TargetKind) -> String {
|
||||||
|
if self.pkg_ref.use_new_structure() {
|
||||||
|
version_id.target().packages_folder(&project_target)
|
||||||
|
} else {
|
||||||
|
"..".to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the folder to store the contents of the package in
|
||||||
|
pub fn container_folder<P: AsRef<Path>>(
|
||||||
|
&self,
|
||||||
|
path: &P,
|
||||||
|
name: &PackageNames,
|
||||||
|
version: &Version,
|
||||||
|
) -> PathBuf {
|
||||||
|
if self.pkg_ref.like_wally() {
|
||||||
|
return path
|
||||||
|
.as_ref()
|
||||||
|
.join(format!(
|
||||||
|
"{}_{}@{}",
|
||||||
|
name.as_str().0,
|
||||||
|
name.as_str().1,
|
||||||
|
version
|
||||||
|
))
|
||||||
|
.join(name.as_str().1);
|
||||||
|
}
|
||||||
|
|
||||||
|
path.as_ref()
|
||||||
|
.join(name.escaped())
|
||||||
|
.join(version.to_string())
|
||||||
|
.join(name.as_str().1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A graph of `DependencyGraphNode`s
|
||||||
|
pub type DependencyGraph = Graph<DependencyGraphNode>;
|
||||||
|
|
||||||
|
/// A downloaded dependency graph node, i.e. a `DependencyGraphNode` with a `Target`
|
||||||
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||||
|
pub struct DownloadedDependencyGraphNode {
|
||||||
|
/// The target of the package
|
||||||
|
pub target: Target,
|
||||||
|
/// The node
|
||||||
|
#[serde(flatten)]
|
||||||
|
pub node: DependencyGraphNode,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A graph of `DownloadedDependencyGraphNode`s
|
||||||
|
pub type DownloadedGraph = Graph<DownloadedDependencyGraphNode>;
|
||||||
|
|
||||||
/// A lockfile
|
/// A lockfile
|
||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||||
pub struct Lockfile {
|
pub struct Lockfile {
|
||||||
/// The name of the package
|
/// The name of the package
|
||||||
pub name: PackageName,
|
pub name: PackageName,
|
||||||
/// The version of the package
|
/// The version of the package
|
||||||
pub version: Version,
|
pub version: Version,
|
||||||
/// The target of the package
|
/// The target of the package
|
||||||
pub target: TargetKind,
|
pub target: TargetKind,
|
||||||
/// The overrides of the package
|
/// The overrides of the package
|
||||||
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
||||||
pub overrides: BTreeMap<OverrideKey, DependencySpecifiers>,
|
pub overrides: BTreeMap<OverrideKey, DependencySpecifiers>,
|
||||||
|
|
||||||
/// The workspace members
|
/// The workspace members
|
||||||
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
||||||
pub workspace: BTreeMap<PackageName, BTreeMap<TargetKind, RelativePathBuf>>,
|
pub workspace: BTreeMap<PackageName, BTreeMap<TargetKind, RelativePathBuf>>,
|
||||||
|
|
||||||
/// The graph of dependencies
|
/// The graph of dependencies
|
||||||
#[serde(default, skip_serializing_if = "DependencyGraph::is_empty")]
|
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
||||||
pub graph: DependencyGraph,
|
pub graph: DownloadedGraph,
|
||||||
}
|
|
||||||
|
|
||||||
/// Old lockfile stuff. Will be removed in a future version.
|
|
||||||
#[deprecated(
|
|
||||||
note = "Intended to be used to migrate old lockfiles to the new format. Will be removed in a future version."
|
|
||||||
)]
|
|
||||||
pub mod old {
|
|
||||||
use crate::{
|
|
||||||
manifest::{
|
|
||||||
overrides::OverrideKey,
|
|
||||||
target::{Target, TargetKind},
|
|
||||||
Alias, DependencyType,
|
|
||||||
},
|
|
||||||
names::{PackageName, PackageNames},
|
|
||||||
source::{
|
|
||||||
ids::{PackageId, VersionId},
|
|
||||||
refs::PackageRefs,
|
|
||||||
specifiers::DependencySpecifiers,
|
|
||||||
},
|
|
||||||
};
|
|
||||||
use relative_path::RelativePathBuf;
|
|
||||||
use semver::Version;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::collections::BTreeMap;
|
|
||||||
|
|
||||||
/// An old dependency graph node
|
|
||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
||||||
pub struct DependencyGraphNodeOld {
|
|
||||||
/// The alias, specifier, and original (as in the manifest) type for the dependency, if it is a direct dependency (i.e. used by the current project)
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
pub direct: Option<(Alias, DependencySpecifiers, DependencyType)>,
|
|
||||||
/// The dependencies of the package
|
|
||||||
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
|
||||||
pub dependencies: BTreeMap<PackageNames, (VersionId, Alias)>,
|
|
||||||
/// The resolved (transformed, for example Peer -> Standard) type of the dependency
|
|
||||||
pub resolved_ty: DependencyType,
|
|
||||||
/// Whether the resolved type should be Peer if this isn't depended on
|
|
||||||
#[serde(default, skip_serializing_if = "std::ops::Not::not")]
|
|
||||||
pub is_peer: bool,
|
|
||||||
/// The package reference
|
|
||||||
pub pkg_ref: PackageRefs,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// A downloaded dependency graph node, i.e. a `DependencyGraphNode` with a `Target`
|
|
||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
||||||
pub struct DownloadedDependencyGraphNodeOld {
|
|
||||||
/// The target of the package
|
|
||||||
pub target: Target,
|
|
||||||
/// The node
|
|
||||||
#[serde(flatten)]
|
|
||||||
pub node: DependencyGraphNodeOld,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// An old version of a lockfile
|
|
||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
||||||
pub struct LockfileOld {
|
|
||||||
/// The name of the package
|
|
||||||
pub name: PackageName,
|
|
||||||
/// The version of the package
|
|
||||||
pub version: Version,
|
|
||||||
/// The target of the package
|
|
||||||
pub target: TargetKind,
|
|
||||||
/// The overrides of the package
|
|
||||||
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
|
||||||
pub overrides: BTreeMap<OverrideKey, DependencySpecifiers>,
|
|
||||||
|
|
||||||
/// The workspace members
|
|
||||||
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
|
||||||
pub workspace: BTreeMap<PackageName, BTreeMap<TargetKind, RelativePathBuf>>,
|
|
||||||
|
|
||||||
/// The graph of dependencies
|
|
||||||
#[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
|
|
||||||
pub graph: BTreeMap<PackageNames, BTreeMap<VersionId, DownloadedDependencyGraphNodeOld>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl LockfileOld {
|
|
||||||
/// Converts this lockfile to a new lockfile
|
|
||||||
pub fn to_new(self) -> super::Lockfile {
|
|
||||||
super::Lockfile {
|
|
||||||
name: self.name,
|
|
||||||
version: self.version,
|
|
||||||
target: self.target,
|
|
||||||
overrides: self.overrides,
|
|
||||||
workspace: self.workspace,
|
|
||||||
graph: self
|
|
||||||
.graph
|
|
||||||
.into_iter()
|
|
||||||
.flat_map(|(name, versions)| {
|
|
||||||
versions.into_iter().map(move |(version, node)| {
|
|
||||||
(
|
|
||||||
PackageId(name.clone(), version),
|
|
||||||
crate::graph::DependencyGraphNode {
|
|
||||||
direct: node.node.direct,
|
|
||||||
dependencies: node
|
|
||||||
.node
|
|
||||||
.dependencies
|
|
||||||
.into_iter()
|
|
||||||
.map(|(name, (version, alias))| {
|
|
||||||
(PackageId(name, version), alias)
|
|
||||||
})
|
|
||||||
.collect(),
|
|
||||||
resolved_ty: node.node.resolved_ty,
|
|
||||||
is_peer: node.node.is_peer,
|
|
||||||
pkg_ref: node.node.pkg_ref,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
})
|
|
||||||
})
|
|
||||||
.collect(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
522
src/main.rs
522
src/main.rs
|
@ -1,340 +1,326 @@
|
||||||
#[cfg(feature = "version-management")]
|
#[cfg(feature = "version-management")]
|
||||||
use crate::cli::version::{check_for_updates, current_version, get_or_download_engine};
|
use crate::cli::version::{check_for_updates, get_or_download_version, TagInfo};
|
||||||
use crate::cli::{auth::get_tokens, display_err, home_dir, HOME_DIR};
|
use crate::cli::{auth::get_tokens, display_err, home_dir, HOME_DIR};
|
||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use clap::{builder::styling::AnsiColor, Parser};
|
use clap::{builder::styling::AnsiColor, Parser};
|
||||||
use fs_err::tokio as fs;
|
use fs_err::tokio as fs;
|
||||||
use indicatif::MultiProgress;
|
use pesde::{matching_globs, AuthConfig, Project, MANIFEST_FILE_NAME};
|
||||||
use pesde::{engine::EngineKind, find_roots, version_matches, AuthConfig, Project};
|
|
||||||
use semver::VersionReq;
|
|
||||||
use std::{
|
use std::{
|
||||||
io,
|
collections::HashSet,
|
||||||
path::{Path, PathBuf},
|
path::{Path, PathBuf},
|
||||||
str::FromStr,
|
|
||||||
sync::Mutex,
|
|
||||||
};
|
};
|
||||||
use tempfile::NamedTempFile;
|
use tempfile::NamedTempFile;
|
||||||
use tracing::instrument;
|
use tracing::instrument;
|
||||||
|
use tracing_indicatif::{filter::IndicatifFilter, IndicatifLayer};
|
||||||
use tracing_subscriber::{
|
use tracing_subscriber::{
|
||||||
filter::LevelFilter, fmt::MakeWriter, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter,
|
filter::LevelFilter, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter, Layer,
|
||||||
};
|
};
|
||||||
|
|
||||||
mod cli;
|
mod cli;
|
||||||
pub mod util;
|
pub mod util;
|
||||||
|
|
||||||
const STYLES: clap::builder::Styles = clap::builder::Styles::styled()
|
const STYLES: clap::builder::Styles = clap::builder::Styles::styled()
|
||||||
.header(AnsiColor::Yellow.on_default().underline())
|
.header(AnsiColor::Yellow.on_default().underline())
|
||||||
.usage(AnsiColor::Yellow.on_default().underline())
|
.usage(AnsiColor::Yellow.on_default().underline())
|
||||||
.literal(AnsiColor::Green.on_default().bold())
|
.literal(AnsiColor::Green.on_default().bold())
|
||||||
.placeholder(AnsiColor::Cyan.on_default());
|
.placeholder(AnsiColor::Cyan.on_default());
|
||||||
|
|
||||||
#[derive(Parser, Debug)]
|
#[derive(Parser, Debug)]
|
||||||
#[clap(
|
#[clap(
|
||||||
version,
|
version,
|
||||||
about = "A package manager for the Luau programming language",
|
about = "A package manager for the Luau programming language",
|
||||||
long_about = "A package manager for the Luau programming language, supporting multiple runtimes including Roblox and Lune"
|
long_about = "A package manager for the Luau programming language, supporting multiple runtimes including Roblox and Lune"
|
||||||
)]
|
)]
|
||||||
#[command(disable_version_flag = true, styles = STYLES)]
|
#[command(disable_version_flag = true, styles = STYLES)]
|
||||||
struct Cli {
|
struct Cli {
|
||||||
/// Print version
|
/// Print version
|
||||||
#[arg(short = 'v', short_alias = 'V', long, action = clap::builder::ArgAction::Version)]
|
#[arg(short = 'v', short_alias = 'V', long, action = clap::builder::ArgAction::Version)]
|
||||||
version: (),
|
version: (),
|
||||||
|
|
||||||
#[command(subcommand)]
|
#[command(subcommand)]
|
||||||
subcommand: cli::commands::Subcommand,
|
subcommand: cli::commands::Subcommand,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(level = "trace")]
|
#[instrument(level = "trace")]
|
||||||
async fn get_linkable_dir(path: &Path) -> PathBuf {
|
async fn get_linkable_dir(path: &Path) -> PathBuf {
|
||||||
let mut curr_path = PathBuf::new();
|
let mut curr_path = PathBuf::new();
|
||||||
let file_to_try = NamedTempFile::new_in(path).expect("failed to create temporary file");
|
let file_to_try = NamedTempFile::new_in(path).expect("failed to create temporary file");
|
||||||
|
|
||||||
let temp_path = tempfile::Builder::new()
|
let temp_path = tempfile::Builder::new()
|
||||||
.make(|_| Ok(()))
|
.make(|_| Ok(()))
|
||||||
.expect("failed to create temporary file")
|
.expect("failed to create temporary file")
|
||||||
.into_temp_path();
|
.into_temp_path();
|
||||||
let temp_file_name = temp_path.file_name().expect("failed to get file name");
|
let temp_file_name = temp_path.file_name().expect("failed to get file name");
|
||||||
|
|
||||||
// C: and \ are different components on Windows
|
// C: and \ are different components on Windows
|
||||||
#[cfg(windows)]
|
#[cfg(windows)]
|
||||||
let components = path.components().map(|c| {
|
let components = path.components().map(|c| {
|
||||||
let mut path = c.as_os_str().to_os_string();
|
let mut path = c.as_os_str().to_os_string();
|
||||||
if let std::path::Component::Prefix(_) = c {
|
if let std::path::Component::Prefix(_) = c {
|
||||||
path.push(std::path::MAIN_SEPARATOR_STR);
|
path.push(std::path::MAIN_SEPARATOR_STR);
|
||||||
}
|
}
|
||||||
|
|
||||||
path
|
path
|
||||||
});
|
});
|
||||||
#[cfg(not(windows))]
|
#[cfg(not(windows))]
|
||||||
let components = path.components().map(|c| c.as_os_str().to_os_string());
|
let components = path.components().map(|c| c.as_os_str().to_os_string());
|
||||||
|
|
||||||
for component in components {
|
for component in components {
|
||||||
curr_path.push(component);
|
curr_path.push(component);
|
||||||
|
|
||||||
let try_path = curr_path.join(temp_file_name);
|
let try_path = curr_path.join(temp_file_name);
|
||||||
|
|
||||||
if fs::hard_link(file_to_try.path(), &try_path).await.is_ok() {
|
if fs::hard_link(file_to_try.path(), &try_path).await.is_ok() {
|
||||||
if let Err(err) = fs::remove_file(&try_path).await {
|
if let Err(err) = fs::remove_file(&try_path).await {
|
||||||
tracing::warn!(
|
tracing::warn!(
|
||||||
"failed to remove temporary file at {}: {err}",
|
"failed to remove temporary file at {}: {err}",
|
||||||
try_path.display()
|
try_path.display()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
return curr_path;
|
return curr_path;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
panic!(
|
panic!(
|
||||||
"couldn't find a linkable directory for any point in {}",
|
"couldn't find a linkable directory for any point in {}",
|
||||||
curr_path.display()
|
curr_path.display()
|
||||||
);
|
);
|
||||||
}
|
|
||||||
|
|
||||||
pub static PROGRESS_BARS: Mutex<Option<MultiProgress>> = Mutex::new(None);
|
|
||||||
|
|
||||||
#[derive(Clone, Copy)]
|
|
||||||
pub struct IndicatifWriter;
|
|
||||||
|
|
||||||
impl IndicatifWriter {
|
|
||||||
fn suspend<F: FnOnce() -> R, R>(f: F) -> R {
|
|
||||||
match *PROGRESS_BARS.lock().unwrap() {
|
|
||||||
Some(ref progress_bars) => progress_bars.suspend(f),
|
|
||||||
None => f(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl io::Write for IndicatifWriter {
|
|
||||||
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
|
|
||||||
Self::suspend(|| io::stderr().write(buf))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn flush(&mut self) -> io::Result<()> {
|
|
||||||
Self::suspend(|| io::stderr().flush())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn write_vectored(&mut self, bufs: &[io::IoSlice<'_>]) -> io::Result<usize> {
|
|
||||||
Self::suspend(|| io::stderr().write_vectored(bufs))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn write_all(&mut self, buf: &[u8]) -> io::Result<()> {
|
|
||||||
Self::suspend(|| io::stderr().write_all(buf))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn write_fmt(&mut self, fmt: std::fmt::Arguments<'_>) -> io::Result<()> {
|
|
||||||
Self::suspend(|| io::stderr().write_fmt(fmt))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a> MakeWriter<'a> for IndicatifWriter {
|
|
||||||
type Writer = IndicatifWriter;
|
|
||||||
|
|
||||||
fn make_writer(&'a self) -> Self::Writer {
|
|
||||||
*self
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn run() -> anyhow::Result<()> {
|
async fn run() -> anyhow::Result<()> {
|
||||||
let cwd = std::env::current_dir().expect("failed to get current working directory");
|
let cwd = std::env::current_dir().expect("failed to get current working directory");
|
||||||
// Unix doesn't return the symlinked path, so we need to get it from the 0 argument
|
|
||||||
#[cfg(unix)]
|
|
||||||
let current_exe = PathBuf::from(std::env::args_os().next().expect("argument 0 not set"));
|
|
||||||
#[cfg(not(unix))]
|
|
||||||
let current_exe = std::env::current_exe().expect("failed to get current executable path");
|
|
||||||
let exe_name = current_exe
|
|
||||||
.file_stem()
|
|
||||||
.unwrap()
|
|
||||||
.to_str()
|
|
||||||
.expect("exe name is not valid utf-8");
|
|
||||||
let exe_name_engine = EngineKind::from_str(exe_name);
|
|
||||||
|
|
||||||
#[cfg(windows)]
|
#[cfg(windows)]
|
||||||
'scripts: {
|
'scripts: {
|
||||||
// if we're an engine, we don't want to run any scripts
|
let exe = std::env::current_exe().expect("failed to get current executable path");
|
||||||
if exe_name_engine.is_ok() {
|
if exe.parent().is_some_and(|parent| {
|
||||||
break 'scripts;
|
parent.file_name().is_some_and(|parent| parent != "bin")
|
||||||
}
|
|| parent
|
||||||
|
.parent()
|
||||||
|
.and_then(|parent| parent.file_name())
|
||||||
|
.is_some_and(|parent| parent != HOME_DIR)
|
||||||
|
}) {
|
||||||
|
break 'scripts;
|
||||||
|
}
|
||||||
|
|
||||||
if let Some(bin_folder) = current_exe.parent() {
|
let exe_name = exe.file_name().unwrap().to_string_lossy();
|
||||||
// we're not in {path}/bin/{exe}
|
let exe_name = exe_name
|
||||||
if bin_folder.file_name().is_some_and(|parent| parent != "bin") {
|
.strip_suffix(std::env::consts::EXE_SUFFIX)
|
||||||
break 'scripts;
|
.unwrap_or(&exe_name);
|
||||||
}
|
|
||||||
|
|
||||||
// we're not in {path}/.pesde/bin/{exe}
|
if exe_name == env!("CARGO_BIN_NAME") {
|
||||||
if bin_folder
|
break 'scripts;
|
||||||
.parent()
|
}
|
||||||
.and_then(|home_folder| home_folder.file_name())
|
|
||||||
.is_some_and(|home_folder| home_folder != HOME_DIR)
|
|
||||||
{
|
|
||||||
break 'scripts;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// the bin script will search for the project root itself, so we do that to ensure
|
// the bin script will search for the project root itself, so we do that to ensure
|
||||||
// consistency across platforms, since the script is executed using a shebang
|
// consistency across platforms, since the script is executed using a shebang
|
||||||
// on unix systems
|
// on unix systems
|
||||||
let status = std::process::Command::new("lune")
|
let status = std::process::Command::new("lune")
|
||||||
.arg("run")
|
.arg("run")
|
||||||
.arg(
|
.arg(
|
||||||
current_exe
|
exe.parent()
|
||||||
.parent()
|
.map(|p| p.join(".impl").join(exe.file_name().unwrap()))
|
||||||
.unwrap_or(¤t_exe)
|
.unwrap_or(exe)
|
||||||
.join(".impl")
|
.with_extension("luau"),
|
||||||
.join(current_exe.file_name().unwrap())
|
)
|
||||||
.with_extension("luau"),
|
.arg("--")
|
||||||
)
|
.args(std::env::args_os().skip(1))
|
||||||
.arg("--")
|
.current_dir(cwd)
|
||||||
.args(std::env::args_os().skip(1))
|
.status()
|
||||||
.current_dir(cwd)
|
.expect("failed to run lune");
|
||||||
.status()
|
|
||||||
.expect("failed to run lune");
|
|
||||||
|
|
||||||
std::process::exit(status.code().unwrap());
|
std::process::exit(status.code().unwrap());
|
||||||
}
|
}
|
||||||
|
|
||||||
let tracing_env_filter = EnvFilter::builder()
|
let indicatif_layer = IndicatifLayer::new().with_filter(IndicatifFilter::new(false));
|
||||||
.with_default_directive(LevelFilter::INFO.into())
|
|
||||||
.from_env_lossy()
|
|
||||||
.add_directive("reqwest=info".parse().unwrap())
|
|
||||||
.add_directive("rustls=info".parse().unwrap())
|
|
||||||
.add_directive("tokio_util=info".parse().unwrap())
|
|
||||||
.add_directive("goblin=info".parse().unwrap())
|
|
||||||
.add_directive("tower=info".parse().unwrap())
|
|
||||||
.add_directive("hyper=info".parse().unwrap())
|
|
||||||
.add_directive("h2=info".parse().unwrap());
|
|
||||||
|
|
||||||
let fmt_layer = tracing_subscriber::fmt::layer().with_writer(IndicatifWriter);
|
let tracing_env_filter = EnvFilter::builder()
|
||||||
|
.with_default_directive(LevelFilter::INFO.into())
|
||||||
|
.from_env_lossy()
|
||||||
|
.add_directive("reqwest=info".parse().unwrap())
|
||||||
|
.add_directive("rustls=info".parse().unwrap())
|
||||||
|
.add_directive("tokio_util=info".parse().unwrap())
|
||||||
|
.add_directive("goblin=info".parse().unwrap())
|
||||||
|
.add_directive("tower=info".parse().unwrap())
|
||||||
|
.add_directive("hyper=info".parse().unwrap())
|
||||||
|
.add_directive("h2=info".parse().unwrap());
|
||||||
|
|
||||||
#[cfg(debug_assertions)]
|
let fmt_layer =
|
||||||
let fmt_layer = fmt_layer.with_timer(tracing_subscriber::fmt::time::uptime());
|
tracing_subscriber::fmt::layer().with_writer(indicatif_layer.inner().get_stderr_writer());
|
||||||
|
|
||||||
#[cfg(not(debug_assertions))]
|
#[cfg(debug_assertions)]
|
||||||
let fmt_layer = fmt_layer
|
let fmt_layer = fmt_layer.with_timer(tracing_subscriber::fmt::time::uptime());
|
||||||
.pretty()
|
|
||||||
.with_timer(())
|
|
||||||
.with_line_number(false)
|
|
||||||
.with_file(false)
|
|
||||||
.with_target(false);
|
|
||||||
|
|
||||||
tracing_subscriber::registry()
|
#[cfg(not(debug_assertions))]
|
||||||
.with(tracing_env_filter)
|
let fmt_layer = fmt_layer
|
||||||
.with(fmt_layer)
|
.pretty()
|
||||||
.init();
|
.with_timer(())
|
||||||
|
.with_line_number(false)
|
||||||
|
.with_file(false)
|
||||||
|
.with_target(false);
|
||||||
|
|
||||||
let (project_root_dir, project_workspace_dir) = find_roots(cwd.clone())
|
tracing_subscriber::registry()
|
||||||
.await
|
.with(tracing_env_filter)
|
||||||
.context("failed to find project root")?;
|
.with(fmt_layer)
|
||||||
|
.with(indicatif_layer)
|
||||||
|
.init();
|
||||||
|
|
||||||
tracing::trace!(
|
let (project_root_dir, project_workspace_dir) = 'finder: {
|
||||||
"project root: {}\nworkspace root: {}",
|
let mut current_path = Some(cwd.clone());
|
||||||
project_root_dir.display(),
|
let mut project_root = None::<PathBuf>;
|
||||||
project_workspace_dir
|
let mut workspace_dir = None::<PathBuf>;
|
||||||
.as_ref()
|
|
||||||
.map_or("none".to_string(), |p| p.display().to_string())
|
|
||||||
);
|
|
||||||
|
|
||||||
let home_dir = home_dir()?;
|
async fn get_workspace_members(path: &Path) -> anyhow::Result<HashSet<PathBuf>> {
|
||||||
let data_dir = home_dir.join("data");
|
let manifest = fs::read_to_string(path.join(MANIFEST_FILE_NAME))
|
||||||
fs::create_dir_all(&data_dir)
|
.await
|
||||||
.await
|
.context("failed to read manifest")?;
|
||||||
.expect("failed to create data directory");
|
let manifest: pesde::manifest::Manifest =
|
||||||
|
toml::from_str(&manifest).context("failed to parse manifest")?;
|
||||||
|
|
||||||
let cas_dir = get_linkable_dir(&project_root_dir).await.join(HOME_DIR);
|
if manifest.workspace_members.is_empty() {
|
||||||
|
return Ok(HashSet::new());
|
||||||
|
}
|
||||||
|
|
||||||
let cas_dir = if cas_dir == home_dir {
|
matching_globs(
|
||||||
&data_dir
|
path,
|
||||||
} else {
|
manifest.workspace_members.iter().map(|s| s.as_str()),
|
||||||
&cas_dir
|
false,
|
||||||
}
|
false,
|
||||||
.join("cas");
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to get workspace members")
|
||||||
|
}
|
||||||
|
|
||||||
tracing::debug!("using cas dir in {}", cas_dir.display());
|
while let Some(path) = current_path {
|
||||||
|
current_path = path.parent().map(|p| p.to_path_buf());
|
||||||
|
|
||||||
let project = Project::new(
|
if !path.join(MANIFEST_FILE_NAME).exists() {
|
||||||
project_root_dir,
|
continue;
|
||||||
project_workspace_dir,
|
}
|
||||||
data_dir,
|
|
||||||
cas_dir,
|
|
||||||
AuthConfig::new().with_tokens(get_tokens().await?.0),
|
|
||||||
);
|
|
||||||
|
|
||||||
let reqwest = {
|
match (project_root.as_ref(), workspace_dir.as_ref()) {
|
||||||
let mut headers = reqwest::header::HeaderMap::new();
|
(Some(project_root), Some(workspace_dir)) => {
|
||||||
|
break 'finder (project_root.clone(), Some(workspace_dir.clone()));
|
||||||
|
}
|
||||||
|
|
||||||
headers.insert(
|
(Some(project_root), None) => {
|
||||||
reqwest::header::ACCEPT,
|
if get_workspace_members(&path).await?.contains(project_root) {
|
||||||
"application/json"
|
workspace_dir = Some(path);
|
||||||
.parse()
|
}
|
||||||
.context("failed to create accept header")?,
|
}
|
||||||
);
|
|
||||||
|
|
||||||
reqwest::Client::builder()
|
(None, None) => {
|
||||||
.user_agent(concat!(
|
if get_workspace_members(&path).await?.contains(&cwd) {
|
||||||
env!("CARGO_PKG_NAME"),
|
// initializing a new member of a workspace
|
||||||
"/",
|
break 'finder (cwd, Some(path));
|
||||||
env!("CARGO_PKG_VERSION")
|
} else {
|
||||||
))
|
project_root = Some(path);
|
||||||
.default_headers(headers)
|
}
|
||||||
.build()?
|
}
|
||||||
};
|
|
||||||
|
|
||||||
#[cfg(feature = "version-management")]
|
(None, Some(_)) => unreachable!(),
|
||||||
'engines: {
|
}
|
||||||
let Ok(engine) = exe_name_engine else {
|
}
|
||||||
break 'engines;
|
|
||||||
};
|
|
||||||
|
|
||||||
let req = match project.deser_manifest().await {
|
// we mustn't expect the project root to be found, as that would
|
||||||
Ok(mut manifest) => manifest.engines.remove(&engine),
|
// disable the ability to run pesde in a non-project directory (for example to init it)
|
||||||
Err(pesde::errors::ManifestReadError::Io(e)) if e.kind() == io::ErrorKind::NotFound => {
|
(project_root.unwrap_or_else(|| cwd.clone()), workspace_dir)
|
||||||
None
|
};
|
||||||
}
|
|
||||||
Err(e) => return Err(e.into()),
|
|
||||||
};
|
|
||||||
|
|
||||||
if engine == EngineKind::Pesde {
|
tracing::trace!(
|
||||||
match &req {
|
"project root: {}\nworkspace root: {}",
|
||||||
// we're already running a compatible version
|
project_root_dir.display(),
|
||||||
Some(req) if version_matches(req, ¤t_version()) => break 'engines,
|
project_workspace_dir
|
||||||
// the user has not requested a specific version, so we'll just use the current one
|
.as_ref()
|
||||||
None => break 'engines,
|
.map_or("none".to_string(), |p| p.display().to_string())
|
||||||
_ => (),
|
);
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let exe_path =
|
let home_dir = home_dir()?;
|
||||||
get_or_download_engine(&reqwest, engine, req.unwrap_or(VersionReq::STAR)).await?;
|
let data_dir = home_dir.join("data");
|
||||||
if exe_path == current_exe {
|
fs::create_dir_all(&data_dir)
|
||||||
anyhow::bail!("engine linker executed by itself")
|
.await
|
||||||
}
|
.expect("failed to create data directory");
|
||||||
|
|
||||||
let status = std::process::Command::new(exe_path)
|
let cas_dir = get_linkable_dir(&project_root_dir).await.join(HOME_DIR);
|
||||||
.args(std::env::args_os().skip(1))
|
|
||||||
.status()
|
|
||||||
.expect("failed to run new version");
|
|
||||||
|
|
||||||
std::process::exit(status.code().unwrap());
|
let cas_dir = if cas_dir == home_dir {
|
||||||
}
|
&data_dir
|
||||||
|
} else {
|
||||||
|
&cas_dir
|
||||||
|
}
|
||||||
|
.join("cas");
|
||||||
|
|
||||||
#[cfg(feature = "version-management")]
|
tracing::debug!("using cas dir in {}", cas_dir.display());
|
||||||
display_err(
|
|
||||||
check_for_updates(&reqwest).await,
|
|
||||||
" while checking for updates",
|
|
||||||
);
|
|
||||||
|
|
||||||
let cli = Cli::parse();
|
let project = Project::new(
|
||||||
|
project_root_dir,
|
||||||
|
project_workspace_dir,
|
||||||
|
data_dir,
|
||||||
|
cas_dir,
|
||||||
|
AuthConfig::new().with_tokens(get_tokens().await?.0),
|
||||||
|
);
|
||||||
|
|
||||||
cli.subcommand.run(project, reqwest).await
|
let reqwest = {
|
||||||
|
let mut headers = reqwest::header::HeaderMap::new();
|
||||||
|
|
||||||
|
headers.insert(
|
||||||
|
reqwest::header::ACCEPT,
|
||||||
|
"application/json"
|
||||||
|
.parse()
|
||||||
|
.context("failed to create accept header")?,
|
||||||
|
);
|
||||||
|
|
||||||
|
reqwest::Client::builder()
|
||||||
|
.user_agent(concat!(
|
||||||
|
env!("CARGO_PKG_NAME"),
|
||||||
|
"/",
|
||||||
|
env!("CARGO_PKG_VERSION")
|
||||||
|
))
|
||||||
|
.default_headers(headers)
|
||||||
|
.build()?
|
||||||
|
};
|
||||||
|
|
||||||
|
#[cfg(feature = "version-management")]
|
||||||
|
{
|
||||||
|
let target_version = project
|
||||||
|
.deser_manifest()
|
||||||
|
.await
|
||||||
|
.ok()
|
||||||
|
.and_then(|manifest| manifest.pesde_version);
|
||||||
|
|
||||||
|
let exe_path = if let Some(version) = target_version {
|
||||||
|
get_or_download_version(&reqwest, &TagInfo::Incomplete(version), false).await?
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
};
|
||||||
|
|
||||||
|
if let Some(exe_path) = exe_path {
|
||||||
|
let status = std::process::Command::new(exe_path)
|
||||||
|
.args(std::env::args_os().skip(1))
|
||||||
|
.status()
|
||||||
|
.expect("failed to run new version");
|
||||||
|
|
||||||
|
std::process::exit(status.code().unwrap());
|
||||||
|
}
|
||||||
|
|
||||||
|
display_err(
|
||||||
|
check_for_updates(&reqwest).await,
|
||||||
|
" while checking for updates",
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let cli = Cli::parse();
|
||||||
|
|
||||||
|
cli.subcommand.run(project, reqwest).await
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
let result = run().await;
|
let result = run().await;
|
||||||
let is_err = result.is_err();
|
let is_err = result.is_err();
|
||||||
display_err(result, "");
|
display_err(result, "");
|
||||||
if is_err {
|
if is_err {
|
||||||
std::process::exit(1);
|
std::process::exit(1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Some files were not shown because too many files have changed in this diff Show more
Loading…
Add table
Reference in a new issue