kodit 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

kodit/source/source_service.py

@@ -6,26 +6,22 @@ system, database operations (via SourceRepository), and provides a clean API for
 source management.
 """
 
-import mimetypes
-import shutil
-import tempfile
-from datetime import UTC, datetime
-from hashlib import sha256
+from datetime import datetime
 from pathlib import Path
 
-import aiofiles
-import git
 import pydantic
 import structlog
-from tqdm import tqdm
 
 from kodit.source.git import is_valid_clone_target
-from kodit.source.ignore import IgnorePatterns
-from kodit.source.source_models import (
-    Author,
-    File,
-    Source,
-    SourceType,
+from kodit.source.source_factories import (
+    FolderFileMetadataExtractor,
+    FolderSourceFactory,
+    FolderWorkingCopyProvider,
+    GitAuthorExtractor,
+    GitFileMetadataExtractor,
+    GitSourceFactory,
+    GitWorkingCopyProvider,
+    NoOpAuthorExtractor,
 )
 from kodit.source.source_repository import SourceRepository
 
@@ -69,6 +65,32 @@ class SourceService:
         self.repository = repository
         self.log = structlog.get_logger(__name__)
 
+        # Initialize factories
+        self._setup_factories()
+
+    def _setup_factories(self) -> None:
+        # Git-specific dependencies
+        git_working_copy = GitWorkingCopyProvider(self.clone_dir)
+        git_metadata_extractor = GitFileMetadataExtractor()
+        git_author_extractor = GitAuthorExtractor(self.repository)
+        self.git_factory = GitSourceFactory(
+            working_copy=git_working_copy,
+            metadata_extractor=git_metadata_extractor,
+            author_extractor=git_author_extractor,
+            repository=self.repository,
+        )
+
+        # Folder-specific dependencies
+        folder_working_copy = FolderWorkingCopyProvider(self.clone_dir)
+        folder_metadata_extractor = FolderFileMetadataExtractor()
+        no_op_author_extractor = NoOpAuthorExtractor()
+        self.folder_factory = FolderSourceFactory(
+            working_copy=folder_working_copy,
+            metadata_extractor=folder_metadata_extractor,
+            author_extractor=no_op_author_extractor,
+            repository=self.repository,
+        )
+
     async def get(self, source_id: int) -> SourceView:
         """Get a source by ID.
 
@@ -92,137 +114,13 @@ class SourceService:
         """Create a new source from a URI or path."""
         # If it's possible to clone it, then do so
         if is_valid_clone_target(uri_or_path_like):
-            return await self._create_git_source(uri_or_path_like)
-
+            source = await self.git_factory.create(uri_or_path_like)
         # Otherwise just treat it as a directory
-        if Path(uri_or_path_like).is_dir():
-            return await self._create_folder_source(Path(uri_or_path_like))
-
-        msg = f"Unsupported source: {uri_or_path_like}"
-        raise ValueError(msg)
-
-    async def _create_folder_source(self, directory: Path) -> SourceView:
-        """Create a folder source.
-
-        Args:
-            directory: The path to the local directory.
-
-        Raises:
-            ValueError: If the folder doesn't exist.
-            SourceAlreadyExistsError: If the folder is already added.
-
-        """
-        # Resolve the directory to an absolute path
-        directory = directory.expanduser().resolve()
-
-        source = await self.repository.get_source_by_uri(directory.as_uri())
-        if source:
-            self.log.info("Source already exists, reusing...", source_id=source.id)
+        elif Path(uri_or_path_like).is_dir():
+            source = await self.folder_factory.create(uri_or_path_like)
         else:
