Skip to content

Commit 82cf159

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 8c94e96 commit 82cf159

File tree

3 files changed

+206
-6
lines changed

3 files changed

+206
-6
lines changed

Documentation/scalar.adoc

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,9 @@ SYNOPSIS
99
--------
1010
[verse]
1111
scalar clone [--single-branch] [--branch <main-branch>] [--full-clone]
12-
[--[no-]src] [--[no-]tags] [--[no-]maintenance] <url> [<enlistment>]
12+
[--[no-]src] [--[no-]tags] [--[no-]maintenance]
13+
[--[no-]src] [--local-cache-path <path>] [--cache-server-url <url>]
14+
<url> [<enlistment>]
1315
scalar list
1416
scalar register [--[no-]maintenance] [<enlistment>]
1517
scalar unregister [<enlistment>]
@@ -102,6 +104,17 @@ cloning. If the HEAD at the remote did not point at any branch when
102104
background maintenance feature. Use the `--no-maintenance` to skip
103105
this configuration.
104106

107+
--local-cache-path <path>::
108+
Override the path to the local cache root directory; Pre-fetched objects
109+
are stored into a repository-dependent subdirectory of that path.
110+
+
111+
The default is `<drive>:\.scalarCache` on Windows (on the same drive as the
112+
clone), and `~/.scalarCache` on macOS.
113+
114+
--cache-server-url <url>::
115+
Retrieve missing objects from the specified remote, which is expected to
116+
understand the GVFS protocol.
117+
105118
List
106119
~~~~
107120

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;
@@ -434,6 +455,87 @@ static int supports_gvfs_protocol(const char *url, char **cache_server_url)
434455
return 0; /* error out quietly */
435456
}
436457

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

