Skip to content

Commit 0528ea5

Browse files
committed
scalar: do initialize gvfs.sharedCache
This finalizes the port of the `QueryVstsInfo()` function: we already taught `gvfs-helper` to access the `vsts/info` endpoint on demand, we implemented proper JSON parsing, and now it is time to hook it all up. To that end, we also provide a default local cache root directory. It works the same way as the .NET version of Scalar: it uses C:\scalarCache on Windows, ~/.scalarCache/ on macOS and ~/.cache/scalar on Linux Modified to include call to is_unattended() that was removed from a previous commit. Signed-off-by: Johannes Schindelin <[email protected]>
1 parent a4ef597 commit 0528ea5

File tree

3 files changed

+205
-6
lines changed

3 files changed

+205
-6
lines changed

Documentation/scalar.adoc

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@ SYNOPSIS
99
--------
1010
[verse]
1111
scalar clone [--single-branch] [--branch <main-branch>] [--full-clone]
12-
[--[no-]src] <url> [<enlistment>]
12+
[--[no-]src] [--local-cache-path <path>] [--cache-server-url <url>]
13+
<url> [<enlistment>]
1314
scalar list
1415
scalar register [<enlistment>]
1516
scalar unregister [<enlistment>]
@@ -97,6 +98,17 @@ cloning. If the HEAD at the remote did not point at any branch when
9798
A sparse-checkout is initialized by default. This behavior can be
9899
turned off via `--full-clone`.
99100

101+
--local-cache-path <path>::
102+
Override the path to the local cache root directory; Pre-fetched objects
103+
are stored into a repository-dependent subdirectory of that path.
104+
+
105+
The default is `<drive>:\.scalarCache` on Windows (on the same drive as the
106+
clone), and `~/.scalarCache` on macOS.
107+
108+
--cache-server-url <url>::
109+
Retrieve missing objects from the specified remote, which is expected to
110+
understand the GVFS protocol.
111+
100112
List
101113
~~~~
102114

