diff --git a/.gitignore b/.gitignore index 18138a95..a518bfca 100644 --- a/.gitignore +++ b/.gitignore @@ -5,6 +5,7 @@ **/*.iml **/.DS_Store **/*.keystore +**/*.cgp **/release.properties **/RecentProject_database* **/agent/ diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 00000000..ad9dd4f3 --- /dev/null +++ b/.gitmodules @@ -0,0 +1,4 @@ +[submodule "ai-core/subprojects/llama.cpp"] + path = ai-core/subprojects/llama.cpp + url = https://github.com/appdevforall/llama.cpp.git + branch = androidide-custom diff --git a/README.md b/README.md index ce14550d..99b5b0ea 100644 --- a/README.md +++ b/README.md @@ -22,6 +22,8 @@ See the official [plugin documentation](https://www.appdevforall.org/codeonthego | [`compose-preview/`](compose-preview/) | Renders Jetpack Compose `@Preview` functions on-device — no full app build or run. | | [`ai-literacy-course/`](ai-literacy-course/) | Bundles Learn AI Anywhere's offline "Introduction to AI" course (26 videos + interactive activities) and plays it full-screen, fully offline. | | [`layout-editor/`](layout-editor/) | Visual drag-and-drop editor for Android XML layouts. | +| [`ai-core/`](ai-core/) | Shared on-device LLM inference backend (bundled llama.cpp AAR) plus a Gemini API backend, exposed to other plugins as a runtime service. | +| [`ai-assistant/`](ai-assistant/) | In-IDE AI chat assistant with tool calling; talks to `ai-core` for inference over local or Gemini models. | ## Building a plugin diff --git a/ai-assistant/.gitignore b/ai-assistant/.gitignore new file mode 100644 index 00000000..5380c6d5 --- /dev/null +++ b/ai-assistant/.gitignore @@ -0,0 +1,3 @@ +**/.cxx/ +build-output.log +**/.kotlin/ diff --git a/ai-assistant/README.md b/ai-assistant/README.md new file mode 100644 index 00000000..54a729ef --- /dev/null +++ b/ai-assistant/README.md @@ -0,0 +1,61 @@ +# AI Assistant plugin for CodeOnTheGo + +An in-IDE AI chat assistant with tool calling. It provides the chat UI, settings, +and the agent tool-loop (read/write files, list/search the project, add +dependencies, run Gradle sync, etc.), and delegates all model inference to the +sibling [`ai-core`](../ai-core/) plugin. + +> This is the **frontend** plugin. It has **no compile-time dependency** on +> `ai-core`; the two communicate at runtime through the CodeOnTheGo plugin +> manager (SharedServices). Install **`ai-core` first**, then `ai-assistant`. + +## Architecture + +``` +┌──────────────────────────┐ +│ ai-assistant (this) │ ← Chat UI, settings, agent tool-loop +└────────────┬─────────────┘ + │ SharedServices (runtime) + ▼ +┌──────────────────────────┐ +│ ai-core │ ← LLM inference: local (llama.cpp) + Gemini +└──────────────────────────┘ +``` + +## Features + +- Chat UI (RecyclerView + Markdown rendering via Markwon) +- Agent tool-loop with per-tool approval dialogs +- Session persistence +- Works with either backend exposed by `ai-core` (on-device GGUF or Gemini API) + +## Building + +Prerequisites: Android SDK (API 33+), JDK 17. Create `local.properties` with +`sdk.dir=...`. This plugin needs no NDK, submodule, or native toolchain. + +```bash +cd ai-assistant +./gradlew assemblePlugin # release -> build/plugin/ai-assistant.cgp +./gradlew assemblePluginDebug # debug variant +``` + +The build resolves `plugin-api.jar` from the repo-root `../libs/`. + +## Installation + +1. Build and install **`ai-core` first** (see [`../ai-core/README.md`](../ai-core/README.md)). +2. Build this plugin, copy `build/plugin/ai-assistant.cgp` to the device. +3. Install via CodeOnTheGo's Plugin Manager, then restart the IDE. +4. Open **AI Settings** to pick a local model or configure a Gemini API key. + +## Key classes + +- `AiAssistantPlugin.kt` — plugin entry point / lifecycle +- `fragments/ChatFragment.kt`, `viewmodel/ChatViewModel.kt` — chat UI + state +- `fragments/AiSettingsFragment.kt`, `viewmodel/AiSettingsViewModel.kt` — model/backend config +- `tool/` — the agent tool-loop (executor, router, per-tool handlers, approval) + +## License + +GPL-3.0 — same as AndroidIDE / CodeOnTheGo. diff --git a/ai-assistant/ai-assistant.html b/ai-assistant/ai-assistant.html new file mode 100644 index 00000000..2d52facc --- /dev/null +++ b/ai-assistant/ai-assistant.html @@ -0,0 +1,129 @@ + + + + + +AI Assistant Plugin — Documentation + + + +

AI Assistant Plugin

+

An AI coding assistant for Code on the Go (CoGo), with + on-device LLM inference and an optional Google Gemini cloud backend.

+ +

1. Executive Overview

+

The AI Assistant brings an interactive coding agent into the CoGo IDE. It + adds an Agent tab where you chat with a language model that can read, + search, and edit files in your project through an approval-gated tool loop. + It ships as two cooperating plugins:

+ +

The two plugins are bridged by a shared service registry + (SharedServices), so the UI plugin discovers the inference + service the core plugin registers at runtime.

+ +

2. Core Functionality

+ + +

3. Technical Architecture

+ 
ai-assistant  (UI, agent tool-loop)
+        │  discovers via SharedServices
+        ▼
+ai-core       (LlmInferenceService)
+        ├── LocalLlmBackend  ──► llama-impl ──► llama.cpp (native, JNI)
+        └── GeminiBackend    ──► Google GenAI SDK (HTTPS)
+ + +

4. Usage

+

Build

+

ai-core and ai-assistant are independent Gradle builds; a + normal build needs no submodule or NDK (the llama.cpp AAR ships prebuilt).

+ 
cd ai-core       && ./gradlew assemblePlugin
+cd ../ai-assistant && ./gradlew assemblePlugin
+

Outputs land at ai-core/build/plugin/ai-core.cgp and + ai-assistant/build/plugin/ai-assistant.cgp. See + ../ai-core/BUILDING.md for full prerequisites.

+

Install & configure

+ + +

5. Key Benefits

+ + + + + + +
BenefitWhy it matters
Offline capableOn-device inference works with no network and keeps code private.
Cloud optionGemini backend offers higher-quality responses when connectivity is available.
Project-awareThe agent operates on your actual files through IDE services, not copy-paste.
Safe by defaultPath containment plus approval prompts keep file edits scoped and consented.
+ +

+ Author: AndroidIDE / appdevforall. See README.md for developer + details and the in-IDE Agent-tab tooltip for quick help.