633+
static int init_shared_object_cache(const char *url,
634+
const char *local_cache_root)
635+
{
636+
struct strbuf buf = STRBUF_INIT;
637+
int res = 0;
638+
char *cache_key = NULL, *shared_cache_path = NULL, *alternates = NULL;
639+
640+
if (!(cache_key = get_cache_key(url))) {
641+
res = error(_("could not determine cache key for '%s'"), url);
642+
goto cleanup;
643+
}
644+
645+
shared_cache_path = xstrfmt("%s/%s", local_cache_root, cache_key);
646+
if (set_config("gvfs.sharedCache=%s", shared_cache_path)) {
647+
res = error(_("could not configure shared cache"));
648+
goto cleanup;
649+
}
650+
651+
strbuf_addf(&buf, "%s/pack", shared_cache_path);
652+
switch (safe_create_leading_directories(the_repository, buf.buf)) {
653+
case SCLD_OK: case SCLD_EXISTS:
654+
break; /* okay */
655+
default:
656+
res = error_errno(_("could not initialize '%s'"), buf.buf);
657+
goto cleanup;
658+
}
659+
660+
alternates = repo_git_path(the_repository, "objects/info/alternates");
661+
write_file(alternates, "%s\n", shared_cache_path);
662+
663+
cleanup:
664+
strbuf_release(&buf);
665+
free(shared_cache_path);
666+
free(cache_key);
667+
free(alternates);
668+
return res;
669+
}
670+
531671
static int cmd_clone(int argc, const char **argv)
532672
{
533673
const char *branch = NULL;
534674
char *branch_to_free = NULL;
535675
int full_clone = 0, single_branch = 0, show_progress = isatty(2);
536676
int src = 1, tags = 1, maintenance = 1;
537-
const char *cache_server_url = NULL;
538-
char *default_cache_server_url = NULL;
677+
const char *cache_server_url = NULL, *local_cache_root = NULL;
678+
char *default_cache_server_url = NULL, *local_cache_root_abs = NULL;
539679
struct option clone_options[] = {
540680
OPT_STRING('b', "branch", &branch, N_("<branch>"),
541681
N_("branch to checkout after clone")),
@@ -553,6 +693,9 @@ static int cmd_clone(int argc, const char **argv)
553693
OPT_STRING(0, "cache-server-url", &cache_server_url,
554694
N_("<url>"),
555695
N_("the url or friendly name of the cache server")),
696+
OPT_STRING(0, "local-cache-path", &local_cache_root,
697+
N_("<path>"),
698+
N_("override the path for the local Scalar cache")),
556699
OPT_END(),
557700
};
558701
const char * const clone_usage[] = {
@@ -594,11 +737,23 @@ static int cmd_clone(int argc, const char **argv)
594737
if (is_directory(enlistment))
595738
die(_("directory '%s' exists already"), enlistment);
596739

740+
ensure_absolute_path(enlistment, &enlistment);
741+
597742
if (src)
598743
dir = xstrfmt("%s/src", enlistment);
599744
else
600745
dir = xstrdup(enlistment);
601746

747+
if (!local_cache_root)
748+
local_cache_root = local_cache_root_abs =
749+
default_cache_root(enlistment);
750+
else
751+
local_cache_root = ensure_absolute_path(local_cache_root,
752+
&local_cache_root_abs);
753+
754+
if (!local_cache_root)
755+
die(_("could not determine local cache root"));
756+
602757
strbuf_reset(&buf);
603758
if (branch)
604759
strbuf_addf(&buf, "init.defaultBranch=%s", branch);
@@ -618,8 +773,28 @@ static int cmd_clone(int argc, const char **argv)
618773

619774
setup_git_directory();
620775

776+
git_config(git_default_config, NULL);
777+
778+
/*
779+
* This `dir_inside_of()` call relies on git_config() having parsed the
780+
* newly-initialized repository config's `core.ignoreCase` value.
781+
*/
782+
if (dir_inside_of(local_cache_root, dir) >= 0) {
783+
struct strbuf path = STRBUF_INIT;
784+
785+
strbuf_addstr(&path, enlistment);
786+
if (chdir("../..") < 0 ||
787+
remove_dir_recursively(&path, 0) < 0)
788+
die(_("'--local-cache-path' cannot be inside the src "
789+
"folder;\nCould not remove '%s'"), enlistment);
790+
791+
die(_("'--local-cache-path' cannot be inside the src folder"));
792+
}
793+
621794
/* common-main already logs `argv` */
622795
trace2_def_repo(the_repository);
796+
trace2_data_intmax("scalar", the_repository, "unattended",
797+
is_unattended());
623798

624799
if (!branch && !(branch = branch_to_free = remote_default_branch(url))) {
625800
res = error(_("failed to get default branch for '%s'"), url);
@@ -649,6 +824,8 @@ static int cmd_clone(int argc, const char **argv)
649824
supports_gvfs_protocol(url, &default_cache_server_url);
650825

651826
if (gvfs_protocol) {
827+
if ((res = init_shared_object_cache(url, local_cache_root)))
828+
goto cleanup;
652829
if (!cache_server_url)
653830
cache_server_url = default_cache_server_url;
654831
if (set_config("core.useGVFSHelper=true") ||
@@ -727,6 +904,7 @@ static int cmd_clone(int argc, const char **argv)
727904
free(dir);
728905
strbuf_release(&buf);
729906
free(default_cache_server_url);
907+
free(local_cache_root_abs);
730908
return res;
731909
}
732910

@@ -1161,6 +1339,12 @@ int cmd_main(int argc, const char **argv)
11611339
struct strbuf scalar_usage = STRBUF_INIT;
11621340
int i;
11631341

1342+
if (is_unattended()) {
1343+
setenv("GIT_ASKPASS", "", 0);
1344+
setenv("GIT_TERMINAL_PROMPT", "false", 0);
1345+
git_config_push_parameter("credential.interactive=false");
1346+
}
1347+
11641348
while (argc > 1 && *argv[1] == '-') {
11651349
if (!strcmp(argv[1], "-C")) {
11661350
if (argc < 3)

0 commit comments

Comments
 (0)