-            # Check if the folder exists
-            if not directory.exists():
-                msg = f"Folder does not exist: {directory}"
-                raise ValueError(msg)
-
-            # Check if the folder is already added
-            if await self.repository.get_source_by_uri(directory.as_uri()):
-                msg = f"Directory already added: {directory}"
-                raise ValueError(msg)
-
-            # Clone into a local directory
-            clone_path = self.clone_dir / directory.as_posix().replace("/", "_")
-            clone_path.mkdir(parents=True, exist_ok=True)
-
-            # Copy all files recursively, preserving directory structure, ignoring
-            # hidden files
-            shutil.copytree(
-                directory,
-                clone_path,
-                ignore=shutil.ignore_patterns(".*"),
-                dirs_exist_ok=True,
-            )
-
-            source = await self.repository.create_source(
-                Source(
-                    uri=directory.as_uri(),
-                    cloned_path=str(clone_path),
-                    source_type=SourceType.FOLDER,
-                ),
-            )
-
-            # Add all files to the source
-            # Count total files for progress bar
-            file_count = sum(1 for _ in clone_path.rglob("*") if _.is_file())
-
-            # Process each file in the source directory
-            for path in tqdm(clone_path.rglob("*"), total=file_count, leave=False):
-                await self._process_file(source, path.absolute())
-
-        return SourceView(
-            id=source.id,
-            uri=source.uri,
-            cloned_path=Path(source.cloned_path),
-            created_at=source.created_at,
-            num_files=await self.repository.num_files_for_source(source.id),
-        )
-
-    async def _create_git_source(self, uri: str) -> SourceView:
-        """Create a git source.
-
-        Args:
-            uri: The URI of the git repository.
-
-        Raises:
-            ValueError: If the repository cloning fails.
-
-        """
-        self.log.debug("Normalising git uri", uri=uri)
-        with tempfile.TemporaryDirectory() as temp_dir:
-            git.Repo.clone_from(uri, temp_dir)
-            remote = git.Repo(temp_dir).remote()
-            uri = remote.url
-
-        self.log.debug("Checking if source already exists", uri=uri)
-        source = await self.repository.get_source_by_uri(uri)
-
-        if source:
-            self.log.info("Source already exists, reusing...", source_id=source.id)
-        else:
-            # Create a unique directory name for the clone
-            clone_path = self.clone_dir / uri.replace("/", "_").replace(":", "_")
-            clone_path.mkdir(parents=True, exist_ok=True)
-
-            try:
-                self.log.info("Cloning repository", uri=uri, clone_path=str(clone_path))
-                git.Repo.clone_from(uri, clone_path)
-            except git.GitCommandError as e:
-                if "already exists and is not an empty directory" in str(e):
-                    self.log.info("Repository already exists, reusing...", uri=uri)
-                else:
-                    msg = f"Failed to clone repository: {e}"
-                    raise ValueError(msg) from e
-
-            self.log.debug("Creating source", uri=uri, clone_path=str(clone_path))
-            source = await self.repository.create_source(
-                Source(
-                    uri=uri,
-                    cloned_path=str(clone_path),
-                    source_type=SourceType.GIT,
-                ),
-            )
-
-            # Get the ignore patterns for this source
-            ignore_patterns = IgnorePatterns(clone_path)
-
-            # Get all files that are not ignored
-            files = [
-                f for f in clone_path.rglob("*") if not ignore_patterns.should_ignore(f)
-            ]
-
-            # Process each file in the source directory
-            self.log.info("Inspecting files", source_id=source.id, num_files=len(files))
-            for path in tqdm(files, total=len(files), leave=False):
-                await self._process_file(source, path.absolute())
+            msg = f"Unsupported source: {uri_or_path_like}"
+            raise ValueError(msg)
 
         return SourceView(
             id=source.id,
@@ -232,81 +130,6 @@ class SourceService:
             num_files=await self.repository.num_files_for_source(source.id),
         )
 
