list.go 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. package stream
  2. import (
  3. "context"
  4. "os"
  5. "github.com/0xJacky/Nginx-UI/internal/config"
  6. "github.com/0xJacky/Nginx-UI/model"
  7. )
  8. // ListOptions represents the options for listing streams
  9. type ListOptions struct {
  10. Search string
  11. Status string
  12. OrderBy string
  13. Sort string
  14. EnvGroupID uint64
  15. }
  16. // GetStreamConfigs retrieves and processes stream configurations with database integration
  17. func GetStreamConfigs(ctx context.Context, options *ListOptions, streams []*model.Stream) ([]config.Config, error) {
  18. // Convert to generic options
  19. genericOptions := &config.GenericListOptions{
  20. Search: options.Search,
  21. Status: options.Status,
  22. OrderBy: options.OrderBy,
  23. Sort: options.Sort,
  24. EnvGroupID: options.EnvGroupID,
  25. IncludeDirs: false, // Filter out directories for stream configurations
  26. }
  27. // Create processor with stream-specific logic
  28. processor := &config.GenericConfigProcessor{
  29. Paths: config.ConfigPaths{
  30. AvailableDir: "streams-available",
  31. EnabledDir: "streams-enabled",
  32. },
  33. StatusMapBuilder: config.DefaultStatusMapBuilder,
  34. ConfigBuilder: buildConfig,
  35. FilterMatcher: config.DefaultFilterMatcher,
  36. }
  37. return config.GetGenericConfigs(ctx, genericOptions, streams, processor)
  38. }
  39. // buildConfig creates a config.Config from file information with stream-specific data
  40. func buildConfig(fileName string, fileInfo os.FileInfo, status config.ConfigStatus, envGroupID uint64, envGroup *model.EnvGroup) config.Config {
  41. indexedStream := GetIndexedStream(fileName)
  42. // Convert proxy targets
  43. proxyTargets := make([]config.ProxyTarget, len(indexedStream.ProxyTargets))
  44. for i, target := range indexedStream.ProxyTargets {
  45. proxyTargets[i] = config.ProxyTarget{
  46. Host: target.Host,
  47. Port: target.Port,
  48. Type: target.Type,
  49. }
  50. }
  51. return config.Config{
  52. Name: fileName,
  53. ModifiedAt: fileInfo.ModTime(),
  54. Size: fileInfo.Size(),
  55. IsDir: fileInfo.IsDir(),
  56. Status: status,
  57. EnvGroupID: envGroupID,
  58. EnvGroup: envGroup,
  59. ProxyTargets: proxyTargets,
  60. }
  61. }