Make reconcile command work for helmrepository and helmchart kinds

pkg/flux/reconcile.go

@@ -86,6 +86,18 @@ func NewReconcileCommand(resource string) *reconcileCommand {
 			groupVersion: sourcev1beta2.GroupVersion,
 			kind:         sourcev1beta2.BucketKind,
 		}
+	case sourcev1beta2.HelmRepositoryKind:
+		return &reconcileCommand{
+			object:       helmRepositoryAdapter{&sourcev1beta2.HelmRepository{}},
+			groupVersion: sourcev1beta2.GroupVersion,
+			kind:         sourcev1beta2.HelmRepositoryKind,
+		}
+	case sourcev1beta2.HelmChartKind:
+		return &reconcileCommand{
+			object:       helmChartAdapter{&sourcev1beta2.HelmChart{}},
+			groupVersion: sourcev1beta2.GroupVersion,
+			kind:         sourcev1beta2.HelmChartKind,
+		}
 	}
 
 	return nil

pkg/flux/source.go

@@ -84,3 +84,43 @@ func (obj bucketAdapter) lastHandledReconcileRequest() string {
 func (obj bucketAdapter) successMessage() string {
 	return fmt.Sprintf("fetched revision %s", obj.Status.Artifact.Revision)
 }
+
+type helmRepositoryAdapter struct {
+	*sourcev1beta2.HelmRepository
+}
+
+func (a helmRepositoryAdapter) asClientObject() client.Object {
+	return a.HelmRepository
+}
+
+func (obj helmRepositoryAdapter) isSuspended() bool {
+	return obj.HelmRepository.Spec.Suspend
+}
+
+func (obj helmRepositoryAdapter) lastHandledReconcileRequest() string {
+	return obj.Status.GetLastHandledReconcileRequest()
+}
+
+func (obj helmRepositoryAdapter) successMessage() string {
+	return fmt.Sprintf("fetched revision %s", obj.Status.Artifact.Revision)
+}
+
+type helmChartAdapter struct {
+	*sourcev1beta2.HelmChart
+}
+
+func (a helmChartAdapter) asClientObject() client.Object {
+	return a.HelmChart
+}
+
+func (obj helmChartAdapter) isSuspended() bool {
+	return obj.HelmChart.Spec.Suspend
+}
+
+func (obj helmChartAdapter) lastHandledReconcileRequest() string {
+	return obj.Status.GetLastHandledReconcileRequest()
+}
+
+func (obj helmChartAdapter) successMessage() string {
+	return fmt.Sprintf("fetched revision %s", obj.Status.Artifact.Revision)
+}