-    async def _process_file(
-        self,
-        source: Source,
-        cloned_file: Path,
-    ) -> None:
-        """Process a single file for indexing."""
-        if not cloned_file.is_file():
-            return
-
-        # If this file exists in a git repository, pull out the file's metadata
-        authors: list[Author] = []
-        first_modified_at: datetime | None = None
-        last_modified_at: datetime | None = None
-        if source.type == SourceType.GIT:
-            # Get the git repository
-            git_repo = git.Repo(source.cloned_path)
-
-            # Get the last commit that touched this file
-            commits = list(
-                git_repo.iter_commits(
-                    paths=str(cloned_file),
-                    all=True,
-                )
-            )
-            if len(commits) > 0:
-                last_modified_at = commits[0].committed_datetime
-                first_modified_at = commits[-1].committed_datetime
-
-            # Get the file's blame
-            blames = git_repo.blame("HEAD", str(cloned_file))
-
-            # Extract the blame's authors
-            actors = [
-                commit.author
-                for blame in blames or []
-                for commit in blame
-                if isinstance(commit, git.Commit)
-            ]
-
-            # Get or create the authors in the database
-            for actor in actors:
-                if actor.name or actor.email:
-                    author = await self.repository.get_or_create_author(
-                        actor.name or "", actor.email or ""
-                    )
-                    authors.append(author)
-
-        # Create the file record
-        async with aiofiles.open(cloned_file, "rb") as f:
-            content = await f.read()
-            mime_type = mimetypes.guess_type(cloned_file)
-            sha = sha256(content).hexdigest()
-
-            # Create file record
-            file = File(
-                created_at=first_modified_at or datetime.now(UTC),
-                updated_at=last_modified_at or datetime.now(UTC),
-                source_id=source.id,
-                cloned_path=str(cloned_file),
-                mime_type=mime_type[0]
-                if mime_type and mime_type[0]
-                else "application/octet-stream",
-                uri=cloned_file.as_uri(),
-                sha256=sha,
-                size_bytes=len(content),
-            )
-
-            await self.repository.create_file(file)
-
-            # Create mapping of authors to the file
-            for author in authors:
-                await self.repository.get_or_create_author_file_mapping(
-                    author_id=author.id, file_id=file.id
-                )
-
     async def list_sources(self) -> list[SourceView]:
         """List all available sources.
 

{kodit-0.2.2.dist-info → kodit-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kodit
-Version: 0.2.2
+Version: 0.2.3
 Summary: Code indexing for better AI code generation
 Project-URL: Homepage, https://docs.helixml.tech/kodit/
 Project-URL: Documentation, https://docs.helixml.tech/kodit/
@@ -37,9 +37,9 @@ Requires-Dist: httpx-retries>=0.3.2
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: openai>=1.82.0
 Requires-Dist: pathspec>=0.12.1
-Requires-Dist: posthog>=4.0.1
 Requires-Dist: pydantic-settings>=2.9.1
 Requires-Dist: pytable-formatter>=0.1.1
+Requires-Dist: rudder-sdk-python>=2.1.4
 Requires-Dist: sentence-transformers>=4.1.0
 Requires-Dist: sqlalchemy[asyncio]>=2.0.40
 Requires-Dist: structlog>=25.3.0

{kodit-0.2.2.dist-info → kodit-0.2.3.dist-info}/RECORD

@@ -1,11 +1,11 @@
 kodit/.gitignore,sha256=ztkjgRwL9Uud1OEi36hGQeDGk3OLK1NfDEO8YqGYy8o,11
 kodit/__init__.py,sha256=aEKHYninUq1yh6jaNfvJBYg-6fenpN132nJt1UU6Jxs,59
-kodit/_version.py,sha256=OjGGK5TcHVG44Y62aAqeJH4CskkZoY9ydbHOtCDew50,511
-kodit/app.py,sha256=Mr5BFHOHx5zppwjC4XPWVvHjwgl1yrKbUjTWXKubJQM,891
-kodit/cli.py,sha256=i7eEt0FdIQGEfXKFte-8fBcZZGE8BPXBp40aGwJDQGI,11323
+kodit/_version.py,sha256=wD8hnA5gV5UmPkQnpT3xR6V2csgj9K5NEADogbLK79M,511
+kodit/app.py,sha256=qKBWJ0VNSY_M6G3VFfAQ0133q5bnS99cUFD0p396taw,1032
+kodit/cli.py,sha256=wKFXGUMX-fDLooaK-3po2TBpNNRBwgSD7BRbUddg-_M,11562
 kodit/config.py,sha256=3yh7hfLSILjZK_qJMhcExwRcrWJ0b5Eb1JjjOvMPJZo,4146
 kodit/database.py,sha256=WB1KpVxUYPgiJGU0gJa2hqytYB8wJEJ5z3WayhWzNMU,2403
-kodit/log.py,sha256=HU1OmuxO4FcVw61k4WW7Y4WM7BrDaeplw1PcBHhuIZY,5434
+kodit/log.py,sha256=sHPHYetlMcKTor2VaFLMyao1_fZ_xhuzqXCAt5F5UMU,8575
 kodit/mcp.py,sha256=QruyPskWB0_x59pkfj5BBeXuR13GMny5TAZEa2j4U9s,5752
 kodit/middleware.py,sha256=I6FOkqG9-8RH5kR1-0ZoQWfE4qLCB8lZYv8H_OCH29o,2714
 kodit/bm25/__init__.py,sha256=j8zyriNWhbwE5Lbybzg1hQAhANlU9mKHWw4beeUR6og,19
@@ -14,19 +14,19 @@ kodit/bm25/keyword_search_service.py,sha256=aBbWQKgQmi2re3EIHdXFS00n7Wj3b2D0pZsL
 kodit/bm25/local_bm25.py,sha256=nokrd_xAeqXi3m68X5P1R5KBhRRB1E2L_J6Zgm26PCg,3869
 kodit/bm25/vectorchord_bm25.py,sha256=0p_FgliaoevB8GLSmzWnV3zUjdcWgCgOKIpLURr7Qfo,6549
 kodit/embedding/__init__.py,sha256=h9NXzDA1r-K23nvBajBV-RJzHJN0p3UJ7UQsmdnOoRw,24
-kodit/embedding/embedding_factory.py,sha256=UbrTl3NEqBBH3ecvRG7vGW5wuvUMbWJEWbAAFALOGqs,2141
+kodit/embedding/embedding_factory.py,sha256=-WuXNleQ_mqdw1E4TczjtOawNeaXKAiDPFqN_XX7Mmg,2419
 kodit/embedding/embedding_models.py,sha256=rN90vSs86dYiqoawcp8E9jtwY31JoJXYfaDlsJK7uqc,656
 kodit/embedding/embedding_repository.py,sha256=-ux3scpBzel8c0pMH9fNOEsSXFIzl-IfgaWrkTb1szo,6907
 kodit/embedding/local_vector_search_service.py,sha256=dgMi8hQNUbYEgHnEYmLIpon4yLduoNUpu7k7VP6sOHI,2042
 kodit/embedding/vector_search_service.py,sha256=pQJ129QjGrAWOXzqkywmgtDRpy8_gtzYgkivyqF9Vrs,1009
 kodit/embedding/vectorchord_vector_search_service.py,sha256=TKNR3HgWHwwWtJ1SsvSaj_BXLJ_uw6Bdr_tpaePMeAA,5383
 kodit/embedding/embedding_provider/__init__.py,sha256=h9NXzDA1r-K23nvBajBV-RJzHJN0p3UJ7UQsmdnOoRw,24
-kodit/embedding/embedding_provider/embedding_provider.py,sha256=IC7fZaZ_ze-DxpxKfK44pRDwHWUQhVIqVKKQ3alO5Qc,1882
+kodit/embedding/embedding_provider/embedding_provider.py,sha256=T6AKMWwzEJ1vDe-cEIg-qxdjhUEZ0PKs9YqQMWaLaQ0,1928
 kodit/embedding/embedding_provider/hash_embedding_provider.py,sha256=nAhlhh8j8PqqCCbhVl26Y8ntFBm2vJBCtB4X04g5Wwg,2638
 kodit/embedding/embedding_provider/local_embedding_provider.py,sha256=WP8lw6XG7v1_5Mw4_rhIOETooYRsxhkwmFaXCqCouQU,1977
 kodit/embedding/embedding_provider/openai_embedding_provider.py,sha256=-phz5FKYM_tI3Q4_3SPzjzIOK3k92Uk52TAOTmoVoWI,2722
 kodit/enrichment/__init__.py,sha256=vBEolHpKaHUhfINX0dSGyAPlvgpLNAer9YzFtdvCB24,18
-kodit/enrichment/enrichment_factory.py,sha256=JbWFNciB6Yf79SFVjG9UhLgCcrXZ1rIJrenU8QmNLBE,1411
+kodit/enrichment/enrichment_factory.py,sha256=AAzvxgjo-FQU5aAm9Zla4DAwUMKGrcw8mQwJsMhIsHY,1566
 kodit/enrichment/enrichment_service.py,sha256=87Sd3gGbEMJYb_wVrHG8L1yGIZmQNR7foUS4_y94azI,977
 kodit/enrichment/enrichment_provider/__init__.py,sha256=klf8iuLVWX4iRz-DZQauFFNAoJC5CByczh48TBZPW-o,27
 kodit/enrichment/enrichment_provider/enrichment_provider.py,sha256=E0H5rq3OENM0yYbA8K_3nSnj5lUHCpoIOqpWLo-2MVU,413
@@ -36,14 +36,14 @@ kodit/indexing/__init__.py,sha256=cPyi2Iej3G1JFWlWr7X80_UrsMaTu5W5rBwgif1B3xo,75
 kodit/indexing/fusion.py,sha256=TZb4fPAedXdEUXzwzOofW98QIOymdbclBOP1KOijuEk,1674
 kodit/indexing/indexing_models.py,sha256=6NX9HVcj6Pu9ePwHC7n-PWSyAgukpJq0nCNmUIigtbo,1282
 kodit/indexing/indexing_repository.py,sha256=dqOS0pxKM6bUjMXWqYukAK8XdiD36OnskFASgZRXRQM,6955
-kodit/indexing/indexing_service.py,sha256=79BZ4yaSJqADkivzjsq1bDCBtbfWikVRC7Fjlp1HmZw,10885
+kodit/indexing/indexing_service.py,sha256=3hW7vbFyabLEkLU-PRoGR49yVLewANdOKlye4GhR-tw,11467
 kodit/migrations/README,sha256=ISVtAOvqvKk_5ThM5ioJE-lMkvf9IbknFUFVU_vPma4,58
 kodit/migrations/__init__.py,sha256=lP5MuwlyWRMO6UcDWnQcQ3G-GYHcFb6rl9gYPHJ1sjo,40
 kodit/migrations/env.py,sha256=w1M7OZh-ZeR2dPHS0ByXAUxQjfZQ8xIzMseWuzLDTWw,2469
 kodit/migrations/script.py.mako,sha256=zWziKtiwYKEWuwPV_HBNHwa9LCT45_bi01-uSNFaOOE,703
-kodit/migrations/versions/42e836b21102_add_authors.py,sha256=KmXlHb_y8bIa_ABNU67zZi13r0DAfHA9G8tjQNkdITM,2638
 kodit/migrations/versions/7c3bbc2ab32b_add_embeddings_table.py,sha256=-61qol9PfQKILCDQRA5jEaats9aGZs9Wdtp-j-38SF4,1644
 kodit/migrations/versions/85155663351e_initial.py,sha256=Cg7zlF871o9ShV5rQMQ1v7hRV7fI59veDY9cjtTrs-8,3306
+kodit/migrations/versions/9e53ea8bb3b0_add_authors.py,sha256=a32Zm8KUQyiiLkjKNPYdaJDgjW6VsV-GhaLnPnK_fpI,3884
 kodit/migrations/versions/__init__.py,sha256=9-lHzptItTzq_fomdIRBegQNm4Znx6pVjwD4MiqRIdo,36
 kodit/migrations/versions/c3f5137d30f5_index_all_the_things.py,sha256=rI8LmjF-I2OMxZ2nOIF_NRmqOLXe45hL_iz_nx97DTQ,1680
 kodit/snippets/__init__.py,sha256=-2coNoCRjTixU9KcP6alpmt7zqf37tCRWH3D7FPJ8dg,48
@@ -58,13 +58,14 @@ kodit/snippets/languages/typescript.scm,sha256=U-ujbbv4tylbUBj9wuhL-e5cW6hmgPCNs
 kodit/source/__init__.py,sha256=1NTZyPdjThVQpZO1Mp1ColVsS7sqYanOVLqnoqV9Ipo,83
 kodit/source/git.py,sha256=CpNczc06SbxpzfQKq76lZFzuol10ZJvTRSzeXW9DFUs,363
 kodit/source/ignore.py,sha256=W7cuIrYlgfu3S1qyoIepXe8PqYmtFv61Tt5RO8cbZbg,1701
-kodit/source/source_models.py,sha256=lCaaoukLlMHuRWJBuYM2nkNKGtFASgbk7ZXq8kp4H5c,3519
-kodit/source/source_repository.py,sha256=4L-W0uE4LOB9LQlefk5f2sgHlsJjj8t33USPxU0na40,4448
-kodit/source/source_service.py,sha256=v-lY-7tsNFCyXo9yCUo7Q00NOWYKGiDB_M2-Hr8hp3U,11391
+kodit/source/source_factories.py,sha256=qTjKBKu62vynHUxf3-lgxQCoj1nGn8aI9ExFUWNk5B8,11678
+kodit/source/source_models.py,sha256=XlgrY2dlpJkOFZQjrwMGwebAcwQ4rSfjshvCrHKA18Q,3916
+kodit/source/source_repository.py,sha256=eme0C3pRqwFZ1ZSbqq4Z6SV9CC6AvRmiOjy3eHQoEN0,5443
+kodit/source/source_service.py,sha256=E1KPG7TrorqdreJVHxZPx8CVLncOxGEvZ5uDQ6yZugo,5050
 kodit/util/__init__.py,sha256=bPu6CtqDWCRGU7VgW2_aiQrCBi8G89FS6k1PjvDajJ0,37
 kodit/util/spinner.py,sha256=R9bzrHtBiIH6IfLbmsIVHL53s8vg-tqW4lwGGALu4dw,1932
-kodit-0.2.2.dist-info/METADATA,sha256=UU1curOx-XMql_IiXty-eoz-MJrd5QdlzfCj7ZoSzhg,5857
-kodit-0.2.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kodit-0.2.2.dist-info/entry_points.txt,sha256=hoTn-1aKyTItjnY91fnO-rV5uaWQLQ-Vi7V5et2IbHY,40
-kodit-0.2.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kodit-0.2.2.dist-info/RECORD,,
+kodit-0.2.3.dist-info/METADATA,sha256=ccZ0bl5PCGFNC30XSr-4ljL-JrkTgTfYlqZUlUPYba8,5867
+kodit-0.2.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kodit-0.2.3.dist-info/entry_points.txt,sha256=hoTn-1aKyTItjnY91fnO-rV5uaWQLQ-Vi7V5et2IbHY,40
+kodit-0.2.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kodit-0.2.3.dist-info/RECORD,,

kodit/migrations/versions/42e836b21102_add_authors.py

@@ -1,64 +0,0 @@
-# ruff: noqa
-"""add authors
-
-Revision ID: 42e836b21102
-Revises: c3f5137d30f5
-Create Date: 2025-06-13 14:48:50.152940
-
-"""
-
-from typing import Sequence, Union
-
-from alembic import op
-import sqlalchemy as sa
-
-
-# revision identifiers, used by Alembic.
-revision: str = '42e836b21102'
-down_revision: Union[str, None] = 'c3f5137d30f5'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    """Upgrade schema."""
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table('authors',
-    sa.Column('name', sa.String(length=255), nullable=False),
-    sa.Column('email', sa.String(length=255), nullable=False),
-    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
-    sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
-    sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
-    sa.PrimaryKeyConstraint('id')
-    )
-    op.create_index(op.f('ix_authors_email'), 'authors', ['email'], unique=True)
-    op.create_index(op.f('ix_authors_name'), 'authors', ['name'], unique=True)
-    op.create_table('author_file_mappings',
-    sa.Column('author_id', sa.Integer(), nullable=False),
-    sa.Column('file_id', sa.Integer(), nullable=False),
-    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
-    sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
-    sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
-    sa.ForeignKeyConstraint(['author_id'], ['authors.id'], ),
-    sa.ForeignKeyConstraint(['file_id'], ['files.id'], ),
-    sa.PrimaryKeyConstraint('id')
-    )
-    op.add_column('files', sa.Column('extension', sa.String(length=255), nullable=False))
-    op.create_index(op.f('ix_files_extension'), 'files', ['extension'], unique=False)
-    op.add_column('sources', sa.Column('type', sa.Enum('UNKNOWN', 'FOLDER', 'GIT', name='sourcetype'), nullable=False))
-    op.create_index(op.f('ix_sources_type'), 'sources', ['type'], unique=False)
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    """Downgrade schema."""
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_index(op.f('ix_sources_type'), table_name='sources')
-    op.drop_column('sources', 'type')
-    op.drop_index(op.f('ix_files_extension'), table_name='files')
-    op.drop_column('files', 'extension')
-    op.drop_table('author_file_mappings')
-    op.drop_index(op.f('ix_authors_name'), table_name='authors')
-    op.drop_index(op.f('ix_authors_email'), table_name='authors')
-    op.drop_table('authors')
-    # ### end Alembic commands ###