refactor: Add helper functions and differentiate processing of one vs multiple input

Author: jcfr

src/cython_cmake/cmake/UseCython.cmake

@@ -127,15 +127,9 @@ function(Cython_compile_pyx)
   set(_language_arg ${_language_${_language}_arg})
   set(_language_extension ${_language_${_language}_extension})
 
-  set(generated_files)
-
-  foreach(_source_file IN LISTS _source_files)
+  function(_compile_pyx _source_file generated_file)
 
-    # Can use cmake_path for CMake 3.20+
-    # cmake_path(GET _source_file STEM _name)
-    get_filename_component(_name "${_source_file}" NAME_WE)
 
-    set(generated_file "${CMAKE_CURRENT_BINARY_DIR}/${_name}.${_language_extension}")
     set_source_files_properties(${generated_file} PROPERTIES GENERATED TRUE)
 
     # Generated depfile is expected to have the ".dep" extension and be located along
@@ -167,9 +161,34 @@ function(Cython_compile_pyx)
       VERBATIM
       COMMENT ${comment}
     )
+  endfunction()
+
+  function(_set_output _input_file _output_var)
+    # Can use cmake_path for CMake 3.20+
+    # cmake_path(GET _input_file STEM basename)
+    get_filename_component(_basename "${_input_file}" NAME_WE)
+
+    set(${_output_var} "${CMAKE_CURRENT_BINARY_DIR}/${_basename}.${_language_extension}" PARENT_SCOPE)
+  endfunction()
+
+  set(generated_files)
+
+  if(input_length EQUAL 1)
+    list(GET_source_files 0 _source_file)
+    _set_output(${_source_file} generated_file)
+    _compile_pyx(${_source_file} ${generated_file})
     list(APPEND generated_files ${generated_file})
-  endforeach()
 
+  else()
+    foreach(_source_file IN LISTS _source_files)
+      _set_output(${_source_file} generated_file)
+      _compile_pyx(${_source_file} ${generated_file})
+      list(APPEND generated_files ${generated_file})
+    endforeach()
+
+  endif()
+
+  # Output variable only if set
   if(_args_OUTPUT_VARIABLE)
     set(_output_variable ${_args_OUTPUT_VARIABLE})
     set(${_output_variable} ${generated_files} PARENT_SCOPE)