Function hyper_scripter::query::do_script_query
source · pub async fn do_script_query<'b>(
script_query: &ScriptQuery,
script_repo: &'b mut ScriptRepo,
finding_filtered: bool,
forbid_prompt: bool
) -> Result<Option<RepoEntry<'b>>>Examples found in repository?
src/query/util.rs (line 152)
146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167
pub async fn do_script_query_strict<'b>(
script_query: &ScriptQuery,
script_repo: &'b mut ScriptRepo,
) -> Result<RepoEntry<'b>> {
// FIXME: 一旦 NLL 進化就修掉這段 unsafe
let ptr = script_repo as *mut ScriptRepo;
if let Some(info) = do_script_query(script_query, script_repo, false, false).await? {
return Ok(info);
}
let script_repo = unsafe { &mut *ptr };
#[cfg(not(feature = "benching"))]
if !script_query.bang {
let filtered = do_script_query(script_query, script_repo, true, true).await?;
if let Some(mut filtered) = filtered {
filtered.update(|script| script.miss()).await?;
return Err(Error::ScriptIsFiltered(filtered.name.key().to_string()));
}
};
Err(Error::ScriptNotFound(script_query.to_string()))
}More examples
src/util/main_util.rs (line 72)
63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141
pub async fn edit_or_create(
edit_query: EditQuery<ScriptQuery>,
script_repo: &'_ mut ScriptRepo,
ty: Option<ScriptFullType>,
tags: EditTagArgs,
) -> Result<(PathBuf, RepoEntry<'_>, Option<ScriptType>)> {
let final_ty: ScriptFullType;
let (script_name, script_path) = if let EditQuery::Query(query) = edit_query {
match query::do_script_query(&query, script_repo, false, false).await {
// TODO: 手動測試文件?
Err(Error::DontFuzz) | Ok(None) => {
if tags.explicit_select {
return Err(RedundantOpt::Selector.into());
}
final_ty = ty.unwrap_or_default();
let name = query.into_script_name()?;
if script_repo.get_mut(&name, Visibility::All).is_some() {
log::error!("與被篩掉的腳本撞名");
return Err(Error::ScriptIsFiltered(name.to_string()));
}
log::debug!("打開新命名腳本:{:?}", name);
let p = path::open_script(&name, &final_ty.ty, None)
.context(format!("打開新命名腳本失敗:{:?}", name))?;
if p.exists() {
if p.is_dir() {
return Err(Error::PathExist(p).context("與目錄撞路徑"));
}
check_path_collision(&p, script_repo)?;
log::warn!("編輯野生腳本!");
} else {
// NOTE: 創建資料夾
if let Some(parent) = p.parent() {
super::handle_fs_res(&[&p], create_dir_all(parent))?;
}
}
(name, p)
}
Ok(Some(entry)) => {
if ty.is_some() {
return Err(RedundantOpt::Type.into());
}
if tags.explicit_tag {
return Err(RedundantOpt::Tag.into());
}
log::debug!("打開既有命名腳本:{:?}", entry.name);
let p = path::open_script(&entry.name, &entry.ty, Some(true))
.context(format!("打開命名腳本失敗:{:?}", entry.name))?;
// NOTE: 直接返回
// FIXME: 一旦 NLL 進化就修掉這段雙重詢問
// return Ok((p, entry));
let n = entry.name.clone();
return Ok((p, script_repo.get_mut(&n, Visibility::All).unwrap(), None));
}
Err(e) => return Err(e),
}
} else {
if tags.explicit_select {
return Err(RedundantOpt::Selector.into());
}
final_ty = ty.unwrap_or_default();
log::debug!("打開新匿名腳本");
path::open_new_anonymous(&final_ty.ty).context("打開新匿名腳本失敗")?
};
log::info!("編輯 {:?}", script_name);
let ScriptFullType { ty, sub } = final_ty;
// 這裡的 or_insert 其實永遠會發生,所以無需用閉包來傳
let entry = script_repo
.entry(&script_name)
.or_insert(
ScriptInfo::builder(0, script_name, ty, tags.content.into_allowed_iter()).build(),
)
.await?;
Ok((script_path, entry, sub))
}