[rayci] support generic '.ci.yaml' suffixes (#277)

so that repo using the tool does not have to mention ray

Author: aslonnie

raycicmd/make.go

@@ -11,22 +11,25 @@ import (
 	yaml "gopkg.in/yaml.v3"
 )
 
+var ciYamlSuffixes = []string{
+	".rayci.yaml", ".rayci.yml",
+	".ci.yaml", ".ci.yml",
+}
+
 func isRayCIYaml(p string) bool {
-	if strings.HasSuffix(p, ".rayci.yaml") {
-		return true
-	}
-	if strings.HasSuffix(p, ".rayci.yml") {
-		return true
+	for _, suffix := range ciYamlSuffixes {
+		if strings.HasSuffix(p, suffix) {
+			return true
+		}
 	}
 	return false
 }
 
 func stripRayCIYamlSuffix(p string) string {
-	if strings.HasSuffix(p, ".rayci.yaml") {
-		return strings.TrimSuffix(p, ".rayci.yaml")
-	}
-	if strings.HasSuffix(p, ".rayci.yml") {
-		return strings.TrimSuffix(p, ".rayci.yml")
+	for _, suffix := range ciYamlSuffixes {
+		if strings.HasSuffix(p, suffix) {
+			return strings.TrimSuffix(p, suffix)
+		}
 	}
 	return p
 }

raycicmd/make_test.go

@@ -15,6 +15,8 @@ func TestIsRayCIYaml(t *testing.T) {
 	for _, f := range []string{
 		"foo.rayci.yaml",
 		"foo.rayci.yml",
+		"foo.ci.yaml",
+		"foo.ci.yml",
 		"dir/foo.rayci.yml",
 	} {
 		if !isRayCIYaml(f) {
@@ -24,6 +26,8 @@ func TestIsRayCIYaml(t *testing.T) {
 
 	for _, f := range []string{
 		"rayci.yaml",
+		"ci.yaml",
+		"ci.yml",
 		"pipeline.build.yaml",
 		"pipeline.tests.yml",
 	} {
@@ -42,6 +46,10 @@ func TestListCIYamlFiles(t *testing.T) {
 		"foo.rayci.yml",
 		"dir/foo.rayci.yml",
 		"pipeline.build.yaml",
+		"foo.ci.yaml",
+		"bar.ci.yaml",
+		"foo.ci.yml",
+		"dir/foo.ci.yml",
 	} {
 		dir := filepath.Join(tmp, filepath.Dir(f))
 		if err := os.MkdirAll(dir, 0o700); err != nil {
@@ -59,7 +67,10 @@ func TestListCIYamlFiles(t *testing.T) {
 	}
 
 	want := []string{
+		"bar.ci.yaml",
 		"bar.rayci.yaml",
+		"foo.ci.yaml",
+		"foo.ci.yml",
 		"foo.rayci.yaml",
 		"foo.rayci.yml",
 	}