+ + diff --git a/ai-assistant/build.gradle.kts b/ai-assistant/build.gradle.kts new file mode 100644 index 00000000..b376da7c --- /dev/null +++ b/ai-assistant/build.gradle.kts @@ -0,0 +1,86 @@ +plugins { + id("com.android.application") + id("org.jetbrains.kotlin.android") + id("com.itsaky.androidide.plugins.build") +} + +pluginBuilder { + pluginName = "ai-assistant" +} + +android { + namespace = "com.itsaky.androidide.plugins.aiassistant" + compileSdk = 34 + + defaultConfig { + applicationId = "com.itsaky.androidide.plugins.aiassistant" + minSdk = 33 + targetSdk = 34 + versionCode = 1 + versionName = "1.0.0" + } + + buildFeatures { + viewBinding = true + } + + buildTypes { + release { + // Disable minification to avoid lambda obfuscation issues with ClassLoader isolation + isMinifyEnabled = false + isShrinkResources = false + signingConfig = signingConfigs.getByName("debug") + proguardFiles(getDefaultProguardFile("proguard-android-optimize.txt"), "proguard-rules.pro") + } + } + + compileOptions { + sourceCompatibility = JavaVersion.VERSION_17 + targetCompatibility = JavaVersion.VERSION_17 + } + + kotlin { + compilerOptions { + jvmTarget.set(org.jetbrains.kotlin.gradle.dsl.JvmTarget.JVM_17) + } + } + + packaging { + resources { + excludes += setOf( + "META-INF/DEPENDENCIES", + "META-INF/LICENSE", + "META-INF/LICENSE.txt", + "META-INF/NOTICE", + "META-INF/NOTICE.txt" + ) + } + } +} + +dependencies { + compileOnly(files("../libs/plugin-api.jar")) + + // Use 'implementation' (not 'compileOnly') for androidx libraries. + // This is required for XML layouts: AAPT2 needs these dependencies at compile-time to process + // resource attributes and resolve xmlns declarations. This is standard across all CoGo plugins + // with XML layouts (random-xkcd, sketch-to-ui-plugin, Beepy). See investigation in Task 4. + implementation("androidx.appcompat:appcompat:1.6.1") + implementation("androidx.fragment:fragment-ktx:1.6.2") + implementation("com.google.android.material:material:1.10.0") + implementation("androidx.recyclerview:recyclerview:1.3.2") + implementation("androidx.constraintlayout:constraintlayout:2.1.4") + + // Markdown rendering - plugin-specific library + implementation("io.noties.markwon:core:4.6.2") + + // JSON serialization for session persistence + implementation("com.google.code.gson:gson:2.10.1") + + // Plugin dependencies are loaded at runtime by the plugin manager + // No explicit compile-time dependency on the ai-core plugin needed + testImplementation("junit:junit:4.13.2") + testImplementation("io.mockk:mockk:1.13.8") + testImplementation("org.jetbrains.kotlinx:kotlinx-coroutines-test:1.7.3") + testImplementation("androidx.arch.core:core-testing:2.2.0") +} diff --git a/ai-assistant/gradle.properties b/ai-assistant/gradle.properties new file mode 100644 index 00000000..fcd58cda --- /dev/null +++ b/ai-assistant/gradle.properties @@ -0,0 +1,10 @@ +android.enableJetifier=false +android.jetifier.ignorelist=common-30.2.2.jar +android.nonTransitiveRClass=false +android.useAndroidX=true +org.gradle.caching=true +org.gradle.configureondemand=true +org.gradle.jvmargs=-Xmx4096M -Dkotlin.daemon.jvm.options\="-Xmx4096M" +org.gradle.parallel=true + +kotlin.code.style=official diff --git a/ai-assistant/gradle/wrapper/gradle-wrapper.jar b/ai-assistant/gradle/wrapper/gradle-wrapper.jar new file mode 100755 index 00000000..d64cd491 Binary files /dev/null and b/ai-assistant/gradle/wrapper/gradle-wrapper.jar differ diff --git a/ai-assistant/gradle/wrapper/gradle-wrapper.properties b/ai-assistant/gradle/wrapper/gradle-wrapper.properties new file mode 100644 index 00000000..692c2dc2 --- /dev/null +++ b/ai-assistant/gradle/wrapper/gradle-wrapper.properties @@ -0,0 +1,7 @@ +distributionBase=GRADLE_USER_HOME +distributionPath=wrapper/dists +distributionUrl=https\://services.gradle.org/distributions/gradle-8.14.4-all.zip +networkTimeout=10000 +validateDistributionUrl=true +zipStoreBase=GRADLE_USER_HOME +zipStorePath=wrapper/dists \ No newline at end of file diff --git a/ai-assistant/gradlew b/ai-assistant/gradlew new file mode 100755 index 00000000..1aa94a42 --- /dev/null +++ b/ai-assistant/gradlew @@ -0,0 +1,249 @@ +#!/bin/sh + +# +# Copyright © 2015-2021 the original authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# https://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +############################################################################## +# +# Gradle start up script for POSIX generated by Gradle. +# +# Important for running: +# +# (1) You need a POSIX-compliant shell to run this script. If your /bin/sh is +# noncompliant, but you have some other compliant shell such as ksh or +# bash, then to run this script, type that shell name before the whole +# command line, like: +# +# ksh Gradle +# +# Busybox and similar reduced shells will NOT work, because this script +# requires all of these POSIX shell features: +# * functions; +# * expansions «$var», «${var}», «${var:-default}», «${var+SET}», +# «${var#prefix}», «${var%suffix}», and «$( cmd )»; +# * compound commands having a testable exit status, especially «case»; +# * various built-in commands including «command», «set», and «ulimit». +# +# Important for patching: +# +# (2) This script targets any POSIX shell, so it avoids extensions provided +# by Bash, Ksh, etc; in particular arrays are avoided. +# +# The "traditional" practice of packing multiple parameters into a +# space-separated string is a well documented source of bugs and security +# problems, so this is (mostly) avoided, by progressively accumulating +# options in "$@", and eventually passing that to Java. +# +# Where the inherited environment variables (DEFAULT_JVM_OPTS, JAVA_OPTS, +# and GRADLE_OPTS) rely on word-splitting, this is performed explicitly; +# see the in-line comments for details. +# +# There are tweaks for specific operating systems such as AIX, CygWin, +# Darwin, MinGW, and NonStop. +# +# (3) This script is generated from the Groovy template +# https://github.com/gradle/gradle/blob/HEAD/subprojects/plugins/src/main/resources/org/gradle/api/internal/plugins/unixStartScript.txt +# within the Gradle project. +# +# You can find Gradle at https://github.com/gradle/gradle/. +# +############################################################################## + +# Attempt to set APP_HOME + +# Resolve links: $0 may be a link +app_path=$0 + +# Need this for daisy-chained symlinks. +while + APP_HOME=${app_path%"${app_path##*/}"} # leaves a trailing /; empty if no leading path + [ -h "$app_path" ] +do + ls=$( ls -ld "$app_path" ) + link=${ls#*' -> '} + case $link in #( + /*) app_path=$link ;; #( + *) app_path=$APP_HOME$link ;; + esac +done + +# This is normally unused +# shellcheck disable=SC2034 +APP_BASE_NAME=${0##*/} +# Discard cd standard output in case $CDPATH is set (https://github.com/gradle/gradle/issues/25036) +APP_HOME=$( cd "${APP_HOME:-./}" > /dev/null && pwd -P ) || exit + +# Use the maximum available, or set MAX_FD != -1 to use that value. +MAX_FD=maximum + +warn () { + echo "$*" +} >&2 + +die () { + echo + echo "$*" + echo + exit 1 +} >&2 + +# OS specific support (must be 'true' or 'false'). +cygwin=false +msys=false +darwin=false +nonstop=false +case "$( uname )" in #( + CYGWIN* ) cygwin=true ;; #( + Darwin* ) darwin=true ;; #( + MSYS* | MINGW* ) msys=true ;; #( + NONSTOP* ) nonstop=true ;; +esac + +CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar + + +# Determine the Java command to use to start the JVM. +if [ -n "$JAVA_HOME" ] ; then + if [ -x "$JAVA_HOME/jre/sh/java" ] ; then + # IBM's JDK on AIX uses strange locations for the executables + JAVACMD=$JAVA_HOME/jre/sh/java + else + JAVACMD=$JAVA_HOME/bin/java + fi + if [ ! -x "$JAVACMD" ] ; then + die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME + +Please set the JAVA_HOME variable in your environment to match the +location of your Java installation." + fi +else + JAVACMD=java + if ! command -v java >/dev/null 2>&1 + then + die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. + +Please set the JAVA_HOME variable in your environment to match the +location of your Java installation." + fi +fi + +# Increase the maximum file descriptors if we can. +if ! "$cygwin" && ! "$darwin" && ! "$nonstop" ; then + case $MAX_FD in #( + max*) + # In POSIX sh, ulimit -H is undefined. That's why the result is checked to see if it worked. + # shellcheck disable=SC2039,SC3045 + MAX_FD=$( ulimit -H -n ) || + warn "Could not query maximum file descriptor limit" + esac + case $MAX_FD in #( + '' | soft) :;; #( + *) + # In POSIX sh, ulimit -n is undefined. That's why the result is checked to see if it worked. + # shellcheck disable=SC2039,SC3045 + ulimit -n "$MAX_FD" || + warn "Could not set maximum file descriptor limit to $MAX_FD" + esac +fi + +# Collect all arguments for the java command, stacking in reverse order: +# * args from the command line +# * the main class name +# * -classpath +# * -D...appname settings +# * --module-path (only if needed) +# * DEFAULT_JVM_OPTS, JAVA_OPTS, and GRADLE_OPTS environment variables. + +# For Cygwin or MSYS, switch paths to Windows format before running java +if "$cygwin" || "$msys" ; then + APP_HOME=$( cygpath --path --mixed "$APP_HOME" ) + CLASSPATH=$( cygpath --path --mixed "$CLASSPATH" ) + + JAVACMD=$( cygpath --unix "$JAVACMD" ) + + # Now convert the arguments - kludge to limit ourselves to /bin/sh + for arg do + if + case $arg in #( + -*) false ;; # don't mess with options #( + /?*) t=${arg#/} t=/${t%%/*} # looks like a POSIX filepath + [ -e "$t" ] ;; #( + *) false ;; + esac + then + arg=$( cygpath --path --ignore --mixed "$arg" ) + fi + # Roll the args list around exactly as many times as the number of + # args, so each arg winds up back in the position where it started, but + # possibly modified. + # + # NB: a `for` loop captures its iteration list before it begins, so + # changing the positional parameters here affects neither the number of + # iterations, nor the values presented in `arg`. + shift # remove old arg + set -- "$@" "$arg" # push replacement arg + done +fi + + +# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. +DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"' + +# Collect all arguments for the java command: +# * DEFAULT_JVM_OPTS, JAVA_OPTS, JAVA_OPTS, and optsEnvironmentVar are not allowed to contain shell fragments, +# and any embedded shellness will be escaped. +# * For example: A user cannot expect ${Hostname} to be expanded, as it is an environment variable and will be +# treated as '${Hostname}' itself on the command line. + +set -- \ + "-Dorg.gradle.appname=$APP_BASE_NAME" \ + -classpath "$CLASSPATH" \ + org.gradle.wrapper.GradleWrapperMain \ + "$@" + +# Stop when "xargs" is not available. +if ! command -v xargs >/dev/null 2>&1 +then + die "xargs is not available" +fi + +# Use "xargs" to parse quoted args. +# +# With -n1 it outputs one arg per line, with the quotes and backslashes removed. +# +# In Bash we could simply go: +# +# readarray ARGS < <( xargs -n1 <<<"$var" ) && +# set -- "${ARGS[@]}" "$@" +# +# but POSIX shell has neither arrays nor command substitution, so instead we +# post-process each arg (as a line of input to sed) to backslash-escape any +# character that might be a shell metacharacter, then use eval to reverse +# that process (while maintaining the separation between arguments), and wrap +# the whole thing up as a single "set" statement. +# +# This will of course break if any of these variables contains a newline or +# an unmatched quote. +# + +eval "set -- $( + printf '%s\n' "$DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS" | + xargs -n1 | + sed ' s~[^-[:alnum:]+,./:=@_]~\\&~g; ' | + tr '\n' ' ' + )" '"$@"' + +exec "$JAVACMD" "$@" diff --git a/ai-assistant/gradlew.bat b/ai-assistant/gradlew.bat new file mode 100755 index 00000000..93e3f59f --- /dev/null +++ b/ai-assistant/gradlew.bat @@ -0,0 +1,92 @@ +@rem +@rem Copyright 2015 the original author or authors. +@rem +@rem Licensed under the Apache License, Version 2.0 (the "License"); +@rem you may not use this file except in compliance with the License. +@rem You may obtain a copy of the License at +@rem +@rem https://www.apache.org/licenses/LICENSE-2.0 +@rem +@rem Unless required by applicable law or agreed to in writing, software +@rem distributed under the License is distributed on an "AS IS" BASIS, +@rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +@rem See the License for the specific language governing permissions and +@rem limitations under the License. +@rem + +@if "%DEBUG%"=="" @echo off +@rem ########################################################################## +@rem +@rem Gradle startup script for Windows +@rem +@rem ########################################################################## + +@rem Set local scope for the variables with windows NT shell +if "%OS%"=="Windows_NT" setlocal + +set DIRNAME=%~dp0 +if "%DIRNAME%"=="" set DIRNAME=. +@rem This is normally unused +set APP_BASE_NAME=%~n0 +set APP_HOME=%DIRNAME% + +@rem Resolve any "." and ".." in APP_HOME to make it shorter. +for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi + +@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. +set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m" + +@rem Find java.exe +if defined JAVA_HOME goto findJavaFromJavaHome + +set JAVA_EXE=java.exe +%JAVA_EXE% -version >NUL 2>&1 +if %ERRORLEVEL% equ 0 goto execute + +echo. +echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. +echo. +echo Please set the JAVA_HOME variable in your environment to match the +echo location of your Java installation. + +goto fail + +:findJavaFromJavaHome +set JAVA_HOME=%JAVA_HOME:"=% +set JAVA_EXE=%JAVA_HOME%/bin/java.exe + +if exist "%JAVA_EXE%" goto execute + +echo. +echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME% +echo. +echo Please set the JAVA_HOME variable in your environment to match the +echo location of your Java installation. + +goto fail + +:execute +@rem Setup the command line + +set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar + + +@rem Execute Gradle +"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %* + +:end +@rem End local scope for the variables with windows NT shell +if %ERRORLEVEL% equ 0 goto mainEnd + +:fail +rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of +rem the _cmd.exe /c_ return code! +set EXIT_CODE=%ERRORLEVEL% +if %EXIT_CODE% equ 0 set EXIT_CODE=1 +if not ""=="%GRADLE_EXIT_CONSOLE%" exit %EXIT_CODE% +exit /b %EXIT_CODE% + +:mainEnd +if "%OS%"=="Windows_NT" endlocal + +:omega diff --git a/ai-assistant/proguard-rules.pro b/ai-assistant/proguard-rules.pro new file mode 100644 index 00000000..f53bb79d --- /dev/null +++ b/ai-assistant/proguard-rules.pro @@ -0,0 +1,53 @@ +# Keep plugin main class +-keep class com.itsaky.androidide.plugins.aiassistant.AiAssistantPlugin { + public (); + public boolean initialize(com.itsaky.androidide.plugins.PluginContext); + public boolean activate(); + public boolean deactivate(); + public void dispose(); + public java.util.List getEditorTabs(); + public java.util.List getContextMenuItems(com.itsaky.androidide.plugins.extensions.ContextMenuContext); + public java.util.List getMainMenuItems(); +} + +# Keep Fragment classes +-keep class com.itsaky.androidide.plugins.aiassistant.fragments.ChatFragment { + public (); +} + +# Keep plugin API related classes +-keep interface com.itsaky.androidide.plugins.IPlugin { + *; +} + +-keep interface com.itsaky.androidide.plugins.extensions.UIExtension { + *; +} + +-keep class com.itsaky.androidide.plugins.extensions.TabItem { + *; +} + +-keep class com.itsaky.androidide.plugins.extensions.MenuItem { + *; +} + +# Keep annotations +-keepattributes *Annotation* +-keepattributes Signature +-keepattributes Exceptions + +# Keep Kotlin lambdas and function types - CRITICAL for TabItem.fragmentFactory +-keep class kotlin.jvm.functions.** { *; } +-keep class kotlin.jvm.internal.** { *; } +-keepclassmembers class ** { + kotlin.jvm.functions.Function0 fragmentFactory; +} + +# Don't obfuscate lambda implementations +-keep class **$$Lambda$* { *; } + +# Keep synthetic methods (lambdas) +-keepclassmembers class * { + synthetic ; +} diff --git a/ai-assistant/settings.gradle.kts b/ai-assistant/settings.gradle.kts new file mode 100644 index 00000000..de2975f0 --- /dev/null +++ b/ai-assistant/settings.gradle.kts @@ -0,0 +1,35 @@ +@file:Suppress("UnstableApiUsage") + +enableFeaturePreview("TYPESAFE_PROJECT_ACCESSORS") + +pluginManagement { + repositories { + gradlePluginPortal() + google() + mavenCentral() + } +} + +buildscript { + repositories { + google() + mavenCentral() + } + dependencies { + classpath(files("../libs/plugin-api.jar")) + classpath(files("../libs/gradle-plugin.jar")) + classpath("com.android.tools.build:gradle:8.13.2") + classpath("org.jetbrains.kotlin:kotlin-gradle-plugin:2.3.0") + } +} + +dependencyResolutionManagement { + repositoriesMode.set(RepositoriesMode.FAIL_ON_PROJECT_REPOS) + repositories { + google() + mavenCentral() + maven { url = uri("https://jitpack.io") } + } +} + +rootProject.name = "ai-assistant" diff --git a/ai-assistant/src/main/AndroidManifest.xml b/ai-assistant/src/main/AndroidManifest.xml new file mode 100644 index 00000000..8b593826 --- /dev/null +++ b/ai-assistant/src/main/AndroidManifest.xml @@ -0,0 +1,61 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + diff --git a/ai-assistant/src/main/assets/docs/index.html b/ai-assistant/src/main/assets/docs/index.html new file mode 100644 index 00000000..fb09f85f --- /dev/null +++ b/ai-assistant/src/main/assets/docs/index.html @@ -0,0 +1,80 @@ + + + + + +AI Assistant — Guide + + + +

AI Assistant — Agent Guide

+ +

The Agent tab is a chat assistant that can read, search, and edit + your project through an approval-gated tool loop. Inference is served by the + companion AI Core plugin, so install AI Core first.

+ +

Choosing a backend

+
    +
  • Local (on-device) — runs a .gguf model via + llama.cpp. Open Settings, pick a model file from your Downloads + folder. Nothing leaves the device.
    • +
  • Gemini (cloud) — enter a Gemini API key in Settings. + Prompts and any file contents the agent reads are sent to Google over + HTTPS.
    • +
+ +

What the agent can do

+
    +
  • Read and search files within the current project.
    • +
  • Create and update files (asks for approval before writing).
    • +
  • Trigger a Gradle sync and read build output.
    • +
  • Add dependencies and generate code from templates.
    • +
+ +
+ File tools are confined to the project root: paths that resolve outside it + are rejected, so the model cannot read or overwrite arbitrary files on the + device. +
+ +

Context-menu actions

+

Select code in the editor and open the context menu for + Explain Code and Generate Code shortcuts.

+ +

Troubleshooting

+
    +
  • "No model configured" — select a .gguf file (Local) + or set an API key (Gemini) in Settings.
    • +
  • Gemini errors — check the API key, network connection, and quota.
    • +
  • Agent tab missing — confirm both AI Core and AI Assistant are + installed and the IDE was restarted.
    • +
+ + diff --git a/ai-assistant/src/main/assets/icon_day.png b/ai-assistant/src/main/assets/icon_day.png new file mode 100644 index 00000000..4cb1dc35 Binary files /dev/null and b/ai-assistant/src/main/assets/icon_day.png differ diff --git a/ai-assistant/src/main/assets/icon_night.png b/ai-assistant/src/main/assets/icon_night.png new file mode 100644 index 00000000..355a5c0e Binary files /dev/null and b/ai-assistant/src/main/assets/icon_night.png differ diff --git a/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/AiAssistantPlugin.kt b/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/AiAssistantPlugin.kt new file mode 100644 index 00000000..92cd5220 --- /dev/null +++ b/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/AiAssistantPlugin.kt @@ -0,0 +1,243 @@ +package com.itsaky.androidide.plugins.aiassistant + +import com.itsaky.androidide.plugins.IPlugin +import com.itsaky.androidide.plugins.PluginContext +import com.itsaky.androidide.plugins.extensions.UIExtension +import com.itsaky.androidide.plugins.extensions.ContextMenuContext +import com.itsaky.androidide.plugins.extensions.DocumentationExtension +import com.itsaky.androidide.plugins.extensions.MenuItem +import com.itsaky.androidide.plugins.extensions.PluginTooltipButton +import com.itsaky.androidide.plugins.extensions.PluginTooltipEntry +import com.itsaky.androidide.plugins.extensions.TabItem +import com.itsaky.androidide.plugins.services.LlmInferenceService +import com.itsaky.androidide.plugins.services.SharedServices +import com.itsaky.androidide.plugins.aiassistant.fragments.ChatFragment +import java.io.File + +class AiAssistantPlugin : IPlugin, UIExtension, DocumentationExtension { + + private lateinit var context: PluginContext + private var llmService: LlmInferenceService? = null + + companion object { + const val TOOLTIP_TAG_TAB = "agent_chat_tab" + + @Volatile + private var pluginContext: PluginContext? = null + + fun getContext(): PluginContext? = pluginContext + } + + override fun initialize(context: PluginContext): Boolean { + this.context = context + pluginContext = context // Store for ChatFragment access + + // Also store in SharedServices so ai-core can access preferences + SharedServices.register(PluginContext::class.java, context) + + context.logger.info("AI Assistant Plugin initializing...") + return true + } + + override fun activate(): Boolean { + // Get LlmInferenceService from SharedServices + llmService = SharedServices.get(LlmInferenceService::class.java) + + if (llmService == null) { + context.logger.warn("LlmInferenceService not available - LOCAL_LLM backend disabled") + context.logger.warn("Install AI Core plugin to enable local LLM support") + } else { + context.logger.info("LlmInferenceService available from SharedServices") + } + + // Migrate chat history and settings on first activation + migrateDataIfNeeded() + + return true + } + + override fun deactivate(): Boolean { + context.logger.info("AI Assistant Plugin deactivating...") + return true + } + + override fun dispose() { + context.logger.info("AI Assistant Plugin disposing...") + + // Release the shared references set up in initialize() so the plugin's + // PluginContext (and everything it holds) can be garbage-collected when + // the plugin is unloaded. + SharedServices.unregister(PluginContext::class.java) + pluginContext = null + llmService = null + } + + // Register Agent tab + override fun getEditorTabs(): List { + return listOf( + TabItem( + id = "agent_chat", + title = "Agent", + order = 100, + fragmentFactory = { ChatFragment() }, + isEnabled = true, + isVisible = true, + tooltipTag = TOOLTIP_TAG_TAB + ) + ) + } + + override fun getContextMenuItems(menuContext: ContextMenuContext): List { + val selectedText = menuContext.selectedText + if (selectedText.isNullOrBlank()) { + return emptyList() + } + + return listOf( + MenuItem( + id = "ai_explain_code", + title = "Explain Code", + isEnabled = true, + isVisible = true, + action = { context.logger.info("Explain Code clicked") } + ), + MenuItem( + id = "ai_generate_code", + title = "Generate Code", + isEnabled = true, + isVisible = true, + action = { context.logger.info("Generate Code clicked") } + ) + ) + } + + override fun getMainMenuItems(): List = emptyList() + + // --- DocumentationExtension: three-tier tooltip help for the Agent tab --- + // + // Tier 1 = `summary` (one-liner shown on long-press) + // Tier 2 = `detail` (HTML paragraph behind "See More") + // Tier 3 = `buttons[].uri` (offline HTML page served from + // src/main/assets/docs/ at localhost) + + override fun getTooltipCategory(): String = "plugin_ai_assistant" + + override fun getTooltipEntries(): List = listOf( + PluginTooltipEntry( + tag = TOOLTIP_TAG_TAB, + summary = "AI Agent: chat with an on-device or Gemini model that can read, search and edit your project.", + detail = """ +

The Agent tab opens a chat assistant backed by the + AI Core plugin. It can answer questions and run an + agentic tool-loop over your project.

+

Backends:

+
    +
  • Local — on-device inference via llama.cpp (select a + .gguf model in Settings).
    • +
  • Gemini — Google's cloud API (needs an API key in + Settings; requests leave the device over HTTPS).
    • +
+

File-editing tools are confined to the current project and + ask for approval before writing.

+ """.trimIndent(), + buttons = listOf( + PluginTooltipButton( + description = "AI Assistant guide", + uri = "index.html", // resolves to plugin//index.html + order = 0 + ) + ) + ) + ) + + /** Subdirectory under src/main/assets/ holding the Tier 3 offline docs. */ + override fun getTier3DocsAssetPath(): String = "docs" + + private fun migrateDataIfNeeded() { + migrateChatHistory() + migrateSettings() + } + + private fun migrateChatHistory() { + try { + val appChatDir = File(context.getAppFilesDir(), "chat_sessions") + val pluginChatDir = File(context.getPluginFilesDir(), "chat_sessions") + + if (appChatDir.exists() && !pluginChatDir.exists()) { + context.logger.info("Migrating chat history from app to plugin storage") + pluginChatDir.mkdirs() + + var migratedCount = 0 + appChatDir.listFiles()?.forEach { file -> + val targetFile = File(pluginChatDir, file.name) + if (!targetFile.exists()) { + file.copyTo(targetFile, overwrite = false) + migratedCount++ + } + } + + context.logger.info("Migrated $migratedCount chat session files") + // Keep original files (don't delete) + } else if (pluginChatDir.exists()) { + context.logger.info("Chat history already migrated") + } + } catch (e: Exception) { + context.logger.error("Failed to migrate chat history", e) + } + } + + private fun migrateSettings() { + try { + val appPrefs = context.getAppSharedPreferences("LlamaPrefs") + if (appPrefs == null) { + context.logger.info("App preferences not found, skipping settings migration") + return + } + + val pluginPrefs = context.getPluginSharedPreferences("AgentSettings") + + val PREF_KEY_AI_BACKEND = "ai_backend_preference" + val PREF_KEY_LOCAL_MODEL_PATH = "local_llm_model_path" + val PREF_KEY_LOCAL_MODEL_SHA256 = "local_llm_model_sha256" + + var migratedCount = 0 + + // Migrate backend preference + if (!pluginPrefs.contains(PREF_KEY_AI_BACKEND)) { + val backend = appPrefs.getString(PREF_KEY_AI_BACKEND, null) + if (backend != null) { + pluginPrefs.edit().putString(PREF_KEY_AI_BACKEND, backend).apply() + migratedCount++ + } + } + + // Migrate model path + if (!pluginPrefs.contains(PREF_KEY_LOCAL_MODEL_PATH)) { + val modelPath = appPrefs.getString(PREF_KEY_LOCAL_MODEL_PATH, null) + if (modelPath != null) { + pluginPrefs.edit().putString(PREF_KEY_LOCAL_MODEL_PATH, modelPath).apply() + migratedCount++ + } + } + + // Migrate model SHA256 + if (!pluginPrefs.contains(PREF_KEY_LOCAL_MODEL_SHA256)) { + val sha256 = appPrefs.getString(PREF_KEY_LOCAL_MODEL_SHA256, null) + if (sha256 != null) { + pluginPrefs.edit().putString(PREF_KEY_LOCAL_MODEL_SHA256, sha256).apply() + migratedCount++ + } + } + + // Note: Encrypted Gemini API key migration handled by EncryptedPrefs + + if (migratedCount > 0) { + context.logger.info("Migrated $migratedCount settings from app to plugin") + } else { + context.logger.info("Settings already migrated") + } + } catch (e: Exception) { + context.logger.error("Failed to migrate settings", e) + } + } +} diff --git a/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/adapters/ChatAdapter.kt b/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/adapters/ChatAdapter.kt new file mode 100644 index 00000000..e9df31bc --- /dev/null +++ b/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/adapters/ChatAdapter.kt @@ -0,0 +1,377 @@ +package com.itsaky.androidide.plugins.aiassistant.adapters + +import android.content.ClipData +import android.content.ClipboardManager +import android.content.Context +import android.view.LayoutInflater +import android.view.View +import android.view.ViewGroup +import android.widget.Button +import android.widget.ImageView +import android.widget.LinearLayout +import android.widget.PopupMenu +import android.widget.ProgressBar +import android.widget.TextView +import android.widget.Toast +import androidx.recyclerview.widget.DiffUtil +import androidx.recyclerview.widget.ListAdapter +import androidx.recyclerview.widget.RecyclerView +import com.itsaky.androidide.plugins.aiassistant.R +import com.itsaky.androidide.plugins.aiassistant.models.ChatMessage +import com.itsaky.androidide.plugins.aiassistant.models.MessageStatus +import com.itsaky.androidide.plugins.aiassistant.models.Sender +import io.noties.markwon.Markwon +import java.text.DecimalFormat +import java.text.SimpleDateFormat +import java.util.Date +import java.util.Locale + +class ChatAdapter( + private val pluginContext: Context, + private val markwon: Markwon, + private val onMessageAction: (action: String, message: ChatMessage) -> Unit +) : ListAdapter(DiffCallback) { + + private val timeFormatter = SimpleDateFormat("h:mm a", Locale.getDefault()) + private val decimalSecondsFormatter = DecimalFormat("0.0") + private val expandedMessageIds = mutableSetOf() + + companion object { + private const val VIEW_TYPE_DEFAULT = 0 + private const val VIEW_TYPE_SYSTEM = 1 + + const val ACTION_EDIT = "edit" + const val ACTION_RETRY = "retry" + const val ACTION_OPEN_SETTINGS = "open_settings" + } + + sealed class MessageViewHolder(view: View) : RecyclerView.ViewHolder(view) + + class DefaultMessageViewHolder(view: View) : MessageViewHolder(view) { + val messageSender: TextView = view.findViewById(R.id.message_sender) + val loadingIndicator: ProgressBar = view.findViewById(R.id.loading_indicator) + val messageContent: TextView = view.findViewById(R.id.message_content) + val messageMetadataContainer: LinearLayout = view.findViewById(R.id.message_metadata_container) + val messageTimestamp: TextView = view.findViewById(R.id.message_timestamp) + val generatingDots: TextView = view.findViewById(R.id.generating_dots) + val messageDuration: TextView = view.findViewById(R.id.message_duration) + val btnRetry: Button = view.findViewById(R.id.btn_retry) + } + + class SystemMessageViewHolder(view: View) : MessageViewHolder(view) { + val messageHeader: LinearLayout = view.findViewById(R.id.message_header) + val messageHeaderTitle: TextView = view.findViewById(R.id.message_header_title) + val expandIcon: ImageView = view.findViewById(R.id.expand_icon) + val messageContent: TextView = view.findViewById(R.id.message_content) + } + + override fun getItemCount(): Int { + val count = super.getItemCount() + android.util.Log.d("ChatAdapter", "getItemCount() = $count") + return count + } + + override fun getItemViewType(position: Int): Int { + val message = getItem(position) + return if (message.sender == Sender.SYSTEM && message.status == MessageStatus.ERROR) { + VIEW_TYPE_DEFAULT + } else if (message.sender == Sender.SYSTEM) { + VIEW_TYPE_SYSTEM + } else { + VIEW_TYPE_DEFAULT + } + } + + override fun onCreateViewHolder(parent: ViewGroup, viewType: Int): RecyclerView.ViewHolder { + android.util.Log.d("ChatAdapter", "onCreateViewHolder called, viewType=$viewType") + // Use plugin context for inflating layouts to access plugin resources + val inflater = LayoutInflater.from(pluginContext) + return when (viewType) { + VIEW_TYPE_SYSTEM -> { + val view = inflater.inflate(R.layout.list_item_chat_system_message, parent, false) + SystemMessageViewHolder(view) + } + else -> { + val view = inflater.inflate(R.layout.list_item_chat_message, parent, false) + DefaultMessageViewHolder(view) + } + } + } + + override fun onBindViewHolder(holder: RecyclerView.ViewHolder, position: Int) { + val message = getItem(position) + when (holder) { + is DefaultMessageViewHolder -> bindDefaultMessage(holder, message) + is SystemMessageViewHolder -> bindSystemMessage(holder, message) + } + } + + override fun onBindViewHolder(holder: RecyclerView.ViewHolder, position: Int, payloads: MutableList) { + if (payloads.isEmpty()) { + // No payload, do full bind + onBindViewHolder(holder, position) + } else { + // Handle payload update + val payload = payloads[0] + if (payload is TextUpdatePayload && holder is DefaultMessageViewHolder) { + val message = getItem(position) + // Only update the text content and status, don't rebind everything + when (payload.status) { + MessageStatus.LOADING -> { + holder.loadingIndicator.visibility = View.VISIBLE + holder.messageContent.visibility = View.GONE + holder.generatingDots.visibility = View.GONE + } + MessageStatus.SENT -> { + holder.loadingIndicator.visibility = View.GONE + holder.messageContent.visibility = View.VISIBLE + markwon.setMarkdown(holder.messageContent, payload.text) + + // Show dots animation for AGENT messages being generated + if (message.sender == Sender.AGENT && message.durationMs == null) { + animateGeneratingDots(holder) + } else { + holder.generatingDots.visibility = View.GONE + } + } + MessageStatus.COMPLETED -> { + holder.loadingIndicator.visibility = View.GONE + holder.messageContent.visibility = View.VISIBLE + holder.generatingDots.visibility = View.GONE + markwon.setMarkdown(holder.messageContent, payload.text) + } + MessageStatus.ERROR -> { + holder.loadingIndicator.visibility = View.GONE + holder.messageContent.visibility = View.VISIBLE + holder.generatingDots.visibility = View.GONE + holder.messageContent.text = payload.text + } + } + } else if (payload is TextUpdatePayload && holder is SystemMessageViewHolder) { + markwon.setMarkdown(holder.messageContent, payload.text) + updateSystemMessageExpansion(holder, getItem(position)) + } else { + // Unknown payload, do full bind + onBindViewHolder(holder, position) + } + } + } + + private fun bindDefaultMessage(holder: DefaultMessageViewHolder, message: ChatMessage) { + holder.messageSender.text = message.sender.name.lowercase(Locale.getDefault()) + .replaceFirstChar { it.titlecase(Locale.getDefault()) } + + holder.itemView.setOnLongClickListener { view -> + if (message.status == MessageStatus.SENT) { + showContextMenu(view, message) + } + true + } + + when (message.status) { + MessageStatus.LOADING -> { + holder.loadingIndicator.visibility = View.VISIBLE + holder.messageContent.visibility = View.GONE + holder.btnRetry.visibility = View.GONE + holder.messageMetadataContainer.visibility = View.GONE + } + MessageStatus.SENT -> { + holder.loadingIndicator.visibility = View.GONE + holder.messageContent.visibility = View.VISIBLE + holder.btnRetry.visibility = View.GONE + markwon.setMarkdown(holder.messageContent, message.text) + updateMessageMetadata(holder, message) + + // Show dots animation for AGENT messages being generated + if (message.sender == Sender.AGENT && message.durationMs == null) { + animateGeneratingDots(holder) + } else { + holder.generatingDots.visibility = View.GONE + } + } + MessageStatus.COMPLETED -> { + holder.loadingIndicator.visibility = View.GONE + holder.messageContent.visibility = View.VISIBLE + holder.btnRetry.visibility = View.GONE + holder.generatingDots.visibility = View.GONE + markwon.setMarkdown(holder.messageContent, message.text) + updateMessageMetadata(holder, message) + } + MessageStatus.ERROR -> { + holder.loadingIndicator.visibility = View.GONE + holder.messageContent.visibility = View.VISIBLE + holder.btnRetry.visibility = View.VISIBLE + holder.generatingDots.visibility = View.GONE + holder.messageContent.text = message.text + if (message.sender == Sender.SYSTEM) { + holder.btnRetry.text = "Open AI Settings" + holder.btnRetry.setOnClickListener { + onMessageAction(ACTION_OPEN_SETTINGS, message) + } + } else { + holder.btnRetry.text = "Retry" + holder.btnRetry.setOnClickListener { + onMessageAction(ACTION_RETRY, message) + } + } + updateMessageMetadata(holder, message) + } + } + } + + private fun bindSystemMessage(holder: SystemMessageViewHolder, message: ChatMessage) { + markwon.setMarkdown(holder.messageContent, message.text) + updateSystemMessageExpansion(holder, message) + + holder.messageHeader.setOnClickListener { + if (!expandedMessageIds.remove(message.id)) { + expandedMessageIds.add(message.id) + } + val pos = holder.bindingAdapterPosition + if (pos != RecyclerView.NO_POSITION) { + notifyItemChanged(pos) + } + } + } + + private fun updateSystemMessageExpansion(holder: SystemMessageViewHolder, message: ChatMessage) { + val isExpanded = expandedMessageIds.contains(message.id) + if (isExpanded) { + holder.messageHeaderTitle.text = "System Log" + holder.messageContent.visibility = View.VISIBLE + holder.expandIcon.rotation = 180f + } else { + holder.messageHeaderTitle.text = createPreview(message.text) + holder.messageContent.visibility = View.GONE + holder.expandIcon.rotation = 0f + } + } + + private fun animateGeneratingDots(holder: DefaultMessageViewHolder) { + holder.generatingDots.visibility = View.VISIBLE + val dotStates = arrayOf(".", "..", "...") + var currentIndex = 0 + + val handler = android.os.Handler(android.os.Looper.getMainLooper()) + val runnable = object : Runnable { + override fun run() { + if (holder.generatingDots.visibility == View.VISIBLE) { + holder.generatingDots.text = dotStates[currentIndex] + currentIndex = (currentIndex + 1) % dotStates.size + handler.postDelayed(this, 500) + } + } + } + handler.post(runnable) + } + + private fun createPreview(rawText: String): String { + val cleanedText = rawText + .replace(Regex("`{1,3}|\\*{1,2}|_"), "") + .replace(Regex("\\s+"), " ") + .trim() + return "Log: $cleanedText" + } + + private fun updateMessageMetadata(holder: DefaultMessageViewHolder, message: ChatMessage) { + val timestampText = formatTimestamp(message.timestamp) + val durationText = formatDuration(message.durationMs) + + val hasTimestamp = timestampText != null + val hasDuration = durationText != null + + if (!hasTimestamp && !hasDuration) { + holder.messageMetadataContainer.visibility = View.GONE + return + } + + holder.messageMetadataContainer.visibility = View.VISIBLE + + if (hasTimestamp) { + holder.messageTimestamp.text = timestampText + holder.messageTimestamp.visibility = View.VISIBLE + } else { + holder.messageTimestamp.visibility = View.GONE + } + + if (hasDuration) { + holder.messageDuration.text = durationText + holder.messageDuration.visibility = View.VISIBLE + } else { + holder.messageDuration.visibility = View.GONE + } + } + + private fun formatTimestamp(timestamp: Long): String? { + if (timestamp <= 0L) return null + return synchronized(timeFormatter) { + timeFormatter.format(Date(timestamp)) + } + } + + private fun formatDuration(durationMs: Long?): String? { + if (durationMs == null || durationMs <= 0) return null + val seconds = durationMs / 1000.0 + return if (seconds < 60) { + "took ${decimalSecondsFormatter.format(seconds)}s" + } else { + val minutes = seconds / 60.0 + "took ${decimalSecondsFormatter.format(minutes)}m" + } + } + + private fun showContextMenu(view: View, message: ChatMessage) { + val context = view.context + val popup = PopupMenu(context, view) + + popup.menu.add(0, 1, 0, "Copy Text") + if (message.sender == Sender.USER) { + popup.menu.add(0, 2, 0, "Edit Message") + } + + popup.setOnMenuItemClickListener { item -> + when (item.itemId) { + 1 -> { + val clipboard = context.getSystemService(Context.CLIPBOARD_SERVICE) as ClipboardManager + val clip = ClipData.newPlainText("chat_message", message.text) + clipboard.setPrimaryClip(clip) + Toast.makeText(context, "Copied", Toast.LENGTH_SHORT).show() + true + } + 2 -> { + onMessageAction(ACTION_EDIT, message) + true + } + else -> false + } + } + popup.show() + } + + override fun onCurrentListChanged(previousList: MutableList, currentList: MutableList) { + super.onCurrentListChanged(previousList, currentList) + expandedMessageIds.clear() + } + + object DiffCallback : DiffUtil.ItemCallback() { + override fun areItemsTheSame(oldItem: ChatMessage, newItem: ChatMessage): Boolean { + return oldItem.id == newItem.id + } + + override fun areContentsTheSame(oldItem: ChatMessage, newItem: ChatMessage): Boolean { + return oldItem == newItem + } + + override fun getChangePayload(oldItem: ChatMessage, newItem: ChatMessage): Any? { + // If only the text or status changed, return a payload to avoid full rebind + if (oldItem.id == newItem.id && + (oldItem.text != newItem.text || oldItem.status != newItem.status)) { + return TextUpdatePayload(newItem.text, newItem.status) + } + return null + } + } + + // Payload for partial updates + data class TextUpdatePayload(val text: String, val status: MessageStatus) +} diff --git a/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/data/ChatStorageManager.kt b/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/data/ChatStorageManager.kt new file mode 100644 index 00000000..621f8640 --- /dev/null +++ b/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/data/ChatStorageManager.kt @@ -0,0 +1,41 @@ +package com.itsaky.androidide.plugins.aiassistant.data + +import android.content.Context +import android.content.SharedPreferences +import com.google.gson.Gson +import com.google.gson.reflect.TypeToken +import com.itsaky.androidide.plugins.aiassistant.models.ChatSession + +class ChatStorageManager(context: Context) { + private val prefs: SharedPreferences = + context.getSharedPreferences("ai_assistant_chats", Context.MODE_PRIVATE) + private val gson = Gson() + + companion object { + private const val KEY_SESSIONS = "chat_sessions" + private const val KEY_CURRENT_SESSION_ID = "current_session_id" + } + + fun saveSessions(sessions: List) { + val json = gson.toJson(sessions) + prefs.edit().putString(KEY_SESSIONS, json).apply() + } + + fun loadSessions(): List { + val json = prefs.getString(KEY_SESSIONS, null) ?: return emptyList() + val type = object : TypeToken>() {}.type + return try { + gson.fromJson(json, type) + } catch (e: Exception) { + emptyList() + } + } + + fun saveCurrentSessionId(sessionId: String?) { + prefs.edit().putString(KEY_CURRENT_SESSION_ID, sessionId).apply() + } + + fun loadCurrentSessionId(): String? { + return prefs.getString(KEY_CURRENT_SESSION_ID, null) + } +} diff --git a/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/fragments/AiSettingsFragment.kt b/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/fragments/AiSettingsFragment.kt new file mode 100644 index 00000000..7ed32d68 --- /dev/null +++ b/ai-assistant/src/main/kotlin/com/itsaky/androidide/plugins/aiassistant/fragments/AiSettingsFragment.kt @@ -0,0 +1,450 @@ +package com.itsaky.androidide.plugins.aiassistant.fragments + +import android.annotation.SuppressLint +import android.content.Context +import android.content.Intent +import android.net.Uri +import android.os.Bundle +import android.view.LayoutInflater +import android.view.View +import android.view.ViewGroup +import android.widget.* +import androidx.activity.result.contract.ActivityResultContracts +import androidx.fragment.app.DialogFragment +import androidx.lifecycle.ViewModelProvider +import com.itsaky.androidide.plugins.PluginContext +import com.itsaky.androidide.plugins.aiassistant.R +import com.itsaky.androidide.plugins.aiassistant.viewmodel.AiBackend +import com.itsaky.androidide.plugins.aiassistant.viewmodel.AiSettingsViewModel +import com.itsaky.androidide.plugins.aiassistant.viewmodel.EngineState +import com.itsaky.androidide.plugins.aiassistant.viewmodel.ModelLoadingState +import java.text.SimpleDateFormat +import java.util.Date +import java.util.Locale + +class AiSettingsFragment : DialogFragment() { + + private lateinit var viewModel: AiSettingsViewModel + private lateinit var settingsToolbar: LinearLayout + private lateinit var backButton: ImageButton + private lateinit var backendSpinner: Spinner + private lateinit var backendSpecificContainer: FrameLayout + + override fun onCreate(savedInstanceState: Bundle?) { + super.onCreate(savedInstanceState) + // Disable Material transitions to avoid resource loading issues + // Plugin uses compileOnly dependencies, so Material transition resources aren't bundled + enterTransition = null + exitTransition = null + } + + private val filePickerLauncher = + registerForActivityResult(ActivityResultContracts.OpenDocument()) { uri: Uri? -> + uri?.let { + try { + val takeFlags = Intent.FLAG_GRANT_READ_URI_PERMISSION + requireContext().contentResolver.takePersistableUriPermission(it, takeFlags) + + val uriString = it.toString() + viewModel.loadModelFromUri(uriString, requireContext()) + Toast.makeText(requireContext(), "Loading model...", Toast.LENGTH_SHORT).show() + } catch (e: Exception) { + Toast.makeText(requireContext(), "Error: ${e.message}", Toast.LENGTH_LONG).show() + } + } + } + + override fun onCreateView( + inflater: LayoutInflater, + container: ViewGroup?, + savedInstanceState: Bundle? + ): View? { + // Get plugin context to ensure proper resource inflation + val pluginContext = getPluginContext()?.androidContext ?: requireContext() + + // Create inflater with plugin context + val pluginInflater = inflater.cloneInContext(pluginContext) + + return pluginInflater.inflate(R.layout.fragment_ai_settings, container, false) + } + + override fun onViewCreated(view: View, savedInstanceState: Bundle?) { + super.onViewCreated(view, savedInstanceState) + + initializeViewModel() + initializeViews(view) + setupToolbar() + setupBackendSelector() + } + + private fun initializeViewModel() { + viewModel = ViewModelProvider( + this, + AiSettingsViewModelFactory { getPluginContext() } + )[AiSettingsViewModel::class.java] + } + + private fun getPluginContext(): PluginContext? { + return com.itsaky.androidide.plugins.aiassistant.AiAssistantPlugin.getContext() + } + + private fun initializeViews(view: View) { + settingsToolbar = view.findViewById(R.id.settings_toolbar) + backButton = view.findViewById(R.id.toolbar_back_button) + backendSpinner = view.findViewById(R.id.backend_autocomplete) + backendSpecificContainer = view.findViewById(R.id.backend_specific_settings_container) + } + + private fun setupToolbar() { + backButton.setOnClickListener { + // Close the dialog + dismiss() + } + } + + private fun setupBackendSelector() { + val backends = viewModel.getAvailableBackends() + val backendNames = backends.map { it.displayName } + val adapter = ArrayAdapter( + requireContext(), + android.R.layout.simple_spinner_item, + backendNames + ) + adapter.setDropDownViewResource(android.R.layout.simple_spinner_dropdown_item) + backendSpinner.adapter = adapter + + val currentBackend = viewModel.getCurrentBackend() + backendSpinner.setSelection(backends.indexOf(currentBackend)) + updateBackendSpecificUi(currentBackend) + + backendSpinner.onItemSelectedListener = object : AdapterView.OnItemSelectedListener { + override fun onItemSelected(parent: AdapterView<*>?, view: View?, position: Int, id: Long) { + val selectedBackend = backends[position] + viewModel.saveBackend(selectedBackend) + updateBackendSpecificUi(selectedBackend) + } + + override fun onNothingSelected(parent: AdapterView<*>?) {} + } + } + + private fun updateBackendSpecificUi(backend: AiBackend) { + backendSpecificContainer.removeAllViews() + + // Use plugin context for inflating layouts + val pluginContext = getPluginContext()?.androidContext ?: requireContext() + + when (backend) { + AiBackend.LOCAL_LLM -> { + val localLlmView = LayoutInflater.from(pluginContext) + .inflate(R.layout.layout_settings_local_llm, backendSpecificContainer, false) + backendSpecificContainer.addView(localLlmView) + setupLocalLlmUi(localLlmView) + } + AiBackend.GEMINI -> { + val geminiApiView = LayoutInflater.from(pluginContext) + .inflate(R.layout.layout_settings_gemini_api, backendSpecificContainer, false) + backendSpecificContainer.addView(geminiApiView) + setupGeminiApiUi(geminiApiView) + } + } + } + + private fun setupLocalLlmUi(view: View) { + val modelPathTextView = view.findViewById(R.id.selected_model_path) + val browseButton = view.findViewById