Haxio/libpb
Haxio
/
libpb
mirror of https://github.com/Hax-io/libpb
1
0
Fork 0
Code Issues Packages Projects Releases Activity

- Added authentication token support on a per-instance basis, this adds `setAuthToken(string)` and `string getAuthToken()` 

- Authentication tokens that are `""` (empty) will not cause headers to be added
This commit is contained in:
Tristan B. Velloza Kildaire 2023-01-02 15:55:36 +02:00
parent 4bb0c225af
commit 3e853e7431
1 changed files with 74 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
source/libpb.d
View File

@ -62,17 +62,50 @@ public final class PocketBaseParsingException : PBException
}
mixin template AuthTokenHeader(alias http, PocketBase pbInstance)
{
// Must be an instance of HTTP from `std.curl`
static assert(__traits(isSame, typeof(http), HTTP));
void InitializeAuthHeader()
{
// Check if the given PocketBase instance as an authToken
if(pbInstance.authToken.length > 0)
{
// Then add the authaorization header
http.addRequestHeader("Authorization", pbInstance.getAuthToken());
}
}
}
public class PocketBase
{
private string pocketBaseURL;
private string authToken;
/**
* Constructs a new PocketBase instance with
* the default settings
*/
this(string pocketBaseURL = "http://127.0.0.1:8090/api/")
this(string pocketBaseURL = "http://127.0.0.1:8090/api/", string authToken = "")
{
this.pocketBaseURL = pocketBaseURL;
this.authToken = authToken;
}
public void setAuthToken(string authToken)
{
if(cmp(authToken, "") != 0)
{
this.authToken = authToken;
}
}
public string getAuthToken()
{
return this.authToken;
}
/**
@ -87,6 +120,11 @@ public class PocketBase
*/
public RecordType[] listRecords(RecordType)(string table, ulong page = 1, ulong perPage = 30, string filter = "")
{
// Set authorization token if setup
HTTP httpSettings = HTTP();
mixin AuthTokenHeader!(httpSettings, this);
InitializeAuthHeader();
RecordType[] recordsOut;
// Compute the query string
@ -117,7 +155,7 @@ public class PocketBase
try
{
string responseData = cast(string)get(pocketBaseURL~"collections/"~table~"/records?"~queryStr);
string responseData = cast(string)get(pocketBaseURL~"collections/"~table~"/records?"~queryStr, httpSettings);
JSONValue responseJSON = parseJSON(responseData);
JSONValue[] returnedItems = responseJSON["items"].array();
foreach(JSONValue returnedItem; returnedItems)
@ -127,6 +165,17 @@ public class PocketBase
return recordsOut;
}
catch(HTTPStatusException e)
{
if(e.status == 403)
{
throw new NotAuthorized(table, null);
}
else
{
throw new NetworkException();
}
}
catch(CurlException e)
{
debug(dbg)
@ -158,7 +207,12 @@ public class PocketBase
RecordType recordOut;
// Set authorization token if setup
HTTP httpSettings = HTTP();
mixin AuthTokenHeader!(httpSettings, this);
InitializeAuthHeader();
// Set the content type
httpSettings.addRequestHeader("Content-Type", "application/json");
// Serialize the record instance
@ -228,9 +282,14 @@ public class PocketBase
{
RecordType recordOut;
// Set authorization token if setup
HTTP httpSettings = HTTP();
mixin AuthTokenHeader!(httpSettings, this);
InitializeAuthHeader();
try
{
string responseData = cast(string)get(pocketBaseURL~"collections/"~table~"/records/"~id);
string responseData = cast(string)get(pocketBaseURL~"collections/"~table~"/records/"~id, httpSettings);
JSONValue responseJSON = parseJSON(responseData);
recordOut = fromJSON!(RecordType)(responseJSON);
@ -274,8 +333,13 @@ public class PocketBase
idAbleCheck(item);
RecordType recordOut;
// Set authorization token if setup
HTTP httpSettings = HTTP();
mixin AuthTokenHeader!(httpSettings, this);
InitializeAuthHeader();
// Set the content type
httpSettings.addRequestHeader("Content-Type", "application/json");
// Serialize the record instance
@ -329,9 +393,14 @@ public class PocketBase
*/
public void deleteRecord(string table, string id)
{
// Set authorization token if setup
HTTP httpSettings = HTTP();
mixin AuthTokenHeader!(httpSettings, this);
InitializeAuthHeader();
try
{
del(pocketBaseURL~"collections/"~table~"/records/"~id);
del(pocketBaseURL~"collections/"~table~"/records/"~id, httpSettings);
}
catch(HTTPStatusException e)
{