Add track_env_var to the proc macro server

This commit is contained in:
Laurențiu Nicola 2020-08-03 13:57:04 +03:00
parent e5f252ade7
commit 3d169bd3f4
6 changed files with 40 additions and 1 deletions

View File

@ -160,6 +160,7 @@ macro_rules! define_handles {
}
define_handles! {
'owned:
FreeFunctions,
TokenStream,
TokenStreamBuilder,
TokenStreamIter,

View File

@ -57,6 +57,10 @@ use std::thread;
macro_rules! with_api {
($S:ident, $self:ident, $m:ident) => {
$m! {
FreeFunctions {
fn drop($self: $S::FreeFunctions);
fn track_env_var(var: &str, value: Option<&str>);
},
TokenStream {
fn drop($self: $S::TokenStream);
fn clone($self: &$S::TokenStream) -> $S::TokenStream;

View File

@ -11,6 +11,8 @@ use super::client::HandleStore;
/// Declare an associated item of one of the traits below, optionally
/// adjusting it (i.e., adding bounds to types and default bodies to methods).
macro_rules! associated_item {
(type FreeFunctions) =>
(type FreeFunctions: 'static;);
(type TokenStream) =>
(type TokenStream: 'static + Clone;);
(type TokenStreamBuilder) =>

View File

@ -924,3 +924,25 @@ impl fmt::Debug for Literal {
self.0.fmt(f)
}
}
pub mod tracked_env {
use std::env::{self, VarError};
use std::ffi::OsStr;
/// Retrieve an environment variable and add it to build dependency info.
/// Build system executing the compiler will know that the variable was accessed during
/// compilation, and will be able to rerun the build when the value of that variable changes.
/// Besides the dependency tracking this function should be equivalent to `env::var` from the
/// standard library, except that the argument must be UTF-8.
pub fn var<K: AsRef<OsStr> + AsRef<str>>(key: K) -> Result<String, VarError> {
use std::ops::Deref;
let key: &str = key.as_ref();
let value = env::var(key);
super::bridge::client::FreeFunctions::track_env_var(
key,
value.as_ref().map(|t| t.deref()).ok(),
);
value
}
}

View File

@ -242,6 +242,8 @@ impl TokenStreamBuilder {
}
}
pub struct FreeFunctions;
#[derive(Clone)]
pub struct TokenStreamIter {
trees: IntoIter<TokenTree>,
@ -254,6 +256,7 @@ pub struct Rustc {
}
impl server::Types for Rustc {
type FreeFunctions = FreeFunctions;
type TokenStream = TokenStream;
type TokenStreamBuilder = TokenStreamBuilder;
type TokenStreamIter = TokenStreamIter;
@ -267,6 +270,13 @@ impl server::Types for Rustc {
type MultiSpan = Vec<Span>;
}
impl server::FreeFunctions for Rustc {
fn track_env_var(&mut self, _var: &str, _value: Option<&str>) {
// FIXME: track env var accesses
// https://github.com/rust-lang/rust/pull/71858
}
}
impl server::TokenStream for Rustc {
fn new(&mut self) -> Self::TokenStream {
Self::TokenStream::new()

View File

@ -7,7 +7,7 @@ use anyhow::{bail, format_err, Context, Result};
use crate::not_bash::{pushd, run};
// Latest stable, feel free to send a PR if this lags behind.
const REQUIRED_RUST_VERSION: u32 = 46;
const REQUIRED_RUST_VERSION: u32 = 47;
pub struct InstallCmd {
pub client: Option<ClientOpt>,