diagnose.c

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ int create_diagnostics_archive(struct repository *r,
186186
struct strvec archiver_args = STRVEC_INIT;
187187
char **argv_copy = NULL;
188188
int stdout_fd = -1, archiver_fd = -1;
189-
char *cache_server_url = NULL;
189+
char *cache_server_url = NULL, *shared_cache = NULL;
190190
struct strbuf buf = STRBUF_INIT;
191191
int res;
192192
struct archive_dir archive_dirs[] = {
@@ -224,8 +224,10 @@ int create_diagnostics_archive(struct repository *r,
224224
strbuf_addf(&buf, "Repository root: %s\n", r->worktree);
225225

226226
repo_config_get_string(r, "gvfs.cache-server", &cache_server_url);
227-
strbuf_addf(&buf, "Cache Server: %s\n\n",
228-
cache_server_url ? cache_server_url : "None");
227+
repo_config_get_string(r, "gvfs.sharedCache", &shared_cache);
228+
strbuf_addf(&buf, "Cache Server: %s\nLocal Cache: %s\n\n",
229+
cache_server_url ? cache_server_url : "None",
230+
shared_cache ? shared_cache : "None");
229231

230232
get_disk_info(&buf);
231233
write_or_die(stdout_fd, buf.buf, buf.len);
@@ -285,6 +287,7 @@ int create_diagnostics_archive(struct repository *r,
285287
strvec_clear(&archiver_args);
286288
strbuf_release(&buf);
287289
free(cache_server_url);
290+
free(shared_cache);
288291

289292
return res;
290293
}

scalar.c

Lines changed: 186 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
#include "git-compat-util.h"
88
#include "abspath.h"
99
#include "gettext.h"
10+
#include "hex.h"
1011
#include "parse-options.h"
1112
#include "config.h"
1213
#include "run-command.h"
@@ -15,11 +16,18 @@
1516
#include "fsmonitor-settings.h"
1617
#include "refs.h"
1718
#include "dir.h"
19+
#include "object-file.h"
1820
#include "packfile.h"
1921
#include "help.h"
2022
#include "setup.h"
23+
#include "wrapper.h"
2124
#include "trace2.h"
2225
#include "json-parser.h"
26+
#include "path.h"
27+
28+
static int is_unattended(void) {
29+
return git_env_bool("Scalar_UNATTENDED", 0);
30+
}
2331

2432
static void setup_enlistment_directory(int argc, const char **argv,
2533
const char * const *usagestr,
@@ -106,6 +114,19 @@ static int run_git(const char *arg, ...)
106114
return res;
107115
}
108116

117+
static const char *ensure_absolute_path(const char *path, char **absolute)
118+
{
119+
struct strbuf buf = STRBUF_INIT;
120+
121+
if (is_absolute_path(path))
122+
return path;
123+
124+
strbuf_realpath_forgiving(&buf, path, 1);
125+
free(*absolute);
126+
*absolute = strbuf_detach(&buf, NULL);
127+
return *absolute;
128+
}
129+
109130
struct scalar_config {
110131
const char *key;
111132
const char *value;
@@ -419,6 +440,87 @@ static int supports_gvfs_protocol(const char *url, char **cache_server_url)
419440
return 0; /* error out quietly */
420441
}
421442

443+
static char *default_cache_root(const char *root)
444+
{
445+
const char *env;
446+
447+
if (is_unattended())
448+
return xstrfmt("%s/.scalarCache", root);
449+
450+
#ifdef WIN32
451+
(void)env;
452+
return xstrfmt("%.*s.scalarCache", offset_1st_component(root), root);
453+
#elif defined(__APPLE__)
454+
if ((env = getenv("HOME")) && *env)
455+
return xstrfmt("%s/.scalarCache", env);
456+
return NULL;
457+
#else
458+
if ((env = getenv("XDG_CACHE_HOME")) && *env)
459+
return xstrfmt("%s/scalar", env);
460+
if ((env = getenv("HOME")) && *env)
461+
return xstrfmt("%s/.cache/scalar", env);
462+
return NULL;
463+
#endif
464+
}
465+
466+
static int get_repository_id(struct json_iterator *it)
467+
{
468+
if (it->type == JSON_STRING &&
469+
!strcasecmp(".repository.id", it->key.buf)) {
470+
*(char **)it->fn_data = strbuf_detach(&it->string_value, NULL);
471+
return 1;
472+
}
473+
474+
return 0;
475+
}
476+
477+
/* Needs to run this in a worktree; gvfs-helper requires a Git repository */
478+
static char *get_cache_key(const char *url)
479+
{
480+
struct child_process cp = CHILD_PROCESS_INIT;
481+
struct strbuf out = STRBUF_INIT;
482+
char *cache_key = NULL;
483+
484+
cp.git_cmd = 1;
485+
strvec_pushl(&cp.args, "gvfs-helper", "--remote", url,
486+
"endpoint", "vsts/info", NULL);
487+
if (!pipe_command(&cp, NULL, 0, &out, 512, NULL, 0)) {
488+
char *id = NULL;
489+
struct json_iterator it =
490+
JSON_ITERATOR_INIT(out.buf, get_repository_id, &id);
491+
492+
if (iterate_json(&it) < 0)
493+
warning("JSON parse error (%s)", out.buf);
494+
else if (id)
495+
cache_key = xstrfmt("id_%s", id);
496+
free(id);
497+
}
498+
499+
if (!cache_key) {
500+
struct strbuf downcased = STRBUF_INIT;
501+
int hash_algo_index = hash_algo_by_name("sha1");
502+
const struct git_hash_algo *hash_algo = hash_algo_index < 0 ?
503+
the_hash_algo : &hash_algos[hash_algo_index];
504+
struct git_hash_ctx ctx;
505+
unsigned char hash[GIT_MAX_RAWSZ];
506+
507+
strbuf_addstr(&downcased, url);
508+
strbuf_tolower(&downcased);
509+
510+
hash_algo->init_fn(&ctx);
511+
hash_algo->update_fn(&ctx, downcased.buf, downcased.len);
512+
hash_algo->final_fn(hash, &ctx);
513+
514+
strbuf_release(&downcased);
515+
516+
cache_key = xstrfmt("url_%s",
517+
hash_to_hex_algop(hash, hash_algo));
518+
}
519+
520+
strbuf_release(&out);
521+
return cache_key;
522+
}
523+
422524
static char *remote_default_branch(const char *url)
423525
{
424526
struct child_process cp = CHILD_PROCESS_INIT;
@@ -513,14 +615,52 @@ void load_builtin_commands(const char *prefix UNUSED,
513615
die("not implemented");
514616
}
515617

618+
static int init_shared_object_cache(const char *url,
619+
const char *local_cache_root)
620+
{
621+
struct strbuf buf = STRBUF_INIT;
622+
int res = 0;
623+
char *cache_key = NULL, *shared_cache_path = NULL, *alternates = NULL;
624+
625+
if (!(cache_key = get_cache_key(url))) {
626+
res = error(_("could not determine cache key for '%s'"), url);
627+
goto cleanup;
628+
}
629+
630+
shared_cache_path = xstrfmt("%s/%s", local_cache_root, cache_key);
631+
if (set_config("gvfs.sharedCache=%s", shared_cache_path)) {
632+
res = error(_("could not configure shared cache"));
633+
goto cleanup;
634+
}
635+
636+
strbuf_addf(&buf, "%s/pack", shared_cache_path);
637+
switch (safe_create_leading_directories(buf.buf)) {
638+
case SCLD_OK: case SCLD_EXISTS:
639+
break; /* okay */
640+
default:
641+
res = error_errno(_("could not initialize '%s'"), buf.buf);
642+
goto cleanup;
643+
}
644+
645+
alternates = repo_git_path(the_repository, "objects/info/alternates");
646+
write_file(alternates, "%s\n", shared_cache_path);
647+
648+
cleanup:
649+
strbuf_release(&buf);
650+
free(shared_cache_path);
651+
free(cache_key);
652+
free(alternates);
653+
return res;
654+
}
655+
516656
static int cmd_clone(int argc, const char **argv)
517657
{
518658
const char *branch = NULL;
519659
char *branch_to_free = NULL;
520660
int full_clone = 0, single_branch = 0, show_progress = isatty(2);
521661
int src = 1, tags = 1;
522-
const char *cache_server_url = NULL;
523-
char *default_cache_server_url = NULL;
662+
const char *cache_server_url = NULL, *local_cache_root = NULL;
663+
char *default_cache_server_url = NULL, *local_cache_root_abs = NULL;
524664
struct option clone_options[] = {
525665
OPT_STRING('b', "branch", &branch, N_("<branch>"),
526666
N_("branch to checkout after clone")),
@@ -536,6 +676,9 @@ static int cmd_clone(int argc, const char **argv)
536676
OPT_STRING(0, "cache-server-url", &cache_server_url,
537677
N_("<url>"),
538678
N_("the url or friendly name of the cache server")),
679+
OPT_STRING(0, "local-cache-path", &local_cache_root,
680+
N_("<path>"),
681+
N_("override the path for the local Scalar cache")),
539682
OPT_END(),
540683
};
541684
const char * const clone_usage[] = {
@@ -577,11 +720,23 @@ static int cmd_clone(int argc, const char **argv)
577720
if (is_directory(enlistment))
578721
die(_("directory '%s' exists already"), enlistment);
579722

723+
ensure_absolute_path(enlistment, &enlistment);
724+
580725
if (src)
581726
dir = xstrfmt("%s/src", enlistment);
582727
else
583728
dir = xstrdup(enlistment);
584729

730+
if (!local_cache_root)
731+
local_cache_root = local_cache_root_abs =
732+
default_cache_root(enlistment);
733+
else
734+
local_cache_root = ensure_absolute_path(local_cache_root,
735+
&local_cache_root_abs);
736+
737+
if (!local_cache_root)
738+
die(_("could not determine local cache root"));
739+
585740
strbuf_reset(&buf);
586741
if (branch)
587742
strbuf_addf(&buf, "init.defaultBranch=%s", branch);
@@ -601,8 +756,28 @@ static int cmd_clone(int argc, const char **argv)
601756

602757
setup_git_directory();
603758

759+
git_config(git_default_config, NULL);
760+
761+
/*
762+
* This `dir_inside_of()` call relies on git_config() having parsed the
763+
* newly-initialized repository config's `core.ignoreCase` value.
764+
*/
765+
if (dir_inside_of(local_cache_root, dir) >= 0) {
766+
struct strbuf path = STRBUF_INIT;
767+
768+
strbuf_addstr(&path, enlistment);
769+
if (chdir("../..") < 0 ||
770+
remove_dir_recursively(&path, 0) < 0)
771+
die(_("'--local-cache-path' cannot be inside the src "
772+
"folder;\nCould not remove '%s'"), enlistment);
773+
774+
die(_("'--local-cache-path' cannot be inside the src folder"));
775+
}
776+
604777
/* common-main already logs `argv` */
605778
trace2_def_repo(the_repository);
779+
trace2_data_intmax("scalar", the_repository, "unattended",
780+
is_unattended());
606781

607782
if (!branch && !(branch = branch_to_free = remote_default_branch(url))) {
608783
res = error(_("failed to get default branch for '%s'"), url);
@@ -632,6 +807,8 @@ static int cmd_clone(int argc, const char **argv)
632807
supports_gvfs_protocol(url, &default_cache_server_url);
633808

634809
if (gvfs_protocol) {
810+
if ((res = init_shared_object_cache(url, local_cache_root)))
811+
goto cleanup;
635812
if (!cache_server_url)
636813
cache_server_url = default_cache_server_url;
637814
if (set_config("core.useGVFSHelper=true") ||
@@ -709,6 +886,7 @@ static int cmd_clone(int argc, const char **argv)
709886
free(dir);
710887
strbuf_release(&buf);
711888
free(default_cache_server_url);
889+
free(local_cache_root_abs);
712890
return res;
713891
}
714892

@@ -1120,6 +1298,12 @@ int cmd_main(int argc, const char **argv)
11201298
struct strbuf scalar_usage = STRBUF_INIT;
11211299
int i;
11221300

1301+
if (is_unattended()) {
1302+
setenv("GIT_ASKPASS", "", 0);
1303+
setenv("GIT_TERMINAL_PROMPT", "false", 0);
1304+
git_config_push_parameter("credential.interactive=false");
1305+
}
1306+
11231307
while (argc > 1 && *argv[1] == '-') {
11241308
if (!strcmp(argv[1], "-C")) {
11251309
if (argc < 3)

0 commit comments

Comments
 (0)