1use ek_core::{Field, SectionMap};
11
12#[derive(Debug, Clone)]
14pub struct AcctSpec {
15 pub broker_id: String,
16 pub user: String,
17 pub td_addr: String,
18 pub md_addr: String,
19 pub app_id: String,
20 pub auth_code: String,
21 pub classes: String,
23 pub md_user: String,
24 pub md_password: Option<String>,
26}
27
28pub fn derived_name(spec: &AcctSpec) -> Result<String, String> {
30 ek_ctp::derived_gateway_name(&spec.broker_id, &spec.user).ok_or_else(|| {
31 format!(
32 "派生注册名失败(broker={:?} 账号={:?}):字符不合名纪律",
33 spec.broker_id, spec.user
34 )
35 })
36}
37
38pub fn unique_gw_name(
53 taken: &[(String, String, String, String)],
54 broker: &str,
55 user: &str,
56 td_addr: &str,
57) -> Result<String, String> {
58 let broker = broker.trim();
59 let user = user.trim();
60 let td = td_addr.trim();
61 let base = ek_ctp::derived_gateway_name(broker, user)
62 .ok_or_else(|| format!("派生注册名失败(broker={broker:?} 账号={user:?}):字符不合名纪律"))?;
63 for (gw, b, u, t) in taken {
64 if b.trim() == broker && u.trim() == user && t.trim() == td {
65 return Err(format!(
66 "该 broker+账号+td 前置已在册(行 {gw},前置 {td});同前置不重复登记"
67 ));
68 }
69 }
70 if !taken.iter().any(|(gw, _, _, _)| gw == &base) {
71 return Ok(base);
72 }
73 for n in 2..=99u32 {
74 let cand = format!("{base}_{n}");
75 if cand.len() > 32 {
76 return Err(format!(
77 "同 broker+账号 的第 {n} 套前置超派生名 32 字符上限({cand})"
78 ));
79 }
80 if !taken.iter().any(|(gw, _, _, _)| gw == &cand) {
81 return Ok(cand);
82 }
83 }
84 Err("同 broker+账号 的前置套数已超 98 条(找不着后缀空位)".to_string())
85}
86
87pub fn taken_accounts(
92 store: &ek_cfg::Store,
93) -> Result<Vec<(String, String, String, String)>, String> {
94 let mut out = Vec::new();
95 for ns in store
96 .list_namespaces()
97 .map_err(|e| format!("档案目录列举失败: {e:?}"))?
98 {
99 let Some(gw) = ns.strip_prefix("ctp.") else {
100 continue;
101 };
102 if let Ok(ek_cfg::LoadOutcome::Loaded { map, .. }) = store.load(&ns) {
103 out.push((
104 gw.to_string(),
105 map.get_plain("broker_id").unwrap_or("").to_string(),
106 map.get_plain("td_user").unwrap_or("").to_string(),
107 map.get_plain("td_addr").unwrap_or("").to_string(),
108 ));
109 }
110 }
111 Ok(out)
112}
113
114pub fn write_account(
122 store: &ek_cfg::Store,
123 spec: &AcctSpec,
124 password: &str,
125) -> Result<String, String> {
126 if spec.broker_id.is_empty()
127 || spec.user.is_empty()
128 || spec.td_addr.is_empty()
129 || spec.app_id.is_empty()
130 || spec.auth_code.is_empty()
131 {
132 return Err("缺必填要素(broker/user/td 前置/app_id/auth_code)".to_string());
133 }
134 if password.trim().is_empty() {
135 return Err("密码为空:档案不接受无密码账户(受理校验同款)".to_string());
136 }
137 let gw = unique_gw_name(
138 &taken_accounts(store)?,
139 &spec.broker_id,
140 &spec.user,
141 &spec.td_addr,
142 )?;
143 let ns = format!("ctp.{gw}");
144 ek_cfg::secret::register(
145 &ns,
146 "gw_td_password",
147 ek_cfg::SecretValue::from_bytes(password.trim().to_owned()),
148 )
149 .map_err(|e| format!("密码登记密库失败: {e:?}"))?;
150 let mut map = SectionMap::new(1)
151 .set("broker_id", Field::plain(spec.broker_id.trim()))
152 .set("td_addr", Field::plain(spec.td_addr.trim()))
153 .set("td_user", Field::plain(spec.user.trim()))
154 .set("td_app_id", Field::plain(spec.app_id.trim()))
155 .set("td_auth_code", Field::plain(spec.auth_code.trim()))
156 .set("td_password", Field::secret("gw_td_password"));
157 if !spec.md_addr.trim().is_empty() {
158 map = map.set("md_addr", Field::plain(spec.md_addr.trim()));
159 if !spec.md_user.trim().is_empty() {
160 map = map.set("md_user", Field::plain(spec.md_user.trim()));
161 }
162 if let Some(mp) = spec.md_password.as_deref() {
163 let mp = mp.trim();
164 if !mp.is_empty() {
165 ek_cfg::secret::register(
166 &ns,
167 "gw_md_password",
168 ek_cfg::SecretValue::from_bytes(mp),
169 )
170 .map_err(|e| format!("行情密码登记密库失败: {e:?}"))?;
171 map = map.set("md_password", Field::secret("gw_md_password"));
172 }
173 }
174 }
175 let classes = spec.classes.trim();
176 if !classes.is_empty() {
177 for c in classes.split(',').map(str::trim) {
178 if c.chars().count() != 1 {
179 return Err(format!(
180 "product_classes 项 {c:?} 非单字符 CTP 码(D-PC-FILTER)"
181 ));
182 }
183 }
184 map = map.set("product_classes", Field::plain(classes));
185 }
186 store
187 .save(&ns, &map)
188 .map_err(|e| format!("档案落盘失败: {e:?}"))?;
189 Ok(gw)
190}
191
192pub fn purge_account_archive(store: &ek_cfg::Store, gw: &str) -> Result<(), String> {
204 purge_ns_direct(store, &format!("ctp.{gw}"))
205}
206
207pub fn purge_ns_direct(store: &ek_cfg::Store, ns: &str) -> Result<(), String> {
210 match store.load(ns) {
211 Ok(ek_cfg::LoadOutcome::Loaded { .. }) => {}
212 Ok(ek_cfg::LoadOutcome::Fresh) => return Ok(()),
213 Err(e) => return Err(format!("档案 {ns} 读不动: {e}")),
214 }
215 store
218 .purge(ns)
219 .map_err(|e| format!("档案 {ns} 清段失败: {e}"))?;
220 ek_cfg::secret::forget_ns(ns).map_err(|e| format!("秘库 {ns} 清理失败: {e}"))?;
221 Ok(())
222}