exp run: add experiment name check.

dvc/repo/experiments/__init__.py

@@ -473,6 +473,21 @@ def _log_reproduced(self, revs: Iterable[str], tmp_dir: bool = False):
             "\tdvc exp branch <exp> <branch>\n"
         )
 
+    def _ref_name_validation(self, name: Optional[str], **kwargs):
+        from .utils import check_ref_format
+
+        if name is None:
+            return
+
+        baseline_sha = self.repo.scm.get_rev()
+        exp_ref = ExpRefInfo(baseline_sha=baseline_sha, name=name)
+        check_ref_format(self.scm, exp_ref)
+
+        reset = kwargs.get("reset", False)
+        force = kwargs.get("force", False)
+        if not (force or reset) and self.scm.get_ref(str(exp_ref)):
+            raise ExperimentExistsError(name)
+
     @scm_locked
     def new(self, *args, checkpoint_resume: Optional[str] = None, **kwargs):
         """Create a new experiment.
@@ -485,7 +500,10 @@ def new(self, *args, checkpoint_resume: Optional[str] = None, **kwargs):
                 *args, resume_rev=checkpoint_resume, **kwargs
             )
 
-        return self._stash_exp(*args, **kwargs)
+        name = kwargs.pop("name", None)
+        self._ref_name_validation(name, **kwargs)
+
+        return self._stash_exp(*args, name=name, **kwargs)
 
     def _resume_checkpoint(
         self, *args, resume_rev: Optional[str] = None, **kwargs

dvc/repo/experiments/utils.py

@@ -157,3 +157,12 @@ def resolve_exp_ref(
         msg.extend([f"\t{info}" for info in exp_ref_list])
         raise InvalidArgumentError("\n".join(msg))
     return exp_ref_list[0]
+
+
+def check_ref_format(scm: "Git", ref: ExpRefInfo):
+    # "/" forbidden, only in dvc exp as we didn't support it for now.
+    if not scm.check_ref_format(str(ref)) or "/" in ref.name:
+        raise InvalidArgumentError(
+            f"Invalid exp name {ref.name}, the exp name must follow rules in "
+            "https://git-scm.com/docs/git-check-ref-format"
+        )

dvc/scm/git/__init__.py

@@ -346,6 +346,7 @@ def get_fs(self, rev: str):
     status = partialmethod(_backend_func, "status")
     merge = partialmethod(_backend_func, "merge")
     validate_git_remote = partialmethod(_backend_func, "validate_git_remote")
+    check_ref_format = partialmethod(_backend_func, "check_ref_format")
 
     def resolve_rev(self, rev: str) -> str:
         from dvc.repo.experiments.utils import exp_refs_by_name

dvc/scm/git/backend/dulwich/__init__.py

@@ -681,3 +681,8 @@ def validate_git_remote(self, url: str, **kwargs):
             os.path.join("", path)
         ):
             raise InvalidRemoteSCMRepo(url)
+
+    def check_ref_format(self, refname: str):
+        from dulwich.refs import check_ref_format
+
+        return check_ref_format(refname.encode())

tests/func/experiments/test_experiments.py

@@ -53,13 +53,15 @@ def test_experiment_exists(tmp_dir, scm, dvc, exp_stage, mocker, workspace):
         tmp_dir=not workspace,
     )
 
+    new_mock = mocker.spy(dvc.experiments, "_stash_exp")
     with pytest.raises(ExperimentExistsError):
         dvc.experiments.run(
             exp_stage.addressing,
             name="foo",
             params=["foo=3"],
             tmp_dir=not workspace,
         )
+    new_mock.assert_not_called()
 
     results = dvc.experiments.run(
         exp_stage.addressing,
@@ -685,3 +687,16 @@ def test_exp_run_recursive(tmp_dir, scm, dvc, run_copy_metrics):
     )
     assert dvc.experiments.run(".", recursive=True)
     assert (tmp_dir / "metric.json").parse() == {"foo": 1}
+
+
+def test_experiment_name_invalid(tmp_dir, scm, dvc, exp_stage, mocker):
+    from dvc.exceptions import InvalidArgumentError
+
+    new_mock = mocker.spy(dvc.experiments, "_stash_exp")
+    with pytest.raises(InvalidArgumentError):
+        dvc.experiments.run(
+            exp_stage.addressing,
+            name="fo^o",
+            params=["foo=3"],
+        )
+    new_mock.assert_not_called()

tests/unit/repo/experiments/test_utils.py

@@ -1,7 +1,8 @@
 import pytest
 
+from dvc.exceptions import InvalidArgumentError
 from dvc.repo.experiments.base import EXPS_NAMESPACE, ExpRefInfo
-from dvc.repo.experiments.utils import resolve_exp_ref
+from dvc.repo.experiments.utils import check_ref_format, resolve_exp_ref
 
 
 def commit_exp_ref(tmp_dir, scm, file="foo", contents="foo", name="foo"):
@@ -25,3 +26,22 @@ def test_resolve_exp_ref(tmp_dir, scm, git_upstream, name_only, use_url):
     remote_ref_info = resolve_exp_ref(scm, "foo" if name_only else ref, remote)
     assert isinstance(remote_ref_info, ExpRefInfo)
     assert str(remote_ref_info) == ref
+
+
+def test_run_check_ref_format(scm):
+    baseline_rev = "b05eecc666734e899f79af228ff49a7ae5a18cc0"
+
+    def fun(name):
+        ref = ExpRefInfo(baseline_rev, name)
+        check_ref_format(scm, ref)
+
+    fun("name")
+
+    # Forbid slash / here because we didn't support it for now.
+    with pytest.raises(InvalidArgumentError):
+        fun("group/name")
+
+    with pytest.raises(InvalidArgumentError):
+        fun("na\05me")
+    with pytest.raises(InvalidArgumentError):
+        fun("na me")