import React from "react";
import BranchView from "../components/BranchView";
import { connect } from "react-redux";
import { compose } from "redux";
import { Redirect, Route, Switch, withRouter } from "react-router-dom";
import { Repository, Branch } from "@scm-manager/ui-types";
import { fetchBranch, getBranch, getFetchBranchFailure, isFetchBranchPending } from "../modules/branches";
import { ErrorNotification, Loading } from "@scm-manager/ui-components";
import { History } from "history";
import { NotFoundError } from "@scm-manager/ui-components";
import queryString from "query-string";

type Props = {
  repository: Repository;
  branchName: string;
  branch: Branch;
  loading: boolean;
  error?: Error;

  // context props
  history: History;
  match: any;
  location: any;

  // dispatch functions
  fetchBranch: (repository: Repository, branchName: string) => void;
};

class BranchRoot extends React.Component<Props> {
  componentDidMount() {
    const { fetchBranch, repository, branchName } = this.props;
    fetchBranch(repository, branchName);
  }

  stripEndingSlash = (url: string) => {
    if (url.endsWith("/")) {
      return url.substring(0, url.length - 1);
    }
    return url;
  };

  matchedUrl = () => {
    return this.stripEndingSlash(this.props.match.url);
  };

  render() {
    const { repository, branch, loading, error, match, location } = this.props;

    const url = this.matchedUrl();

    if (error) {
      if (error instanceof NotFoundError && queryString.parse(location.search).create === "true") {
        return (
          <Redirect
            to={`/repo/${repository.namespace}/${repository.name}/branches/create?name=${match.params.branch}`}
          />
        );
      }

      return <ErrorNotification error={error} />;
    }

    if (loading || !branch) {
      return <Loading />;
    }

    return (
      <Switch>
        <Redirect exact from={url} to={`${url}/info`} />
        <Route path={`${url}/info`} component={() => <BranchView repository={repository} branch={branch} />} />
      </Switch>
    );
  }
}

const mapStateToProps = (state: any, ownProps: Props) => {
  const { repository } = ownProps;
  const branchName = decodeURIComponent(ownProps.match.params.branch);
  const branch = getBranch(state, repository, branchName);
  const loading = isFetchBranchPending(state, repository, branchName);
  const error = getFetchBranchFailure(state, repository, branchName);
  return {
    repository,
    branchName,
    branch,
    loading,
    error
  };
};

const mapDispatchToProps = (dispatch: any) => {
  return {
    fetchBranch: (repository: Repository, branchName: string) => {
      dispatch(fetchBranch(repository, branchName));
    }
  };
};

export default compose(withRouter, connect(mapStateToProps, mapDispatchToProps))(BranchRoot);