| 
1 | 1 | #include "create_secret_functions.hpp"  | 
2 | 2 | #include "s3fs.hpp"  | 
3 |  | -#include "duckdb/main/extension_util.hpp"  | 
 | 3 | +#include "duckdb/main/extension/extension_loader.hpp"  | 
4 | 4 | #include "duckdb/common/local_file_system.hpp"  | 
5 | 5 | 
 
  | 
6 | 6 | namespace duckdb {  | 
7 | 7 | 
 
  | 
8 |  | -void CreateS3SecretFunctions::Register(DatabaseInstance &instance) {  | 
9 |  | -	RegisterCreateSecretFunction(instance, "s3");  | 
10 |  | -	RegisterCreateSecretFunction(instance, "aws");  | 
11 |  | -	RegisterCreateSecretFunction(instance, "r2");  | 
12 |  | -	RegisterCreateSecretFunction(instance, "gcs");  | 
 | 8 | +void CreateS3SecretFunctions::Register(ExtensionLoader &loader) {  | 
 | 9 | +	RegisterCreateSecretFunction(loader, "s3");  | 
 | 10 | +	RegisterCreateSecretFunction(loader, "aws");  | 
 | 11 | +	RegisterCreateSecretFunction(loader, "r2");  | 
 | 12 | +	RegisterCreateSecretFunction(loader, "gcs");  | 
13 | 13 | }  | 
14 | 14 | 
 
  | 
15 | 15 | static Value MapToStruct(const Value &map) {  | 
@@ -220,39 +220,39 @@ void CreateS3SecretFunctions::SetBaseNamedParams(CreateSecretFunction &function,  | 
220 | 220 | 	}  | 
221 | 221 | }  | 
222 | 222 | 
 
  | 
223 |  | -void CreateS3SecretFunctions::RegisterCreateSecretFunction(DatabaseInstance &instance, string type) {  | 
 | 223 | +void CreateS3SecretFunctions::RegisterCreateSecretFunction(ExtensionLoader &loader, string type) {  | 
224 | 224 | 	// Register the new type  | 
225 | 225 | 	SecretType secret_type;  | 
226 | 226 | 	secret_type.name = type;  | 
227 | 227 | 	secret_type.deserializer = KeyValueSecret::Deserialize<KeyValueSecret>;  | 
228 | 228 | 	secret_type.default_provider = "config";  | 
229 | 229 | 	secret_type.extension = "httpfs";  | 
230 | 230 | 
 
  | 
231 |  | -	ExtensionUtil::RegisterSecretType(instance, secret_type);  | 
 | 231 | +	loader.RegisterSecretType(secret_type);  | 
232 | 232 | 
 
  | 
233 | 233 | 	CreateSecretFunction from_empty_config_fun2 = {type, "config", CreateS3SecretFromConfig};  | 
234 | 234 | 	SetBaseNamedParams(from_empty_config_fun2, type);  | 
235 |  | -	ExtensionUtil::RegisterFunction(instance, from_empty_config_fun2);  | 
 | 235 | +	loader.RegisterFunction(from_empty_config_fun2);  | 
236 | 236 | }  | 
237 | 237 | 
 
  | 
238 |  | -void CreateBearerTokenFunctions::Register(DatabaseInstance &instance) {  | 
 | 238 | +void CreateBearerTokenFunctions::Register(ExtensionLoader &loader) {  | 
239 | 239 | 	// HuggingFace secret  | 
240 | 240 | 	SecretType secret_type_hf;  | 
241 | 241 | 	secret_type_hf.name = HUGGINGFACE_TYPE;  | 
242 | 242 | 	secret_type_hf.deserializer = KeyValueSecret::Deserialize<KeyValueSecret>;  | 
243 | 243 | 	secret_type_hf.default_provider = "config";  | 
244 | 244 | 	secret_type_hf.extension = "httpfs";  | 
245 |  | -	ExtensionUtil::RegisterSecretType(instance, secret_type_hf);  | 
 | 245 | +	loader.RegisterSecretType(secret_type_hf);  | 
246 | 246 | 
 
  | 
247 | 247 | 	// Huggingface config provider  | 
248 | 248 | 	CreateSecretFunction hf_config_fun = {HUGGINGFACE_TYPE, "config", CreateBearerSecretFromConfig};  | 
249 | 249 | 	hf_config_fun.named_parameters["token"] = LogicalType::VARCHAR;  | 
250 |  | -	ExtensionUtil::RegisterFunction(instance, hf_config_fun);  | 
 | 250 | +	loader.RegisterFunction(hf_config_fun);  | 
251 | 251 | 
 
  | 
252 | 252 | 	// Huggingface credential_chain provider  | 
253 | 253 | 	CreateSecretFunction hf_cred_fun = {HUGGINGFACE_TYPE, "credential_chain",  | 
254 | 254 | 	                                    CreateHuggingFaceSecretFromCredentialChain};  | 
255 |  | -	ExtensionUtil::RegisterFunction(instance, hf_cred_fun);  | 
 | 255 | +	loader.RegisterFunction(hf_cred_fun);  | 
256 | 256 | }  | 
257 | 257 | 
 
  | 
258 | 258 | unique_ptr<BaseSecret> CreateBearerTokenFunctions::CreateSecretFunctionInternal(ClientContext &context,  | 
 | 
0 commit comments